Refactor generate package

PelicanPlatform · Dec 1, 2023 · 8c9ac17 · 8c9ac17
1 parent 4998245
commit 8c9ac17
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/generate/main.go b/generate/main.go
@@ -0,0 +1,13 @@
+package main
+
+//go:generate go run ../generate
+
+// This should not be included in any release of pelican
+
+// Include more generator functions here but keep them encapsulated
+// in their separate files under `generate` package
+func main() {
+	GenParamEnum()
+	GenParamStruct()
+	GenPlaceholderPathForNext()
+}
diff --git a/generate/next_generator.go b/generate/next_generator.go
@@ -1,5 +1,7 @@
 package main
 
+// This should not be included in any release of pelican
+
 import (
 	"log"
 	"os"

diff --git a/generate/param_generator.go b/generate/param_generator.go
@@ -25,12 +25,6 @@ type TemplateData struct {
 	GeneratedCode string
 }
 
-func main() {
-	GenParamEnum()
-	GenParamStruct()
-	GenPlaceholderPathForNext()
-}
-
 var requiredKeys = [3]string{"description", "default", "type"}
 
 func GenParamEnum() {

diff --git a/param/param.go b/param/param.go
@@ -7,8 +7,6 @@ import (
 	"github.com/spf13/viper"
 )
 
-//go:generate go run ../generate
-
 var (
 	viperConfig *config
 	configMutex sync.RWMutex