kv.go

package kv

import (
        "errors"

        consul "github.com/hashicorp/consul/api"
)

type consulKV struct {
        Client *consul.Client
}

func (c *consulKV) Get(key string) (*Item, error) {
        keyval, _, err := c.Client.KV().Get(key, nil)
        if err != nil {
                return nil, err
        }
        if keyval == nil {
                return nil, errors.New("key not found")
        }

        return &Item{
                Key:   keyval.Key,
                Value: keyval.Value,
        }, nil
}

func (c *consulKV) Del(key string) error {
        _, err := c.Client.KV().Delete(key, nil)
        return err
}

func (c *consulKV) Put(item *Item) error {
        _, err := c.Client.KV().Put(&consul.KVPair{
                Key:   item.Key,
                Value: item.Value,
        }, nil)

        return err
}

func newConsulKV(addrs []string, opt ...Option) KV {
        config := consul.DefaultConfig()
        if len(addrs) > 0 {
                config.Address = addrs[0]
        }

        client, _ := consul.NewClient(config)

        return &consulKV{
                Client: client,