Browse Source

添加多表前缀的支持

lxg 3 years ago
parent
commit
915a72a259
2 changed files with 5 additions and 3 deletions
  1. 4 2
      entity.go
  2. 1 1
      options.go

+ 4 - 2
entity.go

@@ -721,8 +721,10 @@ func newEntity(model Model, opts *Options) *Entity {
 	entity.lruCache, _ = lru.New(50)
 	entity.reflectType = entity.reflectValue.Type()
 	tableName := model.TableName()
-	if opts.RemoveTablePrefix && opts.TablePrefix != "" {
-		tableName = strings.TrimLeft(tableName, opts.TablePrefix)
+	if opts.RemoveTablePrefix {
+		for _, prefix := range opts.TablePrefixes {
+			tableName = strings.TrimPrefix(tableName, prefix)
+		}
 	}
 	entity.singularName = inflector.Singularize(tableName)
 	entity.pluralizeName = inflector.Pluralize(tableName)

+ 1 - 1
options.go

@@ -10,7 +10,7 @@ type Options struct {
 	EnableNamespace   bool
 	DB                *gorm.DB
 	Prefix            string
-	TablePrefix       string
+	TablePrefixes     []string
 	RemoveTablePrefix bool
 	Formatter         *Formatter
 	Middleware        []http.Middleware