gin+gorm+logrus实现增删改查项目源码

新学golang，呕心沥血小几天，转载请务必附上本文连接。

可能有些地方写的不是很好，请在评论区指教一二～

package main

import (
    "fmt"
    "github.com/gin-gonic/gin"
    _ "github.com/go-sql-driver/mysql"
    "github.com/jinzhu/gorm"
    "github.com/sirupsen/logrus"
    "net/http"
    "os"
    "strconv"
    "time"
)

//对应的实体类
type AppConf struct {
    Id      int    `gorm:"type:bigint(20);column:id;primary_key;AUTO_INCREMENT"`
    App_Id   int    `gorm:"type:bigint(20);column:app_id;not null"`
    Key     string `gorm:"type:varchar(255);column:key;not null"`
    Content string `gorm:"type:varchar(1500);column:content;not null"`
    Ctime   int    `gorm:"type:bigint(20);column:ctime;not null"`
    Utime   int    `gorm:"type:bigint(20);column:utime;not null"`
}

//声明全局数据库连接变量
var db *gorm.DB
var log = logrus.New()
func main() {
    defer db.Close()
    engine := gin.Default()
    engine.Use(LoggerToFile())
    routerGroup := engine.Group("/test")

    //查询所有数据
    routerGroup.GET("/appConfs", ListAll)

    //增加数据
    routerGroup.POST("/appConfs", Add)

    //删除数据
    routerGroup.DELETE("/appConfs/:id", Delete)

    //更新数据
    routerGroup.PATCH("/appConfs/:id", Update)
    engine.Run(":9999")

}

//初始化，创建连接
func init() {
    //连接mysql
    db, _ = gorm.Open("mysql", "root:123456(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local")
    LoggerToFile()
    //单对单
    db.SingularTable(true)
    //设置数据库连接池
    db.DB().SetMaxIdleConns(5)
    db.DB().SetMaxOpenConns(10)
}

//查询所有用户信息
func ListAll(context *gin.Context) {
    var (
        //定义数据表条数
        count int
        //为count传参
        appConfCount[] AppConf
    )

    //获取指定表的数据总条数
    err := db.Model(&appConfCount).Count(&count)
    if err != nil {
        context.JSON(http.StatusOK,gin.H{
            "code":401,
            "status":"error",
            "message":"空表 ！",
            "data":"null",
        })
    }
    //创建一个长度为0，容量为count的切片，使其避免扩容。    当时没用gorm，需要存到集合里，才会有扩容。下述不会
    var appConfs = make([]AppConf, 0, count)

    //查询数据表 app_conf
    rowsAffected:= db.Model(&AppConf{}).Find(&appConfs).RowsAffected
    if rowsAffected > 0 {

        context.JSON(http.StatusOK,gin.H{
            "code":200,
            "status":"success",
            "data":appConfs,
        })
    }else {
        context.JSON(http.StatusOK,gin.H{
            "code":401,
            "status":"error",
            "message":"空表 ！",
            "data":"null",
        })
    }
}

//新增数据
func Add(context *gin.Context) {
    var appConf AppConf

    //绑定结构体
    context.Bind(&appConf)

    fmt.Println(appConf)
    //若id一样，则添加失败
    rowsAffected := db.Model(&AppConf{}).Create(&appConf).RowsAffected
    if rowsAffected > 0 {
        context.JSON(http.StatusOK,gin.H{
            "code":200,
            "status":"success",
            "message":"添加成功 ！",
        })
    }else {
        context.JSON(http.StatusOK, gin.H{
            "code":401,
            "status":"error",
            "message": "添加失败",
        })
    }
}

//根据id删除数据
func Delete(context *gin.Context) {
    id := context.Param("id")
    //sting装int
    s,_ := strconv.Atoi(id)
    rowsAffected := db.Model(&AppConf{}).Where(&AppConf{Id: s}).Delete(&AppConf{}).RowsAffected

    //fmt.Println(rowsAffected)
    if rowsAffected > 0 {
        context.JSON(http.StatusOK,gin.H{
            "code":200,
            "status":"success",
            "message":"删除成功",
        })
    } else {
        context.JSON(http.StatusOK,gin.H{
            "code":401,
            "status":"error",
            "message":"删除失败",
        })
    }
}

//根据id修改信息
func Update(context *gin.Context) {
    var appConf AppConf
    id := context.Param("id")
    s,_ := strconv.Atoi(id)
    //绑定结构体
    context.Bind(&appConf)
    rowsAffected := db.Model(&AppConf{}).Where(&AppConf{Id: s}).Update(&appConf).RowsAffected
    if rowsAffected > 0 {
        context.JSON(http.StatusOK,gin.H{
            "code":200,
            "status":"success",
            "message":"更新成功",
        })
    } else {
        context.JSON(http.StatusOK,gin.H{
            "code":401,
            "status":"error",
            "message":"更新失败",
        })
    }
}


func LoggerToFile() gin.HandlerFunc {

    fileName := "./logger.log"
    //写入文件
    src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
    if err != nil {
        fmt.Println("err", err)
    }

    //实例化
    logger := logrus.New()

    //设置输出
    logger.Out = src

    //设置日志级别
    logger.SetLevel(logrus.DebugLevel)

    //处理日期格式

    logger.SetFormatter(&logrus.TextFormatter{
        TimestampFormat:"2006-01-01 15:04:05",
    })

    //设置日志格式
logger.SetFormatter(&logrus.JSONFormatter{})

    return func(c *gin.Context) {
        // 开始时间
        startTime := time.Now()
        // 处理请求
        c.Next()
        // 结束时间
        endTime := time.Now()
        // 执行时间
        latencyTime := endTime.Sub(startTime)
        latencyTime2 := float32(latencyTime) / float32(1000000)
        // 请求方式
        reqMethod := c.Request.Method
        // 请求路由
        reqUri := c.Request.RequestURI
        // 状态码
        statusCode := c.Writer.Status()
        // 请求IP
        clientIP := c.Request.Host
        // 日志格式
        logger.WithFields(logrus.Fields{
            "status_code"  : statusCode,
            "latency_time" : latencyTime2,
            "client_ip"    : clientIP,
            "req_method"   : reqMethod,
            "req_uri"      : reqUri,
        }).Info()
    }
}