d21d770b5b
* reactor sql generation * reactor sql generation * add console & example * optimize unit test & add document * modify default config * remove test file * Revert "remove test file" This reverts commit 81041f9e * fix stringx.go & optimize example * remove unused code
52 lines
1.9 KiB
Go
52 lines
1.9 KiB
Go
package gen
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/model/sql/parser"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util/stringx"
|
|
)
|
|
|
|
type (
|
|
// tableName:user
|
|
// {{prefix}}=cache
|
|
// key:id
|
|
Key struct {
|
|
VarExpression string // cacheUserIdPrefix="cache#user#id#"
|
|
Left string // cacheUserIdPrefix
|
|
Right string // cache#user#id#
|
|
Variable string // userIdKey
|
|
KeyExpression string // userIdKey: = fmt.Sprintf("cache#user#id#%v", userId)
|
|
DataKeyExpression string // userIdKey: = fmt.Sprintf("cache#user#id#%v", data.userId)
|
|
RespKeyExpression string // userIdKey: = fmt.Sprintf("cache#user#id#%v", resp.userId)
|
|
}
|
|
)
|
|
|
|
// key-数据库原始字段名,value-缓存key相关数据
|
|
func genCacheKeys(table parser.Table) (map[string]Key, error) {
|
|
fields := table.Fields
|
|
m := make(map[string]Key)
|
|
camelTableName := table.Name.Snake2Camel()
|
|
lowerStartCamelTableName := stringx.From(camelTableName).LowerStart()
|
|
for _, field := range fields {
|
|
if !field.IsKey {
|
|
continue
|
|
}
|
|
camelFieldName := field.Name.Snake2Camel()
|
|
lowerStartCamelFieldName := stringx.From(camelFieldName).LowerStart()
|
|
left := fmt.Sprintf("cache%s%sPrefix", camelTableName, camelFieldName)
|
|
right := fmt.Sprintf("cache#%s#%s#", camelTableName, lowerStartCamelFieldName)
|
|
variable := fmt.Sprintf("%s%sKey", lowerStartCamelTableName, camelFieldName)
|
|
m[field.Name.Source()] = Key{
|
|
VarExpression: fmt.Sprintf(`%s = "%s"`, left, right),
|
|
Left: left,
|
|
Right: right,
|
|
Variable: variable,
|
|
KeyExpression: fmt.Sprintf(`%s := fmt.Sprintf("%s%s", %s,%s)`, variable, "%s", "%v", left, lowerStartCamelFieldName),
|
|
DataKeyExpression: fmt.Sprintf(`%s := fmt.Sprintf("%s%s",%s, data.%s)`, variable, "%s", "%v", left, camelFieldName),
|
|
RespKeyExpression: fmt.Sprintf(`%s := fmt.Sprintf("%s%s", %s,resp.%s)`, variable, "%s", "%v", left, camelFieldName),
|
|
}
|
|
}
|
|
return m, nil
|
|
}
|