代码:
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" "encoding/json" "fmt" "log" "math" "net/http" "strconv" "github.com/go-chi/chi/v5" _ "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"` } 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 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-chi") w.Write([]byte(fmt.Sprintf("此站接口使用go.chi实现,<a href='api.html' target='_blank'>接口列表</a>"))) } func User_register(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") username := r.URL.Query().Get("username") password := r.URL.Query().Get("password") nickname := r.URL.Query().Get("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() output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}) w.Write([]byte(fmt.Sprintf(string(output)))) } else if result1.Id > 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "用户名已经存在", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } else { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "未知错误", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func User_login(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") username := r.URL.Query().Get("username") password := r.URL.Query().Get("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 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "用户名或者密码错误", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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 output, _ := json.Marshal(&Responseuserinfo{Code: 0, Msg: "", Data: result1}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func User_logout(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") sessionId := r.URL.Query().Get("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 = "" } output, _ := json.Marshal(&Responseuserinfo{Code: 0, Msg: "", Data: result1}) w.Write([]byte(fmt.Sprintf(string(output)))) } func User_info(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") sessionId := r.URL.Query().Get("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) } output, _ := json.Marshal(&Responseuserinfo{Code: 0, Msg: "", Data: result1}) w.Write([]byte(fmt.Sprintf(string(output)))) } func Post_list(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") page, err := strconv.Atoi(r.URL.Query().Get("page")) keyword := r.URL.Query().Get("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 output, _ := json.Marshal(&Responsepostlistpage{Code: 0, Msg: "", Data: result3}) w.Write([]byte(fmt.Sprintf(string(output)))) } func Post_detail(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("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) } output, _ := json.Marshal(&Responsepostinfodetail{Code: 0, Msg: "", Data: result1}) w.Write([]byte(fmt.Sprintf(string(output)))) } func Post_add(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") title := r.URL.Query().Get("title") content := r.URL.Query().Get("content") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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() output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func Post_edit(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("id") title := r.URL.Query().Get("title") content := r.URL.Query().Get("content") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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) } output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func Post_delete(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("id") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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) } output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func Reply_list(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") page, err := strconv.Atoi(r.URL.Query().Get("page")) contentId := r.URL.Query().Get("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 output, _ := json.Marshal(&Responsereplylistpage{Code: 0, Msg: "", Data: result3}) w.Write([]byte(fmt.Sprintf(string(output)))) } func Reply_detail(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("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) } output, _ := json.Marshal(&Responsereplyinfo{Code: 0, Msg: "", Data: result1}) w.Write([]byte(fmt.Sprintf(string(output)))) } func Reply_add(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") contentId := r.URL.Query().Get("contentId") content := r.URL.Query().Get("content") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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() output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: fmt.Sprintf("%d", insertId)}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func Reply_edit(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("id") content := r.URL.Query().Get("content") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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) } output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func Reply_delete(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "go-chi") id := r.URL.Query().Get("id") sessionId := r.URL.Query().Get("sessionId") var Db *sqlx.DB = connectMysql() defer Db.Close() var userinfo1 *userinfo = getloginuserinfo(sessionId) if userinfo1.Id <= 0 { output, _ := json.Marshal(&Responsex{Code: 1, Msg: "请先登录", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } 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) } output, _ := json.Marshal(&Responsex{Code: 0, Msg: "", Data: ""}) w.Write([]byte(fmt.Sprintf(string(output)))) } } func main() { r := chi.NewRouter() 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) fmt.Println("http://127.0.0.1:1097") http.ListenAndServe(":1097", r) }
输出:
D:\workspace\studys\study_bbs>D:\software\go\bin\go.exe run D:\workspace\studys\study_bbs\start_web_chi.go http://127.0.0.1:1097
标签:string,err,chi,db,sessionId,result1,json,go,bbs From: https://www.cnblogs.com/xuxiaobo/p/18392620