blob: 6afc63b186465a6ced2016e7f4b86b2d6009c066 [file] [log] [blame]
/*
* Copyright 2018-present Open Networking Foundation
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package model
import (
"encoding/json"
"fmt"
"github.com/opencord/voltha-go/protos/voltha"
)
type EventBus struct {
client *EventBusClient
topic string
}
var (
IGNORED_CALLBACKS = map[CallbackType]struct{}{
PRE_ADD: {},
GET: {},
POST_LISTCHANGE: {},
PRE_REMOVE: {},
PRE_UPDATE: {},
}
)
func NewEventBus() *EventBus {
bus := &EventBus{
client: NewEventBusClient(),
topic: "model-change-events",
}
return bus
}
//func (bus *EventBus) Advertise(eventType CallbackType, data interface{}, hash string) {
func (bus *EventBus) Advertise(args ...interface{}) interface{} {
eventType := args[0].(CallbackType)
data := args[1]
hash := args[2].(string)
if _, ok := IGNORED_CALLBACKS[eventType]; ok {
fmt.Printf("ignoring event - type:%s, data:%+v\n", eventType, data)
}
var kind voltha.ConfigEventType_ConfigEventType
switch eventType {
case POST_ADD:
kind = voltha.ConfigEventType_add
case POST_REMOVE:
kind = voltha.ConfigEventType_remove
default:
kind = voltha.ConfigEventType_update
}
var msg []byte
var err error
if IsProtoMessage(data) {
if msg, err = json.Marshal(data); err != nil {
fmt.Errorf("problem marshalling data: %+v, err:%s\n", data, err.Error())
}
} else {
msg = data.([]byte)
}
event := voltha.ConfigEvent{
Type: kind,
Hash: hash,
Data: string(msg),
}
bus.client.Publish(bus.topic, event)
return nil
}