95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
)
|
|
|
|
const (
|
|
masterPeerAddressPattern = "%s-master-%d.%s-master-peer:9333"
|
|
masterPeerAddressWithNamespacePattern = "%s-master-%d.%s-master-peer.%s:9333"
|
|
filerServiceAddressWithNamespacePattern = "%s-filer.%s:8888"
|
|
masterServiceAddressWithNamespacePattern = "%s-master.%s:9333"
|
|
)
|
|
|
|
var (
|
|
kubernetesEnvVars = []corev1.EnvVar{
|
|
{
|
|
Name: "POD_IP",
|
|
ValueFrom: &corev1.EnvVarSource{
|
|
FieldRef: &corev1.ObjectFieldSelector{
|
|
FieldPath: "status.podIP",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "POD_NAME",
|
|
ValueFrom: &corev1.EnvVarSource{
|
|
FieldRef: &corev1.ObjectFieldSelector{
|
|
FieldPath: "metadata.name",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "NAMESPACE",
|
|
ValueFrom: &corev1.EnvVarSource{
|
|
FieldRef: &corev1.ObjectFieldSelector{
|
|
FieldPath: "metadata.namespace",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
)
|
|
|
|
func ReconcileResult(err error) (bool, ctrl.Result, error) {
|
|
if err != nil {
|
|
return true, ctrl.Result{}, err
|
|
}
|
|
return false, ctrl.Result{}, nil
|
|
}
|
|
|
|
func getMasterAddresses(name string, replicas int32) []string {
|
|
peersAddresses := make([]string, 0, replicas)
|
|
for i := int32(0); i < replicas; i++ {
|
|
peersAddresses = append(peersAddresses, fmt.Sprintf(masterPeerAddressPattern, name, i, name))
|
|
}
|
|
return peersAddresses
|
|
}
|
|
|
|
func getMasterPeersString(name string, replicas int32) string {
|
|
return strings.Join(getMasterAddresses(name, replicas), ",")
|
|
}
|
|
func getMasterAddressesWithNamespace(name, namespace string, replicas int32) []string {
|
|
peersAddresses := make([]string, 0, replicas)
|
|
for i := int32(0); i < replicas; i++ {
|
|
peersAddresses = append(peersAddresses, fmt.Sprintf(masterPeerAddressWithNamespacePattern, name, i, name, namespace))
|
|
}
|
|
return peersAddresses
|
|
}
|
|
|
|
func getMasterPeersStringWithNamespace(name, namespace string, replicas int32) string {
|
|
return strings.Join(getMasterAddressesWithNamespace(name, namespace, replicas), ",")
|
|
}
|
|
|
|
func getFilerServiceAddressWithNamespace(name, namespace string) string {
|
|
return fmt.Sprintf(filerServiceAddressWithNamespacePattern, name, namespace)
|
|
}
|
|
|
|
func getMasterServiceAddressWithNamespace(name, namespace string) string {
|
|
return fmt.Sprintf(masterServiceAddressWithNamespacePattern, name, namespace)
|
|
}
|
|
|
|
func copyAnnotations(src map[string]string) map[string]string {
|
|
if src == nil {
|
|
return nil
|
|
}
|
|
dst := map[string]string{}
|
|
for k, v := range src {
|
|
dst[k] = v
|
|
}
|
|
return dst
|
|
}
|