a1-go/pkg/restful/restful.go

   1 /*
   2 ==================================================================================
   3   Copyright (c) 2021 Samsung
   4
   5    Licensed under the Apache License, Version 2.0 (the "License");
   6    you may not use this file except in compliance with the License.
   7    You may obtain a copy of the License at
   8
   9        http://www.apache.org/licenses/LICENSE-2.0
  10
  11    Unless required by applicable law or agreed to in writing, software
  12    distributed under the License is distributed on an "AS IS" BASIS,
  13    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14    See the License for the specific language governing permissions and
  15    limitations under the License.
  16
  17    This source code is part of the near-RT RIC (RAN Intelligent Controller)
  18    platform project (RICP).
  19 ==================================================================================
  20 */
  21 package restful
  22
  23 import (
  24         "log"
  25         "os"
  26
  27         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/a1"
  28         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/models"
  29         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi"
  30         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations"
  31         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations/a1_mediator"
  32         "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/resthooks"
  33         "github.com/go-openapi/loads"
  34         "github.com/go-openapi/runtime/middleware"
  35 )
  36
  37 func NewRestful() *Restful {
  38         r := &Restful{
  39                 rh: resthooks.NewResthook(),
  40         }
  41         r.api = r.setupHandler()
  42         return r
  43 }
  44
  45 func (r *Restful) setupHandler() *operations.A1API {
  46         swaggerSpec, err := loads.Embedded(restapi.SwaggerJSON, restapi.FlatSwaggerJSON)
  47         if err != nil {
  48                 os.Exit(1)
  49         }
  50
  51         api := operations.NewA1API(swaggerSpec)
  52         api.A1MediatorA1ControllerGetAllPolicyTypesHandler = a1_mediator.A1ControllerGetAllPolicyTypesHandlerFunc(func(param a1_mediator.A1ControllerGetAllPolicyTypesParams) middleware.Responder {
  53                 a1.Logger.Debug("handler for get all policy type")
  54                 return a1_mediator.NewA1ControllerGetAllPolicyTypesOK().WithPayload(r.rh.GetAllPolicyType())
  55         })
  56
  57         api.A1MediatorA1ControllerCreatePolicyTypeHandler = a1_mediator.A1ControllerCreatePolicyTypeHandlerFunc(func(params a1_mediator.A1ControllerCreatePolicyTypeParams) middleware.Responder {
  58                 a1.Logger.Debug("handler for Create policy type ")
  59                 if err = r.rh.CreatePolicyType(models.PolicyTypeID(params.PolicyTypeID), *params.Body); err == nil {
  60                         //Increase prometheus counter
  61                         return a1_mediator.NewA1ControllerCreatePolicyTypeCreated()
  62                 }
  63                 if r.rh.IsTypeAlready(err) || r.rh.IsTypeMismatch(err) {
  64                         return a1_mediator.NewA1ControllerCreatePolicyTypeBadRequest()
  65                 }
  66                 return a1_mediator.NewA1ControllerCreatePolicyTypeServiceUnavailable()
  67
  68         })
  69
  70         api.A1MediatorA1ControllerGetPolicyTypeHandler = a1_mediator.A1ControllerGetPolicyTypeHandlerFunc(func(params a1_mediator.A1ControllerGetPolicyTypeParams) middleware.Responder {
  71                 a1.Logger.Debug("handler for get policy type from policytypeID")
  72                 return a1_mediator.NewA1ControllerGetPolicyTypeOK().WithPayload(r.rh.GetPolicyType(models.PolicyTypeID(params.PolicyTypeID)))
  73         })
  74
  75         api.A1MediatorA1ControllerCreateOrReplacePolicyInstanceHandler = a1_mediator.A1ControllerCreateOrReplacePolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerCreateOrReplacePolicyInstanceParams) middleware.Responder {
  76                 a1.Logger.Debug("handler for create policy type instance ")
  77                 if err = r.rh.CreatePolicyInstance(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID), params.Body); err == nil {
  78                         return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceAccepted()
  79                 }
  80                 if r.rh.IsValidJson(err) {
  81                         return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceBadRequest()
  82                 }
  83                 return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceServiceUnavailable()
  84
  85         })
  86
  87         api.A1MediatorA1ControllerGetPolicyInstanceHandler = a1_mediator.A1ControllerGetPolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerGetPolicyInstanceParams) middleware.Responder {
  88                 a1.Logger.Debug("handler for get policy instance from policytypeID")
  89                 if resp, err := r.rh.GetPolicyInstance(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID)); err == nil {
  90                         return a1_mediator.NewA1ControllerGetPolicyInstanceOK().WithPayload(resp)
  91                 }
  92                 if r.rh.IsPolicyInstanceNotFound(err) {
  93                         return a1_mediator.NewA1ControllerGetPolicyInstanceNotFound()
  94                 }
  95                 return a1_mediator.NewA1ControllerGetPolicyInstanceServiceUnavailable()
  96         })
  97
  98         api.A1MediatorA1ControllerGetAllInstancesForTypeHandler = a1_mediator.A1ControllerGetAllInstancesForTypeHandlerFunc(func(params a1_mediator.A1ControllerGetAllInstancesForTypeParams) middleware.Responder {
  99                 a1.Logger.Debug("handler for get all policy instance")
 100                 if resp, err := r.rh.GetAllPolicyInstance(models.PolicyTypeID(params.PolicyTypeID)); err == nil {
 101                         if resp != nil {
 102                                 return a1_mediator.NewA1ControllerGetAllInstancesForTypeOK().WithPayload(resp)
 103                         }
 104                 }
 105                 if r.rh.IsPolicyInstanceNotFound(err) {
 106                         return a1_mediator.NewA1ControllerGetPolicyInstanceNotFound()
 107                 }
 108                 return a1_mediator.NewA1ControllerGetAllInstancesForTypeServiceUnavailable()
 109
 110         })
 111
 112         api.A1MediatorA1ControllerDeletePolicyTypeHandler = a1_mediator.A1ControllerDeletePolicyTypeHandlerFunc(func(params a1_mediator.A1ControllerDeletePolicyTypeParams) middleware.Responder {
 113                 a1.Logger.Debug("handler for delete policy type")
 114                 if err := r.rh.DeletePolicyType(models.PolicyTypeID(params.PolicyTypeID)); err != nil {
 115                         if r.rh.CanPolicyTypeBeDeleted(err) {
 116                                 return a1_mediator.NewA1ControllerDeletePolicyTypeBadRequest()
 117                         }
 118                         return a1_mediator.NewA1ControllerDeletePolicyTypeServiceUnavailable()
 119                 }
 120
 121                 return a1_mediator.NewA1ControllerDeletePolicyTypeNoContent()
 122
 123         })
 124
 125         api.A1MediatorA1ControllerGetPolicyInstanceStatusHandler = a1_mediator.A1ControllerGetPolicyInstanceStatusHandlerFunc(func(params a1_mediator.A1ControllerGetPolicyInstanceStatusParams) middleware.Responder {
 126                 a1.Logger.Debug("handler for get policy instance status")
 127                 if resp, err := r.rh.GetPolicyInstanceStatus(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID)); err == nil {
 128                         return a1_mediator.NewA1ControllerGetPolicyInstanceStatusOK().WithPayload(resp)
 129                 }
 130                 if r.rh.IsPolicyInstanceNotFound(err) {
 131                         return a1_mediator.NewA1ControllerGetPolicyInstanceStatusNotFound()
 132                 }
 133                 return a1_mediator.NewA1ControllerGetPolicyInstanceStatusServiceUnavailable()
 134         })
 135
 136         api.A1MediatorA1ControllerDeletePolicyInstanceHandler = a1_mediator.A1ControllerDeletePolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerDeletePolicyInstanceParams) middleware.Responder {
 137                 a1.Logger.Debug("handler for delete policy instance")
 138                 if err := r.rh.DeletePolicyInstance(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID)); err != nil {
 139                         if r.rh.CanPolicyInstanceBeDeleted(err) {
 140                                 return a1_mediator.NewA1ControllerDeletePolicyInstanceNotFound()
 141                         }
 142                         return a1_mediator.NewA1ControllerDeletePolicyInstanceServiceUnavailable()
 143                 }
 144
 145                 return a1_mediator.NewA1ControllerDeletePolicyInstanceAccepted()
 146
 147         })
 148
 149         return api
 150
 151 }
 152
 153 func (r *Restful) Run() {
 154
 155         server := restapi.NewServer(r.api)
 156         defer server.Shutdown()
 157         server.Port = 10000
 158         server.Host = "0.0.0.0"
 159         if err := server.Serve(); err != nil {
 160                 log.Fatal(err.Error())
 161         }
 162 }