goctl added

2020-07-29 17:11:41 +08:00
parent b1975d29a7
commit 121323b8c3
142 changed files with 10690 additions and 0 deletions
--- a/tools/goctl/model/sql/gen/keys.go
+++ b/tools/goctl/model/sql/gen/keys.go
@@ -0,0 +1,106 @@
+package gen
+
+import (
+	"bytes"
+	"strings"
+	"text/template"
+)
+
+var (
+	cacheKeyExpressionTemplate = `cache{{.upperCamelTable}}{{.upperCamelField}}Prefix = "cache#{{.lowerCamelTable}}#{{.lowerCamelField}}#"`
+	keyTemplate                = `{{.lowerCamelField}}Key := fmt.Sprintf("%s%v", {{.define}}, {{.lowerCamelField}})`
+	keyRespTemplate            = `{{.lowerCamelField}}Key := fmt.Sprintf("%s%v", {{.define}}, resp.{{.upperCamelField}})`
+	keyDataTemplate            = `{{.lowerCamelField}}Key := fmt.Sprintf("%s%v", {{.define}}, data.{{.upperCamelField}})`
+)
+
+type (
+	Key struct {
+		Define      string // cacheKey define,如：cacheUserUserIdPrefix
+		Value       string // cacheKey value expression,如：cache#user#userId#
+		Expression  string // cacheKey expression，如:cacheUserUserIdPrefix="cache#user#userId#"
+		KeyVariable string // cacheKey 声明变量，如：userIdKey
+		Key         string // 缓存key的代码,如 userIdKey:=fmt.Sprintf("%s%v", cacheUserUserIdPrefix, userId)
+		DataKey     string // 缓存key的代码,如 userIdKey:=fmt.Sprintf("%s%v", cacheUserUserIdPrefix, data.userId)
+		RespKey     string // 缓存key的代码,如 userIdKey:=fmt.Sprintf("%s%v", cacheUserUserIdPrefix, resp.userId)
+	}
+)
+
+// key-数据库原始字段名,value-缓存key对象
+func genCacheKeys(table *InnerTable) (map[string]Key, error) {
+	fields := table.Fields
+	var m = make(map[string]Key)
+	if !table.ContainsCache {
+		return m, nil
+	}
+	for _, field := range fields {
+		if !field.Cache && !field.IsPrimaryKey {
+			continue
+		}
+		t, err := template.New("keyExpression").Parse(cacheKeyExpressionTemplate)
+		if err != nil {
+			return nil, err
+		}
+		var expressionBuffer = new(bytes.Buffer)
+		err = t.Execute(expressionBuffer, map[string]string{
+			"upperCamelTable": table.UpperCamelCase,
+			"lowerCamelTable": table.LowerCamelCase,
+			"upperCamelField": field.UpperCamelCase,
+			"lowerCamelField": field.LowerCamelCase,
+		})
+		if err != nil {
+			return nil, err
+		}
+		expression := expressionBuffer.String()
+		expressionAr := strings.Split(expression, "=")
+		define := strings.TrimSpace(expressionAr[0])
+		value := strings.TrimSpace(expressionAr[1])
+		t, err = template.New("key").Parse(keyTemplate)
+		if err != nil {
+			return nil, err
+		}
+		var keyBuffer = new(bytes.Buffer)
+		err = t.Execute(keyBuffer, map[string]string{
+			"lowerCamelField": field.LowerCamelCase,
+			"define":          define,
+		})
+		if err != nil {
+			return nil, err
+		}
+		t, err = template.New("keyData").Parse(keyDataTemplate)
+		if err != nil {
+			return nil, err
+		}
+		var keyDataBuffer = new(bytes.Buffer)
+		err = t.Execute(keyDataBuffer, map[string]string{
+			"lowerCamelField": field.LowerCamelCase,
+			"upperCamelField": field.UpperCamelCase,
+			"define":          define,
+		})
+		if err != nil {
+			return nil, err
+		}
+		t, err = template.New("keyResp").Parse(keyRespTemplate)
+		if err != nil {
+			return nil, err
+		}
+		var keyRespBuffer = new(bytes.Buffer)
+		err = t.Execute(keyRespBuffer, map[string]string{
+			"lowerCamelField": field.LowerCamelCase,
+			"upperCamelField": field.UpperCamelCase,
+			"define":          define,
+		})
+		if err != nil {
+			return nil, err
+		}
+		m[field.SnakeCase] = Key{
+			Define:      define,
+			Value:       value,
+			Expression:  expression,
+			KeyVariable: field.LowerCamelCase + "Key",
+			Key:         keyBuffer.String(),
+			DataKey:     keyDataBuffer.String(),
+			RespKey:     keyRespBuffer.String(),
+		}
+	}
+	return m, nil
+}