SEBA-949 support for publishing bbsim events on kafka
Change-Id: I4354cd026bbadc801e4d6d08b2f9cd3462917b4c
diff --git a/internal/common/kafka_utils.go b/internal/common/kafka_utils.go
new file mode 100644
index 0000000..9ddd79c
--- /dev/null
+++ b/internal/common/kafka_utils.go
@@ -0,0 +1,78 @@
+/*
+ * 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 common
+
+import (
+ "encoding/json"
+ "strconv"
+ "time"
+
+ "github.com/Shopify/sarama"
+ log "github.com/sirupsen/logrus"
+)
+
+var producer sarama.AsyncProducer
+var topic string
+
+// Event defines structure for bbsim events
+type Event struct {
+ EventType string
+ OnuSerial string
+ OltID int
+ IntfID int32
+ OnuID int32
+ EpochTime int64
+ Timestamp string
+}
+
+// InitializePublisher initalizes kafka publisher
+func InitializePublisher(oltID int) error {
+
+ var err error
+ sarama.Logger = log.New()
+ // producer config
+ config := sarama.NewConfig()
+ config.Producer.Retry.Max = 5
+ config.Metadata.Retry.Max = 10
+ config.Metadata.Retry.Backoff = 10 * time.Second
+ config.ClientID = "BBSim-OLT-" + strconv.Itoa(oltID)
+ topic = "BBSim-OLT-" + strconv.Itoa(oltID) + "-Events"
+
+ producer, err = sarama.NewAsyncProducer([]string{Options.BBSim.KafkaAddress}, config)
+ return err
+}
+
+// KafkaPublisher receives messages on eventChannel and publish them to kafka
+func KafkaPublisher(eventChannel chan Event) {
+ defer log.Debugf("KafkaPublisher stopped")
+ for {
+ select {
+ case event := <-eventChannel:
+ log.Debugf("Received event on channel %v", event)
+ jsonEvent, err := json.Marshal(event)
+ if err != nil {
+ log.Errorf("Failed to get json event %v", err)
+ continue
+ }
+ producer.Input() <- &sarama.ProducerMessage{
+ Topic: topic,
+ Value: sarama.ByteEncoder(jsonEvent),
+ }
+ log.Debugf("Event sent on kafka")
+ }
+ }
+}
diff --git a/internal/common/options.go b/internal/common/options.go
index d5317e1..f1f5f0a 100644
--- a/internal/common/options.go
+++ b/internal/common/options.go
@@ -70,6 +70,8 @@
LegacyRestApiAddress string `yaml:"legacy_rest_api_address"`
SadisRestAddress string `yaml:"sadis_rest_address"`
SadisServer bool `yaml:"sadis_server"`
+ KafkaAddress string `yaml:"kafka_address"`
+ Events bool `yaml:"enable_events"`
ControlledActivation string `yaml:"controlled_activation"`
EnablePerf bool `yaml:"enable_perf"`
}
@@ -105,6 +107,8 @@
LegacyRestApiAddress: ":50073",
SadisRestAddress: ":50074",
SadisServer: true,
+ KafkaAddress: ":9092",
+ Events: false,
ControlledActivation: "default",
EnablePerf: false,
},
@@ -171,6 +175,8 @@
controlledActivation := flag.String("ca", conf.BBSim.ControlledActivation, "Set the mode for controlled activation of PON ports and ONUs")
enablePerf := flag.Bool("enableperf", conf.BBSim.EnablePerf, "Setting this flag will cause BBSim to not store data like traffic schedulers, flows of ONUs etc..")
+ enableEvents := flag.Bool("enableEvents", conf.BBSim.Events, "Enable sending BBSim events on configured kafka server")
+ kafkaAddress := flag.String("kafkaAddress", conf.BBSim.KafkaAddress, "IP:Port for kafka")
flag.Parse()
conf.Olt.ID = int(*olt_id)
@@ -187,6 +193,8 @@
conf.BBSim.Delay = *delay
conf.BBSim.ControlledActivation = *controlledActivation
conf.BBSim.EnablePerf = *enablePerf
+ conf.BBSim.Events = *enableEvents
+ conf.BBSim.KafkaAddress = *kafkaAddress
// update device id if not set
if conf.Olt.DeviceId == "" {