提交

buildSql.go

@@ -0,0 +1,38 @@
+package gormx
+
+import (
+	"context"
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+// 根据map组装gorm的where
+func BuildWhere(ctx context.Context, m *gorm.DB, where map[string]interface{}) (*gorm.DB, error) {
+	for key, val := range where {
+		switch val.(type) {
+		case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
+			m = m.Where(fmt.Sprintf("%s = ?", key), val)
+		case string:
+			m = m.Where(fmt.Sprintf("%s = ?", key), val)
+		case float32, float64:
+			m = m.Where(fmt.Sprintf("%s = ?", key), val)
+		case bool:
+			m = m.Where(fmt.Sprintf("%s = ?", key), val)
+		case []int, []int8, []int16, []int32, []int64, []uint, []uint8, []uint16, []uint32, []uint64:
+			m = m.Where(fmt.Sprintf("%s in (?)", key), val)
+		case []string:
+			m = m.Where(fmt.Sprintf("%s in (?)", key), val)
+		case []float32, []float64:
+			m = m.Where(fmt.Sprintf("%s in (?)", key), val)
+		case []bool:
+			m = m.Where(fmt.Sprintf("%s in (?)", key), val)
+		case []interface{}:
+			m = m.Where(fmt.Sprintf("%s in (?)", key), val)
+		default:
+			// byte,rune,any,nil
+			return nil, fmt.Errorf("unknown type:%+v %T", val, val)
+		}
+	}
+	return m, nil
+}

gormx.go

@@ -0,0 +1,27 @@
+package gormx
+
+import (
+	"context"
+	"fmt"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
+)
+
+func NewGorm(prefix, user, password, host, database string, port int) *gorm.DB {
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, password, host, port, database)
+
+	g, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   prefix, // 表名前缀，`Article` 的表名应该是 `it_articles`
+			SingularTable: true,   // 使用单数表名，启用该选项，此时，`Article` 的表名应该是 `it_article`
+		},
+		Logger: NewGormxLogger(context.TODO()),
+	})
+	if err != nil {
+		panic(err)
+	}
+
+	return g
+}

logger.go

@@ -0,0 +1,49 @@
+package gormx
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"gorm.io/gorm/logger"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+// TODO:待进一步封装
+
+type GormxLogger struct {
+	logx.Logger
+}
+
+func NewGormxLogger(ctx context.Context) *GormxLogger {
+	return &GormxLogger{
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (g *GormxLogger) LogMode(LogLevel logger.LogLevel) logger.Interface {
+	return g
+}
+
+func (g *GormxLogger) Info(ctx context.Context, msg string, val ...interface{}) {
+	fmt.Println("info", msg, val)
+	g.Logger.Info(msg, val)
+}
+
+func (g *GormxLogger) Warn(ctx context.Context, msg string, val ...interface{}) {
+	fmt.Println("warn", msg, val)
+	g.Logger.Info(msg, val)
+}
+
+func (g *GormxLogger) Error(ctx context.Context, msg string, val ...interface{}) {
+	fmt.Println("error", msg, val)
+	g.Logger.Error(msg, val)
+}
+
+func (g *GormxLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
+
+	sql, rows := fc()
+	fmt.Printf("trace:  begin:%+v, err:%+v, sql:%+v, rows:%+v\n", begin, err, sql, rows)
+
+}