feature 1.1.5 (#411)
This commit is contained in:
anqiansong
GitHub
@@ -2,61 +2,163 @@ package gen
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sort"
|
||||
"strings"
|
||||
|
||||
"github.com/tal-tech/go-zero/tools/goctl/model/sql/parser"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/util/stringx"
|
||||
)
|
||||
|
||||
// Key defines cache key variable for generating code
|
||||
// Key describes cache key
|
||||
type Key struct {
|
||||
// VarExpression likes cacheUserIdPrefix = "cache#User#id#"
|
||||
// VarLeft describes the varible of cache key expression which likes cacheUserIdPrefix
|
||||
VarLeft string
|
||||
// VarRight describes the value of cache key expression which likes "cache#user#id#"
|
||||
VarRight string
|
||||
// VarExpression describes the cache key expression which likes cacheUserIdPrefix = "cache#user#id#"
|
||||
VarExpression string
|
||||
// Left likes cacheUserIdPrefix
|
||||
Left string
|
||||
// Right likes cache#user#id#
|
||||
Right string
|
||||
// Variable likes userIdKey
|
||||
Variable string
|
||||
// KeyExpression likes userIdKey: = fmt.Sprintf("cache#user#id#%v", userId)
|
||||
// KeyLeft describes the varible of key definiation expression which likes userKey
|
||||
KeyLeft string
|
||||
// KeyRight describes the value of key definiation expression which likes fmt.Sprintf("%s%v", cacheUserPrefix, user)
|
||||
KeyRight string
|
||||
// DataKeyRight describes data key likes fmt.Sprintf("%s%v", cacheUserPrefix, data.User)
|
||||
DataKeyRight string
|
||||
// KeyExpression describes key expression likes userKey := fmt.Sprintf("%s%v", cacheUserPrefix, user)
|
||||
KeyExpression string
|
||||
// DataKeyExpression likes userIdKey: = fmt.Sprintf("cache#user#id#%v", data.userId)
|
||||
// DataKeyExpression describes data key expression likes userKey := fmt.Sprintf("%s%v", cacheUserPrefix, data.User)
|
||||
DataKeyExpression string
|
||||
// RespKeyExpression likes userIdKey: = fmt.Sprintf("cache#user#id#%v", resp.userId)
|
||||
RespKeyExpression string
|
||||
// FieldNameJoin describes the filed slice of table
|
||||
FieldNameJoin Join
|
||||
// Fields describes the fields of table
|
||||
Fields []*parser.Field
|
||||
}
|
||||
|
||||
// key-数据库原始字段名,value-缓存key相关数据
|
||||
func genCacheKeys(table parser.Table) (map[string]Key, error) {
|
||||
fields := table.Fields
|
||||
m := make(map[string]Key)
|
||||
camelTableName := table.Name.ToCamel()
|
||||
lowerStartCamelTableName := stringx.From(camelTableName).Untitle()
|
||||
for _, field := range fields {
|
||||
if field.IsUniqueKey || field.IsPrimaryKey {
|
||||
camelFieldName := field.Name.ToCamel()
|
||||
lowerStartCamelFieldName := stringx.From(camelFieldName).Untitle()
|
||||
left := fmt.Sprintf("cache%s%sPrefix", camelTableName, camelFieldName)
|
||||
if strings.ToLower(camelFieldName) == strings.ToLower(camelTableName) {
|
||||
left = fmt.Sprintf("cache%sPrefix", camelTableName)
|
||||
}
|
||||
right := fmt.Sprintf("cache#%s#%s#", camelTableName, lowerStartCamelFieldName)
|
||||
variable := fmt.Sprintf("%s%sKey", lowerStartCamelTableName, camelFieldName)
|
||||
if strings.ToLower(lowerStartCamelTableName) == strings.ToLower(camelFieldName) {
|
||||
variable = fmt.Sprintf("%sKey", lowerStartCamelTableName)
|
||||
}
|
||||
// Join describes an alias of string slice
|
||||
type Join []string
|
||||
|
||||
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),
|
||||
}
|
||||
}
|
||||
func genCacheKeys(table parser.Table) (Key, []Key) {
|
||||
var primaryKey Key
|
||||
var uniqueKey []Key
|
||||
primaryKey = genCacheKey(table.Name, []*parser.Field{&table.PrimaryKey.Field})
|
||||
for _, each := range table.UniqueIndex {
|
||||
uniqueKey = append(uniqueKey, genCacheKey(table.Name, each))
|
||||
}
|
||||
sort.Slice(uniqueKey, func(i, j int) bool {
|
||||
return uniqueKey[i].VarLeft < uniqueKey[j].VarLeft
|
||||
})
|
||||
|
||||
return primaryKey, uniqueKey
|
||||
}
|
||||
|
||||
func genCacheKey(table stringx.String, in []*parser.Field) Key {
|
||||
var (
|
||||
varLeftJoin, varRightJon, fieldNameJoin Join
|
||||
varLeft, varRight, varExpression string
|
||||
|
||||
keyLeftJoin, keyRightJoin, keyRightArgJoin, dataRightJoin Join
|
||||
keyLeft, keyRight, dataKeyRight, keyExpression, dataKeyExpression string
|
||||
)
|
||||
|
||||
varLeftJoin = append(varLeftJoin, "cache", table.Source())
|
||||
varRightJon = append(varRightJon, "cache", table.Source())
|
||||
keyLeftJoin = append(keyLeftJoin, table.Source())
|
||||
|
||||
for _, each := range in {
|
||||
varLeftJoin = append(varLeftJoin, each.Name.Source())
|
||||
varRightJon = append(varRightJon, each.Name.Source())
|
||||
keyLeftJoin = append(keyLeftJoin, each.Name.Source())
|
||||
keyRightJoin = append(keyRightJoin, stringx.From(each.Name.ToCamel()).Untitle())
|
||||
keyRightArgJoin = append(keyRightArgJoin, "%v")
|
||||
dataRightJoin = append(dataRightJoin, "data."+each.Name.ToCamel())
|
||||
fieldNameJoin = append(fieldNameJoin, each.Name.Source())
|
||||
}
|
||||
varLeftJoin = append(varLeftJoin, "prefix")
|
||||
keyLeftJoin = append(keyLeftJoin, "key")
|
||||
|
||||
varLeft = varLeftJoin.Camel().With("").Untitle()
|
||||
varRight = fmt.Sprintf(`"%s"`, varRightJon.Camel().Untitle().With("#").Source()+"#")
|
||||
varExpression = fmt.Sprintf(`%s = %s`, varLeft, varRight)
|
||||
|
||||
keyLeft = keyLeftJoin.Camel().With("").Untitle()
|
||||
keyRight = fmt.Sprintf(`fmt.Sprintf("%s%s", %s, %s)`, "%s", keyRightArgJoin.With("").Source(), varLeft, keyRightJoin.With(", ").Source())
|
||||
dataKeyRight = fmt.Sprintf(`fmt.Sprintf("%s%s", %s, %s)`, "%s", keyRightArgJoin.With("").Source(), varLeft, dataRightJoin.With(", ").Source())
|
||||
keyExpression = fmt.Sprintf("%s := %s", keyLeft, keyRight)
|
||||
dataKeyExpression = fmt.Sprintf("%s := %s", keyLeft, dataKeyRight)
|
||||
|
||||
return Key{
|
||||
VarLeft: varLeft,
|
||||
VarRight: varRight,
|
||||
VarExpression: varExpression,
|
||||
KeyLeft: keyLeft,
|
||||
KeyRight: keyRight,
|
||||
DataKeyRight: dataKeyRight,
|
||||
KeyExpression: keyExpression,
|
||||
DataKeyExpression: dataKeyExpression,
|
||||
Fields: in,
|
||||
FieldNameJoin: fieldNameJoin,
|
||||
}
|
||||
}
|
||||
|
||||
// Title convert items into Title and return
|
||||
func (j Join) Title() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).Title())
|
||||
}
|
||||
|
||||
return m, nil
|
||||
return join
|
||||
}
|
||||
|
||||
// Camel convert items into Camel and return
|
||||
func (j Join) Camel() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).ToCamel())
|
||||
}
|
||||
return join
|
||||
}
|
||||
|
||||
// Snake convert items into Snake and return
|
||||
func (j Join) Snake() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).ToSnake())
|
||||
}
|
||||
|
||||
return join
|
||||
}
|
||||
|
||||
// Snake convert items into Untitle and return
|
||||
func (j Join) Untitle() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).Untitle())
|
||||
}
|
||||
|
||||
return join
|
||||
}
|
||||
|
||||
// Upper convert items into Upper and return
|
||||
func (j Join) Upper() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).Upper())
|
||||
}
|
||||
|
||||
return join
|
||||
}
|
||||
|
||||
// Lower convert items into Lower and return
|
||||
func (j Join) Lower() Join {
|
||||
var join Join
|
||||
for _, each := range j {
|
||||
join = append(join, stringx.From(each).Lower())
|
||||
}
|
||||
|
||||
return join
|
||||
}
|
||||
|
||||
// With convert items into With and return
|
||||
func (j Join) With(sep string) stringx.String {
|
||||
return stringx.From(strings.Join(j, sep))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user