core/oc-lib
6
0
Fork 0
Code Issues 7 Pull Requests Packages Projects Releases Wiki Activity

wait for NATS

Browse Source
This commit is contained in:
mr 2025-06-24 08:49:53 +02:00
parent e84d262f38
commit 2c8dcbe93d
2 changed files with 38 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/api.go
View File

@ -110,14 +110,14 @@ func (a *API) SubscribeRouter(infos []*beego.ControllerInfo) {
} }
} }
} }
nats.SetNATSPub("api", DISCOVERY, discovery) go nats.SetNATSPub("api", DISCOVERY, discovery)
} }
// CheckRemotePeer checks the state of a remote peer // CheckRemotePeer checks the state of a remote peer
func (a *API) CheckRemotePeer(url string) (State, map[string]int) { func (a *API) CheckRemotePeer(url string) (State, map[string]int) {
// Check if the database is up // Check if the database is up
var resp APIStatusResponse var resp APIStatusResponse
caller := NewHTTPCaller(map[DataType]map[METHOD]string{}) // Create a new http caller caller := NewHTTPCaller(map[DataType]map[METHOD]string{}) // Create a new http caller
b, err := caller.CallPost(url, "", map[string]interface{}{}) // Call the status endpoint of the peer b, err := caller.CallPost(url, "", map[string]interface{}{}) // Call the status endpoint of the peer
if err != nil { if err != nil {
return DEAD, map[string]int{} // If the peer is not reachable, return dead return DEAD, map[string]int{} // If the peer is not reachable, return dead

63
tools/nats_caller.go
View File

@ -3,6 +3,7 @@ package tools
import ( import (
"encoding/json" "encoding/json"
"strings" "strings"
"time"
"cloud.o-forge.io/core/oc-lib/config" "cloud.o-forge.io/core/oc-lib/config"
"cloud.o-forge.io/core/oc-lib/logs" "cloud.o-forge.io/core/oc-lib/logs"
@ -53,22 +54,25 @@ func (s *natsCaller) ListenNats(chanName string, exec func(msg map[string]interf
log.Error().Msg(" -> NATS_SERVER is not set") log.Error().Msg(" -> NATS_SERVER is not set")
return return
} }
nc, err := nats.Connect(config.GetConfig().NATSUrl) for {
if err != nil { nc, err := nats.Connect(config.GetConfig().NATSUrl)
log.Error().Msg(" -> Could not reach NATS server : " + err.Error()) if err != nil {
return time.Sleep(1 * time.Minute)
} continue
ch := make(chan *nats.Msg, 64) }
subs, err := nc.ChanSubscribe(chanName, ch) ch := make(chan *nats.Msg, 64)
if err != nil { subs, err := nc.ChanSubscribe(chanName, ch)
log.Error().Msg("Error listening to NATS : " + err.Error()) if err != nil {
} log.Error().Msg("Error listening to NATS : " + err.Error())
defer subs.Unsubscribe() }
defer subs.Unsubscribe()
for msg := range ch { for msg := range ch {
map_mess := map[string]interface{}{} map_mess := map[string]interface{}{}
json.Unmarshal(msg.Data, &map_mess) json.Unmarshal(msg.Data, &map_mess)
exec(map_mess) exec(map_mess)
}
break
} }
} }
@ -77,18 +81,23 @@ func (o *natsCaller) SetNATSPub(dataName string, method NATSMethod, data interfa
if config.GetConfig().NATSUrl == "" { if config.GetConfig().NATSUrl == "" {
return " -> NATS_SERVER is not set" return " -> NATS_SERVER is not set"
} }
nc, err := nats.Connect(config.GetConfig().NATSUrl) for {
if err != nil { nc, err := nats.Connect(config.GetConfig().NATSUrl)
return " -> Could not reach NATS server : " + err.Error() if err != nil {
} time.Sleep(1 * time.Minute)
defer nc.Close() continue
js, err := json.Marshal(data) }
if err != nil { defer nc.Close()
return " -> " + err.Error() js, err := json.Marshal(data)
} if err != nil {
err = nc.Publish(method.GenerateKey(dataName), js) // Publish the message on the NATS server with a channel name based on the data name (or whatever start) and the method return " -> " + err.Error()
if err != nil { }
return " -> " + err.Error() // Return an error if the message could not be published err = nc.Publish(method.GenerateKey(dataName), js) // Publish the message on the NATS server with a channel name based on the data name (or whatever start) and the method
if err != nil {
time.Sleep(1 * time.Minute)
continue
}
break
} }
return "" return ""
} }