From a4ea6380ed70468b1bbaca0328a65686960fcd52 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期一, 20 四月 2020 16:23:30 +0800
Subject: [PATCH] add svLock
---
server.go | 46 +++++++++++++++++++++++++++++++++++-----------
1 files changed, 35 insertions(+), 11 deletions(-)
diff --git a/server.go b/server.go
index 94c6a24..f23728c 100644
--- a/server.go
+++ b/server.go
@@ -1,8 +1,9 @@
package gopherdiscovery
import (
+ "encoding/json"
"log"
- "strings"
+ "sync"
"time"
"golang.org/x/net/context"
@@ -50,6 +51,9 @@
nodes StringSet
// publisher, we are going to publish the changes of the set here
publisher *Publisher
+
+ svInfo map[string][]byte
+ svLock sync.RWMutex
}
type Publisher struct {
@@ -59,7 +63,7 @@
ctx context.Context
sock mangos.Socket
- publishCh chan []string
+ publishCh chan []byte
}
func Server(urlServer string, urlPubSub string, opt Options) (*DiscoveryServer, error) {
@@ -139,6 +143,7 @@
var err error
var msg []byte
var responses StringSet
+ var si ServiceInfo
err = d.sock.Send([]byte("are you ok?"))
if err != nil {
@@ -147,6 +152,7 @@
}
responses = NewStringSet()
+ d.services.svInfo = make(map[string][]byte, 0)
for {
msg, err = d.sock.Recv()
if err != nil {
@@ -155,10 +161,13 @@
d.services.Add(responses)
return
}
- //fmt.Println(" err: ",err.Error())
- //log.Println("DiscoveryServer: Error reading SURVEY responses", err.Error())
} else {
- responses.Add(string(msg))
+ if json.Unmarshal(msg, &si) == nil {
+ responses.Add(si.ServiceId)
+ d.services.svLock.Lock()
+ d.services.svInfo[si.ServiceId] = si.Info
+ d.services.svLock.Unlock()
+ }
}
}
}
@@ -184,14 +193,14 @@
url: url,
sock: sock,
- publishCh: make(chan []string),
+ publishCh: make(chan []byte),
}
go publiser.run()
return publiser, nil
}
-func (p *Publisher) Publish(msg []string) {
+func (p *Publisher) Publish(msg []byte) {
p.publishCh <- msg
}
@@ -202,7 +211,7 @@
close(p.publishCh)
return
case msg := <-p.publishCh:
- err := p.sock.Send([]byte(strings.Join(msg, "|")))
+ err := p.sock.Send(msg)
if err != nil {
log.Println("DiscoveryServer: Error PUBLISHING changes to the socket", err.Error())
}
@@ -214,6 +223,7 @@
s := &Services{
nodes: NewStringSet(),
publisher: publisher,
+ svInfo: make(map[string][]byte, 0),
}
return s
@@ -233,6 +243,20 @@
//s.publisher.Publish(s.nodes.ToSlice())//publish nodes changed
}
-func (d *DiscoveryServer) PublishMsg(msg string){
- d.services.publisher.Publish([]string{msg})
-}
\ No newline at end of file
+func (d *DiscoveryServer) AliveNodes() StringSet {
+ return d.services.nodes
+}
+
+func (d *DiscoveryServer) SvInfo() map[string][]byte {
+ d.services.svLock.Lock()
+ defer d.services.svLock.Unlock()
+ m := make(map[string][]byte)
+ for k,v := range d.services.svInfo {
+ m[k] = v
+ }
+ return m
+}
+
+func (d *DiscoveryServer) PublishMsg(msg string) {
+ d.services.publisher.Publish([]byte(msg))
+}
--
Gitblit v1.8.0