Merge pull request #2 from m3dev/feature-add-basic-features

Add basic features

Dockerfile

@@ -14,7 +14,7 @@ RUN go mod download
 # Copy the go source
 COPY cmd/main.go cmd/main.go
 COPY api/ api/
-COPY internal/controller/ internal/controller/
+COPY internal/ internal/
 
 # Build
 # the GOARCH has not a default value to allow the binary be built according to the host where the command

api/v1alpha1/broom_types.go

@@ -17,19 +17,79 @@ limitations under the License.
 package v1alpha1
 
 import (
+	"fmt"
+	"strconv"
+
+	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
 // EDIT THIS FILE!  THIS IS SCAFFOLDING FOR YOU TO OWN!
 // NOTE: json tags are required.  Any new fields you add must have json tags for the fields to be serialized.
 
+type BroomTarget struct {
+	Name      string            `json:"name,omitempty"`
+	Labels    map[string]string `json:"labels,omitempty"`
+	Namespace string            `json:"namespace,omitempty"`
+}
+
+type BroomAdjustmentType string
+
+const (
+	AddAdjustment BroomAdjustmentType = "Add"
+	MulAdjustment BroomAdjustmentType = "Mul"
+)
+
+type BroomAdjustment struct {
+	Type  BroomAdjustmentType `json:"type"`
+	Value string              `json:"value"`
+}
+
+func (adj BroomAdjustment) IncreaseMemory(m *resource.Quantity) error {
+	switch adj.Type {
+	case AddAdjustment:
+		y, err := resource.ParseQuantity(adj.Value)
+		if err != nil {
+			return fmt.Errorf("unable to parse value to resource.Quantity: %w", err)
+		}
+		m.Add(y)
+	case MulAdjustment:
+		y, err := strconv.Atoi(adj.Value)
+		if err != nil {
+			return fmt.Errorf("unable to parse value to int: %w", err)
+		}
+		m.Mul(int64(y))
+	}
+	return nil
+}
+
+type BroomRestartPolicy string
+
+const (
+	RestartOnOOMPolicy BroomRestartPolicy = "OnOOM"
+	RestartNeverPolicy BroomRestartPolicy = "Never"
+)
+
+type BroomSlackWebhookSecret struct {
+	Namespace string `json:"namespace"`
+	Name      string `json:"name"`
+	Key       string `json:"key"`
+}
+
+type BroomSlackWebhook struct {
+	Secret  BroomSlackWebhookSecret `json:"secret"`
+	Channel string                  `json:"channel,omitempty"`
+}
+
 // BroomSpec defines the desired state of Broom
 type BroomSpec struct {
 	// INSERT ADDITIONAL SPEC FIELDS - desired state of cluster
 	// Important: Run "make" to regenerate code after modifying this file
 
-	// Foo is an example field of Broom. Edit broom_types.go to remove/update
-	Foo string `json:"foo,omitempty"`
+	Target        BroomTarget        `json:"target,omitempty"`
+	Adjustment    BroomAdjustment    `json:"adjustment"`
+	RestartPolicy BroomRestartPolicy `json:"restartPolicy"`
+	SlackWebhook  BroomSlackWebhook  `json:"slackWebhook"`
 }
 
 // BroomStatus defines the observed state of Broom

cmd/main.go

@@ -122,10 +122,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	if err = (&controller.BroomReconciler{
-		Client: mgr.GetClient(),
-		Scheme: mgr.GetScheme(),
-	}).SetupWithManager(mgr); err != nil {
+	if err = controller.New(mgr).SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "Broom")
 		os.Exit(1)
 	}

config/crd/bases/ai.m3.com_brooms.yaml

@@ -39,10 +39,53 @@ spec:
           spec:
             description: BroomSpec defines the desired state of Broom
             properties:
-              foo:
-                description: Foo is an example field of Broom. Edit broom_types.go
-                  to remove/update
+              adjustment:
+                properties:
+                  type:
+                    type: string
+                  value:
+                    type: string
+                required:
+                - type
+                - value
+                type: object
+              restartPolicy:
                 type: string
+              slackWebhook:
+                properties:
+                  channel:
+                    type: string
+                  secret:
+                    properties:
+                      key:
+                        type: string
+                      name:
+                        type: string
+                      namespace:
+                        type: string
+                    required:
+                    - key
+                    - name
+                    - namespace
+                    type: object
+                required:
+                - secret
+                type: object
+              target:
+                properties:
+                  labels:
+                    additionalProperties:
+                      type: string
+                    type: object
+                  name:
+                    type: string
+                  namespace:
+                    type: string
+                type: object
+            required:
+            - adjustment
+            - restartPolicy
+            - slackWebhook
             type: object
           status:
             description: BroomStatus defines the observed state of Broom

config/manager/kustomization.yaml

@@ -1,2 +1,8 @@
 resources:
 - manager.yaml
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+images:
+- name: controller
+  newName: asia-northeast1-docker.pkg.dev/m3ai-ullman-dev/arbok/broom
+  newTag: latest

config/rbac/role.yaml

@@ -4,6 +4,34 @@ kind: ClusterRole
 metadata:
   name: manager-role
 rules:
+- apiGroups:
+  - ""
+  resources:
+  - events
+  verbs:
+  - get
+  - list
+  - watch
+- apiGroups:
+  - ""
+  resources:
+  - pods
+  verbs:
+  - create
+  - delete
+  - get
+  - list
+  - patch
+  - update
+  - watch
+- apiGroups:
+  - ""
+  resources:
+  - secrets
+  verbs:
+  - get
+  - list
+  - watch
 - apiGroups:
   - ai.m3.com
   resources:
@@ -30,3 +58,27 @@ rules:
   - get
   - patch
   - update
+- apiGroups:
+  - batch
+  resources:
+  - cronjobs
+  verbs:
+  - create
+  - delete
+  - get
+  - list
+  - patch
+  - update
+  - watch
+- apiGroups:
+  - batch
+  resources:
+  - jobs
+  verbs:
+  - create
+  - delete
+  - get
+  - list
+  - patch
+  - update
+  - watch

config/samples/ai_v1alpha1_broom.yaml

@@ -9,4 +9,18 @@ metadata:
     app.kubernetes.io/created-by: broom
   name: broom-sample
 spec:
-  # TODO(user): Add fields here
+  target:
+    # name: oom-sample
+    labels:
+      m3.com/use-broom: "true"
+    # namespace: broom
+  adjustment:
+    type: Mul
+    value: "2"
+  restartPolicy: "OnOOM"
+  slackWebhook:
+    secret:
+      namespace: default
+      name: broom
+      key: SLACK_WEBHOOK_URL
+    # channel: "#alert"

config/samples/cronjob.yaml

@@ -0,0 +1,34 @@
+apiVersion: batch/v1
+kind: CronJob
+metadata:
+  name: oom-sample
+  labels:
+    m3.com/use-broom: "true"
+spec:
+  schedule: "*/2 * * * *"
+  jobTemplate:
+    spec:
+      activeDeadlineSeconds: 60
+      backoffLimit: 1
+      template:
+        spec:
+          containers:
+            - name: oom-container
+              image: ubuntu:latest
+              command:
+                - /bin/bash
+                - -c
+              args:
+                - |
+                  echo PID=$$
+                  for i in {0..9}
+                  do
+                    eval a$i'=$(head --bytes 5000000 /dev/zero | cat -v)'
+                    echo $((i++));
+                  done
+              resources:
+                limits:
+                  memory: "100Mi"
+                requests:
+                  memory: "50Mi"
+          restartPolicy: Never

config/samples/secret.yaml

@@ -0,0 +1,7 @@
+apiVersion: v1
+kind: Secret
+metadata:
+  name: broom
+type: Opaque
+data:
+  SLACK_WEBHOOK_URL: TO_BE_SPECIFIED