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.
37 lines
1013 B
37 lines
1013 B
package cluster |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
) |
|
|
|
// leaderElectorFactoryFunc is a LeaderElector factory method type. |
|
type leaderElectorFactoryFunc func(context.Context, ...interface{}) ( |
|
LeaderElector, error) |
|
|
|
var leaderElectorFactories map[string]leaderElectorFactoryFunc |
|
|
|
// RegisterLeaderElectorFactory will register a new LeaderElector factory |
|
// method corresponding to the passed id. |
|
func RegisterLeaderElectorFactory(id string, factory leaderElectorFactoryFunc) { |
|
if leaderElectorFactories == nil { |
|
leaderElectorFactories = make( |
|
map[string]leaderElectorFactoryFunc, |
|
) |
|
} |
|
|
|
leaderElectorFactories[id] = factory |
|
} |
|
|
|
// MakeLeaderElector will constuct a LeaderElector identified by id with the |
|
// passed arguments. |
|
func MakeLeaderElector(ctx context.Context, id string, args ...interface{}) ( |
|
LeaderElector, error) { |
|
|
|
if _, ok := leaderElectorFactories[id]; !ok { |
|
return nil, fmt.Errorf("leader elector factory for '%v' "+ |
|
"not found", id) |
|
} |
|
|
|
return leaderElectorFactories[id](ctx, args...) |
|
}
|
|
|