81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package util
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/kr/pretty"
|
|
|
|
"github.bus.zalan.do/acid/postgres-operator/pkg/spec"
|
|
"k8s.io/client-go/pkg/api/v1"
|
|
"k8s.io/client-go/pkg/types"
|
|
)
|
|
|
|
var passwordChars = []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
|
|
|
|
func init() {
|
|
rand.Seed(int64(time.Now().Unix()))
|
|
}
|
|
|
|
func RandomPassword(n int) string {
|
|
b := make([]byte, n)
|
|
for i := range b {
|
|
b[i] = passwordChars[rand.Intn(len(passwordChars))]
|
|
}
|
|
|
|
return string(b)
|
|
}
|
|
|
|
func NameFromMeta(meta v1.ObjectMeta) types.NamespacedName {
|
|
obj := types.NamespacedName{
|
|
Namespace: meta.Namespace,
|
|
Name: meta.Name,
|
|
}
|
|
|
|
return obj
|
|
}
|
|
|
|
func PodClusterName(pod *v1.Pod) spec.ClusterName {
|
|
if name, ok := pod.Labels["spilo-cluster"]; ok {
|
|
return spec.ClusterName{
|
|
Namespace: pod.Namespace,
|
|
Name: name,
|
|
}
|
|
}
|
|
|
|
return spec.ClusterName{}
|
|
}
|
|
|
|
func PodSpiloRole(pod *v1.Pod) string {
|
|
return pod.Labels["spilo-role"]
|
|
}
|
|
|
|
func ClusterDNSName(clusterName, teamName, hostedZone string) string {
|
|
return fmt.Sprintf("%s.%s.%s", clusterName, teamName, hostedZone)
|
|
}
|
|
|
|
func PGUserPassword(user spec.PgUser) string {
|
|
s := md5.Sum([]byte(user.Password + user.Name))
|
|
|
|
return "md5" + hex.EncodeToString(s[:])
|
|
}
|
|
|
|
func Pretty(x interface{}) (f fmt.Formatter) {
|
|
return pretty.Formatter(x)
|
|
}
|
|
|
|
func PrettyDiff(a, b interface{}) (result string) {
|
|
diff := pretty.Diff(a, b)
|
|
json, err := json.MarshalIndent(diff, "", " ")
|
|
if err != nil {
|
|
result = fmt.Sprintf("%v", diff)
|
|
} else {
|
|
result = string(json)
|
|
}
|
|
return
|
|
}
|