add resp

liuxiaolong

2020-03-11 1fc291e69cf5cabe8cb743d78f77fb34519495f5

 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))
}