增加template insertBatch支持

Author: xfali

parsing/template/dynamic.go

@@ -19,6 +19,10 @@ func dummyWhere(b bool, cond, column string, value interface{}, origin string) s
 	return origin
 }
 
+func commonAdd(a, b int) int {
+	return a + b
+}
+
 func mysqlUpdateSet(b bool, column string, value interface{}, origin string) string {
 	if !b {
 		return origin
@@ -138,16 +142,19 @@ func postgresWhere(b bool, cond, column string, value interface{}, origin string
 var mysqlFuncMap = template.FuncMap{
 	"set":   mysqlUpdateSet,
 	"where": mysqlWhere,
+	"add":   commonAdd,
 }
 
 var postgresFuncMap = template.FuncMap{
 	"set":   postgresUpdateSet,
 	"where": postgresWhere,
+	"add":   commonAdd,
 }
 
 var dummyFuncMap = template.FuncMap{
 	"set":   dummyUpdateSet,
 	"where": dummyWhere,
+	"add":   commonAdd,
 }
 
 var funcMap = map[string]template.FuncMap{

test/postgresql/postgresql.tpl

@@ -14,6 +14,14 @@ INSERT INTO test_table ({{$COLUMNS}})
   )
 {{end}}
 
+{{define "insertBatchTestTable"}}
+{{$size := len . | add -1}}
+INSERT INTO "test_table"("id","username","password")
+VALUES {{range $i, $v := .}}
+({{$v.Id}}, '{{$v.Username}}', '{{$v.Password}}'){{if lt $i $size}},{{end}}
+{{end}}
+{{end}}
+
 {{define "updateTestTable"}}
 UPDATE test_table
 {{set (ne .Username "") "username" .Username "" | set (ne .Password "") "password" .Password}}

test/postgresql/session_test.go

@@ -30,6 +30,18 @@ func TestSessionTpl(t *testing.T) {
 		t.Log(ret)
 	})
 
+	t.Run("insertBatch", func(t *testing.T) {
+		mgr := gobatis.NewSessionManager(connect())
+		sess := mgr.NewSession()
+		var ret int
+		err := sess.Insert("insertBatchTestTable").Param([]TestTable{
+			{Id: 13, Username: "user13", Password: "pw13"},
+			{Id: 14, Username: "user14", Password: "pw14"},
+		}).Result(&ret)
+		t.Log(err)
+		t.Log(ret)
+	})
+
 	t.Run("update", func(t *testing.T) {
 		mgr := gobatis.NewSessionManager(connect())
 		sess := mgr.NewSession()

test/template/sql.tpl

@@ -13,6 +13,15 @@ INSERT INTO `TEST_TABLE` ({{$COLUMNS}})
   )
 {{end}}
 
+{{define "insertBatchTestTable"}}
+{{$COLUMNS := "id, username, password"}}
+{{$size := len . | add -1}}
+INSERT INTO test_table ({{$COLUMNS}})
+  VALUES {{range $i, $v := .}}
+  ({{$v.Id}}, '{{$v.UserName}}', '{{$v.Password}}'){{if lt $i $size}},{{end}}
+  {{end}}
+{{end}}
+
 {{define "updateTestTable"}}
 UPDATE `TEST_TABLE`
   {{set (ne .UserName "") "username" .UserName "" | set (ne .Password "") "password" .Password | set (ne .Status -1) "status" .Status}}

test/template/temp_test.go

@@ -152,6 +152,22 @@ func TestParser2(t *testing.T) {
 		t.Log(md)
 	})
 
+	t.Run("insertBatch", func(t *testing.T) {
+		p, ok := gobatis.FindTemplateSqlParser("insertBatchTestTable")
+		if !ok {
+			t.Fatal(ok)
+		}
+		md, err := p.ParseMetadata("mysql",[]TestTable{
+			{Id: 11, UserName: "user11", Password: "pw11"},
+			{Id: 12, UserName: "user12", Password: "pw12"},
+		})
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		t.Log(md)
+	})
+
 	t.Run("update", func(t *testing.T) {
 		p, ok := gobatis.FindTemplateSqlParser("updateTestTable")
 		if !ok {