添加模板概念测试

ElementType.go

@@ -30,3 +30,12 @@ const (
 	Element_where     ElementType = "where"
 	Element_Include   ElementType = "include"
 )
+
+func isMethodElement(tag ElementType) bool {
+	switch tag {
+	case Element_Insert, Element_Delete, Element_Update, Element_Select,
+		Element_Insert_Templete, Element_Delete_Templete, Element_Update_Templete, Element_Select_Templete:
+		return true
+	}
+	return false
+}

GoMybatis.go

@@ -42,7 +42,7 @@ func WriteMapperPtrByEngine(ptr interface{}, xml []byte, sessionEngine SessionEn
 func WriteMapper(bean reflect.Value, xml []byte, sessionFactory *SessionFactory, templeteDecoder TempleteDecoder, decoder SqlResultDecoder, sqlBuilder SqlBuilder, enableLog bool) {
 	beanCheck(bean, sqlBuilder)
 	var mapperTree = LoadMapperXml(xml)
-	templeteDecoder.DecodeTree(mapperTree)
+	templeteDecoder.DecodeTree(mapperTree, bean.Type())
 	//make a map[method]xml
 	var methodXmlMap = makeMethodXmlMap(bean, mapperTree)
 	var resultMaps = makeResultMaps(mapperTree)

GoMybatisTempleteDecoder.go

@@ -2,31 +2,43 @@ package GoMybatis
 
 import (
 	"bytes"
-	"errors"
 	"github.com/zhuxiujia/GoMybatis/utils"
+	"reflect"
 	"strings"
 )
 
 var equalOperator = []string{"/", "+", "-", "*", "**", "|", "^", "&", "%", "<", ">", ">=", "<=", " in ", " not in ", " or ", "||", " and ", "&&", "==", "!="}
 
 type GoMybatisTempleteDecoder struct {
-	tree map[string]*MapperXml
 }
 
-func (it *GoMybatisTempleteDecoder) DecodeTree(tree map[string]*MapperXml) error {
+func (it *GoMybatisTempleteDecoder) DecodeTree(tree map[string]*MapperXml, beanType reflect.Type) error {
 	if tree == nil {
-		return errors.New("data cant be nil!")
+		return utils.NewError("GoMybatisTempleteDecoder", "decode data map[string]*MapperXml cant be nil!")
 	}
-	it.tree = tree
-	for _, v := range it.tree {
-		it.Decode(v)
+	if beanType.Kind() == reflect.Ptr {
+		beanType = beanType.Elem()
+	}
+
+	for _, v := range tree {
+		var method *reflect.StructField
+		if isMethodElement(v.Tag) {
+			var upperId = utils.UpperFieldFirstName(v.Id)
+			m, haveMethod := beanType.FieldByName(upperId)
+			if haveMethod {
+				method = &m
+			}
+		}
+		it.Decode(method, v, tree)
 	}
 	return nil
 }
 
-func (it *GoMybatisTempleteDecoder) Decode(mapper *MapperXml) error {
+func (it *GoMybatisTempleteDecoder) Decode(method *reflect.StructField, mapper *MapperXml, tree map[string]*MapperXml) error {
+
+	switch mapper.Tag {
 
-	if mapper.Tag == "selectTemplete" {
+	case "selectTemplete":
 		mapper.Tag = Element_Select
 
 		var tables = mapper.Propertys["tables"]
@@ -50,7 +62,9 @@ func (it *GoMybatisTempleteDecoder) Decode(mapper *MapperXml) error {
 			sql.Reset()
 			it.DecodeWheres(wheres, mapper)
 		}
-	} else if mapper.Tag == "insertTemplete" {
+		break
+
+	case "insertTemplete":
 		mapper.Tag = Element_Insert
 
 		var tables = mapper.Propertys["tables"]
@@ -64,7 +78,7 @@ func (it *GoMybatisTempleteDecoder) Decode(mapper *MapperXml) error {
 			inserts = "*?*"
 		}
 
-		var resultMapData = it.tree[resultMap]
+		var resultMapData = tree[resultMap]
 		if resultMapData == nil {
 			panic(utils.NewError("GoMybatisTempleteDecoder", "resultMap not define! id = ", resultMap))
 		}
@@ -132,6 +146,7 @@ func (it *GoMybatisTempleteDecoder) Decode(mapper *MapperXml) error {
 			}
 		}
 		mapper.ElementItems = append(mapper.ElementItems, trimArg)
+		break
 	}
 
 	return nil

TempleteDecoder.go

@@ -1,6 +1,7 @@
 package GoMybatis
 
+import "reflect"
+
 type TempleteDecoder interface {
-	Decode(mapper *MapperXml) error
-	DecodeTree(tree map[string]*MapperXml) error
+	DecodeTree(tree map[string]*MapperXml, beanType reflect.Type) error
 }

example/Example_ActivityMapper.xml

@@ -17,6 +17,7 @@
         <result column="delete_flag" property="deleteFlag" langType="int"  enable_logic_delete="true"  logic_deleted_value="1" logic_undelete_value="0" />
     </resultMap>
 
+    <!--Templete模板标签可以省略id="",框架默认设置<insertTemplete id="insertTemplete"/>-->
     <insertTemplete tables="biz_activity"></insertTemplete>
     <selectTemplete tables="biz_activity" wheres="name?name = #{name},name?name = #{name},name?name = #{name}"></selectTemplete>
     <!--updateTemplete,sets="*"更新全部内容,sets="*?*"判空参数，然后设置对应参数.-->