修改：zgroup 订阅组 共同消费组消息

git-svn-id: svn://47.119.165.148/zhub@92 e63fbceb-bcc3-4977-ac22-735b83d8d0f4

zsub/zgroup.go

@@ -1,28 +1,33 @@
 package zsub
 
-import "sync"
+import (
+	"sync"
+	"sync/atomic"
+)
 
 type ZGroup struct { // ZGroup
 	sync.Mutex
 	conns  []*ZConn
-	offset int
+	offset int32
 	chMsg  chan string // 组消息即时投递
 	ztopic *ZTopic     // 所属topic
 }
 
-func (g *ZGroup) init() {
-	go func() {
+func (g *ZGroup) appendTo(c *ZConn) {
+	c.appendTo(g.conns)
+	go func() { // 每个连接开启一个携程发送数据
 		for {
 			msg, ok := <-g.chMsg
 			if !ok {
 				break
 			}
 
-			if len(g.conns) == 0 {
-				continue
+			err := c.send("message", g.ztopic.topic, msg)
+			if err != nil { // 失败处理
+				g.chMsg <- msg
+				break
 			}
-			g.conns[0].send("message", g.ztopic.topic, msg)
-			g.offset++
+			atomic.AddInt32(&g.offset, 1)
 		}
 	}()
 }

zsub/zsub.go

@@ -53,15 +53,15 @@ func (s *ZSub) subscribe(c *ZConn, topic string) { // 新增订阅 zconn{}
 	zgroup := ztopic.groups[c.groupid] //ZGroup
 	if zgroup == nil {
 		zgroup = &ZGroup{
-			conns:  []*ZConn{},
+			//conns:  []*ZConn{},
 			ztopic: ztopic,
 			chMsg:  make(chan string, 1000),
 		}
-		zgroup.init()
 		ztopic.groups[c.groupid] = zgroup
 	}
 
-	zgroup.conns = c.appendTo(zgroup.conns)
+	//zgroup.conns = c.appendTo(zgroup.conns)
+	zgroup.appendTo(c)
 
 	for i, item := range c.topics {
 		if strings.EqualFold(item, topic) {
@@ -156,6 +156,9 @@ func (s *ZSub) close(c *ZConn) {
 }
 
 func (c *ZConn) appendTo(arr []*ZConn) []*ZConn {
+	if arr == nil {
+		arr = make([]*ZConn, 0)
+	}
 	for i, item := range arr {
 		if item == c {
 			arr = append(arr[:i], arr[i+1:]...)