ClusterCockpit/cc-backend
1
0
Fork 0
mirror of https://github.com/ClusterCockpit/cc-backend synced 2025-02-04 15:25:17 +01:00
Code Issues Packages Releases Activity

fill listener and handler with dev content

Browse Source
This commit is contained in:
Christoph Kluge 2024-08-08 15:42:34 +02:00
parent 162eaaf77a
commit f4a414fa6e
1 changed files with 87 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
internal/natsMessenger/natsMessenger.go
View File

@ -27,7 +27,6 @@ func New(config *schema.NatsConfig) (nm *NatsMessenger, err error) {
return SetupNatsMessenger(config)
}
// StartJobNatsMessage model
type DevNatsMessage struct {
Content string `json:"content"`
}
@ -89,7 +88,7 @@ func SetupNatsMessenger(config *schema.NatsConfig) (nm *NatsMessenger, err error
go nmr.Server.Start()
if !nmr.Server.ReadyForConnections(4 * time.Second) {
if !nmr.Server.ReadyForConnections(3 * time.Second) {
log.Error("nats server not ready for connection")
return nil, fmt.Errorf("nats server not ready for connection")
}
@ -104,15 +103,13 @@ func SetupNatsMessenger(config *schema.NatsConfig) (nm *NatsMessenger, err error
}
// Subscribe
sub, err := startJobListener(nmr.Connection)
nmr.Subscriptions, err = setupSubscriptions(nmr.Connection)
if err != nil {
log.Error("startJobListener subscription error")
log.Error("error when subscribing to channels")
return nil, err
} else {
log.Infof("NATS subscription to 'start-job' on port '%d' established\n", config.Port)
nmr.Subscriptions = append(nmr.Subscriptions, sub)
}
log.Infof("NATS server and subscriptions on port '%d' established\n", config.Port)
return &nmr, nil
}
@ -129,49 +126,111 @@ func (nm *NatsMessenger) StopNatsMessenger() {
log.Info("NATS connections closed and server shut down")
}
// Listeners: Subscribe to specified channels for actions
func setupSubscriptions(conn *nats.Conn) (subs []*nats.Subscription, err error) {
func startJobListener(conn *nats.Conn) (sub *nats.Subscription, err error) {
sub, err = conn.Subscribe("start-job", func(m *nats.Msg) {
var job DevNatsMessage
if err := json.Unmarshal(m.Data, &job); err != nil {
log.Error("Error while unmarshaling raw json nats message content")
if startSub, err := startJobListener(conn); err != nil {
log.Infof("Subscription to 'start-job' failed: %s", err)
} else {
log.Info("Subscribed to 'start-job'")
subs = append(subs, startSub)
}
if err := startJobHandler(job); err != nil {
if stopSub, err := stopJobListener(conn); err != nil {
log.Infof("Subscription to 'stop-job' failed: %s", err)
} else {
log.Info("Subscribed to 'stop-job'")
subs = append(subs, stopSub)
}
if deleteSub, err := deleteJobListener(conn); err == nil {
log.Infof("Subscription to 'delete-job' failed: %s", err)
} else {
log.Info("Subscribed to 'delete-job'")
subs = append(subs, deleteSub)
}
if eventSub, err := jobEventListener(conn); err != nil {
log.Infof("Subscription to 'job-event' failed: %s", err)
} else {
log.Info("Subscribed to 'job-event'")
subs = append(subs, eventSub)
}
return subs, err
}
// Listeners: Subscribe to specified channels and handle with specific handler functions
func startJobListener(conn *nats.Conn) (sub *nats.Subscription, err error) {
return conn.Subscribe("start-job", func(m *nats.Msg) {
var req DevNatsMessage
if err := json.Unmarshal(m.Data, &req); err != nil {
log.Error("Error while unmarshaling raw json nats message content: startJob")
}
if err := startJobHandler(req); err != nil {
log.Errorf("error: %s", err.Error())
}
})
if err != nil {
return nil, err
} else {
return sub, nil
}
}
func (nm *NatsMessenger) stopJobListener(conn *nats.Conn) {
func stopJobListener(conn *nats.Conn) (sub *nats.Subscription, err error) {
return conn.Subscribe("stop-job", func(m *nats.Msg) {
var req DevNatsMessage
if err := json.Unmarshal(m.Data, &req); err != nil {
log.Error("Error while unmarshaling raw json nats message content: stopJob")
}
func (nm *NatsMessenger) deleteJobListener(conn *nats.Conn) {
if err := stopJobHandler(req); err != nil {
log.Errorf("error: %s", err.Error())
}
})
}
func (nm *NatsMessenger) jobEventListener(conn *nats.Conn) {
func deleteJobListener(conn *nats.Conn) (sub *nats.Subscription, err error) {
return conn.Subscribe("delete-job", func(m *nats.Msg) {
var req DevNatsMessage
if err := json.Unmarshal(m.Data, &req); err != nil {
log.Error("Error while unmarshaling raw json nats message content: deleteJob")
}
if err := deleteJobHandler(req); err != nil {
log.Errorf("error: %s", err.Error())
}
})
}
func jobEventListener(conn *nats.Conn) (sub *nats.Subscription, err error) {
return conn.Subscribe("job-event", func(m *nats.Msg) {
var req DevNatsMessage
if err := json.Unmarshal(m.Data, &req); err != nil {
log.Error("Error while unmarshaling raw json nats message content: jobEvent")
}
if err := jobEventHandler(req); err != nil {
log.Errorf("error: %s", err.Error())
}
})
}
// Handlers: Take content of message and perform action, e.g. adding job in db
func startJobHandler(job DevNatsMessage) (err error) {
log.Debugf("CALLED HANDLER FOR startJob: %s", job.Content)
func startJobHandler(req DevNatsMessage) (err error) {
log.Debugf("CALLED HANDLER FOR startJob: %s", req.Content)
return nil
}
func (nm *NatsMessenger) stopJobHandler() {
func stopJobHandler(req DevNatsMessage) (err error) {
log.Debugf("CALLED HANDLER FOR stopJob: %s", req.Content)
return nil
}
func (nm *NatsMessenger) deleteJobHandler() {
func deleteJobHandler(req DevNatsMessage) (err error) {
log.Debugf("CALLED HANDLER FOR deleteJob: %s", req.Content)
return nil
}
func (nm *NatsMessenger) jobEventHandler() {
func jobEventHandler(req DevNatsMessage) (err error) {
log.Debugf("CALLED HANDLER FOR jobEvent: %s", req.Content)
return nil
}