go kit （11） DB

本文已参加「新人创作礼」活动，一起开启掘金创作之路。

kit 工具包的功能以及应用基本OK了 我今天完善一下数据库连接的封装吧，项目中肯定是要用到的，以mysql为例

我使用的是： "github.com/jinzhu/gorm" 这个东西看个人习惯，用哪个不是最重要的，只要能正常使用的就可以

开始： 链接字符串 "root:123456@tcp(127.0.0.1:3306)/test") 这个都没问题吧

剩下的就是一些基本设置了 目前我是设置了一下我想用到的一些设置，其他的设置可以再进行设置

package global

import (
	"fmt"
	"github.com/jinzhu/gorm"
	"github.com/sirupsen/logrus"
	"goKit/model"
	"log"
	"os"
	"path/filepath"
	"time"
)

var DB *gorm.DB

func InitDB(conf *Config) error {

	connStr := fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		conf.DB.UserName,   /* 用户名称 */
		conf.DB.UserPasswd, /* 用户密码 */
		conf.DB.DbIP,       /* 数据库地址 */
		conf.DB.DbPort,     /* 数据库对外接口 */
		conf.DB.DbName) /* 连接数据库名称 */

	// 连接数据库
	DB, err := gorm.Open("mysql", connStr)
	if err != nil {
		return err
	}

	// 自动创建表时，表明不加S
	DB.SingularTable(true)

	// 获取sql 连接
	sqlDB := DB.DB()

	// 设置 sql 连接属性
	// 空闲模式中与数据库最大连接数
	sqlDB.SetMaxIdleConns(50)
	// 设置与数据库的最大连接数
	sqlDB.SetMaxOpenConns(1000)
	// 最长连接时间
	sqlDB.SetConnMaxLifetime(time.Minute)

	// 设置表的属性
	// 创建记录
	DB.Callback().Create().Replace("gorm:update_time_stamp", updateTimeStampForCreateCallback)
	// 更新记录
	DB.Callback().Update().Replace("gorm:update_time_stamp", updateTimeStampForUpdateCallback)
	// 开启日志记录
	DB.LogMode(conf.DB.LogModel)

	// 日志输出到文件
	dbLog, err := createLogFile(conf.DB.DbLogDir, "db")
	if err != nil {
		return err
	}
	// 打印到文件
	DB.SetLogger(log.New(dbLog, "\r\n", 0))

	// 创建或更新表结构
	if false {
		createAndUpdateTable()
	}
	// 数据库第一次运行时的数据初始化
	if true {
	}
	return nil

}

func updateTimeStampForCreateCallback(scope *gorm.Scope) {
	if !scope.HasError() {
		nowTime := time.Now().Unix()

		if createdAtField, ok := scope.FieldByName("CreatedAt"); ok {
			if createdAtField.IsBlank {
				_ = createdAtField.Set(nowTime)
			}
		}

		if updatedAtField, ok := scope.FieldByName("UpdatedAt"); ok {
			if updatedAtField.IsBlank {
				_ = updatedAtField.Set(nowTime)
			}
		}
	}
}

func updateTimeStampForUpdateCallback(scope *gorm.Scope) {
	if _, ok := scope.Get("gorm:update_column"); !ok {
		_ = scope.SetColumn("UpdatedAt", time.Now().Unix())
	}
}
func createLogFile(logDir string, flag string) (*os.File, error) {
	timeStr := time.Now().Format("2006-01-02.")
	timeStr = timeStr + flag + ".log"
	fileName := filepath.Join(logDir, timeStr)
	fmt.Printf("%s log Path is: %s", flag, fileName)
	f, err := os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
	return f, err
}
func createAndUpdateTable() {
	logrus.Infof("正在更新数据库表结构!!")
	DB.AutoMigrate(
		&model.User{},
	)
}

新增：代码地址： gitee.com/weishunuan_…