mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2026-06-14 12:58:42 +02:00
@@ -20,20 +20,12 @@ type busService struct {
|
||||
network string
|
||||
}
|
||||
|
||||
func New(apiURL, apikey, network string, catalystBus *bus.Bus, db *database.Database) error {
|
||||
func New(apiURL, apikey, network string, catalystBus *bus.Bus, db *database.Database) {
|
||||
h := &busService{db: db, apiURL: apiURL, apiKey: apikey, network: network, catalystBus: catalystBus}
|
||||
|
||||
if err := catalystBus.SubscribeRequest(h.logRequest); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := catalystBus.SubscribeResult(h.handleResult); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := catalystBus.SubscribeJob(h.handleJob); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
catalystBus.RequestChannel.Subscribe(h.logRequest)
|
||||
catalystBus.ResultChannel.Subscribe(h.handleResult)
|
||||
catalystBus.JobChannel.Subscribe(h.handleJob)
|
||||
}
|
||||
|
||||
func busContext() context.Context {
|
||||
@@ -47,7 +39,7 @@ func (h *busService) logRequest(msg *bus.RequestMsg) {
|
||||
var logEntries []*model.LogEntry
|
||||
for _, i := range msg.IDs {
|
||||
logEntries = append(logEntries, &model.LogEntry{
|
||||
Type: bus.ChannelRequest,
|
||||
Type: "request",
|
||||
Reference: i.String(),
|
||||
Creator: msg.User,
|
||||
Message: msg.Function,
|
||||
|
||||
+1
-3
@@ -82,9 +82,7 @@ func (h *busService) handleJob(automationMsg *bus.JobMsg) {
|
||||
}
|
||||
}
|
||||
|
||||
if err := h.catalystBus.PublishResult(automationMsg.Automation, result, automationMsg.Origin); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
h.catalystBus.ResultChannel.Publish(&bus.ResultMsg{Automation: automationMsg.Automation, Data: result, Target: automationMsg.Origin})
|
||||
|
||||
if err := h.db.JobComplete(ctx, automationMsg.ID, result); err != nil {
|
||||
log.Println(err)
|
||||
|
||||
Reference in New Issue
Block a user