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

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传