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

代码：

package main
 
import (
    "fmt"
    "log"
    "net/http"
    "encoding/json"
    "github.com/gorilla/mux"
    "github.com/jmoiron/sqlx"
    _ "github.com/go-sql-driver/mysql"
    "database/sql"
    "strconv"
    "math"
    "crypto/md5"
)

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"`
}

type Responsex struct {
    Code int    `json:"code"`
    Msg  string `json:"msg"`
    Data string `json:"data"`
}

type Responseuserinfo struct {
    Code int       `json:"code"`
    Msg  string    `json:"msg"`
    Data *userinfo `json:"data"`
}

type Responsepostlistpage struct {
    Code int           `json:"code"`
    Msg  string        `json:"msg"`
    Data *postlistpage `json:"data"`
}

type Responsepostinfodetail struct {
    Code int             `json:"code"`
    Msg  string          `json:"msg"`
    Data *postinfodetail `json:"data"`
}

type Responsereplylistpage struct {
    Code int            `json:"code"`
    Msg  string         `json:"msg"`
    Data *replylistpage `json:"data"`
}

type Responsereplyinfo struct {
    Code int        `json:"code"`
    Msg  string     `json:"msg"`
    Data *replyinfo `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 jsonEncode(object interface{}) string {
    // 序列化用户对象到JSON
    jsonData, err := json.Marshal(object)
    if err != nil {
        log.Fatal(err)
    }
    
    return string(jsonData)
}

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(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    fmt.Fprintf(w, "此站接口使用go.mux实现，<a href='api.html' target='_blank'>接口列表</a>")
}

func User_register(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")

    username := r.FormValue("username")
    password := r.FormValue("password")
    nickname := r.FormValue("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()
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}))
    } else if result1.Id > 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 1, Msg: "用户名已经存在", Data: ""}))
    } else {
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 1, Msg: "未知错误", Data: ""}))
    }
}

func User_login(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    username := r.FormValue("username")
    password := r.FormValue("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 {
        fmt.Fprintf(w, jsonEncode(Responsex{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
        fmt.Fprintf(w, jsonEncode(Responseuserinfo{Code: 0, Msg: "", Data: result1}))
    }
}

func User_logout(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    sessionId := r.FormValue("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 = ""
    }

    fmt.Fprintf(w, jsonEncode(Responseuserinfo{Code: 0, Msg: "", Data: result1}))
}

func User_info(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    sessionId := r.FormValue("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)
    }

    fmt.Fprintf(w, jsonEncode(Responseuserinfo{Code: 0, Msg: "", Data: result1}))
}

func Post_list(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    page, err := strconv.Atoi(r.FormValue("page"))
    keyword := r.FormValue("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

    fmt.Fprintf(w, jsonEncode(Responsepostlistpage{Code: 0, Msg: "", Data: result3}))
}

func Post_detail(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("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)
    }

    fmt.Fprintf(w, jsonEncode(Responsepostinfodetail{Code: 0, Msg: "", Data: result1}))
}

func Post_add(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    title := r.FormValue("title")
    content := r.FormValue("content")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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()
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}))
    }
}

func Post_edit(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("id")
    title := r.FormValue("title")
    content := r.FormValue("content")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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)
        }
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: ""}))
    }
}

func Post_delete(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("id")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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)
        }
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: ""}))
    }
}

func Reply_list(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    page, err := strconv.Atoi(r.FormValue("page"))
    contentId := r.FormValue("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

    fmt.Fprintf(w, jsonEncode(Responsereplylistpage{Code: 0, Msg: "", Data: result3}))
}

func Reply_detail(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("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)
    }

    fmt.Fprintf(w, jsonEncode(Responsereplyinfo{Code: 0, Msg: "", Data: result1}))
}

func Reply_add(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    contentId := r.FormValue("contentId")
    content := r.FormValue("content")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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()
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}))
    }
}

func Reply_edit(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("id")
    content := r.FormValue("content")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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)
        }
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: ""}))
    }
}

func Reply_delete(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", "go-mux")
    id := r.FormValue("id")
    sessionId := r.FormValue("sessionId")

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

    var userinfo1 *userinfo = getloginuserinfo(sessionId)
    if userinfo1.Id <= 0 {
        fmt.Fprintf(w, jsonEncode(Responsex{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)
        }
        fmt.Fprintf(w, jsonEncode(Responsex{Code: 0, Msg: "", Data: ""}))
    }
}
 
func main() {
    router := mux.NewRouter()

    router.HandleFunc("/", Index_test).Name("")
    router.HandleFunc("/user/register", User_register).Name("")
    router.HandleFunc("/user/login", User_login).Name("")
    router.HandleFunc("/user/logout", User_logout).Name("")
    router.HandleFunc("/user/getuserinfo", User_info).Name("")
    router.HandleFunc("/post/list", Post_list).Name("")
    router.HandleFunc("/post/detail", Post_detail).Name("")
    router.HandleFunc("/post/add", Post_add).Name("")
    router.HandleFunc("/post/edit", Post_edit).Name("")
    router.HandleFunc("/post/delete", Post_delete).Name("")
    router.HandleFunc("/reply/list", Reply_list).Name("")
    router.HandleFunc("/reply/detail", Reply_detail).Name("")
    router.HandleFunc("/reply/add", Reply_add).Name("")
    router.HandleFunc("/reply/edit", Reply_edit).Name("")
    router.HandleFunc("/reply/delete", Reply_delete).Name("")
 
    log.Fatal(http.ListenAndServe(":1101", router))
}

 

输出：