-
Notifications
You must be signed in to change notification settings - Fork 0
/
example_group_a.go
84 lines (70 loc) · 2.48 KB
/
example_group_a.go
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
package pkg
import (
"context"
"fmt"
"strings"
api "github.com/Yu-Jack/wrangler-test/apis/example.group.a/v1alpha1"
ega "github.com/Yu-Jack/wrangler-test/generated/controllers/example.group.a"
v1alpha1 "github.com/Yu-Jack/wrangler-test/generated/controllers/example.group.a/v1alpha1"
"github.com/rancher/wrangler/pkg/leader"
"github.com/sirupsen/logrus"
k8sv1 "k8s.io/api/core/v1"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/kubernetes/scheme"
typedcorev1 "k8s.io/client-go/kubernetes/typed/core/v1"
"k8s.io/client-go/rest"
"k8s.io/client-go/tools/record"
)
type exampleGroupAFactory struct {
egaFactory *ega.Factory
recorder record.EventRecorder
clientSet *kubernetes.Clientset
cronJobClient v1alpha1.CronJobClient
cronJobController v1alpha1.CronJobController
}
func NewExampleGroupAFactory(restConfig *rest.Config) Register {
egaFactory, err := ega.NewFactoryFromConfig(restConfig)
if err != nil {
panic(err)
}
clientSet, err := kubernetes.NewForConfig(restConfig)
if err != nil {
panic(err)
}
eventBroadcaster := record.NewBroadcaster()
eventBroadcaster.StartLogging(logrus.Infof)
eventBroadcaster.StartRecordingToSink(&typedcorev1.EventSinkImpl{Interface: clientSet.CoreV1().Events("example-group-a-operator-test-system")})
recorder := eventBroadcaster.NewRecorder(scheme.Scheme, k8sv1.EventSource{})
cronJobs := egaFactory.Example().V1alpha1().CronJob()
egaf := &exampleGroupAFactory{
egaFactory: egaFactory,
recorder: recorder,
clientSet: clientSet,
cronJobClient: cronJobs,
cronJobController: cronJobs,
}
cronJobs.OnChange(context.Background(), "example-group-a-cronjob-change", egaf.OnChange)
return egaf
}
func (egaf *exampleGroupAFactory) Setup() {
leader.RunOrDie(context.Background(), "", "example-a-controller", egaf.clientSet, func(cb context.Context) {
if err := egaf.egaFactory.Start(context.Background(), 50); err != nil {
panic(err)
}
})
}
func (egaf *exampleGroupAFactory) OnChange(id string, obj *api.CronJob) (*api.CronJob, error) {
if obj == nil {
return obj, nil
}
apiVersion, _ := obj.GroupVersionKind().ToAPIVersionAndKind()
apiPath := fmt.Sprintf("/apis/%s/namespaces/%s/%s/%s", apiVersion, obj.Namespace, api.CronJobResourceName, obj.Name)
apiPath = strings.ToLower(apiPath)
fmt.Println(apiPath)
res, err := egaf.clientSet.RESTClient().Delete().AbsPath(apiPath).DoRaw(context.Background())
if err != nil {
fmt.Println(err)
}
fmt.Println(string(res))
return obj, nil
}