Add info command

This commit is contained in:
bijingrui
2023-03-15 22:20:40 +08:00
committed by finley
parent 11c0385241
commit d0a7a215ba
7 changed files with 105 additions and 43 deletions

View File

@@ -5,6 +5,7 @@ import (
"github.com/hdt3213/godis/config"
"github.com/hdt3213/godis/interface/redis"
"github.com/hdt3213/godis/redis/protocol"
"github.com/hdt3213/godis/tcp"
"os"
"runtime"
"strings"
@@ -25,21 +26,28 @@ func Ping(c redis.Connection, args [][]byte) redis.Reply {
// Info the information of the godis server returned by the INFO command
func Info(c redis.Connection, args [][]byte) redis.Reply {
if len(args) == 1 {
return protocol.MakeBulkReply(GenGodisInfoString())
infoCommandList := [...]string{"server", "client", "cluster"}
var allSection []byte
for _, s := range infoCommandList {
allSection = append(allSection, GenGodisInfoString(s)...)
}
return protocol.MakeBulkReply(allSection)
} else if len(args) == 2 {
section := strings.ToLower(string(args[1]))
switch section {
case "server":
reply := GenGodisInfoString()
reply := GenGodisInfoString("server")
return protocol.MakeBulkReply(reply)
case "client":
return protocol.MakeBulkReply(GenGodisInfoString("client"))
case "cluster":
return protocol.MakeBulkReply(GenGodisInfoString("cluster"))
default:
return protocol.MakeNullBulkReply()
}
} else {
return protocol.MakeErrReply("ERR wrong number of arguments for 'info' command")
}
return &protocol.NullBulkReply{}
return protocol.MakeErrReply("ERR wrong number of arguments for 'info' command")
}
// Auth validate client's password
@@ -65,44 +73,75 @@ func isAuthenticated(c redis.Connection) bool {
return c.GetPassword() == config.Properties.RequirePass
}
func GenGodisInfoString() []byte {
func GenGodisInfoString(section string) []byte {
startUpTimeFromNow := getGodisRunningTime()
s := fmt.Sprintf("# Server\r\n"+
"godis_version:%s\r\n"+
//"godis_git_sha1:%s\r\n"+
//"godis_git_dirty:%d\r\n"+
//"godis_build_id:%s\r\n"+
"godis_mode:%s\r\n"+
"os:%s %s\r\n"+
"arch_bits:%d\r\n"+
//"multiplexing_api:%s\r\n"+
"go_version:%s\r\n"+
"process_id:%d\r\n"+
"run_id:%s\r\n"+
"tcp_port:%d\r\n"+
"uptime_in_seconds:%d\r\n"+
"uptime_in_days:%d\r\n"+
//"hz:%d\r\n"+
//"lru_clock:%d\r\n"+
"config_file:%s\r\n",
godisVersion,
//TODO,
//TODO,
//TODO,
getGodisRunningMode(),
runtime.GOOS, runtime.GOARCH,
32<<(^uint(0)>>63),
//TODO,
runtime.Version(),
os.Getpid(),
config.Properties.RunID,
config.Properties.Port,
startUpTimeFromNow,
startUpTimeFromNow/time.Duration(3600*24),
//TODO,
//TODO,
config.Properties.CfPath)
return []byte(s)
switch section {
case "server":
s := fmt.Sprintf("# Server\r\n"+
"godis_version:%s\r\n"+
//"godis_git_sha1:%s\r\n"+
//"godis_git_dirty:%d\r\n"+
//"godis_build_id:%s\r\n"+
"godis_mode:%s\r\n"+
"os:%s %s\r\n"+
"arch_bits:%d\r\n"+
//"multiplexing_api:%s\r\n"+
"go_version:%s\r\n"+
"process_id:%d\r\n"+
"run_id:%s\r\n"+
"tcp_port:%d\r\n"+
"uptime_in_seconds:%d\r\n"+
"uptime_in_days:%d\r\n"+
//"hz:%d\r\n"+
//"lru_clock:%d\r\n"+
"config_file:%s\r\n",
godisVersion,
//TODO,
//TODO,
//TODO,
getGodisRunningMode(),
runtime.GOOS, runtime.GOARCH,
32<<(^uint(0)>>63),
//TODO,
runtime.Version(),
os.Getpid(),
config.Properties.RunID,
config.Properties.Port,
startUpTimeFromNow,
startUpTimeFromNow/time.Duration(3600*24),
//TODO,
//TODO,
config.Properties.CfPath)
return []byte(s)
case "client":
s := fmt.Sprintf("# Clients\r\n"+
"connected_clients:%d\r\n",
//"client_recent_max_input_buffer:%d\r\n"+
//"client_recent_max_output_buffer:%d\r\n"+
//"blocked_clients:%d\n",
tcp.ClientCounter,
//TODO,
//TODO,
//TODO,
)
return []byte(s)
case "cluster":
if getGodisRunningMode() == config.ClusterMode {
s := fmt.Sprintf("# Cluster\r\n"+
"cluster_enabled:%s\r\n",
"1",
)
return []byte(s)
} else {
s := fmt.Sprintf("# Cluster\r\n"+
"cluster_enabled:%s\r\n",
"0",
)
return []byte(s)
}
}
return []byte("")
}
// getGodisRunningMode return godis running mode