[icn/sdwan.git] /

// SPDX-License-Identifier: Apache-2.0\r
// Copyright (c) 2020 Intel Corporation\r
\r
package contextdb\r
\r
import (\r
        "context"\r
        "encoding/json"\r
        "os"\r
        "time"\r
\r
        pkgerrors "github.com/pkg/errors"\r
        "go.etcd.io/etcd/clientv3"\r
)\r
\r
// EtcdConfig Configuration values needed for Etcd Client\r
type EtcdConfig struct {\r
        Endpoint string\r
        CertFile string\r
        KeyFile  string\r
        CAFile   string\r
}\r
\r
// EtcdClient for Etcd\r
type EtcdClient struct {\r
        cli      *clientv3.Client\r
        endpoint string\r
}\r
\r
// Etcd For Mocking purposes\r
type Etcd interface {\r
        Put(ctx context.Context, key, val string, opts ...clientv3.OpOption) (*clientv3.PutResponse, error)\r
        Get(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.GetResponse, error)\r
        Delete(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.DeleteResponse, error)\r
}\r
\r
var getEtcd = func(e *EtcdClient) Etcd {\r
        return e.cli\r
}\r
\r
// NewEtcdClient function initializes Etcd client\r
func NewEtcdClient(store *clientv3.Client, c EtcdConfig) (ContextDb, error) {\r
        var endpoint string\r
        if store == nil {\r
                endpoint = "http://" + c.Endpoint + ":2379"\r
\r
                etcdClient := clientv3.Config{\r
                        Endpoints:   []string{endpoint},\r
                        DialTimeout: 5 * time.Second,\r
                }\r
                if len(os.Getenv("CONTEXTDB_EMCO_USERNAME")) > 0 && len(os.Getenv("CONTEXTDB_EMCO_PASSWORD")) > 0 {\r
                        etcdClient.Username = os.Getenv("CONTEXTDB_EMCO_USERNAME")\r
                        etcdClient.Password = os.Getenv("CONTEXTDB_EMCO_PASSWORD")\r
                }\r
                var err error\r
                store, err = clientv3.New(etcdClient)\r
                if err != nil {\r
                        return nil, pkgerrors.Errorf("Error creating etcd client: %s", err.Error())\r
                }\r
        }\r
\r
        return &EtcdClient{\r
                cli:      store,\r
                endpoint: endpoint,\r
        }, nil\r
}\r
\r
// Put values in Etcd DB\r
func (e *EtcdClient) Put(key string, value interface{}) error {\r
        cli := getEtcd(e)\r
        if cli == nil {\r
                return pkgerrors.Errorf("Etcd Client not initialized")\r
        }\r
        if key == "" {\r
                return pkgerrors.Errorf("Key is null")\r
        }\r
        if value == nil {\r
                return pkgerrors.Errorf("Value is nil")\r
        }\r
        v, err := json.Marshal(value)\r
        if err != nil {\r
                return pkgerrors.Errorf("Json Marshal error: %s", err.Error())\r
        }\r
        _, err = cli.Put(context.Background(), key, string(v))\r
        if err != nil {\r
                return pkgerrors.Errorf("Error creating etcd entry: %s", err.Error())\r
        }\r
        return nil\r
}\r
\r
// Get values from Etcd DB and decodes from json\r
func (e *EtcdClient) Get(key string, value interface{}) error {\r
        cli := getEtcd(e)\r
        if cli == nil {\r
                return pkgerrors.Errorf("Etcd Client not initialized")\r
        }\r
        if key == "" {\r
                return pkgerrors.Errorf("Key is null")\r
        }\r
        if value == nil {\r
                return pkgerrors.Errorf("Value is nil")\r
        }\r
        getResp, err := cli.Get(context.Background(), key)\r
        if err != nil {\r
                return pkgerrors.Errorf("Error getting etcd entry: %s", err.Error())\r
        }\r
        if getResp.Count == 0 {\r
                return pkgerrors.Errorf("Key doesn't exist")\r
        }\r
        return json.Unmarshal(getResp.Kvs[0].Value, value)\r
}\r
\r
// GetAllKeys values from Etcd DB\r
func (e *EtcdClient) GetAllKeys(key string) ([]string, error) {\r
        cli := getEtcd(e)\r
        if cli == nil {\r
                return nil, pkgerrors.Errorf("Etcd Client not initialized")\r
        }\r
        getResp, err := cli.Get(context.Background(), key, clientv3.WithPrefix())\r
        if err != nil {\r
                return nil, pkgerrors.Errorf("Error getting etcd entry: %s", err.Error())\r
        }\r
        if getResp.Count == 0 {\r
                return nil, pkgerrors.Errorf("Key doesn't exist")\r
        }\r
        var keys []string\r
        for _, ev := range getResp.Kvs {\r
                keys = append(keys, string(ev.Key))\r
        }\r
        return keys, nil\r
}\r
\r
// DeleteAll keys from Etcd DB\r
func (e *EtcdClient) DeleteAll(key string) error {\r
        cli := getEtcd(e)\r
        if cli == nil {\r
                return pkgerrors.Errorf("Etcd Client not initialized")\r
        }\r
        _, err := cli.Delete(context.Background(), key, clientv3.WithPrefix())\r
        if err != nil {\r
                return pkgerrors.Errorf("Delete failed etcd entry: %s", err.Error())\r
        }\r
        return nil\r
}\r
\r
// Delete values from Etcd DB\r
func (e *EtcdClient) Delete(key string) error {\r
        cli := getEtcd(e)\r
        if cli == nil {\r
                return pkgerrors.Errorf("Etcd Client not initialized")\r
        }\r
        _, err := cli.Delete(context.Background(), key)\r
        if err != nil {\r
                return pkgerrors.Errorf("Delete failed etcd entry: %s", err.Error())\r
        }\r
        return nil\r
}\r
\r
// HealthCheck for checking health of the etcd cluster\r
func (e *EtcdClient) HealthCheck() error {\r
        return nil\r
}\r