.
git-svn-id: svn://47.119.165.148/zhub@147 e63fbceb-bcc3-4977-ac22-735b83d8d0f4
This commit is contained in:
@@ -18,8 +18,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Client struct {
|
type Client struct {
|
||||||
lock sync.Mutex // write lock
|
wlock sync.Mutex // write lock
|
||||||
//rlock sync.Mutex // read lock
|
rlock sync.Mutex // read lock
|
||||||
|
|
||||||
appname string // local appname
|
appname string // local appname
|
||||||
addr string // host:port
|
addr string // host:port
|
||||||
@@ -51,7 +51,8 @@ func Create(appname string, addr string, groupid string) (*Client, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
client := Client{
|
client := Client{
|
||||||
lock: sync.Mutex{},
|
wlock: sync.Mutex{},
|
||||||
|
rlock: sync.Mutex{},
|
||||||
appname: appname,
|
appname: appname,
|
||||||
addr: addr,
|
addr: addr,
|
||||||
conn: conn,
|
conn: conn,
|
||||||
@@ -132,8 +133,8 @@ subscribe x y z
|
|||||||
func (c *Client) Subscribe(topic string, fun func(v string)) {
|
func (c *Client) Subscribe(topic string, fun func(v string)) {
|
||||||
c.send("subscribe " + topic)
|
c.send("subscribe " + topic)
|
||||||
if fun != nil {
|
if fun != nil {
|
||||||
c.lock.Lock()
|
c.wlock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.wlock.Unlock()
|
||||||
c.subFun[topic] = fun
|
c.subFun[topic] = fun
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -213,8 +214,8 @@ func (c *Client) Lock(key string, duration int) Lock {
|
|||||||
|
|
||||||
lockChan := make(chan int, 2)
|
lockChan := make(chan int, 2)
|
||||||
go func() {
|
go func() {
|
||||||
c.lock.Lock()
|
c.wlock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.wlock.Unlock()
|
||||||
c.lockFlag[uuid] = &Lock{
|
c.lockFlag[uuid] = &Lock{
|
||||||
Key: key,
|
Key: key,
|
||||||
Uuid: uuid,
|
Uuid: uuid,
|
||||||
@@ -359,8 +360,8 @@ else if len(vs) gt 1 will send message `* + len(vs)+ "\r\n" +"$"+ len(vs[n])+ "
|
|||||||
*/
|
*/
|
||||||
func (c *Client) send(vs ...string) (err error) {
|
func (c *Client) send(vs ...string) (err error) {
|
||||||
//chSend <- vs
|
//chSend <- vs
|
||||||
c.lock.Lock()
|
c.wlock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.wlock.Unlock()
|
||||||
a:
|
a:
|
||||||
if len(vs) == 1 {
|
if len(vs) == 1 {
|
||||||
_, err = c.conn.Write([]byte(vs[0] + "\r\n"))
|
_, err = c.conn.Write([]byte(vs[0] + "\r\n"))
|
||||||
@@ -382,8 +383,8 @@ a:
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Client) receive() {
|
func (c *Client) receive() {
|
||||||
c.lock.Lock()
|
c.rlock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.rlock.Unlock()
|
||||||
|
|
||||||
r := bufio.NewReader(c.conn)
|
r := bufio.NewReader(c.conn)
|
||||||
for {
|
for {
|
||||||
@@ -426,8 +427,8 @@ func (c *Client) receive() {
|
|||||||
if strings.EqualFold(vs[1], "lock") { // message lock Uuid
|
if strings.EqualFold(vs[1], "lock") { // message lock Uuid
|
||||||
go func() {
|
go func() {
|
||||||
log.Println("lock:" + vs[2])
|
log.Println("lock:" + vs[2])
|
||||||
c.lock.Lock()
|
c.wlock.Lock()
|
||||||
defer c.lock.Unlock()
|
defer c.wlock.Unlock()
|
||||||
|
|
||||||
if c.lockFlag[vs[2]] == nil {
|
if c.lockFlag[vs[2]] == nil {
|
||||||
return
|
return
|
||||||
@@ -451,7 +452,7 @@ func (c *Client) receive() {
|
|||||||
|
|
||||||
continue
|
continue
|
||||||
case "+": // +pong, +xxx
|
case "+": // +pong, +xxx
|
||||||
if strings.EqualFold("+ping", string(v)) {
|
if strings.EqualFold("+ping", string(v)) { // 心跳消息回复
|
||||||
c.send("+pong")
|
c.send("+pong")
|
||||||
}
|
}
|
||||||
case "-":
|
case "-":
|
||||||
|
|||||||
@@ -118,7 +118,8 @@ func msgAccept(v Message) {
|
|||||||
zsub.delay(rcmd, c)
|
zsub.delay(rcmd, c)
|
||||||
case "timer":
|
case "timer":
|
||||||
for _, name := range rcmd[1:] {
|
for _, name := range rcmd[1:] {
|
||||||
zsub.timer([]string{"timer", name}, c)
|
zsub.timer([]string{"timer", name}, c) // append to timers
|
||||||
|
c.timers = append(c.timers, name) // append to conns
|
||||||
}
|
}
|
||||||
case "cmd":
|
case "cmd":
|
||||||
if len(rcmd) == 1 {
|
if len(rcmd) == 1 {
|
||||||
|
|||||||
Reference in New Issue
Block a user