java-operator-plugins/pkg/quarkus/v1alpha/scaffolds/internal/templates/controller/controller.go at 7461902fd90278c1fa7a48ac6d27a5997b7825d9 · jmrodri/java-operator-plugins · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package controller

import (
	"fmt"

	"sigs.k8s.io/kubebuilder/v3/pkg/machinery"

	"github.com/operator-framework/java-operator-plugins/pkg/quarkus/v1alpha/scaffolds/internal/templates/util"
)

var _ machinery.Template = &Controller{}

type Controller struct {
	machinery.TemplateMixin

	// Package is the source files package
	Package string

	// Name of the operator used for the main file.
	ClassName string
}

func (f *Controller) SetTemplateDefaults() error {
	if f.ClassName == "" {
		return fmt.Errorf("invalid model name")
	}

	if f.Path == "" {
		f.Path = util.PrependJavaPath(f.ClassName+"Controller.java", util.AsPath(f.Package))
	}

	f.TemplateBody = controllerTemplate

	return nil
}

// TODO: pass in the name of the operator i.e. replace Memcached
const controllerTemplate = `package {{ .Package }};

import io.fabric8.kubernetes.api.model.*;
import io.fabric8.kubernetes.api.model.apps.Deployment;
import io.fabric8.kubernetes.api.model.apps.DeploymentBuilder;
import io.fabric8.kubernetes.api.model.apps.DeploymentSpecBuilder;
import io.fabric8.kubernetes.client.KubernetesClient;
import io.javaoperatorsdk.operator.api.*;
import io.javaoperatorsdk.operator.api.Context;
import io.javaoperatorsdk.operator.processing.event.EventSourceManager;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import org.apache.commons.collections.CollectionUtils;

@Controller
public class {{ .ClassName }}Controller implements ResourceController<{{ .ClassName }}> {

    private final KubernetesClient client;

    public {{ .ClassName }}Controller(KubernetesClient client) {
        this.client = client;
    }

    // TODO Fill in the rest of the controller

    @Override
    public void init(EventSourceManager eventSourceManager) {
        // TODO: fill in init
    }

    @Override
    public UpdateControl<{{ .ClassName }}> createOrUpdateResource(
        {{ .ClassName }} resource, Context<{{ .ClassName }}> context) {
        // TODO: fill in logic

        return UpdateControl.noUpdate();
    }

    @Override
    public DeleteControl deleteResource({{ .ClassName }} resource, Context<{{ .ClassName }}> context) {
        // nothing to do here...
        // framework takes care of deleting the resource object
        // k8s takes care of deleting deployment and pods because of ownerreference set
        return DeleteControl.DEFAULT_DELETE;
    }
}

`