Add WithDialect Option

ego-component · Dec 12, 2023 · dd1ab7c · dd1ab7c
1 parent 50d6e90
commit dd1ab7c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/option.go b/option.go
@@ -1,8 +1,9 @@
 package egorm
 
 import (
-	"github.com/ego-component/egorm/manager"
 	"gorm.io/gorm/schema"
+
+	"github.com/ego-component/egorm/manager"
 )
 
 // Option 可选项
@@ -15,6 +16,13 @@ func WithDSN(dsn string) Option {
 	}
 }
 
+// WithDialect 设置 dialect
+func WithDialect(dialect string) Option {
+	return func(c *Container) {
+		c.config.Dialect = dialect
+	}
+}
+
 // WithDSNParser 设置自定义dsnParser
 func WithDSNParser(parser manager.DSNParser) Option {
 	return func(c *Container) {