增加批量插入生成代码

Author: xfali

gen_proxy_v2.go

@@ -154,6 +154,35 @@ func genV2Proxy(config Config, tableName string, models []common.ModelInfo) {
         builder.WriteString(common.Newline())
         //insert end
 
+        //insert batch
+        builder.WriteString(fmt.Sprintf("func InsertBatch%s(sess *gobatis.Session, models []%s) (int64, int64, error) {", modelName, modelName))
+        builder.WriteString(common.Newline())
+
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString("var ret int64")
+        builder.WriteString(common.Newline())
+
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(fmt.Sprintf(`runner := sess.Insert("insertBatch%s").Param(models)`, modelName))
+        builder.WriteString(common.Newline())
+
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(`err := runner.Result(&ret)`)
+        builder.WriteString(common.Newline())
+
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString(`id := runner.LastInsertId()`)
+        builder.WriteString(common.Newline())
+
+        builder.WriteString(common.ColumnSpace())
+        builder.WriteString("return ret, id, err")
+        builder.WriteString(common.Newline())
+
+        builder.WriteString("}")
+        builder.WriteString(common.Newline())
+        builder.WriteString(common.Newline())
+        //insert batch end
+
         //update
         builder.WriteString(fmt.Sprintf("func Update%s(sess *gobatis.Session, model %s) (int64, error) {", modelName, modelName))
         builder.WriteString(common.Newline())

gen_xml.go

@@ -180,6 +180,48 @@ func buildMapper(builder *strings.Builder, config Config, tableName string, mode
     builder.WriteString(common.Newline())
     //insert end
 
+    //insert batch
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(fmt.Sprintf("<insert id=\"insertBatch%s\">", modelName))
+    builder.WriteString(common.Newline())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(fmt.Sprintf("INSERT INTO %s (%s)", tableName, columns))
+    builder.WriteString(common.Newline())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString("VALUES")
+    builder.WriteString(common.Newline())
+
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(`<foreach item="item" index="index" collection="{0}" open="" separator="," close="">`)
+    builder.WriteString(common.Newline())
+
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString("(")
+    for i := range model {
+        builder.WriteString(fmt.Sprintf("#{item.%s}", common.Column2DynamicName(modelName, model[i].ColumnName)))
+        if i < len(model)-1 {
+            builder.WriteString(",")
+        }
+    }
+    builder.WriteString(")")
+    builder.WriteString(common.Newline())
+
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString(`</foreach>`)
+    builder.WriteString(common.Newline())
+
+    builder.WriteString(common.ColumnSpace())
+    builder.WriteString("</insert>")
+    builder.WriteString(common.Newline())
+    builder.WriteString(common.Newline())
+    //insert end
+
     //update
     builder.WriteString(common.ColumnSpace())
     builder.WriteString(fmt.Sprintf("<update id=\"update%s\">", modelName))