gin框架学习（二）

0.前言

本系列使用的开发环境如下：

• goland
• go 1.17.1

在本小节中，对上次的代码进行了重构，让整个系统的逻辑性更强，方便后续扩展。

1.结构总览

划分了4个包和一个router控制，具体如下：

• model：里面存一些model，如上节写的user。
• controller：存放主要的逻辑代码。
• common：存放一些通用的代码，目前只有数据库连接。
• middleware：存放一些中间件。目前项目中还没有，后面再补上。

2.model层代码

package model

import "gorm.io/gorm"

type User struct {
gorm.Model
Name string
Telephone string `gorm:"unique;not null"`
Password string `gorm:"not null"`
}

3.数据库链接代码

package common

import (
   "fmt"
   "gorm.io/driver/mysql"
   "gorm.io/gorm"
)

var db *gorm.DB

func InitDb(){
   username:="root"
   password:="123456"
   database:="gin_demo"
   charset:="utf8"
   port:="3306"
   host:="localhost"
   dns:=fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=True",
      username,
      password,
      host,
      port,
      database,
      charset)
   temp,err:=gorm.Open(mysql.Open(dns))
   if err != nil {
      panic("can't connect to the mysql"+err.Error())
   }
   db=temp

}

func GetDB()*gorm.DB{
   return db
}

4.控制层代码

package controller

import (
   "gin_test/common"
   "gin_test/model"
   "github.com/gin-gonic/gin"
   "gorm.io/gorm"
)

func Register(c *gin.Context){
   db:=common.GetDB()

   name:=c.Query("name")
   password:=c.Query("password")
   telephone:=c.Query("telephone")
   if len(telephone)!=11{
      c.JSON(400,gin.H{
         "error":"the telephone is not correct",
      })
      return
   }

   if isTelephoneExist(db,telephone){
      c.JSON(400,gin.H{
         "error":"the telephone is already exit",
      })
      return
   }
   db.Create(&model.User{Name: name,Password: password,Telephone: telephone})
   c.JSON(200,gin.H{
      "msg":"succeed",
   })
   return
}


func isTelephoneExist(db *gorm.DB,telephone string)bool{
   var user model.User
   db.Where("password=?",telephone).First(&user)
   if user.ID!=0{
      return true
   }else{
      return false
   }
}

5.router.go代码

package main

import (
   "gin_test/controller"
   "github.com/gin-gonic/gin"
)

func RouterController(router *gin.Engine)*gin.Engine {
   router.GET("/register",controller.Register)


   return router
}

6.main.go代码

package main

import (
   "gin_test/common"
   "gin_test/model"
   "github.com/gin-gonic/gin"
)


func main(){
   common.InitDb()
   db:=common.GetDB()
   db.AutoMigrate(&model.User{})

   router:=gin.Default()
   router=RouterController(router)
   router.Run()
}

6.总结

本节的内容主要是对项目进行了结构上的划分，为后续的开发奠定了基础，接下来将会完成登录操作，同时在登录中加入中间件验证的权限控制。