forked from rook/rook
Yannis Zarkadas
f5a41e9a7c
Code handling the scale down on the side of the controller. Also, some new util functions. Signed-off-by: Yannis Zarkadas <yanniszarkadas@gmail.com>
92 lines
3.4 KiB
Go
92 lines
3.4 KiB
Go
/*
|
||
Copyright 2018 The Rook Authors. All rights reserved.
|
||
|
||
Licensed under the Apache License, Version 2.0 (the "License");
|
||
you may not use this file except in compliance with the License.
|
||
You may obtain a copy of the License at
|
||
|
||
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
||
Unless required by applicable law or agreed to in writing, software
|
||
distributed under the License is distributed on an "AS IS" BASIS,
|
||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
See the License for the specific language governing permissions and
|
||
limitations under the License.
|
||
*/
|
||
|
||
package controller
|
||
|
||
import (
|
||
"fmt"
|
||
cassandrav1alpha1 "github.com/rook/rook/pkg/apis/cassandra.rook.io/v1alpha1"
|
||
"github.com/rook/rook/pkg/operator/cassandra/controller/util"
|
||
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||
)
|
||
|
||
// cleanup deletes all resources remaining because of cluster scale downs
|
||
func (cc *ClusterController) cleanup(c *cassandrav1alpha1.Cluster) error {
|
||
|
||
for _, r := range c.Spec.Datacenter.Racks {
|
||
services, err := cc.serviceLister.Services(c.Namespace).List(util.RackSelector(r, c))
|
||
if err != nil {
|
||
return fmt.Errorf("error listing member services: %s", err.Error())
|
||
}
|
||
// Get rack status. If it doesn't exist, the rack isn't yet created.
|
||
stsName := util.StatefulSetNameForRack(r, c)
|
||
sts, err := cc.statefulSetLister.StatefulSets(c.Namespace).Get(stsName)
|
||
if apierrors.IsNotFound(err) {
|
||
continue
|
||
}
|
||
if err != nil {
|
||
return fmt.Errorf("error getting statefulset %s: %s", stsName, err.Error())
|
||
}
|
||
memberCount := *sts.Spec.Replicas
|
||
memberServiceCount := int32(len(services))
|
||
// If there are more services than members, some services need to be cleaned up
|
||
if memberServiceCount > memberCount {
|
||
maxIndex := memberCount - 1
|
||
for _, svc := range services {
|
||
svcIndex, err := util.IndexFromName(svc.Name)
|
||
if err != nil {
|
||
logger.Errorf("Unexpected error while parsing index from name %s : %s", svc.Name, err.Error())
|
||
continue
|
||
}
|
||
if svcIndex > maxIndex {
|
||
err := cc.cleanupMemberResources(svc.Name, r, c)
|
||
if err != nil {
|
||
return fmt.Errorf("error cleaning up member resources: %s", err.Error())
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
logger.Infof("%s/%s - Successfully cleaned up cluster.", c.Namespace, c.Name)
|
||
return nil
|
||
}
|
||
|
||
// cleanupMemberResources deletes all resources associated with a given member.
|
||
// Currently those are :
|
||
// - A PVC
|
||
// - A ClusterIP Service
|
||
func (cc *ClusterController) cleanupMemberResources(memberName string, r cassandrav1alpha1.RackSpec, c *cassandrav1alpha1.Cluster) error {
|
||
|
||
logger.Infof("%s/%s - Cleaning up resources for member %s", c.Namespace, c.Name, memberName)
|
||
// Delete PVC
|
||
if len(r.Storage.VolumeClaimTemplates) > 0 {
|
||
// PVC naming convention for StatefulSets is <volumeClaimTemplate.Name>-<pod.Name>
|
||
pvcName := fmt.Sprintf("%s-%s", r.Storage.VolumeClaimTemplates[0].Name, memberName)
|
||
err := cc.kubeClient.CoreV1().PersistentVolumeClaims(c.Namespace).Delete(pvcName, &metav1.DeleteOptions{})
|
||
if err != nil && !apierrors.IsNotFound(err) {
|
||
return fmt.Errorf("error deleting pvc %s: %s", pvcName, err.Error())
|
||
}
|
||
}
|
||
|
||
// Delete Member Service
|
||
err := cc.kubeClient.CoreV1().Services(c.Namespace).Delete(memberName, &metav1.DeleteOptions{})
|
||
if err != nil {
|
||
return fmt.Errorf("error deleting member service %s: %s", memberName, err.Error())
|
||
}
|
||
return nil
|
||
}
|