一个练习项目，好玩的bbs-go-iris

代码：

package main

/**
go env -w GO111MODULE=on
go env -w GOPROXY=https://goproxy.cn
go mod init gos
go mod tidy
go mod vendor
*/

import (
    "crypto/md5"
    "database/sql"
    "fmt"
    "log"
    "math"
    "strconv"

    _ "github.com/go-sql-driver/mysql"
    "github.com/jmoiron/sqlx"
    "github.com/kataras/iris/v12"
)

var (
    userName  string = "root"
    password  string = "123456"
    ipAddrees string = "127.0.0.1"
    port      int    = 3306
    dbName    string = "my_bbs"
    charset   string = "utf8mb4"
)

type countinfo struct {
    Count int64 `db:"count" json:"count"`
}

type userinfo struct {
    Id        int    `db:"id" json:"id"`
    Username  string `db:"username" json:"username"`
    Nickname  string `db:"nickname" json:"nickname"`
    AddTime   string `db:"addTime" json:"addTime"`
    SessionId string `db:"sessionId" json:"sessionId"`
}

type postinfo struct {
    Id            int    `db:"id" json:"id"`
    Title         string `db:"title" json:"title"`
    UserId        string `db:"userId" json:"userId"`
    UserNickename string `db:"userNickename" json:"userNickename"`
    ReplyNum      int    `db:"replyNum" json:"replyNum"`
    UpdateTime    string `db:"updateTime" json:"updateTime"`
}

type postinfodetail struct {
    Id            int    `db:"id" json:"id"`
    Title         string `db:"title" json:"title"`
    Content       string `db:"content" json:"content"`
    UserId        int    `db:"userId" json:"userId"`
    UserNickename string `db:"userNickename" json:"userNickename"`
    ReplyNum      int    `db:"replyNum" json:"replyNum"`
    UpdateTime    string `db:"updateTime" json:"updateTime"`
}

type replyinfo struct {
    Id                 int    `db:"id" json:"id"`
    ContentId          int    `db:"contentId" json:"contentId"`
    Content            string `db:"content" json:"content"`
    ReplyUserId        int    `db:"replyUserId" json:"replyUserId"`
    ReplyUserNickename string `db:"replyUserNickename" json:"replyUserNickename"`
    AddTime            string `db:"addTime" json:"addTime"`
}

type postlistpage struct {
    Totalpage int        `db:"totalpage" json:"totalpage"`
    Data      []postinfo `db:"data" json:"data"`
}

type replylistpage struct {
    Totalpage int         `db:"totalpage" json:"totalpage"`
    Data      []replyinfo `db:"data" json:"data"`
}

var pagesize = 20
var secretKey = "saacac3423@21212"

func connectMysql() *sqlx.DB {
    dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s", userName, password, ipAddrees, port, dbName, charset)
    Db, err := sqlx.Open("mysql", dsn)
    if err != nil {
        log.Fatal(err)
    }
    return Db
}

func calcMD5(message string) string {
    data := []byte(message)
    return fmt.Sprintf("%x", md5.Sum(data))
}

func getloginuserinfo(sessionId string) *userinfo {
    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *userinfo = new(userinfo)
    sql1 := "select id,username,nickname,addTime,sessionId from user where sessionId='" + sessionId + "'"
    err := Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    return result1
}

func Index_test(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    ctx.Text("此站接口使用go.iris实现，<a href='api.html' target='_blank'>接口列表</a>")
}

func User_register(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    username := ctx.URLParam("username")
    password := ctx.URLParam("password")
    nickname := ctx.URLParam("nickname")
    passwordMd5 := calcMD5(password)

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *userinfo = new(userinfo)
    sql1 := "select id,username,nickname,addTime,sessionId from user where username='" + username + "'"
    err := Db.Get(result1, sql1)

    if err == sql.ErrNoRows {
        sql2 := "insert into user(username, password, nickname) value('" + username + "', '" + passwordMd5 + "', '" + nickname + "')"
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        insertId, err := result2.LastInsertId()
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": insertId})
    } else if result1.Id > 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "用户名已经存在", "data": ""})
    }
}

func User_login(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    username := ctx.URLParam("username")
    password := ctx.URLParam("password")
    passwordMd5 := calcMD5(password)

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *userinfo = new(userinfo)
    sql1 := "select id,username,nickname,addTime,sessionId from user where username='" + username + "' and password='" + passwordMd5 + "'"
    err := Db.Get(result1, sql1)

    if err == sql.ErrNoRows {
        ctx.JSON(iris.Map{"code": 1, "msg": "用户名或者密码错误", "data": ""})
    } else {
        sessionId := calcMD5(secretKey + strconv.Itoa(result1.Id) + result1.AddTime)
        sql2 := "update user set sessionId='" + sessionId + "' where id=" + strconv.Itoa(result1.Id)
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        result1.SessionId = sessionId
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result1})
    }
}

func User_logout(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *userinfo = new(userinfo)
    sql1 := "select id,username,nickname,addTime,sessionId from user where sessionId='" + sessionId + "'"
    err := Db.Get(result1, sql1)

    if err == sql.ErrNoRows {
        //
    } else {

        sql2 := "update user set sessionId='' where sessionId='" + sessionId + "'"
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        result1.SessionId = ""
    }

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result1})
}

func User_info(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *userinfo = new(userinfo)
    sql1 := "select id,username,nickname,addTime,sessionId from user where sessionId='" + sessionId + "'"
    err := Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result1})
}

func Post_list(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    page, err := strconv.Atoi(ctx.URLParam("page"))
    keyword := ctx.URLParam("keyword")

    if page <= 0 {
        page = 1
    }

    start := (page - 1) * pagesize

    addSql := " isDel=0 "
    if keyword != "" {
        addSql = " isDel=0 and title like '%" + keyword + "%' "
    }

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *countinfo = new(countinfo)
    sql1 := "select count(1) as count from content where " + addSql
    err = Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    var result2 []postinfo
    sql2 := "select id,title,userId,userNickename,replyNum,updateTime from content where " + addSql + " order by updateTime desc limit " + strconv.Itoa(start) + "," + strconv.Itoa(pagesize)
    err = Db.Select(&result2, sql2)
    if err != nil {
        log.Fatal(err)
    }

    totalcount := result1.Count
    totalpage := int(math.Ceil(float64(totalcount) / float64(pagesize)))

    var result3 *postlistpage = new(postlistpage)

    if result2 != nil {
        result3.Data = result2
    } else {
        result3.Data = make([]postinfo, 0)
    }

    result3.Totalpage = totalpage

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result3})
}

func Post_detail(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *postinfodetail = new(postinfodetail)
    sql1 := "select id,title,content,userId,userNickename,replyNum,updateTime from content where isDel=0 and id=" + id
    err := Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result1})
}

func Post_add(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    title := ctx.URLParam("title")
    content := ctx.URLParam("content")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        sql2 := "insert into content(title, content, userId, userNickename) value('" + title + "', '" + content + "', " + strconv.Itoa(userinfo1.Id) + ", '" + userinfo1.Nickname + "')"
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        insertId, err := result2.LastInsertId()
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": insertId})
    }
}

func Post_edit(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")
    title := ctx.URLParam("title")
    content := ctx.URLParam("content")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        sql2 := "update content set title='" + title + "',content='" + content + "',userId=" + strconv.Itoa(userinfo1.Id) + ",userNickename='" + userinfo1.Nickname + "' where id=" + id + " and userId=" + strconv.Itoa(userinfo1.Id)
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": ""})
    }
}

func Post_delete(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        sql2 := "update content set isDel=1 where id=" + id + " and userId=" + strconv.Itoa(userinfo1.Id)
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": ""})
    }
}

func Reply_list(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    page, err := strconv.Atoi(ctx.URLParam("page"))
    contentId := ctx.URLParam("contentId")

    if page <= 0 {
        page = 1
    }

    start := (page - 1) * pagesize

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *countinfo = new(countinfo)
    sql1 := "select count(1) as count from reply where isDel=0 and contentId=" + contentId
    err = Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    var result2 []replyinfo
    sql2 := "select id,content,replyUserId,replyUserNickename,addTime,contentId from reply where isDel=0 and contentId=" + contentId + " order by id asc limit " + strconv.Itoa(start) + "," + strconv.Itoa(pagesize)
    err = Db.Select(&result2, sql2)
    if err != nil {
        log.Fatal(err)
    }

    totalcount := result1.Count
    totalpage := int(math.Ceil(float64(totalcount) / float64(pagesize)))

    var result3 *replylistpage = new(replylistpage)
    if result2 != nil {
        result3.Data = result2
    } else {
        result3.Data = make([]replyinfo, 0)
    }

    result3.Totalpage = totalpage

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result3})
}

func Reply_detail(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var result1 *replyinfo = new(replyinfo)
    sql1 := "select id,content,replyUserId,replyUserNickename,addTime,contentId from reply where isDel=0 and id=" + id
    err := Db.Get(result1, sql1)
    if err != nil {
        log.Fatal(err)
    }

    ctx.JSON(iris.Map{"code": 0, "msg": "", "data": result1})
}

func Reply_add(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    contentId := ctx.URLParam("contentId")
    content := ctx.URLParam("content")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        sql1 := "update content set replyNum=replyNum+1 where id=" + contentId
        result1, err := Db.Exec(sql1)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result1)
        }

        sql2 := "insert into reply(contentId, content, replyUserId, replyUserNickename) value(" + contentId + ", '" + content + "', " + strconv.Itoa(userinfo1.Id) + ", '" + userinfo1.Nickname + "')"
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        insertId, err := result2.LastInsertId()
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": insertId})
    }
}

func Reply_edit(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")
    content := ctx.URLParam("content")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        sql2 := "update reply set content='" + content + "',replyUserId=" + strconv.Itoa(userinfo1.Id) + ",replyUserNickename='" + userinfo1.Nickname + "' where id=" + id + " and replyUserId=" + strconv.Itoa(userinfo1.Id)
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": ""})
    }
}

func Reply_delete(ctx iris.Context) {
    ctx.Header("Server", "go-iris")
    id := ctx.URLParam("id")
    sessionId := ctx.URLParam("sessionId")

    var Db *sqlx.DB = connectMysql()
    defer Db.Close()

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        ctx.JSON(iris.Map{"code": 1, "msg": "请先登录", "data": ""})
    } else {
        var result0 *replyinfo = new(replyinfo)
        sql0 := "select id,content,replyUserId,replyUserNickename,addTime,contentId from reply where isDel=0 and id=" + id
        err := Db.Get(result0, sql0)
        if err != nil {
            log.Fatal(err)
        }

        sql1 := "update content set replyNum=replyNum-1 where id=" + strconv.Itoa(result0.ContentId)
        result1, err := Db.Exec(sql1)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result1)
        }

        sql2 := "update reply set isDel=1 where id=" + id + " and replyUserId=" + strconv.Itoa(userinfo1.Id)
        result2, err := Db.Exec(sql2)
        if err != nil {
            log.Fatal(err)
            log.Fatal(result2)
        }
        ctx.JSON(iris.Map{"code": 0, "msg": "", "data": ""})
    }
}

func main() {
    app := iris.New()

    app.Get("/", Index_test)
    app.Get("/user/register", User_register)
    app.Get("/user/login", User_login)
    app.Get("/user/logout", User_logout)
    app.Get("/user/getuserinfo", User_info)
    app.Get("/post/list", Post_list)
    app.Get("/post/detail", Post_detail)
    app.Get("/post/add", Post_add)
    app.Get("/post/edit", Post_edit)
    app.Get("/post/delete", Post_delete)
    app.Get("/reply/list", Reply_list)
    app.Get("/reply/detail", Reply_detail)
    app.Get("/reply/add", Reply_add)
    app.Get("/reply/edit", Reply_edit)
    app.Get("/reply/delete", Reply_delete)

    app.Run(iris.Addr(":1094"))
}

 

输出：

D:\workspace\studys\study_bbs>D:\software\go\bin\go.exe run D:\workspace\studys\study_bbs\start_web_iris.go
Iris Version: 12.2.11

Now listening on:
> Network:  http://192.168.14.239:1094
> Network:  http://10.0.0.161:1094
> Network:  http://192.168.137.1:1094
> Local:    http://localhost:1094
Application started. Press CTRL+C to shut down.