.
git-svn-id: svn://47.119.165.148/zhub@58 e63fbceb-bcc3-4977-ac22-735b83d8d0f4
This commit is contained in:
260
zdb/rcmd-exec.go
Normal file
260
zdb/rcmd-exec.go
Normal file
@@ -0,0 +1,260 @@
|
||||
package zdb
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/robfig/cron"
|
||||
"log"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
func execCmd(rcmd []string, conn net.Conn) {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
log.Println("execCmd Recovered:", r)
|
||||
}
|
||||
}()
|
||||
if len(rcmd) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("rcmd: " + strings.Join(rcmd, " "))
|
||||
|
||||
if len(rcmd) == 1 {
|
||||
switch strings.ToLower(rcmd[0]) {
|
||||
case "help":
|
||||
conn.Write([]byte("help-start\r\n"))
|
||||
conn.Write(retHelp)
|
||||
conn.Write([]byte("help-end\r\n"))
|
||||
return
|
||||
default:
|
||||
// subscribe|unsubscribe|daly
|
||||
if strings.Index(rcmd[0], "subscribe") == 0 || strings.Index(rcmd[0], "unsubscribe") == 0 || strings.Index(rcmd[0], "daly") == 0 {
|
||||
rcmd = strings.Split(rcmd[0], " ")
|
||||
} else {
|
||||
conn.Write([]byte("-Error: not supported! (tips: send help)\r\n"))
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
cmd := rcmd[0]
|
||||
switch cmd {
|
||||
case "decr":
|
||||
decr(rcmd, conn)
|
||||
case "incr":
|
||||
incr(rcmd, conn)
|
||||
case "get":
|
||||
get(rcmd, conn)
|
||||
case "set":
|
||||
set(rcmd, conn)
|
||||
case "subscribe":
|
||||
subscribe(rcmd, conn)
|
||||
case "unsubscribe":
|
||||
unsubscribe(rcmd, conn)
|
||||
case "publish":
|
||||
publish(rcmd, conn)
|
||||
case "daly":
|
||||
daly(rcmd, conn)
|
||||
case "timer":
|
||||
timer(rcmd, conn)
|
||||
default:
|
||||
conn.Write([]byte("-Error: default not supported:[" + strings.Join(rcmd, " ") + "]\r\n"))
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func timer(rcmd []string, conn net.Conn) {
|
||||
ztimer := zTimer[rcmd[1]]
|
||||
if ztimer == nil {
|
||||
ztimer = &ZTimer{
|
||||
conns: []*net.Conn{},
|
||||
topic: rcmd[1],
|
||||
}
|
||||
zTimer[rcmd[1]] = ztimer
|
||||
}
|
||||
|
||||
_conns := make([]*net.Conn, 0)
|
||||
for _, c := range ztimer.conns {
|
||||
if *&conn == *c {
|
||||
continue
|
||||
}
|
||||
_conns = append(_conns, c)
|
||||
}
|
||||
_conns = append(_conns, &conn)
|
||||
ztimer.conns = _conns
|
||||
|
||||
if !strings.EqualFold(ztimer.expr, rcmd[2]) {
|
||||
ztimer.expr = rcmd[2]
|
||||
if ztimer.cron != nil {
|
||||
ztimer.cron.Stop()
|
||||
}
|
||||
ztimer.cron = func() *cron.Cron {
|
||||
c := cron.New()
|
||||
c.AddFunc(ztimer.expr, func() {
|
||||
fmt.Println(time.Now().Second())
|
||||
for _, conn := range ztimer.conns {
|
||||
send(*conn, "timer", ztimer.topic)
|
||||
}
|
||||
})
|
||||
go c.Run()
|
||||
return c
|
||||
}()
|
||||
}
|
||||
|
||||
zTimer[ztimer.topic] = ztimer
|
||||
fmt.Println("xx")
|
||||
}
|
||||
|
||||
// daly topic valye 100
|
||||
func daly(rcmd []string, conn net.Conn) {
|
||||
if len(rcmd) != 4 {
|
||||
conn.Write([]byte("-Error: subscribe para number!\r\n"))
|
||||
return
|
||||
}
|
||||
|
||||
t, err := strconv.ParseInt(rcmd[3], 10, 64)
|
||||
if err != nil {
|
||||
conn.Write([]byte("-Error: " + strings.Join(rcmd, " ") + "\r\n"))
|
||||
return
|
||||
}
|
||||
|
||||
timer := time.NewTimer(time.Duration(t) * time.Millisecond)
|
||||
select {
|
||||
case <-timer.C:
|
||||
// daly => publish
|
||||
publish(rcmd[0:3], conn)
|
||||
}
|
||||
}
|
||||
|
||||
func decr(rcmd []string, conn net.Conn) {
|
||||
k := rcmd[1]
|
||||
v := zkv[k]
|
||||
if strings.EqualFold(v, "") {
|
||||
v = "0"
|
||||
}
|
||||
_v, err := strconv.Atoi(v)
|
||||
if err != nil {
|
||||
conn.Write([]byte("-Error: " + err.Error() + "\r\n"))
|
||||
}
|
||||
|
||||
v = strconv.Itoa(_v - 1)
|
||||
zkv[k] = v
|
||||
conn.Write([]byte(v + "\r\n"))
|
||||
}
|
||||
|
||||
func incr(rcmd []string, conn net.Conn) {
|
||||
k := rcmd[1]
|
||||
v := zkv[k]
|
||||
if strings.EqualFold(v, "") {
|
||||
v = "0"
|
||||
}
|
||||
_v, err := strconv.Atoi(v)
|
||||
if err != nil {
|
||||
conn.Write([]byte("- Error: " + err.Error() + "\r\n"))
|
||||
}
|
||||
|
||||
v = strconv.Itoa(_v + 1)
|
||||
zkv[k] = v
|
||||
conn.Write([]byte(v + "\r\n"))
|
||||
}
|
||||
|
||||
func get(rcmd []string, conn net.Conn) {
|
||||
k := rcmd[1]
|
||||
v := zkv[k]
|
||||
conn.Write([]byte(v + "\r\n"))
|
||||
}
|
||||
|
||||
func set(rcmd []string, conn net.Conn) {
|
||||
if len(rcmd) != 3 {
|
||||
conn.Write([]byte("-Error: set para number!\r\n"))
|
||||
return
|
||||
}
|
||||
zkv[rcmd[1]] = rcmd[2]
|
||||
conn.Write([]byte("+OK\r\n"))
|
||||
}
|
||||
|
||||
func subscribe(rcmd []string, conn net.Conn) {
|
||||
if len(rcmd) < 2 {
|
||||
conn.Write([]byte("-Error: subscribe para number!\r\n"))
|
||||
return
|
||||
}
|
||||
|
||||
for _, topic := range rcmd[1:] {
|
||||
conns := zsub[topic]
|
||||
if conns == nil {
|
||||
conns = make([]*ConnContext, 0)
|
||||
}
|
||||
|
||||
zsub[topic] = append(conns, &ConnContext{conn: &conn})
|
||||
}
|
||||
}
|
||||
func unsubscribe(rcmd []string, conn net.Conn) {
|
||||
if len(rcmd) < 2 {
|
||||
conn.Write([]byte("-Error: unsubscribe para number!"))
|
||||
return
|
||||
}
|
||||
|
||||
for _, topic := range rcmd[1:] {
|
||||
conns := zsub[topic]
|
||||
if conns == nil || len(conns) == 0 {
|
||||
return
|
||||
}
|
||||
_conns := make([]*ConnContext, 0)
|
||||
for _, c := range conns {
|
||||
if *c.conn == *&conn {
|
||||
continue
|
||||
}
|
||||
_conns = append(_conns, c)
|
||||
}
|
||||
zsub[topic] = _conns
|
||||
}
|
||||
}
|
||||
func publish(rcmd []string, conn net.Conn) {
|
||||
if len(rcmd) < 3 {
|
||||
conn.Write([]byte("-Error: publish para number!\r\n"))
|
||||
return
|
||||
}
|
||||
|
||||
topic := rcmd[1]
|
||||
v := rcmd[2]
|
||||
|
||||
subs := zsub[topic]
|
||||
if subs == nil || len(subs) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
msgs := []string{"message", topic, v}
|
||||
for _, c := range subs {
|
||||
send(*c.conn, msgs...)
|
||||
/*_conn.Write([]byte("*3\r\n"))
|
||||
for _, msg := range msgs {
|
||||
_conn.Write([]byte("$" + strconv.Itoa(len(msg)) + "\r\n"))
|
||||
_conn.Write([]byte(msg + "\r\n"))
|
||||
}*/
|
||||
}
|
||||
}
|
||||
|
||||
var wlock = sync.Mutex{}
|
||||
|
||||
func send(conn net.Conn, vs ...string) (err error) {
|
||||
//chSend <- vs
|
||||
wlock.Lock()
|
||||
defer wlock.Unlock()
|
||||
|
||||
if len(vs) == 1 {
|
||||
_, err = conn.Write([]byte(vs[0] + "\r\n"))
|
||||
} else if len(vs) > 1 {
|
||||
data := "*" + strconv.Itoa(len(vs)) + "\r\n"
|
||||
for _, v := range vs {
|
||||
data += "$" + strconv.Itoa(len(v)) + "\r\n"
|
||||
data += v + "\r\n"
|
||||
}
|
||||
_, err = conn.Write([]byte(data))
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
146
zdb/zdb-server.go
Normal file
146
zdb/zdb-server.go
Normal file
@@ -0,0 +1,146 @@
|
||||
package zdb
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"github.com/robfig/cron"
|
||||
"log"
|
||||
"net"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 消息命令处理 chan
|
||||
var (
|
||||
chmsg = make(chan Message, 10000)
|
||||
zkv = make(map[string]string)
|
||||
zsub = make(map[string][]*ConnContext) // topic -- connx[]
|
||||
retOk = []byte("+OK")
|
||||
zTimer = make(map[string]*ZTimer)
|
||||
retHelp = []byte(
|
||||
"\n--- zdb help ---\n" +
|
||||
"______ _____ _____ \n|___ / | _ \\ | _ \\ \n / / | | | | | |_| | \n / / | | | | | _ { \n / /__ | |_| | | |_| | \n/_____| |_____/ |_____/ \n" +
|
||||
"had supported command:\n" +
|
||||
"1. set:\n" +
|
||||
" eg: set a 1\n" +
|
||||
"2. get:\n" +
|
||||
" eg: get a\n" +
|
||||
"3. subscribe:\n" +
|
||||
" eg: subscribe x y z\n" +
|
||||
"4. unsubscribe:\n" +
|
||||
" eg: unsubscribe x1 y1 z1\n" +
|
||||
"5. publish:\n" +
|
||||
" eg: publish x 123\n" +
|
||||
"6. incr:\n" +
|
||||
" eg: incr a\n" +
|
||||
"7. decr:\n" +
|
||||
" eg: decr a\n" +
|
||||
"--- zdb help ---\n")
|
||||
)
|
||||
|
||||
// 数据封装
|
||||
type Message struct {
|
||||
Conn *net.Conn
|
||||
Rcmd []string
|
||||
}
|
||||
type ConnContext struct {
|
||||
conn *net.Conn
|
||||
groupId string
|
||||
createTime time.Time
|
||||
}
|
||||
|
||||
type ZTimer struct {
|
||||
conns []*net.Conn
|
||||
expr string
|
||||
topic string
|
||||
cron *cron.Cron
|
||||
}
|
||||
|
||||
// ======================================================================
|
||||
|
||||
// zdb 服务启动
|
||||
func ServerStart(host string, port int) {
|
||||
listen, err := net.Listen("tcp", fmt.Sprintf("%s:%d", host, port))
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
return
|
||||
}
|
||||
log.Printf("zdb started listen on: %s:%d \n", host, port)
|
||||
|
||||
// 启动消息监听处理
|
||||
go func() {
|
||||
for {
|
||||
v, ok := <-chmsg
|
||||
if !ok {
|
||||
break
|
||||
}
|
||||
|
||||
execCmd(v.Rcmd, *&*v.Conn)
|
||||
}
|
||||
}()
|
||||
|
||||
for {
|
||||
conn, err := listen.Accept()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
continue
|
||||
}
|
||||
fmt.Println("conn start: ", conn.RemoteAddr())
|
||||
|
||||
go connHandler(conn)
|
||||
}
|
||||
}
|
||||
|
||||
// 连接处理
|
||||
func connHandler(conn net.Conn) {
|
||||
defer func() {
|
||||
for topic, connx := range zsub {
|
||||
_conns := make([]*ConnContext, 0)
|
||||
for t := range connx {
|
||||
if *connx[t].conn == *&conn {
|
||||
continue
|
||||
}
|
||||
_conns = append(_conns, connx[t])
|
||||
}
|
||||
zsub[topic] = _conns
|
||||
}
|
||||
conn.Close()
|
||||
if r := recover(); r != nil {
|
||||
log.Println("connHandler Recovered:", r)
|
||||
}
|
||||
|
||||
fmt.Println("conn end: ", conn.RemoteAddr())
|
||||
}()
|
||||
|
||||
reader := bufio.NewReader(conn)
|
||||
|
||||
for {
|
||||
rcmd := make([]string, 0)
|
||||
line, _, err := reader.ReadLine()
|
||||
// fmt.Println("line:", string(line)) todo 可使用第一行用于协议头
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return
|
||||
}
|
||||
if len(line) == 0 {
|
||||
continue
|
||||
}
|
||||
switch string(line[:1]) {
|
||||
case "*":
|
||||
n, _ := strconv.Atoi(string(line[1:]))
|
||||
for i := 0; i < n; i++ {
|
||||
reader.ReadLine()
|
||||
v, _, _ := reader.ReadLine()
|
||||
rcmd = append(rcmd, string(v))
|
||||
}
|
||||
default:
|
||||
rcmd = append(rcmd, string(line))
|
||||
}
|
||||
|
||||
if len(rcmd) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
chmsg <- Message{Conn: &conn, Rcmd: rcmd}
|
||||
}
|
||||
}
|
||||
7
zdb/zdb_test.go
Normal file
7
zdb/zdb_test.go
Normal file
@@ -0,0 +1,7 @@
|
||||
package zdb
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
ServerStart("127.0.0.1", 1216)
|
||||
}
|
||||
Reference in New Issue
Block a user