优化生成模板

lazygophers · Jun 11, 2024 · 8d16d17 · 8d16d17
1 parent 98d277e
commit 8d16d17
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/codegen/resource.go b/codegen/resource.go
@@ -280,7 +280,7 @@ var DefaultTemplateFunc = template.FuncMap{
 	"PluckInt64":  anyx.PluckInt64,
 	"PluckUint64": anyx.PluckUint64,
 
-	"SliceEmpty": func(ss []string) bool {
+	"StringSliceEmpty": func(ss []string) bool {
 		return len(ss) == 0
 	},
 

diff --git a/codegen/template/rpc_route.gtpl b/codegen/template/rpc_route.gtpl
@@ -1,8 +1,8 @@
 package main
 
-import (
+import ({{ if gt (len .RPCS) 0 }}
 	"github.com/lazygophers/{{ $.PB.GoPackageName }}"
-	"github.com/lazygophers/{{ $.PB.GoPackageName }}/internal/impl"
+	"github.com/lazygophers/{{ $.PB.GoPackageName }}/internal/impl" {{ end }}
 	"github.com/lazygophers/lrpc"
 )