mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-12 20:20:16 +08:00
refactor project structure
This commit is contained in:
hdt3213
333
keys.go
Normal file
333
keys.go
Normal file
@@ -0,0 +1,333 @@
|
||||
package godis
|
||||
|
||||
import (
|
||||
"github.com/hdt3213/godis/datastruct/dict"
|
||||
"github.com/hdt3213/godis/datastruct/list"
|
||||
"github.com/hdt3213/godis/datastruct/set"
|
||||
"github.com/hdt3213/godis/datastruct/sortedset"
|
||||
"github.com/hdt3213/godis/interface/redis"
|
||||
"github.com/hdt3213/godis/lib/wildcard"
|
||||
"github.com/hdt3213/godis/redis/reply"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Del removes a key from db
|
||||
func Del(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) == 0 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'del' command")
|
||||
}
|
||||
keys := make([]string, len(args))
|
||||
for i, v := range args {
|
||||
keys[i] = string(v)
|
||||
}
|
||||
|
||||
db.Locks(keys...)
|
||||
defer db.UnLocks(keys...)
|
||||
|
||||
deleted := db.Removes(keys...)
|
||||
if deleted > 0 {
|
||||
db.AddAof(makeAofCmd("del", args))
|
||||
}
|
||||
return reply.MakeIntReply(int64(deleted))
|
||||
}
|
||||
|
||||
// Exists checks if a is existed in db
|
||||
func Exists(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'exists' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
_, exists := db.GetEntity(key)
|
||||
if exists {
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
// FlushDB removes all data in current db
|
||||
func FlushDB(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 0 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'flushdb' command")
|
||||
}
|
||||
db.Flush()
|
||||
db.AddAof(makeAofCmd("flushdb", args))
|
||||
return &reply.OkReply{}
|
||||
}
|
||||
|
||||
// FlushAll removes all data in all db
|
||||
func FlushAll(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 0 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'flushall' command")
|
||||
}
|
||||
db.Flush()
|
||||
db.AddAof(makeAofCmd("flushdb", args))
|
||||
return &reply.OkReply{}
|
||||
}
|
||||
|
||||
// Type returns the type of entity, including: string, list, hash, set and zset
|
||||
func Type(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'type' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
entity, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeStatusReply("none")
|
||||
}
|
||||
switch entity.Data.(type) {
|
||||
case []byte:
|
||||
return reply.MakeStatusReply("string")
|
||||
case *list.LinkedList:
|
||||
return reply.MakeStatusReply("list")
|
||||
case dict.Dict:
|
||||
return reply.MakeStatusReply("hash")
|
||||
case *set.Set:
|
||||
return reply.MakeStatusReply("set")
|
||||
case *sortedset.SortedSet:
|
||||
return reply.MakeStatusReply("zset")
|
||||
}
|
||||
return &reply.UnknownErrReply{}
|
||||
}
|
||||
|
||||
// Rename a key
|
||||
func Rename(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'rename' command")
|
||||
}
|
||||
src := string(args[0])
|
||||
dest := string(args[1])
|
||||
|
||||
db.Locks(src, dest)
|
||||
defer db.UnLocks(src, dest)
|
||||
|
||||
entity, ok := db.GetEntity(src)
|
||||
if !ok {
|
||||
return reply.MakeErrReply("no such key")
|
||||
}
|
||||
rawTTL, hasTTL := db.ttlMap.Get(src)
|
||||
db.PutEntity(dest, entity)
|
||||
db.Remove(src)
|
||||
if hasTTL {
|
||||
db.Persist(src) // clean src and dest with their ttl
|
||||
db.Persist(dest)
|
||||
expireTime, _ := rawTTL.(time.Time)
|
||||
db.Expire(dest, expireTime)
|
||||
}
|
||||
db.AddAof(makeAofCmd("rename", args))
|
||||
return &reply.OkReply{}
|
||||
}
|
||||
|
||||
// RenameNx a key, only if the new key does not exist
|
||||
func RenameNx(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'renamenx' command")
|
||||
}
|
||||
src := string(args[0])
|
||||
dest := string(args[1])
|
||||
|
||||
db.Locks(src, dest)
|
||||
defer db.UnLocks(src, dest)
|
||||
|
||||
_, ok := db.GetEntity(dest)
|
||||
if ok {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
entity, ok := db.GetEntity(src)
|
||||
if !ok {
|
||||
return reply.MakeErrReply("no such key")
|
||||
}
|
||||
rawTTL, hasTTL := db.ttlMap.Get(src)
|
||||
db.Removes(src, dest) // clean src and dest with their ttl
|
||||
db.PutEntity(dest, entity)
|
||||
if hasTTL {
|
||||
db.Persist(src) // clean src and dest with their ttl
|
||||
db.Persist(dest)
|
||||
expireTime, _ := rawTTL.(time.Time)
|
||||
db.Expire(dest, expireTime)
|
||||
}
|
||||
db.AddAof(makeAofCmd("renamenx", args))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// Expire sets a key's time to live in seconds
|
||||
func Expire(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'expire' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
|
||||
ttlArg, err := strconv.ParseInt(string(args[1]), 10, 64)
|
||||
if err != nil {
|
||||
return reply.MakeErrReply("ERR value is not an integer or out of range")
|
||||
}
|
||||
ttl := time.Duration(ttlArg) * time.Second
|
||||
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
expireAt := time.Now().Add(ttl)
|
||||
db.Expire(key, expireAt)
|
||||
db.AddAof(makeExpireCmd(key, expireAt))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// ExpireAt sets a key's expiration in unix timestamp
|
||||
func ExpireAt(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'expireat' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
|
||||
raw, err := strconv.ParseInt(string(args[1]), 10, 64)
|
||||
if err != nil {
|
||||
return reply.MakeErrReply("ERR value is not an integer or out of range")
|
||||
}
|
||||
expireTime := time.Unix(raw, 0)
|
||||
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
db.Expire(key, expireTime)
|
||||
db.AddAof(makeExpireCmd(key, expireTime))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// PExpire sets a key's time to live in milliseconds
|
||||
func PExpire(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'pexpire' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
|
||||
ttlArg, err := strconv.ParseInt(string(args[1]), 10, 64)
|
||||
if err != nil {
|
||||
return reply.MakeErrReply("ERR value is not an integer or out of range")
|
||||
}
|
||||
ttl := time.Duration(ttlArg) * time.Millisecond
|
||||
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
expireTime := time.Now().Add(ttl)
|
||||
db.Expire(key, expireTime)
|
||||
db.AddAof(makeExpireCmd(key, expireTime))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// PExpireAt sets a key's expiration in unix timestamp specified in milliseconds
|
||||
func PExpireAt(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 2 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'pexpireat' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
|
||||
raw, err := strconv.ParseInt(string(args[1]), 10, 64)
|
||||
if err != nil {
|
||||
return reply.MakeErrReply("ERR value is not an integer or out of range")
|
||||
}
|
||||
expireTime := time.Unix(0, raw*int64(time.Millisecond))
|
||||
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
db.Expire(key, expireTime)
|
||||
|
||||
db.AddAof(makeExpireCmd(key, expireTime))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// TTL returns a key's time to live in seconds
|
||||
func TTL(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'ttl' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(-2)
|
||||
}
|
||||
|
||||
raw, exists := db.ttlMap.Get(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(-1)
|
||||
}
|
||||
expireTime, _ := raw.(time.Time)
|
||||
ttl := expireTime.Sub(time.Now())
|
||||
return reply.MakeIntReply(int64(ttl / time.Second))
|
||||
}
|
||||
|
||||
// PTTL returns a key's time to live in milliseconds
|
||||
func PTTL(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'pttl' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(-2)
|
||||
}
|
||||
|
||||
raw, exists := db.ttlMap.Get(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(-1)
|
||||
}
|
||||
expireTime, _ := raw.(time.Time)
|
||||
ttl := expireTime.Sub(time.Now())
|
||||
return reply.MakeIntReply(int64(ttl / time.Millisecond))
|
||||
}
|
||||
|
||||
// Persist removes expiration from a key
|
||||
func Persist(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'persist' command")
|
||||
}
|
||||
key := string(args[0])
|
||||
_, exists := db.GetEntity(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
_, exists = db.ttlMap.Get(key)
|
||||
if !exists {
|
||||
return reply.MakeIntReply(0)
|
||||
}
|
||||
|
||||
db.Persist(key)
|
||||
db.AddAof(makeAofCmd("persist", args))
|
||||
return reply.MakeIntReply(1)
|
||||
}
|
||||
|
||||
// BGRewriteAOF asynchronously rewrites Append-Only-File
|
||||
func BGRewriteAOF(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 0 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'bgrewriteaof' command")
|
||||
}
|
||||
go db.aofRewrite()
|
||||
return reply.MakeStatusReply("Background append only file rewriting started")
|
||||
}
|
||||
|
||||
// Keys returns all keys matching the given pattern
|
||||
func Keys(db *DB, args [][]byte) redis.Reply {
|
||||
if len(args) != 1 {
|
||||
return reply.MakeErrReply("ERR wrong number of arguments for 'keys' command")
|
||||
}
|
||||
pattern := wildcard.CompilePattern(string(args[0]))
|
||||
result := make([][]byte, 0)
|
||||
db.data.ForEach(func(key string, val interface{}) bool {
|
||||
if pattern.IsMatch(key) {
|
||||
result = append(result, []byte(key))
|
||||
}
|
||||
return true
|
||||
})
|
||||
return reply.MakeMultiBulkReply(result)
|
||||
}
|
||||
Reference in New Issue
Block a user