Golang中gorm+gin的简单使用

自己做记录，方便查看

    package main
    
    import (
        "fmt"
        "github.com/gin-gonic/gin"
        _ "github.com/go-sql-driver/mysql"
        "github.com/jinzhu/gorm"
    )
    
    type User struct {
        Id       int    `json:"id",gorm:"auto-increment"`
        Name     string `json:"name"`
        Tel      string `json:"tel"`
        Password string `json:"password"`
    }
    
    type ResponseData struct {
        Code    int         `json:"code"`
        Status  string      `json:"status"`
        Message string      `json:"message"`
        Data    interface{} `json:"data"`
    }
    
    var db *gorm.DB
    var err error
    
    func main() {
        // db
        db, err = gorm.Open("mysql", "root:03140314@/golang_test?charset=utf8&parseTime=True&loc=Local")
        checkErr()
        //db.SingularTable(true)
        db.AutoMigrate(&User{})
        defer db.Close()
        // http
        router := gin.Default()
        v1 := router.Group("/api/v1")
        {
            // user
            v1.POST("/user/login", login)
            v1.POST("/user/register", register)
            v1.GET("/user/get_info", getUserInfo)
        }
        router.Run()
    }
    
    func checkErr() {
        if err != nil {
            fmt.Println(err)
            err = nil
        }
    }
    
    // post form
    func login(c *gin.Context) {
        // 获取表单数据 tel，password
        tel := c.PostForm("tel")
        psd := c.PostForm("password")
        if len(tel) == 0 || len(psd) == 0 {
            response := ResponseData{
                Code:    50001,
                Status:  "error",
                Message: "账号或密码不能为空",
                Data:    "",
            }
            c.JSON(200, response)
        } else {
            var user User
            db.Where("tel=?", tel).First(&user)
            if user.Tel == "" {
                response := ResponseData{
                    Code:    50001,
                    Status:  "error",
                    Message: "用户不存在",
                    Data:    "",
                }
                c.JSON(200, response)
            } else {
                if user.Password == psd {
                    response := ResponseData{
                        Code:    200,
                        Status:  "success",
                        Message: "登录成功",
                        Data:    "",
                    }
                    c.JSON(200, response)
                } else {
                    response := ResponseData{
                        Code:    5001,
                        Status:  "error",
                        Message: "密码错误",
                        Data:    "",
                    }
                    c.JSON(200, response)
                }
            }
        }
    }
    
    // post form
    func register(c *gin.Context) {
        tel := c.PostForm("tel")
        psd := c.PostForm("password")
        if len(tel) == 0 || len(psd) == 0 {
            response := ResponseData{
                Code:    50001,
                Status:  "error",
                Message: "账号或密码不能为空",
                Data:    "",
            }
            c.JSON(200, response)
            return
        }
        var user User
        db.Where("tel=?", tel).First(&user)
        if user.Tel == tel {
            response := ResponseData{
                Code:    50001,
                Status:  "error",
                Message: "手机号已注册",
                Data:    "",
            }
            c.JSON(200, response)
        } else {
            newUser := User{Tel: tel, Password: psd}
            db.Create(&newUser)
            response := ResponseData{
                Code:    200,
                Status:  "success",
                Message: "注册成功",
                Data:    "",
            }
            c.JSON(200, response)
        }
    }
    
    // get url
    func getUserInfo(c *gin.Context) {
        id := c.Query("id")
        if id == "" {
            response := ResponseData{
                Code:    50001,
                Status:  "error",
                Message: "参数错误",
                Data:    "",
            }
            c.JSON(200, response)
            return
        }
        var user User
        db.First(&user, id)
        if user.Id > 0 {
            response := ResponseData{
                Code:    200,
                Status:  "success",
                Message: "",
                Data:    user,
            }
            c.JSON(200, response)
        } else {
            response := ResponseData{
                Code:    50001,
                Status:  "error",
                Message: "用户不存在",
                Data:    "",
            }
            c.JSON(200, response)
        }
    }