自动生成model操作数据库方法

Author: xfali

README.md

@@ -56,18 +56,40 @@ gobatis-cmd -driver=mysql -host=localhost -port=3306 -user=test -pw=test -db=tes
 
 例子：
 ```
-package test_package
+package test
 
-import "time"
+import (
+	"github.com/xfali/gobatis"
+	"time"
+)
 
 type TestTable struct {
-    //TableName gobatis.ModelName `test_table`
-    Id int64 `xfield:"id"`
-    Username string `xfield:"username"`
-    Password string `xfield:"password"`
-    UpdateTime time.Time `xfield:"update_time"`
+	//TableName gobatis.ModelName `test_table`
+	Id         int       `xfield:"id"`
+	Username   string    `xfield:"username"`
+	Password   string    `xfield:"password"`
+	Createtime time.Time `xfield:"createtime"`
+}
+
+func (m *TestTable) Select(sess *gobatis.Session) ([]TestTable, error) {
+	return SelectTestTable(sess, *m)
+}
+
+func (m *TestTable) Count(sess *gobatis.Session) (int64, error) {
+	return SelectTestTableCount(sess, *m)
+}
+
+func (m *TestTable) Insert(sess *gobatis.Session) (int64, int64, error) {
+	return InsertTestTable(sess, *m)
 }
 
+func (m *TestTable) Update(sess *gobatis.Session) (int64, error) {
+	return UpdateTestTable(sess, *m)
+}
+
+func (m *TestTable) Delete(sess *gobatis.Session) (int64, error) {
+	return DeleteTestTable(sess, *m)
+}
 ```
 
 ### 2、xml文件

gen_model.go

@@ -54,11 +54,19 @@ func genModel(config Config, tableName string, model []common.ModelInfo) {
             builder.WriteString(common.Newline())
             builder.WriteString(common.Newline())
 
+            builder.WriteString("import (")
+            builder.WriteString(common.Newline())
+            builder.WriteString(common.ColumnSpace())
+            builder.WriteString(`"github.com/xfali/gobatis"`)
+            builder.WriteString(common.Newline())
             if findTime(model) {
-                builder.WriteString("import \"time\"")
-                builder.WriteString(common.Newline())
+                builder.WriteString(common.ColumnSpace())
+                builder.WriteString(`"time"`)
                 builder.WriteString(common.Newline())
             }
+            builder.WriteString(")")
+            builder.WriteString(common.Newline())
+            builder.WriteString(common.Newline())
         }
 
         builder.WriteString("type ")
@@ -82,6 +90,57 @@ func genModel(config Config, tableName string, model []common.ModelInfo) {
         }
         builder.WriteString("}")
         builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+
+        //select
+        builder.WriteString(fmt.Sprintf("func (m *%s) Select(sess *gobatis.Session) ([]%s, error) {", modelName, modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf("return Select%s(sess, *m)", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+
+        //count
+        builder.WriteString(fmt.Sprintf("func (m *%s) Count(sess *gobatis.Session) (int64, error) {", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf("return Select%sCount(sess, *m)", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+
+        //insert
+        builder.WriteString(fmt.Sprintf("func (m *%s) Insert(sess *gobatis.Session) (int64, int64, error) {", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf("return Insert%s(sess, *m)", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+
+        //update
+        builder.WriteString(fmt.Sprintf("func (m *%s) Update(sess *gobatis.Session) (int64, error) {", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf("return Update%s(sess, *m)", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+
+        //delete
+        builder.WriteString(fmt.Sprintf("func (m *%s) Delete(sess *gobatis.Session) (int64, error) {", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf("return Delete%s(sess, *m)", modelName))
+        builder.WriteString(common.Newline())
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
 
         io.Write(modelFile, []byte(builder.String()))
     }