blob: 25d568e3328d335ff88251061d25e04a25d3bc58 [file] [log] [blame]
/*
* Copyright 2018-present Open Networking Foundation
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package model
import (
"errors"
"fmt"
"github.com/opencord/voltha-go/common/log"
"github.com/opencord/voltha-go/db/kvstore"
"strconv"
"time"
)
//TODO: missing cache stuff
//TODO: missing retry stuff
//TODO: missing proper logging
type Backend struct {
Client kvstore.Client
StoreType string
Host string
Port int
Timeout int
PathPrefix string
}
func NewBackend(storeType string, host string, port int, timeout int, pathPrefix string) *Backend {
var err error
b := &Backend{
StoreType: storeType,
Host: host,
Port: port,
Timeout: timeout,
PathPrefix: pathPrefix,
}
address := host + ":" + strconv.Itoa(port)
if b.Client, err = b.newClient(address, timeout); err != nil {
log.Errorf("failed to create a new kv Client - %s", err.Error())
}
return b
}
func (b *Backend) newClient(address string, timeout int) (kvstore.Client, error) {
switch b.StoreType {
case "consul":
return kvstore.NewConsulClient(address, timeout)
case "etcd":
return kvstore.NewEtcdClient(address, timeout)
}
return nil, errors.New("Unsupported KV store")
}
func (b *Backend) makePath(key string) string {
path := fmt.Sprintf("%s/%s", b.PathPrefix, key)
return path
}
func (b *Backend) List(key string) (map[string]*kvstore.KVPair, error) {
return b.Client.List(b.makePath(key), b.Timeout)
}
func (b *Backend) Get(key string) (*kvstore.KVPair, error) {
formattedPath := b.makePath(key)
log.Debugf("Get key: %s, path: %s", key, formattedPath)
start := time.Now()
err, pair := b.Client.Get(formattedPath, b.Timeout)
stop := time.Now()
GetProfiling().AddToDatabaseRetrieveTime(stop.Sub(start).Seconds())
return err, pair
}
func (b *Backend) Put(key string, value interface{}) error {
formattedPath := b.makePath(key)
log.Debugf("Put key: %s, value: %+v, path: %s", key, string(value.([]byte)), formattedPath)
return b.Client.Put(formattedPath, value, b.Timeout)
}
func (b *Backend) Delete(key string) error {
return b.Client.Delete(b.makePath(key), b.Timeout)
}