seaweedfs-operator/controllers/controller_master.go

128 lines
4.4 KiB
Go
Raw Normal View History

2020-08-02 06:27:46 +00:00
package controllers
import (
"context"
appsv1 "k8s.io/api/apps/v1"
"k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/types"
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
2020-08-02 06:27:46 +00:00
seaweedv1 "github.com/seaweedfs/seaweedfs-operator/api/v1"
label "github.com/seaweedfs/seaweedfs-operator/controllers/label"
2020-08-02 06:27:46 +00:00
)
func (r *SeaweedReconciler) ensureMaster(seaweedCR *seaweedv1.Seaweed) (done bool, result ctrl.Result, err error) {
_ = context.Background()
_ = r.Log.WithValues("seaweed", seaweedCR.Name)
if done, result, err = r.ensureMasterPeerService(seaweedCR); done {
return
}
if done, result, err = r.ensureMasterService(seaweedCR); done {
2020-10-15 04:45:53 +00:00
return
2020-08-02 06:27:46 +00:00
}
if done, result, err = r.ensureMasterConfigMap(seaweedCR); done {
return
}
if done, result, err = r.ensureMasterStatefulSet(seaweedCR); done {
2020-10-15 04:45:53 +00:00
return
2020-08-02 06:27:46 +00:00
}
2020-10-15 04:45:53 +00:00
return
2020-08-02 06:27:46 +00:00
}
func (r *SeaweedReconciler) ensureMasterStatefulSet(seaweedCR *seaweedv1.Seaweed) (bool, ctrl.Result, error) {
ctx := context.Background()
log := r.Log.WithValues("sw-master-statefulset", seaweedCR.Name)
masterStatefulSet := &appsv1.StatefulSet{}
2020-08-03 05:48:15 +00:00
err := r.Get(ctx, types.NamespacedName{Name: seaweedCR.Name + "-master", Namespace: seaweedCR.Namespace}, masterStatefulSet)
2020-08-02 06:27:46 +00:00
if err != nil && errors.IsNotFound(err) {
// Define a new deployment
dep := r.createMasterStatefulSet(seaweedCR)
if err := controllerutil.SetControllerReference(seaweedCR, dep, r.Scheme); err != nil {
return ReconcileResult(err)
}
2020-08-02 06:27:46 +00:00
log.Info("Creating a new master statefulset", "Namespace", dep.Namespace, "Name", dep.Name)
err = r.Create(ctx, dep)
if err != nil {
2020-10-17 09:25:23 +00:00
log.Error(err, "Failed to create master statefulset", "Namespace", dep.Namespace, "Name", dep.Name)
2020-10-15 04:45:53 +00:00
return ReconcileResult(err)
2020-08-02 06:27:46 +00:00
}
// StatefulSet created successfully - return and requeue
2020-10-15 04:45:53 +00:00
return ReconcileResult(err)
2020-08-02 06:27:46 +00:00
} else if err != nil {
log.Error(err, "Failed to get Deployment")
2020-10-15 04:45:53 +00:00
return ReconcileResult(err)
2020-08-02 06:27:46 +00:00
}
2020-10-17 09:25:23 +00:00
log.Info("master version " + masterStatefulSet.Spec.Template.Spec.Containers[0].Image + " expected " + seaweedCR.Spec.Image)
if masterStatefulSet.Spec.Template.Spec.Containers[0].Image != seaweedCR.Spec.Image {
masterStatefulSet.Spec.Template.Spec.Containers[0].Image = seaweedCR.Spec.Image
if err = r.Update(ctx, masterStatefulSet); err != nil {
log.Error(err, "Failed to update master statefulset", "Namespace", masterStatefulSet.Namespace, "Name", masterStatefulSet.Name)
return ReconcileResult(err)
}
// Deployment created successfully - return and requeue
return ReconcileResult(err)
}
2020-08-03 05:48:15 +00:00
log.Info("Get master stateful set " + masterStatefulSet.Name)
2020-10-15 04:45:53 +00:00
return ReconcileResult(err)
2020-08-02 06:27:46 +00:00
}
func (r *SeaweedReconciler) ensureMasterConfigMap(seaweedCR *seaweedv1.Seaweed) (bool, ctrl.Result, error) {
log := r.Log.WithValues("sw-master-configmap", seaweedCR.Name)
masterConfigMap := r.createMasterConfigMap(seaweedCR)
if err := controllerutil.SetControllerReference(seaweedCR, masterConfigMap, r.Scheme); err != nil {
return ReconcileResult(err)
}
_, err := r.CreateOrUpdateConfigMap(masterConfigMap)
log.Info("Get master ConfigMap " + masterConfigMap.Name)
return ReconcileResult(err)
}
func (r *SeaweedReconciler) ensureMasterService(seaweedCR *seaweedv1.Seaweed) (bool, ctrl.Result, error) {
2020-08-02 06:27:46 +00:00
log := r.Log.WithValues("sw-master-service", seaweedCR.Name)
2020-10-18 08:59:38 +00:00
masterService := r.createMasterService(seaweedCR)
if err := controllerutil.SetControllerReference(seaweedCR, masterService, r.Scheme); err != nil {
return ReconcileResult(err)
}
2020-10-18 08:59:38 +00:00
_, err := r.CreateOrUpdateService(masterService)
2020-08-02 06:27:46 +00:00
log.Info("Get master service " + masterService.Name)
2020-10-15 04:45:53 +00:00
return ReconcileResult(err)
2020-08-02 06:27:46 +00:00
}
func (r *SeaweedReconciler) ensureMasterPeerService(seaweedCR *seaweedv1.Seaweed) (bool, ctrl.Result, error) {
log := r.Log.WithValues("sw-master-peer-service", seaweedCR.Name)
masterPeerService := r.createMasterPeerService(seaweedCR)
if err := controllerutil.SetControllerReference(seaweedCR, masterPeerService, r.Scheme); err != nil {
return ReconcileResult(err)
}
_, err := r.CreateOrUpdateService(masterPeerService)
log.Info("Get master peer service " + masterPeerService.Name)
return ReconcileResult(err)
}
2020-08-02 06:27:46 +00:00
func labelsForMaster(name string) map[string]string {
return map[string]string{
label.ManagedByLabelKey: "seaweedfs-operator",
label.NameLabelKey: "seaweedfs",
label.ComponentLabelKey: "master",
label.InstanceLabelKey: name,
}
2020-08-02 06:27:46 +00:00
}