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)
	}
}