You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.6 KiB
Go
68 lines
2.6 KiB
Go
package gen
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"text/template"
|
|
|
|
sqltemplate "github.com/tal-tech/go-zero/tools/goctl/model/sql/template"
|
|
)
|
|
|
|
func genFineOneByField(table *InnerTable) (string, error) {
|
|
t, err := template.New("findOneByField").Parse(sqltemplate.FindOneByField)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
list := make([]string, 0)
|
|
for _, field := range table.Fields {
|
|
if field.IsPrimaryKey {
|
|
continue
|
|
}
|
|
if field.QueryType != QueryOne {
|
|
continue
|
|
}
|
|
fineOneByFieldBuffer := new(bytes.Buffer)
|
|
upperFields := make([]string, 0)
|
|
in := make([]string, 0)
|
|
expressionFields := make([]string, 0)
|
|
expressionValuesFields := make([]string, 0)
|
|
upperFields = append(upperFields, field.UpperCamelCase)
|
|
in = append(in, field.LowerCamelCase+" "+field.DataType)
|
|
expressionFields = append(expressionFields, field.SnakeCase+" = ?")
|
|
expressionValuesFields = append(expressionValuesFields, field.LowerCamelCase)
|
|
for _, withField := range field.WithFields {
|
|
upperFields = append(upperFields, withField.UpperCamelCase)
|
|
in = append(in, withField.LowerCamelCase+" "+withField.DataType)
|
|
expressionFields = append(expressionFields, withField.SnakeCase+" = ?")
|
|
expressionValuesFields = append(expressionValuesFields, withField.LowerCamelCase)
|
|
}
|
|
err = t.Execute(fineOneByFieldBuffer, map[string]interface{}{
|
|
"in": strings.Join(in, ","),
|
|
"upperObject": table.UpperCamelCase,
|
|
"upperFields": strings.Join(upperFields, "And"),
|
|
"onlyOneFiled": len(field.WithFields) == 0,
|
|
"withCache": field.Cache,
|
|
"containsCache": table.ContainsCache,
|
|
"lowerObject": table.LowerCamelCase,
|
|
"lowerField": field.LowerCamelCase,
|
|
"snakeField": field.SnakeCase,
|
|
"lowerPrimaryKey": table.PrimaryField.LowerCamelCase,
|
|
"UpperPrimaryKey": table.PrimaryField.UpperCamelCase,
|
|
"primaryKeyDefine": table.CacheKey[table.PrimaryField.SnakeCase].Define,
|
|
"primarySnakeField": table.PrimaryField.SnakeCase,
|
|
"primaryDataType": table.PrimaryField.DataType,
|
|
"primaryDataTypeString": table.PrimaryField.DataType == "string",
|
|
"upperObjectKey": table.PrimaryField.UpperCamelCase,
|
|
"cacheKey": table.CacheKey[field.SnakeCase].Key,
|
|
"cacheKeyVariable": table.CacheKey[field.SnakeCase].KeyVariable,
|
|
"expression": strings.Join(expressionFields, " AND "),
|
|
"expressionValues": strings.Join(expressionValuesFields, ", "),
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
list = append(list, fineOneByFieldBuffer.String())
|
|
}
|
|
return strings.Join(list, ""), nil
|
|
}
|