refactoring hash function to global template funcs

Signed-off-by: Chris Hein <me@chrishein.com>
kubernetes-sigs · Nov 11, 2020 · f50f78b · f50f78b
1 parent c158f4f
commit f50f78b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 41 deletions.
diff --git a/pkg/model/file/funcmap.go b/pkg/model/file/funcmap.go
@@ -17,6 +17,8 @@ limitations under the License.
 package file
 
 import (
+	"fmt"
+	"hash/fnv"
 	"strings"
 	"text/template"
 )
@@ -26,5 +28,15 @@ func DefaultFuncMap() template.FuncMap {
 	return template.FuncMap{
 		"title": strings.Title,
 		"lower": strings.ToLower,
+		"hash":  hash,
 	}
 }
+
+// hash will generate a random string useful for generating a unique string
+func hash(s string) (string, error) {
+	hasher := fnv.New32a()
+	if _, err := hasher.Write([]byte(s)); err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("%x", hasher.Sum(nil)), nil
+}
diff --git a/pkg/plugin/v2/scaffolds/internal/templates/main.go b/pkg/plugin/v2/scaffolds/internal/templates/main.go
@@ -18,17 +18,14 @@ package templates
 
 import (
 	"fmt"
-	"hash/fnv"
 	"path/filepath"
-	"text/template"
 
 	"sigs.k8s.io/kubebuilder/v2/pkg/model/file"
 )
 
 const defaultMainPath = "main.go"
 
 var _ file.Template = &Main{}
-var _ file.UseCustomFuncMap = &Main{}
 
 // Main scaffolds a file that defines the controller manager entry point
 type Main struct {
@@ -53,21 +50,6 @@ func (f *Main) SetTemplateDefaults() error {
 	return nil
 }
 
-func hash(s string) (string, error) {
-	hasher := fnv.New32a()
-	if _, err := hasher.Write([]byte(s)); err != nil {
-		return "", err
-	}
-	return fmt.Sprintf("%x", hasher.Sum(nil)), nil
-}
-
-// GetFuncMap implements file.UseCustomFuncMap
-func (f *Main) GetFuncMap() template.FuncMap {
-	fm := file.DefaultFuncMap()
-	fm["hash"] = hash
-	return fm
-}
-
 var _ file.Inserter = &MainUpdater{}
 
 // MainUpdater updates main.go to run Controllers
@@ -241,13 +223,13 @@ func main() {
 		"Enabling this will ensure there is only one active controller manager.")
 	flag.Parse()
 
-	ctrl.SetLogger(zap.New(zap.UseDevMode(true))) 
+	ctrl.SetLogger(zap.New(zap.UseDevMode(true)))
 
 	mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
 		Scheme:             scheme,
 		MetricsBindAddress: metricsAddr,
-		Port:               9443, 
-		LeaderElection:     enableLeaderElection, 
+		Port:               9443,
+		LeaderElection:     enableLeaderElection,
 		LeaderElectionID:   "{{ hash .Repo }}.{{ .Domain }}",
 	})
 	if err != nil {

diff --git a/pkg/plugin/v3/scaffolds/internal/templates/main.go b/pkg/plugin/v3/scaffolds/internal/templates/main.go
@@ -18,17 +18,14 @@ package templates
 
 import (
 	"fmt"
-	"hash/fnv"
 	"path/filepath"
-	"text/template"
 
 	"sigs.k8s.io/kubebuilder/v2/pkg/model/file"
 )
 
 const defaultMainPath = "main.go"
 
 var _ file.Template = &Main{}
-var _ file.UseCustomFuncMap = &Main{}
 
 // Main scaffolds a file that defines the controller manager entry point
 type Main struct {
@@ -53,21 +50,6 @@ func (f *Main) SetTemplateDefaults() error {
 	return nil
 }
 
-func hash(s string) (string, error) {
-	hasher := fnv.New32a()
-	if _, err := hasher.Write([]byte(s)); err != nil {
-		return "", err
-	}
-	return fmt.Sprintf("%x", hasher.Sum(nil)), nil
-}
-
-// GetFuncMap implements file.UseCustomFuncMap
-func (f *Main) GetFuncMap() template.FuncMap {
-	fm := file.DefaultFuncMap()
-	fm["hash"] = hash
-	return fm
-}
-
 var _ file.Inserter = &MainUpdater{}
 
 // MainUpdater updates main.go to run Controllers
@@ -249,8 +231,8 @@ func main() {
 	mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
 		Scheme:             scheme,
 		MetricsBindAddress: metricsAddr,
-		Port:               9443, 
-		LeaderElection:     enableLeaderElection, 
+		Port:               9443,
+		LeaderElection:     enableLeaderElection,
 		LeaderElectionID:   "{{ hash .Repo }}.{{ .Domain }}",
 	})
 	if err != nil {