自己做记录，方便查看
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)
}
}
作者：小小程序员Eric
链接：https://www.jianshu.com/p/149e4abb6def
随时随地看视频
