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         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi"
  28         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations"
  29         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
  30         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/resthooks"
  31         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
  32         "github.com/go-openapi/loads"
  33         "github.com/go-openapi/runtime/middleware"
  34 )
  35
  36 func NewRestful() *Restful {
  37         r := &Restful{
  38                 rh: resthooks.NewResthook(
  39                         ph.NewHealthChecker(),
  40                 ),
  41         }
  42         r.setupHandler()
  43         return r
  44 }
  45
  46 func (r *Restful) setupHandler() {
  47         swaggerSpec, err := loads.Embedded(restapi.SwaggerJSON, restapi.FlatSwaggerJSON)
  48         if err != nil {
  49                 os.Exit(1)
  50         }
  51
  52         api := operations.NewRICDMSAPI(swaggerSpec)
  53
  54         api.HealthGetHealthCheckHandler = health.GetHealthCheckHandlerFunc(func(ghcp health.GetHealthCheckParams) middleware.Responder {
  55                 ricdms.Logger.Debug("==> HealthCheck API invoked.")
  56                 resp := r.rh.GetDMSHealth()
  57                 return resp
  58         })
  59
  60         r.api = api
  61 }
  62
  63 func (r *Restful) Run() {
  64         server := restapi.NewServer(r.api)
  65         defer server.Shutdown()
  66         server.Port = 8000
  67         server.Host = "0.0.0.0"
  68         ricdms.Logger.Info("Starting server at : %s:%d", server.Host, server.Port)
  69         if err := server.Serve(); err != nil {
  70                 log.Fatal(err.Error())
  71         }
  72 }