cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2018-present Open Networking Foundation |
| 3 | |
| 4 | * 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 |
| 7 | |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | |
| 10 | * 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. |
| 15 | */ |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 16 | |
| 17 | //Package main invokes the application |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 18 | package main |
| 19 | |
| 20 | import ( |
| 21 | "context" |
| 22 | "errors" |
| 23 | "fmt" |
kdarapu | 381c690 | 2019-07-31 18:23:16 +0530 | [diff] [blame] | 24 | "os" |
| 25 | "os/signal" |
| 26 | "strconv" |
| 27 | "syscall" |
| 28 | "time" |
| 29 | |
Scott Baker | 5129015 | 2019-10-24 14:23:20 -0700 | [diff] [blame] | 30 | "github.com/opencord/voltha-lib-go/v2/pkg/adapters/adapterif" |
kdarapu | 381c690 | 2019-07-31 18:23:16 +0530 | [diff] [blame] | 31 | |
Scott Baker | 5129015 | 2019-10-24 14:23:20 -0700 | [diff] [blame] | 32 | "github.com/opencord/voltha-lib-go/v2/pkg/adapters" |
| 33 | com "github.com/opencord/voltha-lib-go/v2/pkg/adapters/common" |
| 34 | "github.com/opencord/voltha-lib-go/v2/pkg/db/kvstore" |
| 35 | "github.com/opencord/voltha-lib-go/v2/pkg/kafka" |
| 36 | "github.com/opencord/voltha-lib-go/v2/pkg/log" |
| 37 | "github.com/opencord/voltha-lib-go/v2/pkg/probe" |
Girish Gowdru | 0c588b2 | 2019-04-23 23:24:56 -0400 | [diff] [blame] | 38 | ac "github.com/opencord/voltha-openolt-adapter/adaptercore" |
| 39 | "github.com/opencord/voltha-openolt-adapter/config" |
Matt Jeanneret | 0c9ae28 | 2019-07-18 18:14:28 -0400 | [diff] [blame] | 40 | "github.com/opencord/voltha-openolt-adapter/config/version" |
Scott Baker | c6e54cb | 2019-11-04 09:31:25 -0800 | [diff] [blame] | 41 | ic "github.com/opencord/voltha-protos/v2/go/inter_container" |
| 42 | "github.com/opencord/voltha-protos/v2/go/voltha" |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 43 | ) |
| 44 | |
| 45 | type adapter struct { |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 46 | instanceID string |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 47 | config *config.AdapterFlags |
| 48 | iAdapter adapters.IAdapter |
| 49 | kafkaClient kafka.Client |
| 50 | kvClient kvstore.Client |
| 51 | kip *kafka.InterContainerProxy |
kdarapu | 381c690 | 2019-07-31 18:23:16 +0530 | [diff] [blame] | 52 | coreProxy adapterif.CoreProxy |
| 53 | adapterProxy adapterif.AdapterProxy |
| 54 | eventProxy adapterif.EventProxy |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 55 | halted bool |
| 56 | exitChannel chan int |
| 57 | receiverChannels []<-chan *ic.InterContainerMessage |
| 58 | } |
| 59 | |
| 60 | func init() { |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 61 | _, _ = log.AddPackage(log.JSON, log.DebugLevel, nil) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 62 | } |
| 63 | |
| 64 | func newAdapter(cf *config.AdapterFlags) *adapter { |
| 65 | var a adapter |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 66 | a.instanceID = cf.InstanceID |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 67 | a.config = cf |
| 68 | a.halted = false |
| 69 | a.exitChannel = make(chan int, 1) |
| 70 | a.receiverChannels = make([]<-chan *ic.InterContainerMessage, 0) |
| 71 | return &a |
| 72 | } |
| 73 | |
| 74 | func (a *adapter) start(ctx context.Context) { |
| 75 | log.Info("Starting Core Adapter components") |
| 76 | var err error |
| 77 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 78 | var p *probe.Probe |
| 79 | if value := ctx.Value(probe.ProbeContextKey); value != nil { |
| 80 | if _, ok := value.(*probe.Probe); ok { |
| 81 | p = value.(*probe.Probe) |
| 82 | p.RegisterService( |
| 83 | "message-bus", |
| 84 | "kv-store", |
| 85 | "container-proxy", |
| 86 | "core-request-handler", |
| 87 | "register-with-core", |
| 88 | ) |
| 89 | } |
| 90 | } |
| 91 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 92 | // Setup KV Client |
| 93 | log.Debugw("create-kv-client", log.Fields{"kvstore": a.config.KVStoreType}) |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 94 | if err = a.setKVClient(); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 95 | log.Fatal("error-setting-kv-client") |
| 96 | } |
| 97 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 98 | if p != nil { |
| 99 | p.UpdateStatus("kv-store", probe.ServiceStatusRunning) |
| 100 | } |
| 101 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 102 | // Setup Kafka Client |
| 103 | if a.kafkaClient, err = newKafkaClient("sarama", a.config.KafkaAdapterHost, a.config.KafkaAdapterPort); err != nil { |
| 104 | log.Fatal("Unsupported-common-client") |
| 105 | } |
| 106 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 107 | if p != nil { |
| 108 | p.UpdateStatus("message-bus", probe.ServiceStatusRunning) |
| 109 | } |
| 110 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 111 | // Start the common InterContainer Proxy - retries indefinitely |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 112 | if a.kip, err = a.startInterContainerProxy(ctx, -1); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 113 | log.Fatal("error-starting-inter-container-proxy") |
| 114 | } |
| 115 | |
| 116 | // Create the core proxy to handle requests to the Core |
| 117 | a.coreProxy = com.NewCoreProxy(a.kip, a.config.Topic, a.config.CoreTopic) |
| 118 | |
| 119 | // Create the adaptor proxy to handle request between olt and onu |
| 120 | a.adapterProxy = com.NewAdapterProxy(a.kip, "brcm_openomci_onu", a.config.CoreTopic) |
| 121 | |
Devmalya Paul | fb990a5 | 2019-07-09 10:01:49 -0400 | [diff] [blame] | 122 | // Create the event proxy to post events to KAFKA |
| 123 | a.eventProxy = com.NewEventProxy(com.MsgClient(a.kafkaClient), com.MsgTopic(kafka.Topic{Name: a.config.EventTopic})) |
| 124 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 125 | // Create the open OLT adapter |
kdarapu | 381c690 | 2019-07-31 18:23:16 +0530 | [diff] [blame] | 126 | if a.iAdapter, err = a.startOpenOLT(ctx, a.kip, a.coreProxy, a.adapterProxy, a.eventProxy, |
| 127 | a.config.OnuNumber, |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 128 | a.config.KVStoreHost, a.config.KVStorePort, a.config.KVStoreType); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 129 | log.Fatal("error-starting-inter-container-proxy") |
| 130 | } |
| 131 | |
| 132 | // Register the core request handler |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 133 | if err = a.setupRequestHandler(ctx, a.instanceID, a.iAdapter); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 134 | log.Fatal("error-setting-core-request-handler") |
| 135 | } |
| 136 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 137 | // Register this adapter to the Core - retries indefinitely |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 138 | if err = a.registerWithCore(ctx, -1); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 139 | log.Fatal("error-registering-with-core") |
| 140 | } |
| 141 | } |
| 142 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 143 | func (a *adapter) stop() { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 144 | // Stop leadership tracking |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 145 | a.halted = true |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 146 | |
| 147 | // send exit signal |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 148 | a.exitChannel <- 0 |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 149 | |
| 150 | // Cleanup - applies only if we had a kvClient |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 151 | if a.kvClient != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 152 | // Release all reservations |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 153 | if err := a.kvClient.ReleaseAllReservations(); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 154 | log.Infow("fail-to-release-all-reservations", log.Fields{"error": err}) |
| 155 | } |
| 156 | // Close the DB connection |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 157 | a.kvClient.Close() |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 158 | } |
| 159 | |
| 160 | // TODO: More cleanup |
| 161 | } |
| 162 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 163 | func newKVClient(storeType, address string, timeout int) (kvstore.Client, error) { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 164 | |
| 165 | log.Infow("kv-store-type", log.Fields{"store": storeType}) |
| 166 | switch storeType { |
| 167 | case "consul": |
| 168 | return kvstore.NewConsulClient(address, timeout) |
| 169 | case "etcd": |
| 170 | return kvstore.NewEtcdClient(address, timeout) |
| 171 | } |
| 172 | return nil, errors.New("unsupported-kv-store") |
| 173 | } |
| 174 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 175 | func newKafkaClient(clientType, host string, port int) (kafka.Client, error) { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 176 | |
| 177 | log.Infow("common-client-type", log.Fields{"client": clientType}) |
| 178 | switch clientType { |
| 179 | case "sarama": |
| 180 | return kafka.NewSaramaClient( |
| 181 | kafka.Host(host), |
| 182 | kafka.Port(port), |
| 183 | kafka.ProducerReturnOnErrors(true), |
| 184 | kafka.ProducerReturnOnSuccess(true), |
| 185 | kafka.ProducerMaxRetries(6), |
Abhilash S.L | 3b49463 | 2019-07-16 15:51:09 +0530 | [diff] [blame] | 186 | kafka.ProducerRetryBackoff(time.Millisecond*30), |
| 187 | kafka.MetadatMaxRetries(15)), nil |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 188 | } |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 189 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 190 | return nil, errors.New("unsupported-client-type") |
| 191 | } |
| 192 | |
| 193 | func (a *adapter) setKVClient() error { |
| 194 | addr := a.config.KVStoreHost + ":" + strconv.Itoa(a.config.KVStorePort) |
| 195 | client, err := newKVClient(a.config.KVStoreType, addr, a.config.KVStoreTimeout) |
| 196 | if err != nil { |
| 197 | a.kvClient = nil |
| 198 | log.Error(err) |
| 199 | return err |
| 200 | } |
| 201 | a.kvClient = client |
| 202 | return nil |
| 203 | } |
| 204 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 205 | func (a *adapter) startInterContainerProxy(ctx context.Context, retries int) (*kafka.InterContainerProxy, error) { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 206 | log.Infow("starting-intercontainer-messaging-proxy", log.Fields{"host": a.config.KafkaAdapterHost, |
| 207 | "port": a.config.KafkaAdapterPort, "topic": a.config.Topic}) |
| 208 | var err error |
| 209 | var kip *kafka.InterContainerProxy |
| 210 | if kip, err = kafka.NewInterContainerProxy( |
| 211 | kafka.InterContainerHost(a.config.KafkaAdapterHost), |
| 212 | kafka.InterContainerPort(a.config.KafkaAdapterPort), |
| 213 | kafka.MsgClient(a.kafkaClient), |
| 214 | kafka.DefaultTopic(&kafka.Topic{Name: a.config.Topic})); err != nil { |
| 215 | log.Errorw("fail-to-create-common-proxy", log.Fields{"error": err}) |
| 216 | return nil, err |
| 217 | } |
| 218 | count := 0 |
| 219 | for { |
| 220 | if err = kip.Start(); err != nil { |
| 221 | log.Warnw("error-starting-messaging-proxy", log.Fields{"error": err}) |
| 222 | if retries == count { |
| 223 | return nil, err |
| 224 | } |
| 225 | count = +1 |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 226 | // Take a nap before retrying |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 227 | time.Sleep(2 * time.Second) |
| 228 | } else { |
| 229 | break |
| 230 | } |
| 231 | } |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 232 | probe.UpdateStatusFromContext(ctx, "container-proxy", probe.ServiceStatusRunning) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 233 | log.Info("common-messaging-proxy-created") |
| 234 | return kip, nil |
| 235 | } |
| 236 | |
kdarapu | 381c690 | 2019-07-31 18:23:16 +0530 | [diff] [blame] | 237 | func (a *adapter) startOpenOLT(ctx context.Context, kip *kafka.InterContainerProxy, |
| 238 | cp adapterif.CoreProxy, ap adapterif.AdapterProxy, ep adapterif.EventProxy, onuNumber int, kvStoreHost string, |
| 239 | kvStorePort int, KVStoreType string) (*ac.OpenOLT, error) { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 240 | log.Info("starting-open-olt") |
| 241 | var err error |
Devmalya Paul | fb990a5 | 2019-07-09 10:01:49 -0400 | [diff] [blame] | 242 | sOLT := ac.NewOpenOLT(ctx, a.kip, cp, ap, ep, onuNumber, kvStoreHost, kvStorePort, KVStoreType) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 243 | |
| 244 | if err = sOLT.Start(ctx); err != nil { |
| 245 | log.Fatalw("error-starting-messaging-proxy", log.Fields{"error": err}) |
| 246 | return nil, err |
| 247 | } |
| 248 | |
| 249 | log.Info("open-olt-started") |
| 250 | return sOLT, nil |
| 251 | } |
| 252 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 253 | func (a *adapter) setupRequestHandler(ctx context.Context, coreInstanceID string, iadapter adapters.IAdapter) error { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 254 | log.Info("setting-request-handler") |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 255 | requestProxy := com.NewRequestHandlerProxy(coreInstanceID, iadapter, a.coreProxy) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 256 | if err := a.kip.SubscribeWithRequestHandlerInterface(kafka.Topic{Name: a.config.Topic}, requestProxy); err != nil { |
| 257 | log.Errorw("request-handler-setup-failed", log.Fields{"error": err}) |
| 258 | return err |
| 259 | |
| 260 | } |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 261 | probe.UpdateStatusFromContext(ctx, "core-request-handler", probe.ServiceStatusRunning) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 262 | log.Info("request-handler-setup-done") |
| 263 | return nil |
| 264 | } |
| 265 | |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 266 | func (a *adapter) registerWithCore(ctx context.Context, retries int) error { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 267 | log.Info("registering-with-core") |
Girish Gowdru | 0c588b2 | 2019-04-23 23:24:56 -0400 | [diff] [blame] | 268 | adapterDescription := &voltha.Adapter{Id: "openolt", // Unique name for the device type |
Matt Jeanneret | f880eb6 | 2019-07-16 20:08:03 -0400 | [diff] [blame] | 269 | Vendor: "VOLTHA OpenOLT", |
| 270 | Version: version.VersionInfo.Version} |
Girish Gowdru | 0c588b2 | 2019-04-23 23:24:56 -0400 | [diff] [blame] | 271 | types := []*voltha.DeviceType{{Id: "openolt", |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 272 | Adapter: "openolt", // Name of the adapter that handles device type |
Girish Gowdru | 0c588b2 | 2019-04-23 23:24:56 -0400 | [diff] [blame] | 273 | AcceptsBulkFlowUpdate: false, // Currently openolt adapter does not support bulk flow handling |
| 274 | AcceptsAddRemoveFlowUpdates: true}} |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 275 | deviceTypes := &voltha.DeviceTypes{Items: types} |
| 276 | count := 0 |
| 277 | for { |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 278 | if err := a.coreProxy.RegisterAdapter(context.TODO(), adapterDescription, deviceTypes); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 279 | log.Warnw("registering-with-core-failed", log.Fields{"error": err}) |
| 280 | if retries == count { |
| 281 | return err |
| 282 | } |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 283 | count++ |
| 284 | // Take a nap before retrying |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 285 | time.Sleep(2 * time.Second) |
| 286 | } else { |
| 287 | break |
| 288 | } |
| 289 | } |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 290 | probe.UpdateStatusFromContext(ctx, "register-with-core", probe.ServiceStatusRunning) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 291 | log.Info("registered-with-core") |
| 292 | return nil |
| 293 | } |
| 294 | |
| 295 | func waitForExit() int { |
| 296 | signalChannel := make(chan os.Signal, 1) |
| 297 | signal.Notify(signalChannel, |
| 298 | syscall.SIGHUP, |
| 299 | syscall.SIGINT, |
| 300 | syscall.SIGTERM, |
| 301 | syscall.SIGQUIT) |
| 302 | |
| 303 | exitChannel := make(chan int) |
| 304 | |
| 305 | go func() { |
| 306 | s := <-signalChannel |
| 307 | switch s { |
| 308 | case syscall.SIGHUP, |
| 309 | syscall.SIGINT, |
| 310 | syscall.SIGTERM, |
| 311 | syscall.SIGQUIT: |
| 312 | log.Infow("closing-signal-received", log.Fields{"signal": s}) |
| 313 | exitChannel <- 0 |
| 314 | default: |
| 315 | log.Infow("unexpected-signal-received", log.Fields{"signal": s}) |
| 316 | exitChannel <- 1 |
| 317 | } |
| 318 | }() |
| 319 | |
| 320 | code := <-exitChannel |
| 321 | return code |
| 322 | } |
| 323 | |
| 324 | func printBanner() { |
| 325 | fmt.Println(" ____ ____ _ _______ ") |
| 326 | fmt.Println(" / _ \\ / __\\| | |__ __|") |
| 327 | fmt.Println(" | | | |_ __ ___ _ __ | | | | | | | ") |
| 328 | fmt.Println(" | | | | '_\\ / _\\ '_\\ | | | | | | | ") |
| 329 | fmt.Println(" | |__| | |_) | __/ | | || |__| | |____| | ") |
| 330 | fmt.Println(" \\____/| .__/\\___|_| |_|\\____/|______|_| ") |
| 331 | fmt.Println(" | | ") |
| 332 | fmt.Println(" |_| ") |
| 333 | fmt.Println(" ") |
| 334 | } |
| 335 | |
Matt Jeanneret | f880eb6 | 2019-07-16 20:08:03 -0400 | [diff] [blame] | 336 | func printVersion() { |
| 337 | fmt.Println("VOLTHA OpenOLT Adapter") |
| 338 | fmt.Println(version.VersionInfo.String(" ")) |
| 339 | } |
| 340 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 341 | func main() { |
| 342 | start := time.Now() |
| 343 | |
| 344 | cf := config.NewAdapterFlags() |
| 345 | cf.ParseCommandArguments() |
| 346 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 347 | // Setup logging |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 348 | |
Girish Gowdru | 6a80bbd | 2019-07-02 07:36:09 -0700 | [diff] [blame] | 349 | // Setup default logger - applies for packages that do not have specific logger set |
Hardik Windlass | b9c869b | 2019-10-10 08:34:32 +0000 | [diff] [blame] | 350 | if _, err := log.SetDefaultLogger(log.JSON, cf.LogLevel, log.Fields{"instanceId": cf.InstanceID}); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 351 | log.With(log.Fields{"error": err}).Fatal("Cannot setup logging") |
| 352 | } |
| 353 | |
| 354 | // Update all loggers (provisionned via init) with a common field |
Hardik Windlass | b9c869b | 2019-10-10 08:34:32 +0000 | [diff] [blame] | 355 | if err := log.UpdateAllLoggers(log.Fields{"instanceId": cf.InstanceID}); err != nil { |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 356 | log.With(log.Fields{"error": err}).Fatal("Cannot setup logging") |
| 357 | } |
| 358 | |
Scott Baker | 5129015 | 2019-10-24 14:23:20 -0700 | [diff] [blame] | 359 | log.SetPackageLogLevel("github.com/opencord/voltha-lib-go/v2/pkg/adapters/common", log.DebugLevel) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 360 | |
| 361 | defer log.CleanUp() |
| 362 | |
Matt Jeanneret | f880eb6 | 2019-07-16 20:08:03 -0400 | [diff] [blame] | 363 | // Print version / build information and exit |
| 364 | if cf.DisplayVersionOnly { |
| 365 | printVersion() |
| 366 | return |
| 367 | } |
| 368 | |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 369 | // Print banner if specified |
| 370 | if cf.Banner { |
| 371 | printBanner() |
| 372 | } |
| 373 | |
| 374 | log.Infow("config", log.Fields{"config": *cf}) |
| 375 | |
| 376 | ctx, cancel := context.WithCancel(context.Background()) |
| 377 | defer cancel() |
| 378 | |
| 379 | ad := newAdapter(cf) |
Rohan Agrawal | 828bf4e | 2019-10-22 10:13:19 +0000 | [diff] [blame] | 380 | |
| 381 | p := &probe.Probe{} |
| 382 | go p.ListenAndServe(fmt.Sprintf("%s:%d", ad.config.ProbeHost, ad.config.ProbePort)) |
| 383 | |
| 384 | probeCtx := context.WithValue(ctx, probe.ProbeContextKey, p) |
| 385 | |
| 386 | go ad.start(probeCtx) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 387 | |
| 388 | code := waitForExit() |
| 389 | log.Infow("received-a-closing-signal", log.Fields{"code": code}) |
| 390 | |
| 391 | // Cleanup before leaving |
| 392 | ad.stop() |
| 393 | |
| 394 | elapsed := time.Since(start) |
Hardik Windlass | b9c869b | 2019-10-10 08:34:32 +0000 | [diff] [blame] | 395 | log.Infow("run-time", log.Fields{"instanceId": ad.config.InstanceID, "time": elapsed / time.Second}) |
cuilin2018 | 7b2a8c3 | 2019-03-26 19:52:28 -0700 | [diff] [blame] | 396 | } |