78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
|
package remote
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/md5"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/armon/consul-api"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ConsulRemoteClient implements the RemoteClient interface
|
||
|
|
// for an Consul compatible server.
|
||
|
|
type ConsulRemoteClient struct {
|
||
|
|
client *consulapi.Client
|
||
|
|
path string // KV path
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewConsulRemoteClient(conf map[string]string) (*ConsulRemoteClient, error) {
|
||
|
|
client := &ConsulRemoteClient{}
|
||
|
|
if err := client.validateConfig(conf); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return client, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConsulRemoteClient) validateConfig(conf map[string]string) (err error) {
|
||
|
|
config := consulapi.DefaultConfig()
|
||
|
|
if token, ok := conf["token"]; ok && token != "" {
|
||
|
|
config.Token = token
|
||
|
|
}
|
||
|
|
if addr, ok := conf["address"]; ok && addr != "" {
|
||
|
|
config.Address = addr
|
||
|
|
}
|
||
|
|
path, ok := conf["path"]
|
||
|
|
if !ok || path == "" {
|
||
|
|
return fmt.Errorf("missing 'path' configuration")
|
||
|
|
}
|
||
|
|
c.path = path
|
||
|
|
c.client, err = consulapi.NewClient(config)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConsulRemoteClient) GetState() (*RemoteStatePayload, error) {
|
||
|
|
kv := c.client.KV()
|
||
|
|
pair, _, err := kv.Get(c.path, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if pair == nil {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create the payload
|
||
|
|
payload := &RemoteStatePayload{
|
||
|
|
State: pair.Value,
|
||
|
|
}
|
||
|
|
|
||
|
|
// Generate the MD5
|
||
|
|
hash := md5.Sum(payload.State)
|
||
|
|
payload.MD5 = hash[:md5.Size]
|
||
|
|
return payload, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConsulRemoteClient) PutState(state []byte, force bool) error {
|
||
|
|
pair := &consulapi.KVPair{
|
||
|
|
Key: c.path,
|
||
|
|
Value: state,
|
||
|
|
}
|
||
|
|
kv := c.client.KV()
|
||
|
|
_, err := kv.Put(pair, nil)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConsulRemoteClient) DeleteState() error {
|
||
|
|
kv := c.client.KV()
|
||
|
|
_, err := kv.Delete(c.path, nil)
|
||
|
|
return err
|
||
|
|
}
|