ridv/gorealis
1
0
Fork 0
Code Issues Pull requests Projects Releases 23 Packages Wiki Activity Actions

Adding port override back in.

Browse source
This commit is contained in:
Renan DelValle 2018-04-12 18:11:07 -07:00
parent 1e1b2e48b7
commit 280db21d33
No known key found for this signature in database
GPG key ID: C240AD6D6F443EC9
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
zk.go
View file

@ -42,6 +42,7 @@ type zkConfig struct {
timeout time.Duration timeout time.Duration
logger Logger logger Logger
auroraSchemeOverride *string auroraSchemeOverride *string
auroraPortOverride *int
} }
type ZKOpt func(z *zkConfig) type ZKOpt func(z *zkConfig)
@ -82,6 +83,12 @@ func ZKAuroraSchemeOverride(scheme string) ZKOpt {
} }
} }
func ZKAuroraPortOverride(port int) ZKOpt {
return func(z *zkConfig) {
z.auroraPortOverride = &port
}
}
// Retrieves current Aurora leader from ZK. // Retrieves current Aurora leader from ZK.
func LeaderFromZK(cluster Cluster) (string, error) { func LeaderFromZK(cluster Cluster) (string, error) {
return LeaderFromZKOpts(ZKEndpoints(strings.Split(cluster.ZK, ",")...), ZKPath(cluster.SchedZKPath)) return LeaderFromZKOpts(ZKEndpoints(strings.Split(cluster.ZK, ",")...), ZKPath(cluster.SchedZKPath))
@ -167,7 +174,11 @@ func LeaderFromZKOpts(options ...ZKOpt) (string, error) {
host = v.Host host = v.Host
port = strconv.Itoa(v.Port) if config.auroraPortOverride == nil {
port = strconv.Itoa(v.Port)
} else {
port = strconv.Itoa(*config.auroraPortOverride)
}
} }
leaderURL = scheme + "://" + host + ":" + port leaderURL = scheme + "://" + host + ":" + port