代码拉取完成,页面将自动刷新
package rosedb
import (
"bytes"
"sync"
"time"
"github.com/roseduan/rosedb/ds/hash"
"github.com/roseduan/rosedb/storage"
)
// HashIdx hash index.
type HashIdx struct {
mu *sync.RWMutex
indexes *hash.Hash
}
// create a new hash index.
func newHashIdx() *HashIdx {
return &HashIdx{indexes: hash.New(), mu: new(sync.RWMutex)}
}
// HSet sets field in the hash stored at key to value. If key does not exist, a new key holding a hash is created.
// If field already exists in the hash, it is overwritten.
// Return num of elements in hash of the specified key.
func (db *RoseDB) HSet(key []byte, field []byte, value []byte) (res int, err error) {
if err = db.checkKeyValue(key, value); err != nil {
return
}
// If the existed value is the same as the set value, nothing will be done.
oldVal := db.HGet(key, field)
if bytes.Compare(oldVal, value) == 0 {
return
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
e := storage.NewEntry(key, value, field, Hash, HashHSet)
if err = db.store(e); err != nil {
return
}
res = db.hashIndex.indexes.HSet(string(key), string(field), value)
return
}
// HSetNx Sets field in the hash stored at key to value, only if field does not yet exist.
// If key does not exist, a new key holding a hash is created. If field already exists, this operation has no effect.
// Return if the operation is successful.
func (db *RoseDB) HSetNx(key, field, value []byte) (res int, err error) {
if err = db.checkKeyValue(key, value); err != nil {
return
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
if res = db.hashIndex.indexes.HSetNx(string(key), string(field), value); res == 1 {
e := storage.NewEntry(key, value, field, Hash, HashHSet)
if err = db.store(e); err != nil {
return
}
}
return
}
// HGet returns the value associated with field in the hash stored at key.
func (db *RoseDB) HGet(key, field []byte) []byte {
if err := db.checkKeyValue(key, nil); err != nil {
return nil
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return nil
}
return db.hashIndex.indexes.HGet(string(key), string(field))
}
// HGetAll returns all fields and values of the hash stored at key.
// In the returned value, every field name is followed by its value, so the length of the reply is twice the size of the hash.
func (db *RoseDB) HGetAll(key []byte) [][]byte {
if err := db.checkKeyValue(key, nil); err != nil {
return nil
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return nil
}
return db.hashIndex.indexes.HGetAll(string(key))
}
// HMSet set multiple hash fields to multiple values
func (db *RoseDB) HMSet(key []byte, values ...[]byte) error {
if len(values)%2 != 0 {
return ErrWrongNumberOfArgs
}
if err := db.checkKeyValue(key, nil); err != nil {
return err
}
fields := make([][]byte, 0)
for i := 0; i < len(values); i += 2 {
fields = append(fields, values[i])
}
existVals := db.HMGet(key, fields...)
// field1 value1 field2 value2 ...
insertVals := make([][]byte, 0)
if existVals == nil {
// existVals means key expired
insertVals = values
} else {
for i := 0; i < len(existVals); i++ {
// If the existed value is the same as the set value, pass this field and value
if bytes.Compare(values[i*2+1], existVals[i]) != 0 {
insertVals = append(insertVals, fields[i], values[i*2+1])
}
}
}
// check all fields and values.
if err := db.checkKeyValue(key, insertVals...); err != nil {
return err
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
for i := 0; i < len(insertVals); i += 2 {
e := storage.NewEntry(key, insertVals[i+1], insertVals[i], Hash, HashHSet)
if err := db.store(e); err != nil {
return err
}
db.hashIndex.indexes.HSet(string(key), string(insertVals[i]), insertVals[i+1])
}
return nil
}
// HMGet get the values of all the given hash fields
func (db *RoseDB) HMGet(key []byte, fields ...[]byte) [][]byte {
if err := db.checkKeyValue(key, nil); err != nil {
return nil
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return nil
}
values := make([][]byte, 0)
for _, field := range fields {
value := db.hashIndex.indexes.HGet(string(key), string(field))
values = append(values, value)
}
return values
}
// HDel removes the specified fields from the hash stored at key.
// Specified fields that do not exist within this hash are ignored.
// If key does not exist, it is treated as an empty hash and this command returns false.
func (db *RoseDB) HDel(key []byte, field ...[]byte) (res int, err error) {
if err = db.checkKeyValue(key, nil); err != nil {
return
}
if field == nil || len(field) == 0 {
return
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
for _, f := range field {
if ok := db.hashIndex.indexes.HDel(string(key), string(f)); ok == 1 {
e := storage.NewEntry(key, nil, f, Hash, HashHDel)
if err = db.store(e); err != nil {
return
}
res++
}
}
return
}
// HKeyExists returns if the key is existed in hash.
func (db *RoseDB) HKeyExists(key []byte) (ok bool) {
if err := db.checkKeyValue(key, nil); err != nil {
return
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return
}
return db.hashIndex.indexes.HKeyExists(string(key))
}
// HExists returns if field is an existing field in the hash stored at key.
func (db *RoseDB) HExists(key, field []byte) (ok bool) {
if err := db.checkKeyValue(key, nil); err != nil {
return
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return
}
return db.hashIndex.indexes.HExists(string(key), string(field))
}
// HLen returns the number of fields contained in the hash stored at key.
func (db *RoseDB) HLen(key []byte) int {
if err := db.checkKeyValue(key, nil); err != nil {
return 0
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return 0
}
return db.hashIndex.indexes.HLen(string(key))
}
// HKeys returns all field names in the hash stored at key.
func (db *RoseDB) HKeys(key []byte) (val []string) {
if err := db.checkKeyValue(key, nil); err != nil {
return
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return nil
}
return db.hashIndex.indexes.HKeys(string(key))
}
// HVals returns all values in the hash stored at key.
func (db *RoseDB) HVals(key []byte) (val [][]byte) {
if err := db.checkKeyValue(key, nil); err != nil {
return
}
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return nil
}
return db.hashIndex.indexes.HVals(string(key))
}
// HClear clear the key in hash.
func (db *RoseDB) HClear(key []byte) (err error) {
if err = db.checkKeyValue(key, nil); err != nil {
return
}
if !db.HKeyExists(key) {
return ErrKeyNotExist
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
e := storage.NewEntryNoExtra(key, nil, Hash, HashHClear)
if err := db.store(e); err != nil {
return err
}
db.hashIndex.indexes.HClear(string(key))
delete(db.expires[Hash], string(key))
return
}
// HExpire set expired time for a hash key.
func (db *RoseDB) HExpire(key []byte, duration int64) (err error) {
if duration <= 0 {
return ErrInvalidTTL
}
if err = db.checkKeyValue(key, nil); err != nil {
return
}
if !db.HKeyExists(key) {
return ErrKeyNotExist
}
db.hashIndex.mu.Lock()
defer db.hashIndex.mu.Unlock()
deadline := time.Now().Unix() + duration
e := storage.NewEntryWithExpire(key, nil, deadline, Hash, HashHExpire)
if err := db.store(e); err != nil {
return err
}
db.expires[Hash][string(key)] = deadline
return
}
// HTTL return time to live for the key.
func (db *RoseDB) HTTL(key []byte) (ttl int64) {
db.hashIndex.mu.RLock()
defer db.hashIndex.mu.RUnlock()
if db.checkExpired(key, Hash) {
return
}
deadline, exist := db.expires[Hash][string(key)]
if !exist {
return
}
return deadline - time.Now().Unix()
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。