zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Feature add template version (#1152)

Browse Source
This commit is contained in:
anqiansong
2021-10-29 09:55:41 +08:00
committed by GitHub
parent bd26783b33
commit 9c48e9ceab
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/goctl/model/sql/gen/insert.go
View File

@@ -13,6 +13,8 @@ import (
func genInsert(table Table, withCache, postgreSql bool) (string, string, error) { func genInsert(table Table, withCache, postgreSql bool) (string, string, error) {
keySet := collection.NewSet() keySet := collection.NewSet()
keyVariableSet := collection.NewSet() keyVariableSet := collection.NewSet()
keySet.AddStr(table.PrimaryCacheKey.DataKeyExpression)
keyVariableSet.AddStr(table.PrimaryCacheKey.KeyLeft)
for _, key := range table.UniqueCacheKey { for _, key := range table.UniqueCacheKey {
keySet.AddStr(key.DataKeyExpression) keySet.AddStr(key.DataKeyExpression)
keyVariableSet.AddStr(key.KeyLeft) keyVariableSet.AddStr(key.KeyLeft)

3
tools/goctl/util/file.go
View File

@@ -10,6 +10,7 @@ import (
"strings" "strings"
"github.com/logrusorgru/aurora" "github.com/logrusorgru/aurora"
"github.com/tal-tech/go-zero/tools/goctl/internal/version"
) )
// NL defines a new line // NL defines a new line
@@ -88,7 +89,7 @@ func GetTemplateDir(category string) (string, error) {
return "", err return "", err
} }
return filepath.Join(goctlHome, category), nil return filepath.Join(goctlHome, version.GetGoctlVersion(), category), nil
} }
// InitTemplates creates template files GoctlHome where could get it by GetGoctlHome // InitTemplates creates template files GoctlHome where could get it by GetGoctlHome