2 // ========================LICENSE_START=================================
5 // Copyright (C) 2021: Nordix Foundation
7 // Licensed under the Apache License, Version 2.0 (the "License");
8 // you may not use this file except in compliance with the License.
9 // You may obtain a copy of the License at
11 // http://www.apache.org/licenses/LICENSE-2.0
13 // Unless required by applicable law or agreed to in writing, software
14 // distributed under the License is distributed on an "AS IS" BASIS,
15 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 // See the License for the specific language governing permissions and
17 // limitations under the License.
18 // ========================LICENSE_END===================================
32 "github.com/gorilla/mux"
33 log "github.com/sirupsen/logrus"
34 "oransc.org/usecase/oruclosedloop/internal/config"
35 "oransc.org/usecase/oruclosedloop/internal/linkfailure"
36 "oransc.org/usecase/oruclosedloop/internal/repository"
37 "oransc.org/usecase/oruclosedloop/internal/restclient"
40 const jobId = "14e7bb84-a44d-44c1-90b7-6995a92ad43c"
41 var job_definition interface{}
43 var jobRegistrationInfo = struct {
44 InfoTypeID string `json:"info_type_id"`
45 JobResultURI string `json:"job_result_uri"`
46 JobOwner string `json:"job_owner"`
47 JobDefinition interface{} `json:"job_definition"`
49 InfoTypeID: "STD_Fault_Messages",
51 JobOwner: "O-RU Closed Loop Usecase",
52 JobDefinition: job_definition,
55 var client restclient.HTTPClient
56 var configuration *config.Config
57 var linkfailureConfig linkfailure.Configuration
58 var lookupService repository.LookupService
59 var consumerPort string
67 configuration = config.New()
69 log.SetLevel(configuration.LogLevel)
70 log.Debug("Using configuration: ", configuration)
72 consumerPort = fmt.Sprint(configuration.ConsumerPort)
73 jobRegistrationInfo.JobResultURI = configuration.ConsumerHost + ":" + consumerPort
75 linkfailureConfig = linkfailure.Configuration{
76 SDNRAddress: configuration.SDNRAddress,
77 SDNRUser: configuration.SDNRUser,
78 SDNRPassword: configuration.SDNPassword,
83 if err := validateConfiguration(configuration); err != nil {
84 log.Fatalf("Unable to start consumer due to configuration error: %v", err)
87 csvFileHelper := repository.NewCsvFileHelperImpl()
88 if initErr := initializeLookupService(csvFileHelper, configuration.ORUToODUMapFile); initErr != nil {
89 log.Fatalf("Unable to create LookupService due to inability to get O-RU-ID to O-DU-ID map. Cause: %v", initErr)
92 var cert tls.Certificate
93 if c, err := restclient.CreateClientCertificate(configuration.ConsumerCertPath, configuration.ConsumerKeyPath); err == nil {
96 log.Fatalf("Stopping producer due to error: %v", err)
98 client = restclient.CreateRetryClient(cert)
102 os.Exit(1) // If the startServer function exits, it is because there has been a failure in the server, so we exit.
106 deleteOnShutdown(make(chan os.Signal, 1))
113 func validateConfiguration(configuration *config.Config) error {
114 if configuration.ConsumerHost == "" || configuration.ConsumerPort == 0 {
115 return fmt.Errorf("consumer host and port must be provided")
118 if configuration.ConsumerCertPath == "" || configuration.ConsumerKeyPath == "" {
119 return fmt.Errorf("missing CONSUMER_CERT and/or CONSUMER_KEY")
125 func initializeLookupService(csvFileHelper repository.CsvFileHelper, csvFile string) error {
126 lookupService = repository.NewLookupServiceImpl(csvFileHelper, csvFile)
127 return lookupService.Init()
130 func getRouter() *mux.Router {
131 messageHandler := linkfailure.NewLinkFailureHandler(lookupService, linkfailureConfig, client)
134 r.HandleFunc("/", messageHandler.MessagesHandler).Methods(http.MethodPost).Name("messageHandler")
135 r.HandleFunc("/status", statusHandler).Methods(http.MethodGet).Name("status")
136 r.HandleFunc("/admin/start", startHandler).Methods(http.MethodPost).Name("start")
137 r.HandleFunc("/admin/stop", stopHandler).Methods(http.MethodPost).Name("stop")
144 if restclient.IsUrlSecure(configuration.ConsumerHost) {
145 err = http.ListenAndServeTLS(fmt.Sprintf(":%v", configuration.ConsumerPort), configuration.ConsumerCertPath, configuration.ConsumerKeyPath, getRouter())
147 err = http.ListenAndServe(fmt.Sprintf(":%v", configuration.ConsumerPort), getRouter())
150 log.Errorf("Server stopped unintentionally due to: %v. Deleteing job.", err)
151 if deleteErr := deleteJob(); deleteErr != nil {
152 log.Errorf("Unable to delete consumer job due to: %v. Please remove job %v manually.", deleteErr, jobId)
157 func keepConsumerAlive() {
158 forever := make(chan int)
162 func startHandler(w http.ResponseWriter, r *http.Request) {
163 body, _ := json.Marshal(jobRegistrationInfo)
164 putErr := restclient.PutWithoutAuth(configuration.InfoCoordinatorAddress+"/data-consumer/v1/info-jobs/"+jobId, body, client)
166 http.Error(w, fmt.Sprintf("Unable to register consumer job due to: %v.", putErr), http.StatusBadRequest)
169 log.Debug("Registered job.")
173 func stopHandler(w http.ResponseWriter, r *http.Request) {
174 deleteErr := deleteJob()
175 if deleteErr != nil {
176 http.Error(w, fmt.Sprintf("Unable to delete consumer job due to: %v. Please remove job %v manually.", deleteErr, jobId), http.StatusBadRequest)
179 log.Debug("Deleted job.")
183 func statusHandler(w http.ResponseWriter, r *http.Request) {
184 runStatus := "started"
186 runStatus = "stopped"
188 fmt.Fprintf(w, `{"status": "%v"}`, runStatus)
191 func deleteOnShutdown(s chan os.Signal) {
192 signal.Notify(s, os.Interrupt)
193 signal.Notify(s, syscall.SIGTERM)
195 log.Info("Shutting down gracefully.")
196 if err := deleteJob(); err != nil {
197 log.Errorf("Unable to delete job on shutdown due to: %v. Please remove job %v manually.", err, jobId)
201 func deleteJob() error {
202 return restclient.Delete(configuration.InfoCoordinatorAddress+"/data-consumer/v1/info-jobs/"+jobId, client)