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 }