forked from mirror/redcon
remove bytes example
This commit is contained in:
parent
e1ee7d9f1e
commit
88057ab8af
|
@ -1,79 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"sync"
|
||||
|
||||
"github.com/tidwall/redcon"
|
||||
)
|
||||
|
||||
var addr = ":6380"
|
||||
|
||||
func main() {
|
||||
var mu sync.RWMutex
|
||||
var items = make(map[string][]byte)
|
||||
go log.Printf("started server at %s", addr)
|
||||
err := redcon.ListenAndServeBytes(addr,
|
||||
func(conn redcon.Conn, commands [][][]byte) {
|
||||
for _, args := range commands {
|
||||
switch string(args[0]) {
|
||||
default:
|
||||
conn.WriteError("ERR unknown command '" + string(args[0]) + "'")
|
||||
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 '" + string(args[0]) + "' command")
|
||||
continue
|
||||
}
|
||||
mu.Lock()
|
||||
items[string(args[1])] = args[2]
|
||||
mu.Unlock()
|
||||
conn.WriteString("OK")
|
||||
case "get":
|
||||
if len(args) != 2 {
|
||||
conn.WriteError("ERR wrong number of arguments for '" + string(args[0]) + "' command")
|
||||
continue
|
||||
}
|
||||
mu.RLock()
|
||||
val, ok := items[string(args[1])]
|
||||
mu.RUnlock()
|
||||
if !ok {
|
||||
conn.WriteNull()
|
||||
} else {
|
||||
conn.WriteBulkBytes(val)
|
||||
}
|
||||
case "del":
|
||||
if len(args) != 2 {
|
||||
conn.WriteError("ERR wrong number of arguments for '" + string(args[0]) + "' command")
|
||||
continue
|
||||
}
|
||||
mu.Lock()
|
||||
_, ok := items[string(args[1])]
|
||||
delete(items, string(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)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue