blob: 038af157eaf249c37f4a253a600a023094a71931 [file] [log] [blame]
cuilin20187b2a8c32019-03-26 19:52:28 -07001/*
cbabu116b73f2019-12-10 17:56:32 +05302* Copyright 2018-present Open Networking Foundation
cuilin20187b2a8c32019-03-26 19:52:28 -07003
cbabu116b73f2019-12-10 17:56:32 +05304* Licensed under the Apache License, Version 2.0 (the "License");
5* you may not use this file except in compliance with the License.
6* You may obtain a copy of the License at
cuilin20187b2a8c32019-03-26 19:52:28 -07007
cbabu116b73f2019-12-10 17:56:32 +05308* http://www.apache.org/licenses/LICENSE-2.0
cuilin20187b2a8c32019-03-26 19:52:28 -07009
cbabu116b73f2019-12-10 17:56:32 +053010* Unless required by applicable law or agreed to in writing, software
11* distributed under the License is distributed on an "AS IS" BASIS,
12* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13* See the License for the specific language governing permissions and
14* limitations under the License.
cuilin20187b2a8c32019-03-26 19:52:28 -070015 */
Girish Gowdru6a80bbd2019-07-02 07:36:09 -070016
17//Package main invokes the application
cuilin20187b2a8c32019-03-26 19:52:28 -070018package main
19
20import (
21 "context"
22 "errors"
23 "fmt"
kdarapu381c6902019-07-31 18:23:16 +053024 "os"
25 "os/signal"
kdarapu381c6902019-07-31 18:23:16 +053026 "syscall"
27 "time"
28
Esin Karamanccb714b2019-11-29 15:02:06 +000029 "github.com/opencord/voltha-lib-go/v3/pkg/adapters/adapterif"
kdarapu381c6902019-07-31 18:23:16 +053030
Esin Karamanccb714b2019-11-29 15:02:06 +000031 "github.com/opencord/voltha-lib-go/v3/pkg/adapters"
32 com "github.com/opencord/voltha-lib-go/v3/pkg/adapters/common"
divyadesaia37f78b2020-02-07 12:41:22 +000033 conf "github.com/opencord/voltha-lib-go/v3/pkg/config"
Esin Karamanccb714b2019-11-29 15:02:06 +000034 "github.com/opencord/voltha-lib-go/v3/pkg/db/kvstore"
35 "github.com/opencord/voltha-lib-go/v3/pkg/kafka"
36 "github.com/opencord/voltha-lib-go/v3/pkg/log"
37 "github.com/opencord/voltha-lib-go/v3/pkg/probe"
Scott Bakerdbd960e2020-02-28 08:57:51 -080038 "github.com/opencord/voltha-lib-go/v3/pkg/version"
39 "github.com/opencord/voltha-openolt-adapter/internal/pkg/config"
40 ac "github.com/opencord/voltha-openolt-adapter/internal/pkg/core"
Esin Karamanccb714b2019-11-29 15:02:06 +000041 ic "github.com/opencord/voltha-protos/v3/go/inter_container"
42 "github.com/opencord/voltha-protos/v3/go/voltha"
cuilin20187b2a8c32019-03-26 19:52:28 -070043)
44
45type adapter struct {
Girish Gowdru6a80bbd2019-07-02 07:36:09 -070046 instanceID string
cuilin20187b2a8c32019-03-26 19:52:28 -070047 config *config.AdapterFlags
48 iAdapter adapters.IAdapter
49 kafkaClient kafka.Client
50 kvClient kvstore.Client
npujarec5762e2020-01-01 14:08:48 +053051 kip kafka.InterContainerProxy
kdarapu381c6902019-07-31 18:23:16 +053052 coreProxy adapterif.CoreProxy
53 adapterProxy adapterif.AdapterProxy
54 eventProxy adapterif.EventProxy
cuilin20187b2a8c32019-03-26 19:52:28 -070055 halted bool
56 exitChannel chan int
57 receiverChannels []<-chan *ic.InterContainerMessage
58}
59
cuilin20187b2a8c32019-03-26 19:52:28 -070060func newAdapter(cf *config.AdapterFlags) *adapter {
61 var a adapter
Girish Gowdru6a80bbd2019-07-02 07:36:09 -070062 a.instanceID = cf.InstanceID
cuilin20187b2a8c32019-03-26 19:52:28 -070063 a.config = cf
64 a.halted = false
65 a.exitChannel = make(chan int, 1)
66 a.receiverChannels = make([]<-chan *ic.InterContainerMessage, 0)
67 return &a
68}
69
70func (a *adapter) start(ctx context.Context) {
Neha Sharma96b7bf22020-06-15 10:37:32 +000071 logger.Info(ctx, "Starting Core Adapter components")
cuilin20187b2a8c32019-03-26 19:52:28 -070072 var err error
73
Rohan Agrawal828bf4e2019-10-22 10:13:19 +000074 var p *probe.Probe
75 if value := ctx.Value(probe.ProbeContextKey); value != nil {
76 if _, ok := value.(*probe.Probe); ok {
77 p = value.(*probe.Probe)
78 p.RegisterService(
Neha Sharma96b7bf22020-06-15 10:37:32 +000079 ctx,
Rohan Agrawal828bf4e2019-10-22 10:13:19 +000080 "message-bus",
81 "kv-store",
82 "container-proxy",
83 "core-request-handler",
84 "register-with-core",
85 )
86 }
87 }
88
cuilin20187b2a8c32019-03-26 19:52:28 -070089 // Setup KV Client
Neha Sharma96b7bf22020-06-15 10:37:32 +000090 logger.Debugw(ctx, "create-kv-client", log.Fields{"kvstore": a.config.KVStoreType})
91 if err = a.setKVClient(ctx); err != nil {
92 logger.Fatalw(ctx, "error-setting-kv-client", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -070093 }
94
Rohan Agrawal828bf4e2019-10-22 10:13:19 +000095 if p != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +000096 p.UpdateStatus(ctx, "kv-store", probe.ServiceStatusRunning)
Rohan Agrawal828bf4e2019-10-22 10:13:19 +000097 }
98
divyadesaia37f78b2020-02-07 12:41:22 +000099 // Setup Log Config
Neha Sharma96b7bf22020-06-15 10:37:32 +0000100 cm := conf.NewConfigManager(ctx, a.kvClient, a.config.KVStoreType, a.config.KVStoreAddress, a.config.KVStoreTimeout)
divyadesaid26f6b12020-03-19 06:30:28 +0000101 go conf.StartLogLevelConfigProcessing(cm, ctx)
Girish Kumar935f7af2020-08-18 11:59:42 +0000102 go conf.StartLogFeaturesConfigProcessing(cm, ctx)
divyadesaia37f78b2020-02-07 12:41:22 +0000103
cuilin20187b2a8c32019-03-26 19:52:28 -0700104 // Setup Kafka Client
Neha Sharma96b7bf22020-06-15 10:37:32 +0000105 if a.kafkaClient, err = newKafkaClient(ctx, "sarama", a.config.KafkaAdapterAddress); err != nil {
106 logger.Fatalw(ctx, "Unsupported-common-client", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700107 }
108
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000109 if p != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000110 p.UpdateStatus(ctx, "message-bus", probe.ServiceStatusRunning)
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000111 }
112
Matteo Scandolo3ad5d2b2020-04-02 17:02:04 -0700113 // setup endpointManager
114
cuilin20187b2a8c32019-03-26 19:52:28 -0700115 // Start the common InterContainer Proxy - retries indefinitely
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000116 if a.kip, err = a.startInterContainerProxy(ctx, -1); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000117 logger.Fatal(ctx, "error-starting-inter-container-proxy")
cuilin20187b2a8c32019-03-26 19:52:28 -0700118 }
119
120 // Create the core proxy to handle requests to the Core
Neha Sharma96b7bf22020-06-15 10:37:32 +0000121 a.coreProxy = com.NewCoreProxy(ctx, a.kip, a.config.Topic, a.config.CoreTopic)
cuilin20187b2a8c32019-03-26 19:52:28 -0700122
123 // Create the adaptor proxy to handle request between olt and onu
Matteo Scandolo46654682020-08-05 11:46:37 -0700124 a.adapterProxy = com.NewAdapterProxy(ctx, a.kip, a.config.CoreTopic, cm.Backend)
cuilin20187b2a8c32019-03-26 19:52:28 -0700125
Devmalya Paulfb990a52019-07-09 10:01:49 -0400126 // Create the event proxy to post events to KAFKA
127 a.eventProxy = com.NewEventProxy(com.MsgClient(a.kafkaClient), com.MsgTopic(kafka.Topic{Name: a.config.EventTopic}))
128
cuilin20187b2a8c32019-03-26 19:52:28 -0700129 // Create the open OLT adapter
Girish Kumarf26e4882020-03-05 06:49:10 +0000130 if a.iAdapter, err = a.startOpenOLT(ctx, a.kip, a.coreProxy, a.adapterProxy, a.eventProxy, a.config); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000131 logger.Fatalw(ctx, "error-starting-openolt", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700132 }
133
134 // Register the core request handler
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000135 if err = a.setupRequestHandler(ctx, a.instanceID, a.iAdapter); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000136 logger.Fatalw(ctx, "error-setting-core-request-handler", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700137 }
138
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700139 // Register this adapter to the Core - retries indefinitely
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000140 if err = a.registerWithCore(ctx, -1); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000141 logger.Fatal(ctx, "error-registering-with-core")
cuilin20187b2a8c32019-03-26 19:52:28 -0700142 }
cbabu95f21522019-11-13 14:25:18 +0100143
cbabu116b73f2019-12-10 17:56:32 +0530144 // check the readiness and liveliness and update the probe status
145 a.checkServicesReadiness(ctx)
cbabu95f21522019-11-13 14:25:18 +0100146}
147
148/**
149This function checks the liveliness and readiness of the kakfa and kv-client services
150and update the status in the probe.
151*/
cbabu116b73f2019-12-10 17:56:32 +0530152func (a *adapter) checkServicesReadiness(ctx context.Context) {
153 // checks the kafka readiness
154 go a.checkKafkaReadiness(ctx)
155
156 // checks the kv-store readiness
157 go a.checkKvStoreReadiness(ctx)
158}
159
160/**
161This function checks the liveliness and readiness of the kv-store service
162and update the status in the probe.
163*/
164func (a *adapter) checkKvStoreReadiness(ctx context.Context) {
165 // dividing the live probe interval by 2 to get updated status every 30s
166 timeout := a.config.LiveProbeInterval / 2
167 kvStoreChannel := make(chan bool, 1)
168
169 // Default false to check the liveliness.
170 kvStoreChannel <- false
cbabu95f21522019-11-13 14:25:18 +0100171 for {
cbabu116b73f2019-12-10 17:56:32 +0530172 timeoutTimer := time.NewTimer(timeout)
173 select {
174 case liveliness := <-kvStoreChannel:
175 if !liveliness {
176 // kv-store not reachable or down, updating the status to not ready state
177 probe.UpdateStatusFromContext(ctx, "kv-store", probe.ServiceStatusNotReady)
178 timeout = a.config.NotLiveProbeInterval
179 } else {
180 // kv-store is reachable , updating the status to running state
181 probe.UpdateStatusFromContext(ctx, "kv-store", probe.ServiceStatusRunning)
182 timeout = a.config.LiveProbeInterval / 2
183 }
184 // Check if the timer has expired or not
185 if !timeoutTimer.Stop() {
186 <-timeoutTimer.C
187 }
188 case <-timeoutTimer.C:
Girish Kumarbeadc112020-02-26 18:41:02 +0000189 // Check the status of the kv-store. Use timeout of 2 seconds to avoid forever blocking
Neha Sharma96b7bf22020-06-15 10:37:32 +0000190 logger.Info(ctx, "kv-store liveliness-recheck")
Girish Kumarbeadc112020-02-26 18:41:02 +0000191 timeoutCtx, cancelFunc := context.WithTimeout(ctx, 2*time.Second)
192
193 kvStoreChannel <- a.kvClient.IsConnectionUp(timeoutCtx)
194 // Cleanup cancel func resources
195 cancelFunc()
cbabu95f21522019-11-13 14:25:18 +0100196 }
cbabu116b73f2019-12-10 17:56:32 +0530197 }
198}
199
200/**
201This function checks the liveliness and readiness of the kafka service
202and update the status in the probe.
203*/
204func (a *adapter) checkKafkaReadiness(ctx context.Context) {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000205 livelinessChannel := a.kafkaClient.EnableLivenessChannel(ctx, true)
206 healthinessChannel := a.kafkaClient.EnableHealthinessChannel(ctx, true)
cbabu116b73f2019-12-10 17:56:32 +0530207 timeout := a.config.LiveProbeInterval
Scott Bakere701b862020-02-20 16:19:16 -0800208 failed := false
cbabu116b73f2019-12-10 17:56:32 +0530209 for {
210 timeoutTimer := time.NewTimer(timeout)
211
212 select {
Scott Baker86fce9a2019-12-12 09:47:17 -0800213 case healthiness := <-healthinessChannel:
214 if !healthiness {
Scott Bakere701b862020-02-20 16:19:16 -0800215 // This will eventually cause K8s to restart the container, and will do
216 // so in a way that allows cleanup to continue, rather than an immediate
217 // panic and exit here.
218 probe.UpdateStatusFromContext(ctx, "message-bus", probe.ServiceStatusFailed)
219 failed = true
220 }
221 // Check if the timer has expired or not
222 if !timeoutTimer.Stop() {
223 <-timeoutTimer.C
Scott Baker86fce9a2019-12-12 09:47:17 -0800224 }
cbabu116b73f2019-12-10 17:56:32 +0530225 case liveliness := <-livelinessChannel:
Scott Bakere701b862020-02-20 16:19:16 -0800226 if failed {
227 // Failures of the message bus are permanent and can't ever be recovered from,
228 // so make sure we never inadvertently reset a failed state back to unready.
229 } else if !liveliness {
cbabu116b73f2019-12-10 17:56:32 +0530230 // kafka not reachable or down, updating the status to not ready state
231 probe.UpdateStatusFromContext(ctx, "message-bus", probe.ServiceStatusNotReady)
232 timeout = a.config.NotLiveProbeInterval
233 } else {
234 // kafka is reachable , updating the status to running state
235 probe.UpdateStatusFromContext(ctx, "message-bus", probe.ServiceStatusRunning)
236 timeout = a.config.LiveProbeInterval
237 }
238 // Check if the timer has expired or not
239 if !timeoutTimer.Stop() {
240 <-timeoutTimer.C
241 }
242 case <-timeoutTimer.C:
Neha Sharma96b7bf22020-06-15 10:37:32 +0000243 logger.Info(ctx, "kafka-proxy-liveness-recheck")
cbabu116b73f2019-12-10 17:56:32 +0530244 // send the liveness probe in a goroutine; we don't want to deadlock ourselves as
245 // the liveness probe may wait (and block) writing to our channel.
Neha Sharma96b7bf22020-06-15 10:37:32 +0000246 err := a.kafkaClient.SendLiveness(ctx)
cbabu116b73f2019-12-10 17:56:32 +0530247 if err != nil {
248 // Catch possible error case if sending liveness after Sarama has been stopped.
Neha Sharma96b7bf22020-06-15 10:37:32 +0000249 logger.Warnw(ctx, "error-kafka-send-liveness", log.Fields{"error": err})
cbabu116b73f2019-12-10 17:56:32 +0530250 }
cbabu95f21522019-11-13 14:25:18 +0100251 }
cbabu95f21522019-11-13 14:25:18 +0100252 }
cuilin20187b2a8c32019-03-26 19:52:28 -0700253}
254
npujarec5762e2020-01-01 14:08:48 +0530255func (a *adapter) stop(ctx context.Context) {
cuilin20187b2a8c32019-03-26 19:52:28 -0700256 // Stop leadership tracking
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700257 a.halted = true
cuilin20187b2a8c32019-03-26 19:52:28 -0700258
259 // send exit signal
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700260 a.exitChannel <- 0
cuilin20187b2a8c32019-03-26 19:52:28 -0700261
262 // Cleanup - applies only if we had a kvClient
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700263 if a.kvClient != nil {
cuilin20187b2a8c32019-03-26 19:52:28 -0700264 // Release all reservations
npujarec5762e2020-01-01 14:08:48 +0530265 if err := a.kvClient.ReleaseAllReservations(ctx); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000266 logger.Infow(ctx, "fail-to-release-all-reservations", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700267 }
268 // Close the DB connection
Neha Sharma96b7bf22020-06-15 10:37:32 +0000269 a.kvClient.Close(ctx)
cuilin20187b2a8c32019-03-26 19:52:28 -0700270 }
271
Scott Bakere701b862020-02-20 16:19:16 -0800272 if a.kip != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000273 a.kip.Stop(ctx)
Scott Bakere701b862020-02-20 16:19:16 -0800274 }
275
cuilin20187b2a8c32019-03-26 19:52:28 -0700276 // TODO: More cleanup
277}
278
Neha Sharma96b7bf22020-06-15 10:37:32 +0000279func newKVClient(ctx context.Context, storeType, address string, timeout time.Duration) (kvstore.Client, error) {
cuilin20187b2a8c32019-03-26 19:52:28 -0700280
Neha Sharma96b7bf22020-06-15 10:37:32 +0000281 logger.Infow(ctx, "kv-store-type", log.Fields{"store": storeType})
cuilin20187b2a8c32019-03-26 19:52:28 -0700282 switch storeType {
283 case "consul":
Neha Sharma96b7bf22020-06-15 10:37:32 +0000284 return kvstore.NewConsulClient(ctx, address, timeout)
cuilin20187b2a8c32019-03-26 19:52:28 -0700285 case "etcd":
Neha Sharma96b7bf22020-06-15 10:37:32 +0000286 return kvstore.NewEtcdClient(ctx, address, timeout, log.FatalLevel)
cuilin20187b2a8c32019-03-26 19:52:28 -0700287 }
288 return nil, errors.New("unsupported-kv-store")
289}
290
Neha Sharma96b7bf22020-06-15 10:37:32 +0000291func newKafkaClient(ctx context.Context, clientType, address string) (kafka.Client, error) {
cuilin20187b2a8c32019-03-26 19:52:28 -0700292
Neha Sharma96b7bf22020-06-15 10:37:32 +0000293 logger.Infow(ctx, "common-client-type", log.Fields{"client": clientType})
cuilin20187b2a8c32019-03-26 19:52:28 -0700294 switch clientType {
295 case "sarama":
296 return kafka.NewSaramaClient(
Neha Sharma3f221ae2020-04-29 19:02:12 +0000297 kafka.Address(address),
cuilin20187b2a8c32019-03-26 19:52:28 -0700298 kafka.ProducerReturnOnErrors(true),
299 kafka.ProducerReturnOnSuccess(true),
300 kafka.ProducerMaxRetries(6),
Abhilash S.L3b494632019-07-16 15:51:09 +0530301 kafka.ProducerRetryBackoff(time.Millisecond*30),
302 kafka.MetadatMaxRetries(15)), nil
cuilin20187b2a8c32019-03-26 19:52:28 -0700303 }
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700304
cuilin20187b2a8c32019-03-26 19:52:28 -0700305 return nil, errors.New("unsupported-client-type")
306}
307
Neha Sharma96b7bf22020-06-15 10:37:32 +0000308func (a *adapter) setKVClient(ctx context.Context) error {
309 client, err := newKVClient(ctx, a.config.KVStoreType, a.config.KVStoreAddress, a.config.KVStoreTimeout)
cuilin20187b2a8c32019-03-26 19:52:28 -0700310 if err != nil {
311 a.kvClient = nil
cuilin20187b2a8c32019-03-26 19:52:28 -0700312 return err
313 }
314 a.kvClient = client
divyadesaia37f78b2020-02-07 12:41:22 +0000315
cuilin20187b2a8c32019-03-26 19:52:28 -0700316 return nil
317}
318
npujarec5762e2020-01-01 14:08:48 +0530319func (a *adapter) startInterContainerProxy(ctx context.Context, retries int) (kafka.InterContainerProxy, error) {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000320 logger.Infow(ctx, "starting-intercontainer-messaging-proxy", log.Fields{"address": a.config.KafkaAdapterAddress,
Neha Sharma3f221ae2020-04-29 19:02:12 +0000321 "topic": a.config.Topic})
cuilin20187b2a8c32019-03-26 19:52:28 -0700322 var err error
npujarec5762e2020-01-01 14:08:48 +0530323 kip := kafka.NewInterContainerProxy(
Neha Sharma3f221ae2020-04-29 19:02:12 +0000324 kafka.InterContainerAddress(a.config.KafkaAdapterAddress),
cuilin20187b2a8c32019-03-26 19:52:28 -0700325 kafka.MsgClient(a.kafkaClient),
npujarec5762e2020-01-01 14:08:48 +0530326 kafka.DefaultTopic(&kafka.Topic{Name: a.config.Topic}))
cuilin20187b2a8c32019-03-26 19:52:28 -0700327 count := 0
328 for {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000329 if err = kip.Start(ctx); err != nil {
330 logger.Warnw(ctx, "error-starting-messaging-proxy", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700331 if retries == count {
332 return nil, err
333 }
334 count = +1
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700335 // Take a nap before retrying
cuilin20187b2a8c32019-03-26 19:52:28 -0700336 time.Sleep(2 * time.Second)
337 } else {
338 break
339 }
340 }
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000341 probe.UpdateStatusFromContext(ctx, "container-proxy", probe.ServiceStatusRunning)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000342 logger.Info(ctx, "common-messaging-proxy-created")
cuilin20187b2a8c32019-03-26 19:52:28 -0700343 return kip, nil
344}
345
npujarec5762e2020-01-01 14:08:48 +0530346func (a *adapter) startOpenOLT(ctx context.Context, kip kafka.InterContainerProxy,
Abhilash Laxmeshwarf9942e92020-01-07 15:32:44 +0530347 cp adapterif.CoreProxy, ap adapterif.AdapterProxy, ep adapterif.EventProxy,
348 cfg *config.AdapterFlags) (*ac.OpenOLT, error) {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000349 logger.Info(ctx, "starting-open-olt")
cuilin20187b2a8c32019-03-26 19:52:28 -0700350 var err error
Abhilash Laxmeshwarf9942e92020-01-07 15:32:44 +0530351 sOLT := ac.NewOpenOLT(ctx, a.kip, cp, ap, ep, cfg)
cuilin20187b2a8c32019-03-26 19:52:28 -0700352
353 if err = sOLT.Start(ctx); err != nil {
cuilin20187b2a8c32019-03-26 19:52:28 -0700354 return nil, err
355 }
356
Neha Sharma96b7bf22020-06-15 10:37:32 +0000357 logger.Info(ctx, "open-olt-started")
cuilin20187b2a8c32019-03-26 19:52:28 -0700358 return sOLT, nil
359}
360
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000361func (a *adapter) setupRequestHandler(ctx context.Context, coreInstanceID string, iadapter adapters.IAdapter) error {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000362 logger.Info(ctx, "setting-request-handler")
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700363 requestProxy := com.NewRequestHandlerProxy(coreInstanceID, iadapter, a.coreProxy)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000364 if err := a.kip.SubscribeWithRequestHandlerInterface(ctx, kafka.Topic{Name: a.config.Topic}, requestProxy); err != nil {
cuilin20187b2a8c32019-03-26 19:52:28 -0700365 return err
366
367 }
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000368 probe.UpdateStatusFromContext(ctx, "core-request-handler", probe.ServiceStatusRunning)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000369 logger.Info(ctx, "request-handler-setup-done")
cuilin20187b2a8c32019-03-26 19:52:28 -0700370 return nil
371}
372
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000373func (a *adapter) registerWithCore(ctx context.Context, retries int) error {
Matteo Scandolo3ad5d2b2020-04-02 17:02:04 -0700374 adapterID := fmt.Sprintf("openolt_%d", a.config.CurrentReplica)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000375 logger.Infow(ctx, "registering-with-core", log.Fields{
Matteo Scandolo3ad5d2b2020-04-02 17:02:04 -0700376 "adapterID": adapterID,
377 "currentReplica": a.config.CurrentReplica,
378 "totalReplicas": a.config.TotalReplicas,
379 })
380 adapterDescription := &voltha.Adapter{
381 Id: adapterID, // Unique name for the device type
Matt Jeanneretf880eb62019-07-16 20:08:03 -0400382 Vendor: "VOLTHA OpenOLT",
Matteo Scandolo3ad5d2b2020-04-02 17:02:04 -0700383 Version: version.VersionInfo.Version,
384 // TODO once we'll be ready to support multiple versions of the OpenOLT adapter
385 // the Endpoint will have to change to `openolt_<currentReplica`>
serkant.uluderya5e3528d2020-05-22 19:31:07 -0700386 Endpoint: a.config.Topic,
Matteo Scandolo3ad5d2b2020-04-02 17:02:04 -0700387 Type: "openolt",
388 CurrentReplica: int32(a.config.CurrentReplica),
389 TotalReplicas: int32(a.config.TotalReplicas),
390 }
391 types := []*voltha.DeviceType{{
392 Id: "openolt",
393 Adapter: "openolt", // Type of the adapter that handles device type
Girish Gowdru0c588b22019-04-23 23:24:56 -0400394 AcceptsBulkFlowUpdate: false, // Currently openolt adapter does not support bulk flow handling
395 AcceptsAddRemoveFlowUpdates: true}}
cuilin20187b2a8c32019-03-26 19:52:28 -0700396 deviceTypes := &voltha.DeviceTypes{Items: types}
397 count := 0
398 for {
Neha Sharma8f4e4322020-08-06 10:51:53 +0000399 if err := a.coreProxy.RegisterAdapter(log.WithSpanFromContext(context.TODO(), ctx), adapterDescription, deviceTypes); err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000400 logger.Warnw(ctx, "registering-with-core-failed", log.Fields{"error": err})
cuilin20187b2a8c32019-03-26 19:52:28 -0700401 if retries == count {
402 return err
403 }
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700404 count++
405 // Take a nap before retrying
cuilin20187b2a8c32019-03-26 19:52:28 -0700406 time.Sleep(2 * time.Second)
407 } else {
408 break
409 }
410 }
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000411 probe.UpdateStatusFromContext(ctx, "register-with-core", probe.ServiceStatusRunning)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000412 logger.Info(ctx, "registered-with-core")
cuilin20187b2a8c32019-03-26 19:52:28 -0700413 return nil
414}
415
Neha Sharma96b7bf22020-06-15 10:37:32 +0000416func waitForExit(ctx context.Context) int {
cuilin20187b2a8c32019-03-26 19:52:28 -0700417 signalChannel := make(chan os.Signal, 1)
418 signal.Notify(signalChannel,
419 syscall.SIGHUP,
420 syscall.SIGINT,
421 syscall.SIGTERM,
422 syscall.SIGQUIT)
423
424 exitChannel := make(chan int)
425
426 go func() {
427 s := <-signalChannel
428 switch s {
429 case syscall.SIGHUP,
430 syscall.SIGINT,
431 syscall.SIGTERM,
432 syscall.SIGQUIT:
Neha Sharma96b7bf22020-06-15 10:37:32 +0000433 logger.Infow(ctx, "closing-signal-received", log.Fields{"signal": s})
cuilin20187b2a8c32019-03-26 19:52:28 -0700434 exitChannel <- 0
435 default:
Neha Sharma96b7bf22020-06-15 10:37:32 +0000436 logger.Infow(ctx, "unexpected-signal-received", log.Fields{"signal": s})
cuilin20187b2a8c32019-03-26 19:52:28 -0700437 exitChannel <- 1
438 }
439 }()
440
441 code := <-exitChannel
442 return code
443}
444
445func printBanner() {
David K. Bainbridge794735f2020-02-11 21:01:37 -0800446 fmt.Println(` ____ ____ _ _______ `)
447 fmt.Println(` / _ \ / __ \| | |__ __|`)
448 fmt.Println(` | | | |_ __ ___ _ __ | | | | | | | `)
449 fmt.Println(` | | | | '_ \ / _ \ '_ \ | | | | | | | `)
450 fmt.Println(` | |__| | |_) | __/ | | || |__| | |____| | `)
451 fmt.Println(` \____/| .__/ \___|_| |_| \____/|______|_| `)
452 fmt.Println(` | | `)
453 fmt.Println(` |_| `)
454 fmt.Println(` `)
cuilin20187b2a8c32019-03-26 19:52:28 -0700455}
456
Matt Jeanneretf880eb62019-07-16 20:08:03 -0400457func printVersion() {
458 fmt.Println("VOLTHA OpenOLT Adapter")
459 fmt.Println(version.VersionInfo.String(" "))
460}
461
cuilin20187b2a8c32019-03-26 19:52:28 -0700462func main() {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000463 ctx := context.Background()
cuilin20187b2a8c32019-03-26 19:52:28 -0700464 start := time.Now()
465
466 cf := config.NewAdapterFlags()
467 cf.ParseCommandArguments()
468
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700469 // Setup logging
cuilin20187b2a8c32019-03-26 19:52:28 -0700470
Rohan Agrawal02f784d2020-02-14 09:34:02 +0000471 logLevel, err := log.StringToLogLevel(cf.LogLevel)
472 if err != nil {
Neha Sharma96b7bf22020-06-15 10:37:32 +0000473 logger.Fatalf(ctx, "Cannot setup logging, %s", err)
Rohan Agrawal02f784d2020-02-14 09:34:02 +0000474 }
Rohan Agrawal2488f192020-01-31 09:26:55 +0000475
Girish Gowdru6a80bbd2019-07-02 07:36:09 -0700476 // Setup default logger - applies for packages that do not have specific logger set
Rohan Agrawal02f784d2020-02-14 09:34:02 +0000477 if _, err := log.SetDefaultLogger(log.JSON, logLevel, log.Fields{"instanceId": cf.InstanceID}); err != nil {
Girish Kumara1ea2aa2020-08-19 18:14:22 +0000478 logger.With(log.Fields{"error": err}).Fatal(ctx, "Cannot setup logging")
cuilin20187b2a8c32019-03-26 19:52:28 -0700479 }
480
481 // Update all loggers (provisionned via init) with a common field
Hardik Windlassb9c869b2019-10-10 08:34:32 +0000482 if err := log.UpdateAllLoggers(log.Fields{"instanceId": cf.InstanceID}); err != nil {
Girish Kumara1ea2aa2020-08-19 18:14:22 +0000483 logger.With(log.Fields{"error": err}).Fatal(ctx, "Cannot setup logging")
cuilin20187b2a8c32019-03-26 19:52:28 -0700484 }
485
Rohan Agrawal02f784d2020-02-14 09:34:02 +0000486 log.SetAllLogLevel(logLevel)
Rohan Agrawal93bced32020-02-11 10:16:01 +0000487
Matteo Scandolo8f2b9572020-02-28 15:35:23 -0800488 realMain()
489
Kent Hagermane6ff1012020-07-14 15:07:53 -0400490 defer func() {
491 err := log.CleanUp()
492 if err != nil {
493 logger.Errorw(context.Background(), "unable-to-flush-any-buffered-log-entries", log.Fields{"error": err})
494 }
495 }()
cuilin20187b2a8c32019-03-26 19:52:28 -0700496
Matt Jeanneretf880eb62019-07-16 20:08:03 -0400497 // Print version / build information and exit
498 if cf.DisplayVersionOnly {
499 printVersion()
500 return
501 }
502
cuilin20187b2a8c32019-03-26 19:52:28 -0700503 // Print banner if specified
504 if cf.Banner {
505 printBanner()
506 }
507
Neha Sharma96b7bf22020-06-15 10:37:32 +0000508 logger.Infow(ctx, "config", log.Fields{"config": *cf})
cuilin20187b2a8c32019-03-26 19:52:28 -0700509
510 ctx, cancel := context.WithCancel(context.Background())
511 defer cancel()
512
513 ad := newAdapter(cf)
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000514
515 p := &probe.Probe{}
Neha Sharma96b7bf22020-06-15 10:37:32 +0000516 go p.ListenAndServe(ctx, ad.config.ProbeAddress)
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000517
518 probeCtx := context.WithValue(ctx, probe.ProbeContextKey, p)
519
Girish Kumar935f7af2020-08-18 11:59:42 +0000520 closer, err := log.GetGlobalLFM().InitTracingAndLogCorrelation(cf.TraceEnabled, cf.TraceAgentAddress, cf.LogCorrelationEnabled)
Girish Kumar11e15972020-06-15 14:51:10 +0000521 if err != nil {
522 logger.Warnw(ctx, "unable-to-initialize-tracing-and-log-correlation-module", log.Fields{"error": err})
523 } else {
524 defer log.TerminateTracing(closer)
525 }
526
Rohan Agrawal828bf4e2019-10-22 10:13:19 +0000527 go ad.start(probeCtx)
cuilin20187b2a8c32019-03-26 19:52:28 -0700528
Neha Sharma96b7bf22020-06-15 10:37:32 +0000529 code := waitForExit(ctx)
530 logger.Infow(ctx, "received-a-closing-signal", log.Fields{"code": code})
cuilin20187b2a8c32019-03-26 19:52:28 -0700531
532 // Cleanup before leaving
npujarec5762e2020-01-01 14:08:48 +0530533 ad.stop(ctx)
cuilin20187b2a8c32019-03-26 19:52:28 -0700534
535 elapsed := time.Since(start)
Neha Sharma96b7bf22020-06-15 10:37:32 +0000536 logger.Infow(ctx, "run-time", log.Fields{"instanceId": ad.config.InstanceID, "time": elapsed / time.Second})
cuilin20187b2a8c32019-03-26 19:52:28 -0700537}