🎉 初始化项目

server/initialize/internal/gorm.go

@@ -0,0 +1,31 @@
+package internal
+
+import (
+	"time"
+
+	"git.echol.cn/loser/st/server/config"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+)
+
+var Gorm = new(_gorm)
+
+type _gorm struct{}
+
+// Config gorm 自定义配置
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (g *_gorm) Config(general config.GeneralDB) *gorm.Config {
+	return &gorm.Config{
+		Logger: logger.New(NewWriter(general), logger.Config{
+			SlowThreshold: 200 * time.Millisecond,
+			LogLevel:      general.LogLevel(),
+			Colorful:      true,
+		}),
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   general.Prefix,
+			SingularTable: general.Singular,
+		},
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
+}

server/initialize/internal/gorm_logger_writer.go

@@ -0,0 +1,41 @@
+package internal
+
+import (
+	"fmt"
+	"git.echol.cn/loser/st/server/config"
+	"git.echol.cn/loser/st/server/global"
+	"gorm.io/gorm/logger"
+)
+
+type Writer struct {
+	config config.GeneralDB
+	writer logger.Writer
+}
+
+func NewWriter(config config.GeneralDB) *Writer {
+	return &Writer{config: config}
+}
+
+// Printf 格式化打印日志
+func (c *Writer) Printf(message string, data ...any) {
+
+	// 当有日志时候均需要输出到控制台
+	fmt.Printf(message, data...)
+
+	// 当开启了zap的情况，会打印到日志记录
+	if c.config.LogZap {
+		switch c.config.LogLevel() {
+		case logger.Silent:
+			global.GVA_LOG.Debug(fmt.Sprintf(message, data...))
+		case logger.Error:
+			global.GVA_LOG.Error(fmt.Sprintf(message, data...))
+		case logger.Warn:
+			global.GVA_LOG.Warn(fmt.Sprintf(message, data...))
+		case logger.Info:
+			global.GVA_LOG.Info(fmt.Sprintf(message, data...))
+		default:
+			global.GVA_LOG.Info(fmt.Sprintf(message, data...))
+		}
+		return
+	}
+}

server/initialize/internal/mongo.go

@@ -0,0 +1,29 @@
+package internal
+
+import (
+	"context"
+	"fmt"
+	"github.com/qiniu/qmgo/options"
+	"go.mongodb.org/mongo-driver/event"
+	opt "go.mongodb.org/mongo-driver/mongo/options"
+	"go.uber.org/zap"
+)
+
+var Mongo = new(mongo)
+
+type mongo struct{}
+
+func (m *mongo) GetClientOptions() []options.ClientOptions {
+	cmdMonitor := &event.CommandMonitor{
+		Started: func(ctx context.Context, event *event.CommandStartedEvent) {
+			zap.L().Info(fmt.Sprintf("[MongoDB][RequestID:%d][database:%s] %s\n", event.RequestID, event.DatabaseName, event.Command), zap.String("business", "mongo"))
+		},
+		Succeeded: func(ctx context.Context, event *event.CommandSucceededEvent) {
+			zap.L().Info(fmt.Sprintf("[MongoDB][RequestID:%d] [%s] %s\n", event.RequestID, event.Duration.String(), event.Reply), zap.String("business", "mongo"))
+		},
+		Failed: func(ctx context.Context, event *event.CommandFailedEvent) {
+			zap.L().Error(fmt.Sprintf("[MongoDB][RequestID:%d] [%s] %s\n", event.RequestID, event.Duration.String(), event.Failure), zap.String("business", "mongo"))
+		},
+	}
+	return []options.ClientOptions{{ClientOptions: &opt.ClientOptions{Monitor: cmdMonitor}}}
+}