From 0a399af5ae2b8d95c3e4ea3c5b66a11e04c5a1f4 Mon Sep 17 00:00:00 2001 From: Hongchao Deng Date: Wed, 7 Feb 2018 15:06:14 -0800 Subject: [PATCH] generator: generate workqueue --- pkg/generator/generator.go | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/pkg/generator/generator.go b/pkg/generator/generator.go index b8c6cf1e11..ecddabd930 100644 --- a/pkg/generator/generator.go +++ b/pkg/generator/generator.go @@ -59,7 +59,11 @@ func (g *Generator) Render() error { return err } - err = g.genWorkqueue(controllerDir) + err = g.genController(controllerDir, "controller.go", templates.ControllerTemplate) + if err != nil { + return err + } + err = g.genController(controllerDir, "workqueue.go", templates.WorkqueueTemplate) if err != nil { return err } @@ -67,14 +71,14 @@ func (g *Generator) Render() error { return nil } -func (g *Generator) genWorkqueue(controllerDir string) error { - f, err := os.OpenFile(filepath.Join(controllerDir, "controller.go"), os.O_WRONLY|os.O_CREATE|os.O_TRUNC, defaultFileMode) +func (g *Generator) genController(controllerDir, filename, tmpl string) error { + f, err := os.OpenFile(filepath.Join(controllerDir, filename), os.O_WRONLY|os.O_CREATE|os.O_TRUNC, defaultFileMode) if err != nil { return err } defer f.Close() - t, err := template.New("controller").Parse(templates.ControllerTemplate) + t, err := template.New(filename).Parse(tmpl) if err != nil { return err } @@ -82,5 +86,6 @@ func (g *Generator) genWorkqueue(controllerDir string) error { if err != nil { return err } + return nil }