forked from rook/rook
Yannis Zarkadas
a9178ccb59
The sidecar now has a control loop, which watches its ClusterIP Service for the `cassandra.rook.io/decommissioned` label. If it finds it, then it decommissions and updates the label to value "true". Signed-off-by: Yannis Zarkadas <yanniszarkadas@gmail.com>
54 lines
1.8 KiB
Go
54 lines
1.8 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 sidecar
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/rook/rook/pkg/operator/cassandra/constants"
|
||
"github.com/rook/rook/pkg/operator/cassandra/controller/util"
|
||
"github.com/yanniszark/go-nodetool/nodetool"
|
||
"k8s.io/api/core/v1"
|
||
)
|
||
|
||
func (m *MemberController) Sync(memberService *v1.Service) error {
|
||
|
||
// Check if member must decommission
|
||
if decommission, ok := memberService.Labels[constants.DecommissionLabel]; ok {
|
||
// Check if member has already decommissioned
|
||
if decommission == constants.LabelValueTrue {
|
||
return nil
|
||
}
|
||
// Else, decommission member
|
||
if err := m.nodetool.Decommission(); err != nil {
|
||
m.logger.Errorf("Error during decommission: %s", err.Error())
|
||
}
|
||
// Confirm memberService has been decommissioned
|
||
if opMode, err := m.nodetool.OperationMode(); err != nil || opMode != nodetool.NodeOperationModeDecommissioned {
|
||
return fmt.Errorf("error during decommission, operation mode: %s, error: %v", opMode, err)
|
||
}
|
||
// Update Label
|
||
old := memberService.DeepCopy()
|
||
memberService.Labels[constants.DecommissionLabel] = constants.LabelValueTrue
|
||
if err := util.PatchService(old, memberService, m.kubeClient); err != nil {
|
||
return fmt.Errorf("error patching MemberService, %s", err.Error())
|
||
}
|
||
|
||
}
|
||
|
||
return nil
|
||
}
|