From 1fc291e69cf5cabe8cb743d78f77fb34519495f5 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 11 三月 2020 14:17:51 +0800
Subject: [PATCH] add resp
---
server.go | 18 +++++++++++++++++-
1 files changed, 17 insertions(+), 1 deletions(-)
diff --git a/server.go b/server.go
index deddb13..01a3b74 100644
--- a/server.go
+++ b/server.go
@@ -1,6 +1,7 @@
package gopherdiscovery
import (
+ "encoding/json"
"log"
"time"
@@ -49,6 +50,8 @@
nodes StringSet
// publisher, we are going to publish the changes of the set here
publisher *Publisher
+
+ svInfo map[string]interface{}
}
type Publisher struct {
@@ -138,6 +141,7 @@
var err error
var msg []byte
var responses StringSet
+ var si ServiceInfo
err = d.sock.Send([]byte("are you ok?"))
if err != nil {
@@ -155,7 +159,10 @@
return
}
} else {
- responses.Add(string(msg))
+ if json.Unmarshal(msg, &si) == nil {
+ responses.Add(si.ServiceId)
+ d.services.svInfo[si.ServiceId] = si.Info
+ }
}
}
}
@@ -211,6 +218,7 @@
s := &Services{
nodes: NewStringSet(),
publisher: publisher,
+ svInfo: make(map[string]interface{}, 0),
}
return s
@@ -234,6 +242,14 @@
return d.services.nodes
}
+func (d *DiscoveryServer) SvInfo() map[string]interface{} {
+ m := make(map[string]interface{})
+ 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