gin-vue-demo/main.go

package main

import (
	"fmt"
	"log"
	"math/rand"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

type User struct {
	gorm.Model
	Name      string `json:"name";gorm:"type:varchar(20);not null`
	Telephone string `json:"telephone";gorm:"type:varchar(11);not null;unique"`
	Password  string `json:"password";gorm:"type:varchar(20);not null"`
}

func main() {
	db := initDB()

	r := gin.Default()
	r.POST("/api/auth/register", func(ctx *gin.Context) {
		// 获取参数
		name := ctx.Query("name")
		telephone := ctx.Query("telephone")
		password := ctx.Query("password")

		// 数据验证
		if len(telephone) != 11 {
			ctx.JSON(http.StatusUnprocessableEntity, gin.H{
				"code":    http.StatusUnprocessableEntity,
				"message": "手机号必须是11位",
			})
			return
		}
		if len(password) < 6 {
			ctx.JSON(http.StatusUnprocessableEntity, gin.H{
				"code":    http.StatusUnprocessableEntity,
				"message": "密码必须大于6位",
			})
			return
		}
		// 如果名字为空则返回10为随机字符串
		if name == "" {
			name = RandomString(10)
		}

		log.Println(name, telephone, password)

		// 判断手机号是否存在
		if isExistTelephone(db, telephone) {
			ctx.JSON(http.StatusUnprocessableEntity, gin.H{
				"code":    http.StatusUnprocessableEntity,
				"message": "手机号已存在",
			})
			return
		}

		// 创建用户
		newUser := User{
			Name:      name,
			Telephone: telephone,
			Password:  password,
		}
		db.Create(&newUser)

		// 返回结果
		ctx.JSON(200, gin.H{
			"code":    http.StatusOK,
			"message": "注册成功",
		})
	})
	panic(r.Run("127.0.0.1:8080"))
}

// 随机字符串
func RandomString(length int) string {
	str := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
	bytes := []byte(str)
	result := []byte{}
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	for i := 0; i < length; i++ {
		result = append(result, bytes[r.Intn(len(bytes))])
	}
	return string(result)
}

func initDB() *gorm.DB {
	// 配置 MySQL 连接参数
	username := "root"       // 账号
	password := "jiaobaba"   // 密码
	host := "127.0.0.1"      // 数据库地址，可以是Ip或者域名
	port := 3306             // 数据库端口
	Dbname := "gin-vue-demo" // 数据库名
	timeout := "10s"         // 连接超时，10秒

	// 拼接下 dsn 参数, dsn 格式可以参考上面的语法，这里使用 Sprintf 动态拼接 dsn 参数，因为一般数据库连接参数，我们都是保存在配置文件里面，需要从配置文件加载参数，然后拼接 dsn。
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local&timeout=%s",
		username,
		password,
		host,
		port,
		Dbname,
		timeout)

	// 连接 Mysql, 获得 DB 类型实例，用于后面的数据库读写操作。
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("连接数据库失败, error=" + err.Error())
	}

	db.AutoMigrate(&User{})
	return db
}

func isExistTelephone(db *gorm.DB, telephone string) bool {
	var user User
	db.Where("telephone = ?", telephone).First(&user)
	return user.ID != 0
}