init

client.go

@@ -0,0 +1,81 @@
+package notify
+
+import (
+	"net"
+	"strings"
+	"time"
+)
+
+var connc net.Conn
+
+var clientSign map[string]chan string
+
+func init() {
+	clientSign = make(map[string]chan string)
+}
+
+func Notify(key string) chan string {
+	if _, ok := clientSign[key]; !ok {
+		ch := make(chan string, 5)
+		clientSign[key] = ch
+	}
+	return clientSign[key]
+}
+
+func store(key, value string) {
+	if _, ok := clientSign[key]; !ok {
+		ch := make(chan string, 5)
+		ch <- value
+		clientSign[key] = ch
+		return
+	}
+	clientSign[key] <- value
+}
+
+func NewNotifyC(netype, value string) error {
+	var err error
+	connc, err = net.Dial(netype, value)
+	if err != nil {
+		return err
+	}
+	go cnotify()
+	go func() {
+		for {
+			buf := make([]byte, 8192)
+			n, err := connc.Read(buf)
+			Queue.ParseMessage(buf[0:n], connc)
+			if err != nil {
+				connc.Close()
+				notifychan <- 0
+				go NewNotifyC(netype, value)
+				break
+			}
+		}
+	}()
+	return nil
+}
+
+func Send(name string) error {
+	_, err := connc.Write(Queue.BuildMessage(name))
+	return err
+}
+
+func cnotify() {
+	for {
+		select {
+		case <-notifychan:
+			break
+		default:
+		}
+		data, err := Queue.RestoreOne()
+		if err != nil {
+			time.Sleep(time.Millisecond * 20)
+			continue
+		}
+		strs := strings.SplitN(data.Msg, "||", 2)
+		if len(strs) < 2 {
+			continue
+		}
+		go store(strs[0], strs[1])
+	}
+}