forked from mirror/redcon
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/tidwall/redcon"
|
|
)
|
|
|
|
var addr = ":6380"
|
|
|
|
func main() {
|
|
var mu sync.RWMutex
|
|
var items = make(map[string]string)
|
|
go log.Printf("started server at %s", addr)
|
|
err := redcon.ListenAndServe(addr,
|
|
func(conn redcon.Conn, commands [][]string) {
|
|
for _, args := range commands {
|
|
switch strings.ToLower(args[0]) {
|
|
default:
|
|
conn.WriteError("ERR unknown command '" + args[0] + "'")
|
|
case "hijack":
|
|
hconn := conn.Hijack()
|
|
log.Printf("connection is hijacked")
|
|
go func() {
|
|
defer hconn.Close()
|
|
hconn.WriteString("OK")
|
|
hconn.Flush()
|
|
}()
|
|
return
|
|
case "ping":
|
|
conn.WriteString("PONG")
|
|
case "quit":
|
|
conn.WriteString("OK")
|
|
conn.Close()
|
|
case "set":
|
|
if len(args) != 3 {
|
|
conn.WriteError("ERR wrong number of arguments for '" + args[0] + "' command")
|
|
continue
|
|
}
|
|
mu.Lock()
|
|
items[args[1]] = args[2]
|
|
mu.Unlock()
|
|
conn.WriteString("OK")
|
|
case "get":
|
|
if len(args) != 2 {
|
|
conn.WriteError("ERR wrong number of arguments for '" + args[0] + "' command")
|
|
continue
|
|
}
|
|
mu.RLock()
|
|
val, ok := items[args[1]]
|
|
mu.RUnlock()
|
|
if !ok {
|
|
conn.WriteNull()
|
|
} else {
|
|
conn.WriteBulk(val)
|
|
}
|
|
case "del":
|
|
if len(args) != 2 {
|
|
conn.WriteError("ERR wrong number of arguments for '" + args[0] + "' command")
|
|
continue
|
|
}
|
|
mu.Lock()
|
|
_, ok := items[args[1]]
|
|
delete(items, args[1])
|
|
mu.Unlock()
|
|
if !ok {
|
|
conn.WriteInt(0)
|
|
} else {
|
|
conn.WriteInt(1)
|
|
}
|
|
}
|
|
}
|
|
},
|
|
func(conn redcon.Conn) bool {
|
|
// use this function to accept or deny the connection.
|
|
// log.Printf("accept: %s", conn.RemoteAddr())
|
|
return true
|
|
},
|
|
func(conn redcon.Conn, err error) {
|
|
// this is called when the connection has been closed
|
|
// log.Printf("closed: %s, err: %v", conn.RemoteAddr(), err)
|
|
},
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|