35 lines
989 B
Go
35 lines
989 B
Go
package provider
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/containous/traefik/safe"
|
|
"github.com/containous/traefik/types"
|
|
"github.com/docker/libkv/store"
|
|
"github.com/docker/libkv/store/zookeeper"
|
|
)
|
|
|
|
var _ Provider = (*Zookepper)(nil)
|
|
|
|
// Zookepper holds configurations of the Zookepper provider.
|
|
type Zookepper struct {
|
|
Kv `mapstructure:",squash"`
|
|
}
|
|
|
|
// Provide allows the provider to provide configurations to traefik
|
|
// using the given configuration channel.
|
|
func (provider *Zookepper) Provide(configurationChan chan<- types.ConfigMessage, pool *safe.Pool, constraints types.Constraints) error {
|
|
store, err := provider.CreateStore()
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to Connect to KV store: %v", err)
|
|
}
|
|
provider.kvclient = store
|
|
return provider.provide(configurationChan, pool, constraints)
|
|
}
|
|
|
|
// CreateStore creates the KV store
|
|
func (provider *Zookepper) CreateStore() (store.Store, error) {
|
|
provider.storeType = store.ZK
|
|
zookeeper.Register()
|
|
return provider.createStore()
|
|
}
|