mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-04 16:32:41 +08:00
optimize code structure
This commit is contained in:
hdt3213
@@ -1,226 +0,0 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/hdt3213/godis/aof"
|
||||
"github.com/hdt3213/godis/config"
|
||||
"github.com/hdt3213/godis/interface/database"
|
||||
"github.com/hdt3213/godis/interface/redis"
|
||||
"github.com/hdt3213/godis/lib/logger"
|
||||
"github.com/hdt3213/godis/lib/utils"
|
||||
"github.com/hdt3213/godis/pubsub"
|
||||
"github.com/hdt3213/godis/redis/protocol"
|
||||
"runtime/debug"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// MultiDB is a set of multiple database set
|
||||
type MultiDB struct {
|
||||
dbSet []*DB
|
||||
|
||||
// handle publish/subscribe
|
||||
hub *pubsub.Hub
|
||||
// handle aof persistence
|
||||
aofHandler *aof.Handler
|
||||
}
|
||||
|
||||
// NewStandaloneServer creates a standalone redis server, with multi database and all other funtions
|
||||
func NewStandaloneServer() *MultiDB {
|
||||
mdb := &MultiDB{}
|
||||
if config.Properties.Databases == 0 {
|
||||
config.Properties.Databases = 16
|
||||
}
|
||||
mdb.dbSet = make([]*DB, config.Properties.Databases)
|
||||
for i := range mdb.dbSet {
|
||||
singleDB := makeDB()
|
||||
singleDB.index = i
|
||||
mdb.dbSet[i] = singleDB
|
||||
}
|
||||
mdb.hub = pubsub.MakeHub()
|
||||
if config.Properties.AppendOnly {
|
||||
aofHandler, err := aof.NewAOFHandler(mdb, func() database.EmbedDB {
|
||||
return MakeBasicMultiDB()
|
||||
})
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
mdb.aofHandler = aofHandler
|
||||
for _, db := range mdb.dbSet {
|
||||
// avoid closure
|
||||
singleDB := db
|
||||
singleDB.addAof = func(line CmdLine) {
|
||||
mdb.aofHandler.AddAof(singleDB.index, line)
|
||||
}
|
||||
}
|
||||
}
|
||||
return mdb
|
||||
}
|
||||
|
||||
// MakeBasicMultiDB create a MultiDB only with basic abilities for aof rewrite and other usages
|
||||
func MakeBasicMultiDB() *MultiDB {
|
||||
mdb := &MultiDB{}
|
||||
mdb.dbSet = make([]*DB, config.Properties.Databases)
|
||||
for i := range mdb.dbSet {
|
||||
mdb.dbSet[i] = makeBasicDB()
|
||||
}
|
||||
return mdb
|
||||
}
|
||||
|
||||
// Exec executes command
|
||||
// parameter `cmdLine` contains command and its arguments, for example: "set key value"
|
||||
func (mdb *MultiDB) Exec(c redis.Connection, cmdLine [][]byte) (result redis.Reply) {
|
||||
defer func() {
|
||||
if err := recover(); err != nil {
|
||||
logger.Warn(fmt.Sprintf("error occurs: %v\n%s", err, string(debug.Stack())))
|
||||
result = &protocol.UnknownErrReply{}
|
||||
}
|
||||
}()
|
||||
|
||||
cmdName := strings.ToLower(string(cmdLine[0]))
|
||||
// authenticate
|
||||
if cmdName == "auth" {
|
||||
return Auth(c, cmdLine[1:])
|
||||
}
|
||||
if !isAuthenticated(c) {
|
||||
return protocol.MakeErrReply("NOAUTH Authentication required")
|
||||
}
|
||||
|
||||
// todo: merge special commands into router
|
||||
// special commands
|
||||
if cmdName == "subscribe" {
|
||||
if len(cmdLine) < 2 {
|
||||
return protocol.MakeArgNumErrReply("subscribe")
|
||||
}
|
||||
return pubsub.Subscribe(mdb.hub, c, cmdLine[1:])
|
||||
} else if cmdName == "publish" {
|
||||
return pubsub.Publish(mdb.hub, cmdLine[1:])
|
||||
} else if cmdName == "unsubscribe" {
|
||||
return pubsub.UnSubscribe(mdb.hub, c, cmdLine[1:])
|
||||
} else if cmdName == "bgrewriteaof" {
|
||||
// aof.go imports router.go, router.go cannot import BGRewriteAOF from aof.go
|
||||
return BGRewriteAOF(mdb, cmdLine[1:])
|
||||
} else if cmdName == "rewriteaof" {
|
||||
return RewriteAOF(mdb, cmdLine[1:])
|
||||
} else if cmdName == "flushall" {
|
||||
return mdb.flushAll()
|
||||
} else if cmdName == "select" {
|
||||
if c != nil && c.InMultiState() {
|
||||
return protocol.MakeErrReply("cannot select database within multi")
|
||||
}
|
||||
if len(cmdLine) != 2 {
|
||||
return protocol.MakeArgNumErrReply("select")
|
||||
}
|
||||
return execSelect(c, mdb, cmdLine[1:])
|
||||
}
|
||||
// todo: support multi database transaction
|
||||
|
||||
// normal commands
|
||||
dbIndex := c.GetDBIndex()
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
return protocol.MakeErrReply("ERR DB index is out of range")
|
||||
}
|
||||
selectedDB := mdb.dbSet[dbIndex]
|
||||
return selectedDB.Exec(c, cmdLine)
|
||||
}
|
||||
|
||||
// AfterClientClose does some clean after client close connection
|
||||
func (mdb *MultiDB) AfterClientClose(c redis.Connection) {
|
||||
pubsub.UnsubscribeAll(mdb.hub, c)
|
||||
}
|
||||
|
||||
// Close graceful shutdown database
|
||||
func (mdb *MultiDB) Close() {
|
||||
if mdb.aofHandler != nil {
|
||||
mdb.aofHandler.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func execSelect(c redis.Connection, mdb *MultiDB, args [][]byte) redis.Reply {
|
||||
dbIndex, err := strconv.Atoi(string(args[0]))
|
||||
if err != nil {
|
||||
return protocol.MakeErrReply("ERR invalid DB index")
|
||||
}
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
return protocol.MakeErrReply("ERR DB index is out of range")
|
||||
}
|
||||
c.SelectDB(dbIndex)
|
||||
return protocol.MakeOkReply()
|
||||
}
|
||||
|
||||
func (mdb *MultiDB) flushAll() redis.Reply {
|
||||
for _, db := range mdb.dbSet {
|
||||
db.Flush()
|
||||
}
|
||||
if mdb.aofHandler != nil {
|
||||
mdb.aofHandler.AddAof(0, utils.ToCmdLine("FlushAll"))
|
||||
}
|
||||
return &protocol.OkReply{}
|
||||
}
|
||||
|
||||
// ForEach traverses all the keys in the given database
|
||||
func (mdb *MultiDB) ForEach(dbIndex int, cb func(key string, data *database.DataEntity, expiration *time.Time) bool) {
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
return
|
||||
}
|
||||
db := mdb.dbSet[dbIndex]
|
||||
db.ForEach(cb)
|
||||
}
|
||||
|
||||
// ExecMulti executes multi commands transaction Atomically and Isolated
|
||||
func (mdb *MultiDB) ExecMulti(conn redis.Connection, watching map[string]uint32, cmdLines []CmdLine) redis.Reply {
|
||||
if conn.GetDBIndex() >= len(mdb.dbSet) {
|
||||
return protocol.MakeErrReply("ERR DB index is out of range")
|
||||
}
|
||||
db := mdb.dbSet[conn.GetDBIndex()]
|
||||
return db.ExecMulti(conn, watching, cmdLines)
|
||||
}
|
||||
|
||||
// RWLocks lock keys for writing and reading
|
||||
func (mdb *MultiDB) RWLocks(dbIndex int, writeKeys []string, readKeys []string) {
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
panic("ERR DB index is out of range")
|
||||
}
|
||||
db := mdb.dbSet[dbIndex]
|
||||
db.RWLocks(writeKeys, readKeys)
|
||||
}
|
||||
|
||||
// RWUnLocks unlock keys for writing and reading
|
||||
func (mdb *MultiDB) RWUnLocks(dbIndex int, writeKeys []string, readKeys []string) {
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
panic("ERR DB index is out of range")
|
||||
}
|
||||
db := mdb.dbSet[dbIndex]
|
||||
db.RWUnLocks(writeKeys, readKeys)
|
||||
}
|
||||
|
||||
// GetUndoLogs return rollback commands
|
||||
func (mdb *MultiDB) GetUndoLogs(dbIndex int, cmdLine [][]byte) []CmdLine {
|
||||
if dbIndex >= len(mdb.dbSet) {
|
||||
panic("ERR DB index is out of range")
|
||||
}
|
||||
db := mdb.dbSet[dbIndex]
|
||||
return db.GetUndoLogs(cmdLine)
|
||||
}
|
||||
|
||||
// ExecWithLock executes normal commands, invoker should provide locks
|
||||
func (mdb *MultiDB) ExecWithLock(conn redis.Connection, cmdLine [][]byte) redis.Reply {
|
||||
if conn.GetDBIndex() >= len(mdb.dbSet) {
|
||||
panic("ERR DB index is out of range")
|
||||
}
|
||||
db := mdb.dbSet[conn.GetDBIndex()]
|
||||
return db.execWithLock(cmdLine)
|
||||
}
|
||||
|
||||
// BGRewriteAOF asynchronously rewrites Append-Only-File
|
||||
func BGRewriteAOF(db *MultiDB, args [][]byte) redis.Reply {
|
||||
go db.aofHandler.Rewrite()
|
||||
return protocol.MakeStatusReply("Background append only file rewriting started")
|
||||
}
|
||||
|
||||
// RewriteAOF start Append-Only-File rewriting and blocked until it finished
|
||||
func RewriteAOF(db *MultiDB, args [][]byte) redis.Reply {
|
||||
db.aofHandler.Rewrite()
|
||||
return protocol.MakeStatusReply("Background append only file rewriting started")
|
||||
}
|
||||
Reference in New Issue
Block a user