pkg/restful/restful.go

   1 //==================================================================================
   2 //  Copyright (c) 2022 Samsung
   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 //   This source code is part of the near-RT RIC (RAN Intelligent Controller)
  17 //   platform project (RICP).
  18 //==================================================================================
  19 //
  20 package restful
  21
  22 import (
  23         "log"
  24         "os"
  25
  26         ph "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
  27         po "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/onboard"
  28         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi"
  29         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations"
  30         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
  31         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/onboard"
  32         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/resthooks"
  33         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
  34         "github.com/go-openapi/loads"
  35         "github.com/go-openapi/runtime/middleware"
  36 )
  37
  38 func NewRestful() *Restful {
  39         r := &Restful{
  40                 rh: resthooks.NewResthook(
  41                         ph.NewHealthChecker(),
  42                         po.NewOnboarder(),
  43                 ),
  44         }
  45         r.setupHandler()
  46         return r
  47 }
  48
  49 func (r *Restful) setupHandler() {
  50         swaggerSpec, err := loads.Embedded(restapi.SwaggerJSON, restapi.FlatSwaggerJSON)
  51         if err != nil {
  52                 os.Exit(1)
  53         }
  54
  55         api := operations.NewRICDMSAPI(swaggerSpec)
  56
  57         api.HealthGetHealthCheckHandler = health.GetHealthCheckHandlerFunc(func(ghcp health.GetHealthCheckParams) middleware.Responder {
  58                 ricdms.Logger.Debug("==> HealthCheck API invoked.")
  59                 resp := r.rh.GetDMSHealth()
  60                 return resp
  61         })
  62
  63         api.OnboardPostOnboardxAppsHandler = onboard.PostOnboardxAppsHandlerFunc(func(poap onboard.PostOnboardxAppsParams) middleware.Responder {
  64                 ricdms.Logger.Debug("==> onboard API invoked.")
  65                 resp := r.rh.OnBoard(poap.Body)
  66                 return resp
  67         })
  68
  69         r.api = api
  70 }
  71
  72 func (r *Restful) Run() {
  73         server := restapi.NewServer(r.api)
  74         defer server.Shutdown()
  75         server.Port = 8000
  76         server.Host = "0.0.0.0"
  77         ricdms.Logger.Info("Starting server at : %s:%d", server.Host, server.Port)
  78         if err := server.Serve(); err != nil {
  79                 log.Fatal(err.Error())
  80         }
  81 }