update sql

Author: hiyasame

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/go-playground/locales v0.14.0 // indirect
 	github.com/go-playground/universal-translator v0.18.0 // indirect
 	github.com/go-playground/validator/v10 v10.11.0 // indirect
+	github.com/go-sql-driver/mysql v1.6.0 // indirect
 	github.com/goccy/go-json v0.9.11 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/leodido/go-urn v1.2.1 // indirect

go.sum

@@ -12,6 +12,8 @@ github.com/go-playground/universal-translator v0.18.0 h1:82dyy6p4OuJq4/CByFNOn/j
 github.com/go-playground/universal-translator v0.18.0/go.mod h1:UvRDBj+xPUEGrFYl+lu/H90nyDXpg0fqeB/AQUGNTVA=
 github.com/go-playground/validator/v10 v10.11.0 h1:0W+xRM511GY47Yy3bZUbJVitCNg2BOGlCyvTqsp/xIw=
 github.com/go-playground/validator/v10 v10.11.0/go.mod h1:i+3WkQ1FvaUjjxh1kSvIA4dMGDBiPU55YFDl0WbKdWU=
+github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
+github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
 github.com/goccy/go-json v0.9.11 h1:/pAaQDLHEoCq/5FFmSKBswWmK6H0e8g4159Kc/X/nqk=
 github.com/goccy/go-json v0.9.11/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
 github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=

main.go

@@ -1,7 +1,9 @@
 package main
 
-import ginlearn "learning2/gin"
+import (
+	"learning2/sql"
+)
 
 func main() {
-	ginlearn.Run()
+	sql.Run()
 }

sql/db.go

@@ -0,0 +1,18 @@
+package sql
+
+import (
+	"database/sql"
+	"fmt"
+	"log"
+)
+
+var dB *sql.DB
+
+func InitializeDB(dbName, root, pwd, ipAndPort, charset string) {
+	daraSourceName := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=%s&parseTime=True", root, pwd, ipAndPort, dbName, charset)
+	db, err := sql.Open("mysql", daraSourceName)
+	if err != nil {
+		log.Fatal(err)
+	}
+	dB = db
+}

sql/main.go

@@ -0,0 +1,17 @@
+package sql
+
+import (
+	"github.com/gin-gonic/gin"
+	_ "github.com/go-sql-driver/mysql"
+)
+
+func Run() {
+	r := gin.Default()
+	// 链接数据库
+	InitializeDB("relearn_mysql", "root", "", "", "utf8")
+	InitRouter(r)
+	err := r.Run()
+	if err != nil {
+		return
+	}
+}

sql/middleware.go

@@ -0,0 +1,18 @@
+package sql
+
+import (
+	"github.com/gin-gonic/gin"
+)
+
+var auth gin.HandlerFunc = func(context *gin.Context) {
+	value, err := context.Cookie("gin_cookie")
+	if err != nil {
+		context.JSON(403, gin.H{
+			"message": "认证失败,没有cookie",
+		})
+		context.Abort()
+	} else {
+		context.Set("cookie", value)
+		context.Next()
+	}
+}

sql/router.go

@@ -0,0 +1,99 @@
+package sql
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"log"
+	"math/rand"
+	"time"
+)
+
+var r *gin.Engine
+
+func InitRouter(e *gin.Engine) {
+	r = e
+	InitAuth()
+}
+
+func InitAuth() {
+	r.POST("/register", func(context *gin.Context) {
+		name := context.PostForm("name")
+		password := context.PostForm("password")
+		spQuestion := context.PostForm("sp_question")
+		spAnswer := context.PostForm("sp_answer")
+		stmt, err := dB.Prepare("insert into login_plus(id, name, password_md5, password_salt, sp_question, sp_answer) values (?,?,?,?,?,?)")
+		if err != nil {
+			log.Fatal(err)
+			return
+		}
+		salt := GenerateUUIDStr()
+		passwordMd5 := MD5(password + salt)
+		_, err = stmt.Exec(0, name, passwordMd5, salt, spQuestion, spAnswer)
+		if err != nil {
+			log.Fatal(err)
+			return
+		}
+		context.JSON(200, map[string]string{
+			"code":    "0",
+			"message": "注册成功",
+		})
+	})
+	r.POST("/login", func(context *gin.Context) {
+		name := context.PostForm("name")
+		password := context.PostForm("password")
+		var passwordMd5 string
+		var passwordSalt string
+		stmt, err := dB.Prepare("select password_md5, password_salt from login_plus where name = ?")
+		if err != nil {
+			log.Fatal(err)
+			return
+		}
+		rows, err := stmt.Query(name)
+		if err != nil {
+			log.Fatal(err)
+			return
+		}
+		defer rows.Close()
+		rows.Next()
+		err = rows.Scan(&passwordMd5, &passwordSalt)
+		if err != nil {
+			log.Fatal(err)
+			return
+		}
+		if passwordMd5 == MD5(password+passwordSalt) {
+			context.SetCookie("gin_cookie", "", 3600, "/", "", false, true)
+			context.JSON(200, map[string]string{
+				"code":    "0",
+				"message": "登录成功",
+			})
+		} else {
+			context.JSON(200, map[string]string{
+				"code":    "-1",
+				"message": "账号或密码错误",
+			})
+		}
+	})
+}
+
+func GenerateUUIDStr() string {
+	rand.Seed(time.Now().Unix())
+	var randomBytes = make([]byte, 16)
+	for i := 0; i < 16; i++ {
+		randomBytes[i] = byte(rand.Intn(128))
+	}
+	// 摘自 java uuid 的生成
+	randomBytes[6] &= 0x0f /* clear version        */
+	randomBytes[6] |= 0x40 /* set to version 4     */
+	randomBytes[8] &= 0x3f /* clear variant        */
+	randomBytes[8] |= 0x80 /* set to IETF variant  */
+	return hex.EncodeToString(randomBytes)
+}
+
+func MD5(str string) string {
+	data := []byte(str) //切片
+	has := md5.Sum(data)
+	md5str := fmt.Sprintf("%x", has) //将[]byte转成16进制
+	return md5str
+}