87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
seaweedv1 "github.com/seaweedfs/seaweedfs-operator/api/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
)
|
|
|
|
const (
|
|
masterPeerAddressPattern = "%s-master-%d.%s-master-peer.%s:9333"
|
|
filerPeerAddressPattern = "%s-filer-%d.%s-filer-peer:8888"
|
|
)
|
|
|
|
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 getFilerAddresses(name string, replicas int32) []string {
|
|
peersAddresses := make([]string, 0, replicas)
|
|
for i := int32(0); i < replicas; i++ {
|
|
peersAddresses = append(peersAddresses, fmt.Sprintf(filerPeerAddressPattern, name, i, name))
|
|
}
|
|
return peersAddresses
|
|
}
|
|
|
|
func getFilerPeersString(name string, replicas int32) string {
|
|
return strings.Join(getFilerAddresses(name, replicas), ",")
|
|
}
|
|
|
|
func getMasterAddresses(namespace string, 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, namespace))
|
|
}
|
|
return peersAddresses
|
|
}
|
|
|
|
func getMasterPeersString(m *seaweedv1.Seaweed) string {
|
|
return strings.Join(getMasterAddresses(m.Namespace, m.Name, m.Spec.Master.Replicas), ",")
|
|
}
|
|
|
|
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
|
|
}
|