kv.go

package kv

import (
	"database/sql"
	"errors"
	"time"

	"github.com/rubiojr/kv/driver/mysql"
	"github.com/rubiojr/kv/driver/sqlite"
	"github.com/rubiojr/kv/types"
)

const TABLE_NAME = "key_values"
const MAX_KEY_LENGTH = 255
const MAX_VALUE_LENGTH = 65535

type KV struct{}

type Database interface {
	Init(tableName string, urn string) error

	Get(key string) ([]byte, error)
	MGet(...string) ([][]byte, error)

	Set(key string, value []byte, expireAt *time.Time) error
	MSet(kvs types.KeyValues, expireAt *time.Time) error

	Del(key string) error
	MDel(keys ...string) error

	Exists(key string) (bool, error)
	MExists(keys ...string) ([]bool, error)

	Raw() *sql.DB
}

func New(driver string, urn string) (Database, error) {
	var db Database
	var err error
	switch driver {
	case "sqlite":
		db = &sqlite.Database{}
		err = db.Init(TABLE_NAME, urn)
	case "mysql":
		db = &mysql.Database{}
		err = db.Init(TABLE_NAME, urn)
	default:
		err = errors.New("driver not supported")
	}

	return db, err
}