blob: b5ac6930a8ba5ab3554c1337eaa35dfb1d7e8f05 [file] [log] [blame]
Elia Battistonc8d0d462022-02-22 16:30:51 +01001/*
2* Copyright 2022-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 */
16
17package main
18
19import (
20 "context"
21 "fmt"
22 "os"
23 "os/signal"
24 "syscall"
25 "time"
26
27 "github.com/opencord/voltha-lib-go/v7/pkg/log"
28 "github.com/opencord/voltha-lib-go/v7/pkg/probe"
29 "github.com/opencord/voltha-lib-go/v7/pkg/version"
Elia Battistonac8d23f2022-03-14 17:54:56 +010030 "github.com/opencord/voltha-northbound-bbf-adapter/internal/clients"
Elia Battistonc8d0d462022-02-22 16:30:51 +010031 "github.com/opencord/voltha-northbound-bbf-adapter/internal/config"
Elia Battistone1cecb22022-03-21 10:05:25 +010032 "github.com/opencord/voltha-northbound-bbf-adapter/internal/core"
Elia Battistonac8d23f2022-03-14 17:54:56 +010033 "github.com/opencord/voltha-northbound-bbf-adapter/internal/sysrepo"
Elia Battistonc8d0d462022-02-22 16:30:51 +010034)
35
Elia Battistonbe9edc12022-03-09 11:35:58 +010036//String for readiness probe services
Elia Battistonc8d0d462022-02-22 16:30:51 +010037const (
38 bbfAdapterService = "bbf-adapter-service"
Elia Battistonac8d23f2022-03-14 17:54:56 +010039 sysrepoService = "sysrepo"
Elia Battistonc8d0d462022-02-22 16:30:51 +010040)
41
Elia Battistonbe9edc12022-03-09 11:35:58 +010042type bbfAdapter struct {
43 conf *config.BBFAdapterConfig
44 volthaNbiClient *clients.VolthaNbiClient
45 oltAppClient *clients.OltAppClient
Elia Battistonac8d23f2022-03-14 17:54:56 +010046 sysrepoPlugin *sysrepo.SysrepoPlugin
Elia Battistonbe9edc12022-03-09 11:35:58 +010047}
48
49func newBbfAdapter(conf *config.BBFAdapterConfig) *bbfAdapter {
50 return &bbfAdapter{
51 conf: conf,
52 }
53}
54
Elia Battistone1cecb22022-03-21 10:05:25 +010055func (a *bbfAdapter) start(ctx context.Context) {
Elia Battistonbe9edc12022-03-09 11:35:58 +010056 var err error
57
58 //Connect to the voltha northbound api
59 a.volthaNbiClient = clients.NewVolthaNbiClient(a.conf.VolthaNbiEndpoint)
60 if err = a.volthaNbiClient.Connect(ctx, a.conf.TlsEnabled, a.conf.TlsVerify); err != nil {
61 logger.Fatalw(ctx, "failed-to-open-voltha-nbi-grpc-connection", log.Fields{"err": err})
62 } else {
63 probe.UpdateStatusFromContext(ctx, a.conf.VolthaNbiEndpoint, probe.ServiceStatusRunning)
64 }
65
66 //Check if the REST APIs of the olt app are reachable
67 a.oltAppClient = clients.NewOltAppClient(a.conf.OnosRestEndpoint, a.conf.OnosUser, a.conf.OnosPassword)
68 if err := a.oltAppClient.CheckConnection(ctx); err != nil {
69 logger.Fatalw(ctx, "failed-to-connect-to-onos-olt-app-api", log.Fields{"err": err})
70 } else {
71 probe.UpdateStatusFromContext(ctx, a.conf.OnosRestEndpoint, probe.ServiceStatusRunning)
72 }
73
Elia Battistone1cecb22022-03-21 10:05:25 +010074 //Create the global adapter that will be used by callbacks
75 core.AdapterInstance = core.NewVolthaYangAdapter(a.volthaNbiClient, a.oltAppClient)
76
Elia Battistonac8d23f2022-03-14 17:54:56 +010077 //Load sysrepo plugin
Elia Battiston589addb2022-04-04 16:40:01 +020078 a.sysrepoPlugin, err = sysrepo.StartNewPlugin(ctx, a.conf.SchemaMountFilePath)
Elia Battistonac8d23f2022-03-14 17:54:56 +010079 if err != nil {
80 logger.Fatalw(ctx, "failed-to-start-sysrepo-plugin", log.Fields{"err": err})
81 } else {
82 probe.UpdateStatusFromContext(ctx, sysrepoService, probe.ServiceStatusRunning)
83 }
84
Elia Battistonbe9edc12022-03-09 11:35:58 +010085 //Set the service as running, making the adapter finally ready
86 probe.UpdateStatusFromContext(ctx, bbfAdapterService, probe.ServiceStatusRunning)
87 logger.Info(ctx, "bbf-adapter-ready")
Elia Battistonbe9edc12022-03-09 11:35:58 +010088}
89
90//Close all connections of the adapter
Elia Battistonac8d23f2022-03-14 17:54:56 +010091func (a *bbfAdapter) cleanup(ctx context.Context) {
Elia Battistone1cecb22022-03-21 10:05:25 +010092 core.AdapterInstance = nil
93
Elia Battistonac8d23f2022-03-14 17:54:56 +010094 a.volthaNbiClient.Close(ctx)
Elia Battistone1cecb22022-03-21 10:05:25 +010095
Elia Battistonac8d23f2022-03-14 17:54:56 +010096 err := a.sysrepoPlugin.Stop(ctx)
97 if err != nil {
98 logger.Errorw(ctx, "failed-to-stop-sysrepo-plugin", log.Fields{"err": err})
99 }
Elia Battistone1cecb22022-03-21 10:05:25 +0100100
101 probe.UpdateStatusFromContext(ctx, bbfAdapterService, probe.ServiceStatusStopped)
Elia Battistonbe9edc12022-03-09 11:35:58 +0100102}
103
Elia Battistonc8d0d462022-02-22 16:30:51 +0100104func printBanner() {
105 fmt.Println(" ____ ____ ______ _ _ ")
106 fmt.Println(" | _ \\| _ \\| ____| /\\ | | | | ")
107 fmt.Println(" | |_) | |_) | |__ / \\ __| | __ _ _ __ | |_ ___ _ __ ")
108 fmt.Println(" | _ <| _ <| __| / /\\ \\ / _` |/ _` | '_ \\| __/ _ \\ '__|")
109 fmt.Println(" | |_) | |_) | | / ____ \\ (_| | (_| | |_) | || __/ | ")
110 fmt.Println(" |____/|____/|_| /_/ \\_\\__,_|\\__,_| .__/ \\__\\___|_| ")
111 fmt.Println(" | | ")
112 fmt.Println(" |_| ")
113}
114
115func printVersion() {
116 fmt.Println("VOLTHA Northbound BBF Adapter")
117 fmt.Println(version.VersionInfo.String(" "))
118}
119
120func waitForExit(ctx context.Context) int {
121 signalChannel := make(chan os.Signal, 1)
122 signal.Notify(signalChannel,
123 syscall.SIGHUP,
124 syscall.SIGINT,
125 syscall.SIGTERM,
126 syscall.SIGQUIT)
127
128 exitChannel := make(chan int)
129
130 go func() {
131 s := <-signalChannel
132 switch s {
133 case syscall.SIGHUP,
134 syscall.SIGINT,
135 syscall.SIGTERM,
136 syscall.SIGQUIT:
137 logger.Infow(ctx, "closing-signal-received", log.Fields{"signal": s})
138 exitChannel <- 0
139 default:
140 logger.Infow(ctx, "unexpected-signal-received", log.Fields{"signal": s})
141 exitChannel <- 1
142 }
143 }()
144
145 code := <-exitChannel
146 return code
147}
148
149func main() {
Elia Battistonbe9edc12022-03-09 11:35:58 +0100150 ctx, cancelCtx := context.WithCancel(context.Background())
151
Elia Battistonc8d0d462022-02-22 16:30:51 +0100152 start := time.Now()
153
154 conf := config.LoadConfig(ctx)
155
156 //Logging
157 logLevel, err := log.StringToLogLevel(conf.LogLevel)
158 if err != nil {
159 logger.Fatalf(ctx, "Cannot setup logging, %s", err)
160 }
161
162 // Setup default logger - applies for packages that do not have specific logger set
163 if _, err := log.SetDefaultLogger(log.JSON, logLevel, log.Fields{}); err != nil {
164 logger.With(log.Fields{"error": err}).Fatal(ctx, "Cannot setup logging")
165 }
166
167 // Update all loggers (provisionned via init) with a common field
168 if err := log.UpdateAllLoggers(log.Fields{}); err != nil {
169 logger.With(log.Fields{"error": err}).Fatal(ctx, "Cannot setup logging")
170 }
171
172 log.SetAllLogLevel(logLevel)
173
174 defer func() {
175 err := log.CleanUp()
176 if err != nil {
177 logger.Errorw(context.Background(), "unable-to-flush-any-buffered-log-entries", log.Fields{"error": err})
178 }
179 }()
180
181 // Print version and exit
182 if conf.PrintVersion {
183 printVersion()
184 return
185 }
186
187 // Print banner if specified
188 if conf.PrintBanner {
189 printBanner()
190 }
191
192 logger.Infow(ctx, "config", log.Fields{"config": *conf})
193
194 p := &probe.Probe{}
195 go p.ListenAndServe(ctx, conf.ProbeAddress)
196
Elia Battistonbe9edc12022-03-09 11:35:58 +0100197 //Register all services that will need to be initialized before considering the adapter ready
Elia Battistonc8d0d462022-02-22 16:30:51 +0100198 probeCtx := context.WithValue(ctx, probe.ProbeContextKey, p)
Elia Battistonbe9edc12022-03-09 11:35:58 +0100199 p.RegisterService(
200 ctx,
201 bbfAdapterService,
202 conf.VolthaNbiEndpoint,
203 conf.OnosRestEndpoint,
Elia Battistonac8d23f2022-03-14 17:54:56 +0100204 sysrepoService,
Elia Battistonbe9edc12022-03-09 11:35:58 +0100205 )
206
Elia Battistonc8d0d462022-02-22 16:30:51 +0100207 closer, err := log.GetGlobalLFM().InitTracingAndLogCorrelation(conf.TraceEnabled, conf.TraceAgentAddress, conf.LogCorrelationEnabled)
208 if err != nil {
209 logger.Warnw(ctx, "unable-to-initialize-tracing-and-log-correlation-module", log.Fields{"error": err})
210 } else {
211 defer log.TerminateTracing(closer)
212 }
213
Elia Battistonbe9edc12022-03-09 11:35:58 +0100214 adapter := newBbfAdapter(conf)
Elia Battistonc8d0d462022-02-22 16:30:51 +0100215
Elia Battistonbe9edc12022-03-09 11:35:58 +0100216 //Run the adapter
Elia Battistone1cecb22022-03-21 10:05:25 +0100217 adapter.start(probeCtx)
Elia Battistonac8d23f2022-03-14 17:54:56 +0100218 defer adapter.cleanup(probeCtx)
Elia Battistonc8d0d462022-02-22 16:30:51 +0100219
Elia Battistonbe9edc12022-03-09 11:35:58 +0100220 //Wait a signal to stop execution
Elia Battistonc8d0d462022-02-22 16:30:51 +0100221 code := waitForExit(ctx)
222 logger.Infow(ctx, "received-a-closing-signal", log.Fields{"code": code})
223
Elia Battistonbe9edc12022-03-09 11:35:58 +0100224 //Stop everything that waits for the context to be done
225 cancelCtx()
Elia Battistonbe9edc12022-03-09 11:35:58 +0100226
Elia Battistonc8d0d462022-02-22 16:30:51 +0100227 elapsed := time.Since(start)
228 logger.Infow(ctx, "run-time", log.Fields{"time": elapsed.Seconds()})
229}