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

代码：

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"
    "net/http"
    "strconv"

    "github.com/gin-gonic/gin"
    _ "github.com/go-sql-driver/mysql"
    "github.com/jmoiron/sqlx"
)

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(c *gin.Context) {
    c.Header("Server", "go-gin")
    c.String(http.StatusOK, "此站接口使用go.gin实现，<a href='api.html' target='_blank'>接口列表</a>")
}

func User_register(c *gin.Context) {
    c.Header("Server", "go-gin")
    username := c.Query("username")
    password := c.Query("password")
    nickname := c.Query("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()
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": insertId})
    } else if result1.Id > 0 {
        c.JSON(http.StatusOK, gin.H{"code": 1, "msg": "用户名已经存在", "data": ""})
    }
}

func User_login(c *gin.Context) {
    c.Header("Server", "go-gin")
    username := c.Query("username")
    password := c.Query("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 {
        c.JSON(http.StatusOK, gin.H{"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
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result1})
    }
}

func User_logout(c *gin.Context) {
    c.Header("Server", "go-gin")
    sessionId := c.Query("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 = ""
    }

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result1})
}

func User_info(c *gin.Context) {
    c.Header("Server", "go-gin")
    sessionId := c.Query("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)
    }

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result1})
}

func Post_list(c *gin.Context) {
    c.Header("Server", "go-gin")
    page, err := strconv.Atoi(c.Query("page"))
    keyword := c.Query("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

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result3})
}

func Post_detail(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("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)
    }

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result1})
}

func Post_add(c *gin.Context) {
    c.Header("Server", "go-gin")
    title := c.Query("title")
    content := c.Query("content")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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()
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": insertId})
    }
}

func Post_edit(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("id")
    title := c.Query("title")
    content := c.Query("content")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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)
        }
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": ""})
    }
}

func Post_delete(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("id")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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)
        }
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": ""})
    }
}

func Reply_list(c *gin.Context) {
    c.Header("Server", "go-gin")
    page, err := strconv.Atoi(c.Query("page"))
    contentId := c.Query("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

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result3})
}

func Reply_detail(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("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)
    }

    c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": result1})
}

func Reply_add(c *gin.Context) {
    c.Header("Server", "go-gin")
    contentId := c.Query("contentId")
    content := c.Query("content")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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()
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": insertId})
    }
}

func Reply_edit(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("id")
    content := c.Query("content")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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)
        }
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": ""})
    }
}

func Reply_delete(c *gin.Context) {
    c.Header("Server", "go-gin")
    id := c.Query("id")
    sessionId := c.Query("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        c.JSON(http.StatusOK, gin.H{"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)
        }
        c.JSON(http.StatusOK, gin.H{"code": 0, "msg": "", "data": ""})
    }
}

func main() {
    r := gin.Default()

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

    r.Run(":1085")
}

 

输出：

D:\workspace\studys\study_pys\pc_app\dist>cd D:\workspace\studys\study_bbs\

D:\workspace\studys\study_bbs>D:\software\go\bin\go.exe run D:\workspace\studys\study_bbs\start_web_gin.go
[GIN-debug] [WARNING] Creating an Engine instance with the Logger and Recovery middleware already attached.

[GIN-debug] [WARNING] Running in "debug" mode. Switch to "release" mode in production.
 - using env:   export GIN_MODE=release
 - using code:  gin.SetMode(gin.ReleaseMode)

[GIN-debug] GET    /                         --> main.Index_test (3 handlers)
[GIN-debug] GET    /user/register            --> main.User_register (3 handlers)
[GIN-debug] GET    /user/login               --> main.User_login (3 handlers)
[GIN-debug] GET    /user/logout              --> main.User_logout (3 handlers)
[GIN-debug] GET    /user/getuserinfo         --> main.User_info (3 handlers)
[GIN-debug] GET    /post/list                --> main.Post_list (3 handlers)
[GIN-debug] GET    /post/detail              --> main.Post_detail (3 handlers)
[GIN-debug] GET    /post/add                 --> main.Post_add (3 handlers)
[GIN-debug] GET    /post/edit                --> main.Post_edit (3 handlers)
[GIN-debug] GET    /post/delete              --> main.Post_delete (3 handlers)
[GIN-debug] GET    /reply/list               --> main.Reply_list (3 handlers)
[GIN-debug] GET    /reply/detail             --> main.Reply_detail (3 handlers)
[GIN-debug] GET    /reply/add                --> main.Reply_add (3 handlers)
[GIN-debug] GET    /reply/edit               --> main.Reply_edit (3 handlers)
[GIN-debug] GET    /reply/delete             --> main.Reply_delete (3 handlers)
[GIN-debug] [WARNING] You trusted all proxies, this is NOT safe. We recommend you to set a value.
Please check https://pkg.go.dev/github.com/gin-gonic/gin#readme-don-t-trust-all-proxies for details.
[GIN-debug] Listening and serving HTTP on :1085