No discovery needed
This commit is contained in:
@@ -2,10 +2,8 @@ package tools
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"cloud.o-forge.io/core/oc-lib/logs"
|
||||
"github.com/nats-io/nats.go"
|
||||
)
|
||||
|
||||
@@ -14,7 +12,6 @@ type NATSMethod int
|
||||
const (
|
||||
REMOVE NATSMethod = iota
|
||||
CREATE
|
||||
DISCOVERY
|
||||
)
|
||||
|
||||
func NameToMethod(name string) NATSMethod {
|
||||
@@ -59,39 +56,3 @@ func (o *natsCaller) SetNATSPub(dataName string, method NATSMethod, data interfa
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
type NATSObjectInterface interface {
|
||||
Serialize() map[string]interface{}
|
||||
}
|
||||
|
||||
func (o *natsCaller) DiscoveryNATS(name string, model NATSObjectInterface) error {
|
||||
if GetConfig().NATSUrl == "" {
|
||||
return errors.New("NATS_SERVER is not set")
|
||||
}
|
||||
nc, err := nats.Connect(GetConfig().NATSUrl)
|
||||
if err != nil {
|
||||
return errors.New("Could not connect to NATS")
|
||||
}
|
||||
go o.listenForChange(nc, model, DISCOVERY.GenerateKey("ask"))
|
||||
go o.listenForChange(nc, model, DISCOVERY.GenerateKey(name))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (o *natsCaller) listenForChange(nc *nats.Conn, model NATSObjectInterface, chanName string) {
|
||||
api := API{}
|
||||
logger := logs.GetLogger()
|
||||
ch := make(chan *nats.Msg, 64)
|
||||
subs, err := nc.ChanSubscribe(chanName, ch)
|
||||
if err != nil {
|
||||
logger.Error().Msg("Error listening to NATS : " + err.Error())
|
||||
return
|
||||
}
|
||||
defer subs.Unsubscribe()
|
||||
for msg := range ch {
|
||||
logger.Info().Msg("Received message from NATS : " + string(msg.Data))
|
||||
m := model.Serialize()
|
||||
s, _, _ := api.GetState()
|
||||
m["state"] = s
|
||||
o.SetNATSPub("answer", DISCOVERY, m)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user