✨ Init

initialize/internal/gorm.go

@@ -0,0 +1,62 @@
+package internal
+
+import (
+	"gorm.io/gorm/schema"
+	"log"
+	"os"
+	"time"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"miniapp/global"
+)
+
+type DBBASE interface {
+	GetLogMode() string
+}
+
+var Gorm = new(_gorm)
+
+type _gorm struct{}
+
+// Config gorm 自定义配置
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (g *_gorm) Config(prefix string, singular bool) *gorm.Config {
+	config := &gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   prefix,
+			SingularTable: singular,
+		},
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
+	_default := logger.New(log.New(os.Stdout, "\r\n", log.LstdFlags), logger.Config{
+		SlowThreshold: 200 * time.Millisecond,
+		LogLevel:      logger.Warn,
+		Colorful:      true,
+	})
+	var logMode DBBASE
+	switch global.GVA_CONFIG.System.DbType {
+	case "mysql":
+		logMode = &global.GVA_CONFIG.Mysql
+	case "pgsql":
+		logMode = &global.GVA_CONFIG.Pgsql
+	case "oracle":
+		logMode = &global.GVA_CONFIG.Oracle
+	default:
+		logMode = &global.GVA_CONFIG.Mysql
+	}
+
+	switch logMode.GetLogMode() {
+	case "silent", "Silent":
+		config.Logger = _default.LogMode(logger.Silent)
+	case "error", "Error":
+		config.Logger = _default.LogMode(logger.Error)
+	case "warn", "Warn":
+		config.Logger = _default.LogMode(logger.Warn)
+	case "info", "Info":
+		config.Logger = _default.LogMode(logger.Info)
+	default:
+		config.Logger = _default.LogMode(logger.Info)
+	}
+	return config
+}

initialize/internal/logger.go

@@ -0,0 +1,35 @@
+package internal
+
+import (
+	"fmt"
+
+	"gorm.io/gorm/logger"
+	"miniapp/global"
+)
+
+type writer struct {
+	logger.Writer
+}
+
+// NewWriter writer 构造函数
+// Author [SliverHorn](https://github.com/SliverHorn)
+func NewWriter(w logger.Writer) *writer {
+	return &writer{Writer: w}
+}
+
+// Printf 格式化打印日志
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (w *writer) Printf(message string, data ...interface{}) {
+	var logZap bool
+	switch global.GVA_CONFIG.System.DbType {
+	case "mysql":
+		logZap = global.GVA_CONFIG.Mysql.LogZap
+	case "pgsql":
+		logZap = global.GVA_CONFIG.Pgsql.LogZap
+	}
+	if logZap {
+		global.GVA_LOG.Info(fmt.Sprintf(message+"\n", data...))
+	} else {
+		w.Writer.Printf(message, data...)
+	}
+}