76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
rConcernS, ok := os.LookupEnv("RCONCERN")
|
|
if !ok {
|
|
rConcernS = "1"
|
|
}
|
|
wConcernS, ok := os.LookupEnv("WCONCERN")
|
|
if !ok {
|
|
wConcernS = "1"
|
|
}
|
|
rConcern, err := strconv.Atoi(rConcernS)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
wConcern, err := strconv.Atoi(wConcernS)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
client, err := New(rConcern, wConcern, strings.Split(os.Getenv("ADDR"), ",")...)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
reader := bufio.NewReader(os.Stdin)
|
|
for {
|
|
fmt.Print("> ")
|
|
line, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
do(client, strings.TrimSpace(line))
|
|
}
|
|
}
|
|
|
|
func do(client *Client, line string) {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
log.Printf("err: %v", err)
|
|
}
|
|
}()
|
|
switch strings.ToLower(strings.Split(line, " ")[0]) {
|
|
case "get":
|
|
get(client, line)
|
|
case "set":
|
|
set(client, line)
|
|
case "put":
|
|
set(client, line)
|
|
default:
|
|
log.Printf("unknown command %v", line)
|
|
}
|
|
}
|
|
|
|
func get(client *Client, line string) {
|
|
words := strings.Split(line, " ")
|
|
key := words[1]
|
|
v, err := client.Get(key)
|
|
log.Printf("get: %v: %s", err, v)
|
|
}
|
|
|
|
func set(client *Client, line string) {
|
|
words := strings.Split(line, " ")
|
|
key := words[1]
|
|
value := strings.Join(words[2:], " ")
|
|
err := client.Set(key, []byte(value))
|
|
log.Printf("set: %v", err)
|
|
}
|