Explorar el Código

Feature add template version (#1152)

anqiansong hace 3 años
padre
commit
9c48e9ceab
Se han modificado 2 ficheros con 4 adiciones y 1 borrados
  1. 2 0
      tools/goctl/model/sql/gen/insert.go
  2. 2 1
      tools/goctl/util/file.go

+ 2 - 0
tools/goctl/model/sql/gen/insert.go

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

+ 2 - 1
tools/goctl/util/file.go

@@ -10,6 +10,7 @@ import (
 	"strings"
 
 	"github.com/logrusorgru/aurora"
+	"github.com/tal-tech/go-zero/tools/goctl/internal/version"
 )
 
 // NL defines a new line
@@ -88,7 +89,7 @@ func GetTemplateDir(category string) (string, error) {
 		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