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         ch "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/charts"
  27         ph "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
  28         po "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/onboard"
  29         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi"
  30         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations"
  31         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/charts"
  32         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
  33         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/onboard"
  34         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/resthooks"
  35         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
  36         "github.com/go-openapi/loads"
  37         "github.com/go-openapi/runtime/middleware"
  38 )
  39
  40 func NewRestful() *Restful {
  41         r := &Restful{
  42                 rh: resthooks.NewResthook(
  43                         ph.NewHealthChecker(),
  44                         po.NewOnboarder(),
  45                         ch.NewChartmgr(),
  46                 ),
  47         }
  48         r.setupHandler()
  49         return r
  50 }
  51
  52 func (r *Restful) setupHandler() {
  53         swaggerSpec, err := loads.Embedded(restapi.SwaggerJSON, restapi.FlatSwaggerJSON)
  54         if err != nil {
  55                 os.Exit(1)
  56         }
  57
  58         api := operations.NewRICDMSAPI(swaggerSpec)
  59
  60         api.HealthGetHealthCheckHandler = health.GetHealthCheckHandlerFunc(func(ghcp health.GetHealthCheckParams) middleware.Responder {
  61                 ricdms.Logger.Debug("==> HealthCheck API invoked.")
  62                 resp := r.rh.GetDMSHealth()
  63                 return resp
  64         })
  65
  66         api.OnboardPostOnboardxAppsHandler = onboard.PostOnboardxAppsHandlerFunc(func(poap onboard.PostOnboardxAppsParams) middleware.Responder {
  67                 ricdms.Logger.Debug("==> onboard API invoked.")
  68                 resp := r.rh.OnBoard(poap.Body)
  69                 return resp
  70         })
  71
  72         api.ChartsGetChartsListHandler = charts.GetChartsListHandlerFunc(func(param charts.GetChartsListParams) middleware.Responder {
  73                 ricdms.Logger.Debug("==> GetChartList")
  74                 resp := r.rh.GetCharts()
  75                 return resp
  76         })
  77
  78         api.ChartsDownloadHelmChartHandler = charts.DownloadHelmChartHandlerFunc(func(param charts.DownloadHelmChartParams) middleware.Responder {
  79                 ricdms.Logger.Debug("==> Download helm chart")
  80                 resp := r.rh.DownloadChart(param.XAppName, param.Version)
  81                 return resp
  82         })
  83
  84         r.api = api
  85 }
  86
  87 func (r *Restful) Run() {
  88         server := restapi.NewServer(r.api)
  89         defer server.Shutdown()
  90         server.Port = 8000
  91         server.Host = "0.0.0.0"
  92         ricdms.Logger.Info("Starting server at : %s:%d", server.Host, server.Port)
  93         if err := server.Serve(); err != nil {
  94                 log.Fatal(err.Error())
  95         }
  96 }