mirror of https://github.com/ledisdb/ledisdb.git
422 lines
7.2 KiB
Go
422 lines
7.2 KiB
Go
package ledis
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"github.com/siddontang/go-leveldb/leveldb"
|
|
)
|
|
|
|
type FVPair struct {
|
|
Field []byte
|
|
Value []byte
|
|
}
|
|
|
|
var errHashKey = errors.New("invalid hash key")
|
|
var errHSizeKey = errors.New("invalid hsize key")
|
|
|
|
const (
|
|
hashStartSep byte = ':'
|
|
hashStopSep byte = hashStartSep + 1
|
|
)
|
|
|
|
func checkHashKFSize(key []byte, field []byte) error {
|
|
if len(key) > MaxKeySize {
|
|
return ErrKeySize
|
|
} else if len(field) > MaxHashFieldSize {
|
|
return ErrHashFieldSize
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *DB) hEncodeSizeKey(key []byte) []byte {
|
|
buf := make([]byte, len(key)+2)
|
|
|
|
buf[0] = db.index
|
|
buf[1] = hSizeType
|
|
|
|
copy(buf[2:], key)
|
|
return buf
|
|
}
|
|
|
|
func (db *DB) hDecodeSizeKey(ek []byte) ([]byte, error) {
|
|
if len(ek) < 2 || ek[0] != db.index || ek[1] != hSizeType {
|
|
return nil, errHSizeKey
|
|
}
|
|
|
|
return ek[2:], nil
|
|
}
|
|
|
|
func (db *DB) hEncodeHashKey(key []byte, field []byte) []byte {
|
|
buf := make([]byte, len(key)+len(field)+1+1+2+1)
|
|
|
|
pos := 0
|
|
buf[pos] = db.index
|
|
pos++
|
|
buf[pos] = hashType
|
|
pos++
|
|
|
|
binary.BigEndian.PutUint16(buf[pos:], uint16(len(key)))
|
|
pos += 2
|
|
|
|
copy(buf[pos:], key)
|
|
pos += len(key)
|
|
|
|
buf[pos] = hashStartSep
|
|
pos++
|
|
copy(buf[pos:], field)
|
|
|
|
return buf
|
|
}
|
|
|
|
func (db *DB) hDecodeHashKey(ek []byte) ([]byte, []byte, error) {
|
|
if len(ek) < 5 || ek[0] != db.index || ek[1] != hashType {
|
|
return nil, nil, errHashKey
|
|
}
|
|
|
|
pos := 2
|
|
keyLen := int(binary.BigEndian.Uint16(ek[pos:]))
|
|
pos += 2
|
|
|
|
if keyLen+5 > len(ek) {
|
|
return nil, nil, errHashKey
|
|
}
|
|
|
|
key := ek[pos : pos+keyLen]
|
|
pos += keyLen
|
|
|
|
if ek[pos] != hashStartSep {
|
|
return nil, nil, errHashKey
|
|
}
|
|
|
|
pos++
|
|
field := ek[pos:]
|
|
return key, field, nil
|
|
}
|
|
|
|
func (db *DB) hEncodeStartKey(key []byte) []byte {
|
|
return db.hEncodeHashKey(key, nil)
|
|
}
|
|
|
|
func (db *DB) hEncodeStopKey(key []byte) []byte {
|
|
k := db.hEncodeHashKey(key, nil)
|
|
|
|
k[len(k)-1] = hashStopSep
|
|
|
|
return k
|
|
}
|
|
|
|
func (db *DB) HLen(key []byte) (int64, error) {
|
|
return Int64(db.db.Get(db.hEncodeSizeKey(key)))
|
|
}
|
|
|
|
func (db *DB) hSetItem(key []byte, field []byte, value []byte) (int64, error) {
|
|
t := db.hashTx
|
|
|
|
ek := db.hEncodeHashKey(key, field)
|
|
|
|
var n int64 = 1
|
|
if v, _ := db.db.Get(ek); v != nil {
|
|
n = 0
|
|
} else {
|
|
if _, err := db.hIncrSize(key, 1); err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
|
|
t.Put(ek, value)
|
|
return n, nil
|
|
}
|
|
|
|
func (db *DB) HSet(key []byte, field []byte, value []byte) (int64, error) {
|
|
if err := checkHashKFSize(key, field); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
t := db.hashTx
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
n, err := db.hSetItem(key, field, value)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
//todo add binlog
|
|
|
|
err = t.Commit()
|
|
return n, err
|
|
}
|
|
|
|
func (db *DB) HGet(key []byte, field []byte) ([]byte, error) {
|
|
if err := checkHashKFSize(key, field); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db.db.Get(db.hEncodeHashKey(key, field))
|
|
}
|
|
|
|
func (db *DB) HMset(key []byte, args ...FVPair) error {
|
|
t := db.hashTx
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
var err error
|
|
var ek []byte
|
|
var num int64 = 0
|
|
for i := 0; i < len(args); i++ {
|
|
if err := checkHashKFSize(key, args[i].Field); err != nil {
|
|
return err
|
|
}
|
|
|
|
ek = db.hEncodeHashKey(key, args[i].Field)
|
|
|
|
if v, err := db.db.Get(ek); err != nil {
|
|
return err
|
|
} else if v == nil {
|
|
num++
|
|
}
|
|
|
|
t.Put(ek, args[i].Value)
|
|
}
|
|
|
|
if _, err = db.hIncrSize(key, num); err != nil {
|
|
return err
|
|
}
|
|
|
|
//todo add binglog
|
|
err = t.Commit()
|
|
return err
|
|
}
|
|
|
|
func (db *DB) HMget(key []byte, args [][]byte) ([]interface{}, error) {
|
|
var ek []byte
|
|
var v []byte
|
|
var err error
|
|
|
|
r := make([]interface{}, len(args))
|
|
for i := 0; i < len(args); i++ {
|
|
if err := checkHashKFSize(key, args[i]); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ek = db.hEncodeHashKey(key, args[i])
|
|
|
|
if v, err = db.db.Get(ek); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r[i] = v
|
|
}
|
|
|
|
return r, nil
|
|
}
|
|
|
|
func (db *DB) HDel(key []byte, args [][]byte) (int64, error) {
|
|
t := db.hashTx
|
|
|
|
var ek []byte
|
|
var v []byte
|
|
var err error
|
|
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
var num int64 = 0
|
|
for i := 0; i < len(args); i++ {
|
|
if err := checkHashKFSize(key, args[i]); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
ek = db.hEncodeHashKey(key, args[i])
|
|
|
|
if v, err = db.db.Get(ek); err != nil {
|
|
return 0, err
|
|
} else if v == nil {
|
|
continue
|
|
} else {
|
|
num++
|
|
t.Delete(ek)
|
|
}
|
|
}
|
|
|
|
if _, err = db.hIncrSize(key, -num); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
err = t.Commit()
|
|
|
|
return num, err
|
|
}
|
|
|
|
func (db *DB) hIncrSize(key []byte, delta int64) (int64, error) {
|
|
t := db.hashTx
|
|
sk := db.hEncodeSizeKey(key)
|
|
|
|
var err error
|
|
var size int64 = 0
|
|
if size, err = Int64(db.db.Get(sk)); err != nil {
|
|
return 0, err
|
|
} else {
|
|
size += delta
|
|
if size <= 0 {
|
|
size = 0
|
|
t.Delete(sk)
|
|
} else {
|
|
t.Put(sk, PutInt64(size))
|
|
}
|
|
}
|
|
|
|
return size, nil
|
|
}
|
|
|
|
func (db *DB) HIncrBy(key []byte, field []byte, delta int64) (int64, error) {
|
|
if err := checkHashKFSize(key, field); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
t := db.hashTx
|
|
var ek []byte
|
|
var err error
|
|
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
ek = db.hEncodeHashKey(key, field)
|
|
|
|
var n int64 = 0
|
|
if n, err = StrInt64(db.db.Get(ek)); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
n += delta
|
|
|
|
_, err = db.hSetItem(key, field, StrPutInt64(n))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
err = t.Commit()
|
|
|
|
return n, err
|
|
}
|
|
|
|
func (db *DB) HGetAll(key []byte) ([]interface{}, error) {
|
|
if err := checkKeySize(key); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
start := db.hEncodeStartKey(key)
|
|
stop := db.hEncodeStopKey(key)
|
|
|
|
v := make([]interface{}, 0, 16)
|
|
|
|
it := db.db.Iterator(start, stop, leveldb.RangeROpen, 0, -1)
|
|
for ; it.Valid(); it.Next() {
|
|
_, k, err := db.hDecodeHashKey(it.Key())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
v = append(v, k)
|
|
v = append(v, it.Value())
|
|
}
|
|
|
|
it.Close()
|
|
|
|
return v, nil
|
|
}
|
|
|
|
func (db *DB) HKeys(key []byte) ([]interface{}, error) {
|
|
if err := checkKeySize(key); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
start := db.hEncodeStartKey(key)
|
|
stop := db.hEncodeStopKey(key)
|
|
|
|
v := make([]interface{}, 0, 16)
|
|
|
|
it := db.db.Iterator(start, stop, leveldb.RangeROpen, 0, -1)
|
|
for ; it.Valid(); it.Next() {
|
|
_, k, err := db.hDecodeHashKey(it.Key())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
v = append(v, k)
|
|
}
|
|
|
|
it.Close()
|
|
|
|
return v, nil
|
|
}
|
|
|
|
func (db *DB) HValues(key []byte) ([]interface{}, error) {
|
|
if err := checkKeySize(key); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
start := db.hEncodeStartKey(key)
|
|
stop := db.hEncodeStopKey(key)
|
|
|
|
v := make([]interface{}, 0, 16)
|
|
|
|
it := db.db.Iterator(start, stop, leveldb.RangeROpen, 0, -1)
|
|
for ; it.Valid(); it.Next() {
|
|
v = append(v, it.Value())
|
|
}
|
|
|
|
it.Close()
|
|
|
|
return v, nil
|
|
}
|
|
|
|
func (db *DB) HClear(key []byte) (int64, error) {
|
|
if err := checkKeySize(key); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
sk := db.hEncodeSizeKey(key)
|
|
start := db.hEncodeStartKey(key)
|
|
stop := db.hEncodeStopKey(key)
|
|
|
|
t := db.hashTx
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
var num int64 = 0
|
|
it := db.db.Iterator(start, stop, leveldb.RangeROpen, 0, -1)
|
|
for ; it.Valid(); it.Next() {
|
|
t.Delete(it.Key())
|
|
num++
|
|
}
|
|
|
|
it.Close()
|
|
|
|
t.Delete(sk)
|
|
|
|
err := t.Commit()
|
|
return num, err
|
|
}
|
|
|
|
func (db *DB) HFlush() (drop int64, err error) {
|
|
t := db.kvTx
|
|
t.Lock()
|
|
defer t.Unlock()
|
|
|
|
minKey := make([]byte, 2)
|
|
minKey[0] = db.index
|
|
minKey[1] = hashType
|
|
|
|
maxKey := make([]byte, 2)
|
|
maxKey[0] = db.index
|
|
maxKey[1] = hSizeType + 1
|
|
|
|
it := db.db.Iterator(minKey, maxKey, leveldb.RangeROpen, 0, -1)
|
|
for ; it.Valid(); it.Next() {
|
|
t.Delete(it.Key())
|
|
drop++
|
|
}
|
|
|
|
err = t.Commit()
|
|
return
|
|
}
|