You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.6 KiB
54 lines
1.6 KiB
package zk |
|
|
|
import ( |
|
"crypto/sha1" |
|
"encoding/base64" |
|
"fmt" |
|
"math/rand" |
|
"strconv" |
|
"strings" |
|
) |
|
|
|
// AuthACL produces an ACL list containing a single ACL which uses the |
|
// provided permissions, with the scheme "auth", and ID "", which is used |
|
// by ZooKeeper to represent any authenticated user. |
|
func AuthACL(perms int32) []ACL { |
|
return []ACL{{perms, "auth", ""}} |
|
} |
|
|
|
// WorldACL produces an ACL list containing a single ACL which uses the |
|
// provided permissions, with the scheme "world", and ID "anyone", which |
|
// is used by ZooKeeper to represent any user at all. |
|
func WorldACL(perms int32) []ACL { |
|
return []ACL{{perms, "world", "anyone"}} |
|
} |
|
|
|
func DigestACL(perms int32, user, password string) []ACL { |
|
userPass := []byte(fmt.Sprintf("%s:%s", user, password)) |
|
h := sha1.New() |
|
if n, err := h.Write(userPass); err != nil || n != len(userPass) { |
|
panic("SHA1 failed") |
|
} |
|
digest := base64.StdEncoding.EncodeToString(h.Sum(nil)) |
|
return []ACL{{perms, "digest", fmt.Sprintf("%s:%s", user, digest)}} |
|
} |
|
|
|
// FormatServers takes a slice of addresses, and makes sure they are in a format |
|
// that resembles <addr>:<port>. If the server has no port provided, the |
|
// DefaultPort constant is added to the end. |
|
func FormatServers(servers []string) []string { |
|
for i := range servers { |
|
if !strings.Contains(servers[i], ":") { |
|
servers[i] = servers[i] + ":" + strconv.Itoa(DefaultPort) |
|
} |
|
} |
|
return servers |
|
} |
|
|
|
// stringShuffle performs a Fisher-Yates shuffle on a slice of strings |
|
func stringShuffle(s []string) { |
|
for i := len(s) - 1; i > 0; i-- { |
|
j := rand.Intn(i + 1) |
|
s[i], s[j] = s[j], s[i] |
|
} |
|
}
|
|
|