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 }