Improve bus (#3)

* Improve bus
* Add ticket log

bus/bus.go

@@ -4,16 +4,7 @@ import (
 	"encoding/json"
 	"log"
 
-	"github.com/arangodb/go-driver"
 	emitter "github.com/emitter-io/go/v2"
-
-	"github.com/SecurityBrewery/catalyst/generated/models"
-)
-
-const (
-	channelUpdate = "data"
-	channelJob    = "job"
-	channelResult = "result"
 )
 
 type Bus struct {
@@ -22,25 +13,13 @@ type Bus struct {
 }
 
 type Config struct {
-	Host         string
-	Key          string
-	resultBusKey string
-	jobBusKey    string
-	dataBusKey   string
-	APIUrl       string
-}
-
-type JobMsg struct {
-	ID         string          `json:"id"`
-	Automation string          `json:"automation"`
-	Origin     *models.Origin  `json:"origin"`
-	Message    *models.Message `json:"message"`
-}
-
-type ResultMsg struct {
-	Automation string                 `json:"automation"`
-	Data       map[string]interface{} `json:"data,omitempty"`
-	Target     *models.Origin         `json:"target"`
+	Host                 string
+	Key                  string
+	databaseUpdateBusKey string
+	jobBusKey            string
+	resultBusKey         string
+	requestKey           string
+	APIUrl               string
 }
 
 func New(c *Config) (*Bus, error) {
@@ -51,7 +30,7 @@ func New(c *Config) (*Bus, error) {
 		return nil, err
 	}
 
-	c.dataBusKey, err = client.GenerateKey(c.Key, channelUpdate+"/", "rwls", 0)
+	c.databaseUpdateBusKey, err = client.GenerateKey(c.Key, channelDatabaseUpdate+"/", "rwls", 0)
 	if err != nil {
 		return nil, err
 	}
@@ -63,30 +42,14 @@ func New(c *Config) (*Bus, error) {
 	if err != nil {
 		return nil, err
 	}
+	c.requestKey, err = client.GenerateKey(c.Key, ChannelRequest+"/", "rwls", 0)
+	if err != nil {
+		return nil, err
+	}
 
 	return &Bus{config: c, client: client}, err
 }
 
-func (b *Bus) PublishUpdate(ids []driver.DocumentID) error {
-	return b.jsonPublish(ids, channelUpdate, b.config.dataBusKey)
-}
-
-func (b *Bus) PublishJob(id, automation string, payload interface{}, context *models.Context, origin *models.Origin) error {
-	return b.jsonPublish(&JobMsg{
-		ID:         id,
-		Automation: automation,
-		Origin:     origin,
-		Message: &models.Message{
-			Context: context,
-			Payload: payload,
-		},
-	}, channelJob, b.config.jobBusKey)
-}
-
-func (b *Bus) PublishResult(automation string, data map[string]interface{}, target *models.Origin) error {
-	return b.jsonPublish(&ResultMsg{Automation: automation, Data: data, Target: target}, channelResult, b.config.resultBusKey)
-}
-
 func (b *Bus) jsonPublish(msg interface{}, channel, key string) error {
 	payload, err := json.Marshal(msg)
 	if err != nil {
@@ -96,39 +59,6 @@ func (b *Bus) jsonPublish(msg interface{}, channel, key string) error {
 	return b.client.Publish(key, channel, payload)
 }
 
-func (b *Bus) SubscribeUpdate(f func(ids []driver.DocumentID)) error {
-	return b.safeSubscribe(b.config.dataBusKey, channelUpdate, func(c *emitter.Client, m emitter.Message) {
-		var msg []driver.DocumentID
-		if err := json.Unmarshal(m.Payload(), &msg); err != nil {
-			log.Println(err)
-			return
-		}
-		go f(msg)
-	})
-}
-
-func (b *Bus) SubscribeJob(f func(msg *JobMsg)) error {
-	return b.safeSubscribe(b.config.jobBusKey, channelJob, func(c *emitter.Client, m emitter.Message) {
-		var msg JobMsg
-		if err := json.Unmarshal(m.Payload(), &msg); err != nil {
-			log.Println(err)
-			return
-		}
-		go f(&msg)
-	})
-}
-
-func (b *Bus) SubscribeResult(f func(msg *ResultMsg)) error {
-	return b.safeSubscribe(b.config.resultBusKey, channelResult, func(c *emitter.Client, m emitter.Message) {
-		var msg ResultMsg
-		if err := json.Unmarshal(m.Payload(), &msg); err != nil {
-			log.Println(err)
-			return
-		}
-		go f(&msg)
-	})
-}
-
 func (b *Bus) safeSubscribe(key, channel string, handler func(c *emitter.Client, m emitter.Message)) error {
 	defer func() {
 		if r := recover(); r != nil {