增加mapper文件扫描方法

Author: xfali

sqlmanager.go

@@ -13,6 +13,8 @@ import (
 	"github.com/xfali/gobatis/parsing/sqlparser"
 	"github.com/xfali/gobatis/parsing/template"
 	"github.com/xfali/gobatis/parsing/xml"
+	"os"
+	"path/filepath"
 )
 
 type sqlManager struct {
@@ -65,4 +67,31 @@ func DynamicParserFactory(sql string) (sqlparser.SqlParser, error) {
 
 func TemplateParserFactory(sql string) (sqlparser.SqlParser, error) {
 	return template.CreateParser([]byte(sql))
+}
+
+func ScanMapperFile(dir string) error {
+	return filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			filename := filepath.Base(path)
+			length := len(filename)
+			if length > 4 {
+				if filename[length-4:] == ".xml" {
+					err := RegisterMapperFile(path)
+					if err != nil {
+						return err
+					}
+				}
+				if filename[length-4:] == ".tpl" {
+					err := RegisterTemplateFile(path)
+					if err != nil {
+						return err
+					}
+				}
+			}
+		}
+		return nil
+	})
 }

test/sqlmanager/manager_test.go

@@ -0,0 +1,18 @@
+// Copyright (C) 2019-2020, Xiongfa Li.
+// @author xiongfa.li
+// @version V1.0
+// Description: 
+
+package sqlmanager
+
+import (
+    "github.com/xfali/gobatis"
+    "testing"
+)
+
+func TestManager(t *testing.T) {
+    err := gobatis.ScanMapperFile("./x")
+    if err != nil {
+        t.Fatal(err)
+    }
+}

test/sqlmanager/x/tpl/x.tpl

@@ -0,0 +1,36 @@
+{{define "namespace"}}test{{end}}
+
+{{define "selectTestTable"}}
+SELECT id,username,password,createtime FROM test_table
+{{where .Id "AND" "id = " (arg .Id) "" | where .Username "AND" "username = " (arg .Username) | where .Password "AND" "password = " (arg .Password) | where .Createtime "AND" "createtime = " (arg .Createtime)}}
+{{end}}
+
+{{define "selectTestTableCount"}}
+SELECT COUNT(*) FROM test_table
+{{where .Id "AND" "id = " (arg .Id) "" | where .Username "AND" "username = " (arg .Username) | where .Password "AND" "password = " (arg .Password) | where .Createtime "AND" "createtime = " (arg .Createtime)}}
+{{end}}
+
+{{define "insertTestTable"}}
+INSERT INTO test_table(id,username,password,createtime)
+VALUES(
+{{arg .Id}}, {{arg .Username}}, {{arg .Password}}, {{arg .Createtime}})
+{{end}}
+
+{{define "insertBatchTestTable"}}
+{{$size := len . | add -1}}
+INSERT INTO test_table(id,username,password,createtime)
+VALUES {{range $i, $v := .}}
+({{arg $v.Id}}, {{arg $v.Username}}, {{arg $v.Password}}, {{arg $v.Createtime}}){{if lt $i $size}},{{end}}
+{{end}}
+{{end}}
+
+{{define "updateTestTable"}}
+UPDATE test_table
+{{set .Id "id = " (arg .Id) "" | set .Username "username = " (arg .Username) | set .Password "password = " (arg .Password) | set .Createtime "createtime = " (arg .Createtime)}}
+{{where .Id "AND" "id = " (arg .Id) ""}}
+{{end}}
+
+{{define "deleteTestTable"}}
+DELETE FROM test_table
+{{where .Id "AND" "id = " (arg .Id) "" | where .Username "AND" "username = " (arg .Username) | where .Password "AND" "password = " (arg .Password) | where .Createtime "AND" "createtime = " (arg .Createtime)}}
+{{end}}

test/sqlmanager/x/xml/x.xml

@@ -0,0 +1,61 @@
+<mapper namespace="test">
+    <sql id="columns_id">id,username,password,createtime</sql>
+
+    <select id="selectTestTable">
+        SELECT <include refid="columns_id"> </include> FROM test_table
+        <where>
+            <if test="{TestTable.id} != nil and {TestTable.id} != 0">AND id = #{TestTable.id} </if>
+            <if test="{TestTable.username} != nil">AND username = #{TestTable.username} </if>
+            <if test="{TestTable.password} != nil">AND password = #{TestTable.password} </if>
+            <if test="{TestTable.createtime} != nil">AND createtime = #{TestTable.createtime} </if>
+        </where>
+    </select>
+
+    <select id="selectTestTableCount">
+        SELECT COUNT(*) FROM test_table
+        <where>
+            <if test="{TestTable.id} != nil and {TestTable.id} != 0">AND id = #{TestTable.id} </if>
+            <if test="{TestTable.username} != nil">AND username = #{TestTable.username} </if>
+            <if test="{TestTable.password} != nil">AND password = #{TestTable.password} </if>
+            <if test="{TestTable.createtime} != nil">AND createtime = #{TestTable.createtime} </if>
+        </where>
+    </select>
+
+    <insert id="insertTestTable">
+        INSERT INTO test_table (id,username,password,createtime)
+        VALUES(
+        #{TestTable.id},
+        #{TestTable.username},
+        #{TestTable.password},
+        #{TestTable.createtime}
+        )
+    </insert>
+
+    <insert id="insertBatchTestTable">
+        INSERT INTO test_table (id,username,password,createtime)
+        VALUES
+        <foreach item="item" index="index" collection="{0}" open="" separator="," close="">
+            (#{item.TestTable.id},#{item.TestTable.username},#{item.TestTable.password},#{item.TestTable.createtime})
+        </foreach>
+    </insert>
+
+    <update id="updateTestTable">
+        UPDATE test_table
+        <set>
+            <if test="{TestTable.username} != nil"> username = #{TestTable.username} </if>
+            <if test="{TestTable.password} != nil"> password = #{TestTable.password} </if>
+            <if test="{TestTable.createtime} != nil"> createtime = #{TestTable.createtime} </if>
+        </set>
+        WHERE id = #{TestTable.id}
+    </update>
+
+    <delete id="deleteTestTable">
+        DELETE FROM test_table
+        <where>
+            <if test="{TestTable.id} != nil and {TestTable.id} != 0">AND id = #{TestTable.id} </if>
+            <if test="{TestTable.username} != nil">AND username = #{TestTable.username} </if>
+            <if test="{TestTable.password} != nil">AND password = #{TestTable.password} </if>
+            <if test="{TestTable.createtime} != nil">AND createtime = #{TestTable.createtime} </if>
+        </where>
+    </delete>
+</mapper>