Adding code generation library changes

Signed-off-by: Christopher Hein <me@christopherhein.com>
amazon-archives · Nov 28, 2018 · 049f785 · 049f785
1 parent d1d2eef
commit 049f785
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 23 deletions.
diff --git a/code-generation/pkg/codegen/assets/base.go.templ b/code-generation/pkg/codegen/assets/base.go.templ
@@ -3,28 +3,37 @@ package base
 import (
   "context"
   "github.com/awslabs/aws-service-operator/pkg/config"
+  "github.com/awslabs/aws-service-operator/pkg/queuemanager"
   {{- range $index, $element := .Items}}
   "github.com/awslabs/aws-service-operator/pkg/operators/{{$element.Spec.Resource.Name}}"
   {{- end}}
 )
 
 type base struct {
   config *config.Config
+  queueManager *queuemanager.QueueManager
+  {{- range $index, $element := .Items}}
+  {{$element.Spec.Resource.Name}} *{{$element.Spec.Resource.Name}}.Operator
+  {{- end}}
 }
 
 func New(
   config *config.Config,
+  queueManager *queuemanager.QueueManager,
 ) *base {
   return &base{
     config: config,
+    queueManager: queueManager,
+    {{- range $index, $element := .Items}}
+    {{$element.Spec.Resource.Name}}: {{$element.Spec.Resource.Name}}.NewOperator(config, queueManager),
+    {{- end}}
   }
 }
 
 func (b *base) Watch(ctx context.Context, namespace string) {
   {{- range $index, $element := .Items}}
   if b.config.Resources["{{$element.Spec.Resource.Name}}"] {
-    {{$element.Spec.Resource.Name}}operator := {{$element.Spec.Resource.Name}}.NewOperator(b.config)
-  	go {{$element.Spec.Resource.Name}}operator.StartWatch(ctx, namespace)
+  	go b.{{$element.Spec.Resource.Name}}.StartWatch(ctx, namespace)
   }
 	{{- end}}
 

diff --git a/code-generation/pkg/codegen/assets/operator.go.templ b/code-generation/pkg/codegen/assets/operator.go.templ
@@ -14,12 +14,13 @@ import (
 	{{- end}}
 
 	"github.com/awslabs/aws-service-operator/pkg/config"
-	{{- if .Spec.Queue}}
-    "github.com/awslabs/aws-service-operator/pkg/queue"
-    corev1 "k8s.io/api/core/v1"
-    "github.com/iancoleman/strcase"
-    "strings"
-		awsclient "github.com/awslabs/aws-service-operator/pkg/client/clientset/versioned/typed/service-operator.aws/v1alpha1"
+  "github.com/awslabs/aws-service-operator/pkg/queuemanager"
+  {{- if .Spec.Queue}}
+  corev1 "k8s.io/api/core/v1"
+  "github.com/iancoleman/strcase"
+  "strings"
+	awsclient "github.com/awslabs/aws-service-operator/pkg/client/clientset/versioned/typed/service-operator.aws/v1alpha1"
+	"github.com/awslabs/aws-service-operator/pkg/queue"
 	{{- end}}
 	"github.com/awslabs/aws-service-operator/pkg/operator"
 	"k8s.io/client-go/tools/cache"
@@ -34,12 +35,23 @@ import (
 type Operator struct {
 	config       *config.Config
 	topicARN     string
+	queueManager *queuemanager.QueueManager
 }
 
 // NewOperator create controller for watching object store custom resources created
-func NewOperator(config *config.Config) *Operator {
+func NewOperator(config *config.Config, queueManager *queuemanager.QueueManager) *Operator {
+	{{- if .Spec.Queue}}
+	queuectrl := queue.New(config, config.AWSClientset, 10)
+	topicARN, _ := queuectrl.Register("{{.Spec.Resource.Name}}")
+	queueManager.Add(topicARN, queuemanager.HandlerFunc(QueueUpdater))
+	{{- end}}
+
 	return &Operator{
 		config:       config,
+		{{- if .Spec.Queue}}
+		topicARN: topicARN,
+		{{- end}}
+		queueManager: queueManager,
 	}
 }
 
@@ -50,19 +62,14 @@ func (c *Operator) StartWatch(ctx context.Context, namespace string) {
 		UpdateFunc: c.onUpdate,
 		DeleteFunc: c.onDelete,
 	}
-	{{- if .Spec.Queue}}
-	queuectrl := queue.New(c.config, c.config.AWSClientset, 1)
-	c.topicARN, _, _, _ = queuectrl.Register("{{.Spec.Resource.Name}}", &awsV1alpha1.{{.Spec.Kind}}{})
-	go queuectrl.StartWatch(queue.HandlerFunc(QueueUpdater), ctx.Done())
-	{{- end}}
 
 	oper := operator.New("{{.Spec.Resource.Plural}}", namespace, resourceHandlers, c.config.AWSClientset.RESTClient())
 	oper.Watch(&awsV1alpha1.{{.Spec.Kind}}{}, ctx.Done())
 }
 
 {{- if .Spec.Queue}}
 // QueueUpdater will take the messages from the queue and process them
-func QueueUpdater(config *config.Config, msg *queue.MessageBody) error {
+func QueueUpdater(config *config.Config, msg *queuemanager.MessageBody) error {
 	logger := config.Logger
 	var name, namespace string
 	if msg.Updatable {