47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package etcdv2
|
|
|
|
import (
|
|
"context"
|
|
"crypto/md5"
|
|
"fmt"
|
|
|
|
etcdapi "github.com/coreos/etcd/client"
|
|
"github.com/hashicorp/terraform/states/remote"
|
|
)
|
|
|
|
// EtcdClient is a remote client that stores data in etcd.
|
|
type EtcdClient struct {
|
|
Client etcdapi.Client
|
|
Path string
|
|
}
|
|
|
|
func (c *EtcdClient) Get() (*remote.Payload, error) {
|
|
resp, err := etcdapi.NewKeysAPI(c.Client).Get(context.Background(), c.Path, &etcdapi.GetOptions{Quorum: true})
|
|
if err != nil {
|
|
if err, ok := err.(etcdapi.Error); ok && err.Code == etcdapi.ErrorCodeKeyNotFound {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
if resp.Node.Dir {
|
|
return nil, fmt.Errorf("path is a directory")
|
|
}
|
|
|
|
data := []byte(resp.Node.Value)
|
|
md5 := md5.Sum(data)
|
|
return &remote.Payload{
|
|
Data: data,
|
|
MD5: md5[:],
|
|
}, nil
|
|
}
|
|
|
|
func (c *EtcdClient) Put(data []byte) error {
|
|
_, err := etcdapi.NewKeysAPI(c.Client).Set(context.Background(), c.Path, string(data), nil)
|
|
return err
|
|
}
|
|
|
|
func (c *EtcdClient) Delete() error {
|
|
_, err := etcdapi.NewKeysAPI(c.Client).Delete(context.Background(), c.Path, nil)
|
|
return err
|
|
}
|