增加register参数，自动生成注册mapper代码（默认为false）。

由于gobatis已增加ScanMapperDir方法，可通过此方法注册整个目录的mapper文件，故不再需要依次注册。

Author: xfali

cmd/gobatis-cmd/main.go

@@ -39,6 +39,7 @@ func main() {
     keyword := flag.Bool("keyword", false, "with Keyword escape")
     namespace := flag.String("namespace", "", "namespace")
     confFile := flag.String("f", "", "config file")
+    register := flag.Bool("register", false, "add register code")
     flag.Parse()
 
     dbDriver := db.GetDriver(*driver)
@@ -65,6 +66,7 @@ func main() {
         conf.MapperFile = *mapper
         conf.Plugin = *plugin
         conf.Keyword = *keyword
+        conf.Register = *register
         conf.TableName = *tableName
         conf.DBName = *dbName
         conf.Host = *host

configs/gobatis-conf.json

@@ -8,6 +8,7 @@
   "mapperFile": "xml",
   "plugin": "",
   "keyword": false,
+  "register": false,
   "tableName": "",
   "dbName": "test_db",
   "host": "localhost",

internal/pkg/generator/gen_proxy_v2.go

@@ -70,18 +70,20 @@ func GenV2Proxy(config config.Config, tableName string, models []common.ModelInf
         builder.WriteString("gobatis.RegisterModel(&modelV)")
         builder.WriteString(common.Newline())
 
-        if config.MapperFile == "xml" {
-            builder.WriteString(common.ColumnSpace())
-            builder.WriteString(fmt.Sprintf("gobatis.RegisterMapperFile(\"%sxml/%s_mapper.xml\")", config.Path, strings.ToLower(tableName)))
-            builder.WriteString(common.Newline())
-        } else if config.MapperFile == "go" {
-            builder.WriteString(common.ColumnSpace())
-            builder.WriteString(fmt.Sprintf("gobatis.RegisterMapperData([]byte(%sMapper))", modelName))
-            builder.WriteString(common.Newline())
-        } else if config.MapperFile == "template" {
-            builder.WriteString(common.ColumnSpace())
-            builder.WriteString(fmt.Sprintf("gobatis.RegisterTemplateFile(\"%stemplate/%s_mapper.tmpl\")", config.Path, strings.ToLower(tableName)))
-            builder.WriteString(common.Newline())
+        if config.Register {
+            if config.MapperFile == "xml" {
+                builder.WriteString(common.ColumnSpace())
+                builder.WriteString(fmt.Sprintf("gobatis.RegisterMapperFile(\"%sxml/%s_mapper.xml\")", config.Path, strings.ToLower(tableName)))
+                builder.WriteString(common.Newline())
+            } else if config.MapperFile == "go" {
+                builder.WriteString(common.ColumnSpace())
+                builder.WriteString(fmt.Sprintf("gobatis.RegisterMapperData([]byte(%sMapper))", modelName))
+                builder.WriteString(common.Newline())
+            } else if config.MapperFile == "template" {
+                builder.WriteString(common.ColumnSpace())
+                builder.WriteString(fmt.Sprintf("gobatis.RegisterTemplateFile(\"%stemplate/%s_mapper.tmpl\")", config.Path, strings.ToLower(tableName)))
+                builder.WriteString(common.Newline())
+            }
         }
 
         builder.WriteString("}")

pkg/config/config.go

@@ -15,6 +15,7 @@ type Config struct {
     MapperFile  string `json:"mapperFile"`
     Plugin      string `json:"plugin"`
     Keyword     bool   `json:"keyword"`
+    Register    bool   `json:"register"`
 }
 
 type FileConfig struct {