servicemanager/main.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2023-2024: OpenInfra Foundation Europe
   6 //   %%
   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
  10 //
  11 //        http://www.apache.org/licenses/LICENSE-2.0
  12 //
  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===================================
  19 //
  20
  21 package main
  22
  23 import (
  24         "fmt"
  25         "net/http"
  26
  27         "github.com/getkin/kin-openapi/openapi3"
  28
  29         "github.com/deepmap/oapi-codegen/pkg/middleware"
  30         echo "github.com/labstack/echo/v4"
  31         echomiddleware "github.com/labstack/echo/v4/middleware"
  32         log "github.com/sirupsen/logrus"
  33
  34         "oransc.org/nonrtric/servicemanager/internal/common29122"
  35         "oransc.org/nonrtric/servicemanager/internal/envreader"
  36
  37         "oransc.org/nonrtric/servicemanager/internal/discoverserviceapi"
  38         "oransc.org/nonrtric/servicemanager/internal/invokermanagementapi"
  39         "oransc.org/nonrtric/servicemanager/internal/providermanagementapi"
  40         "oransc.org/nonrtric/servicemanager/internal/publishserviceapi"
  41
  42         "oransc.org/nonrtric/servicemanager/internal/discoverservice"
  43         "oransc.org/nonrtric/servicemanager/internal/invokermanagement"
  44         "oransc.org/nonrtric/servicemanager/internal/providermanagement"
  45         "oransc.org/nonrtric/servicemanager/internal/publishservice"
  46 )
  47
  48 func main() {
  49         myEnv, myPorts, err := envreader.ReadDotEnv()
  50         if err != nil {
  51                 log.Fatal("error loading environment file")
  52                 return
  53         }
  54
  55         e, err := getEcho(myEnv, myPorts)
  56         if err != nil {
  57                 log.Fatal("getEcho fatal error")
  58                 return
  59         }
  60
  61         port := myPorts["SERVICE_MANAGER_PORT"]
  62
  63         go startWebServer(e, port)
  64         log.Info("server started and listening on port: ", port)
  65         keepServerAlive()
  66 }
  67
  68 func getEcho(myEnv map[string]string, myPorts map[string]int) (*echo.Echo, error) {
  69         e := echo.New()
  70         // Log all requests
  71         e.Use(echomiddleware.Logger())
  72
  73         capifProtocol := myEnv["CAPIF_PROTOCOL"]
  74         capifIPv4 := common29122.Ipv4Addr(myEnv["CAPIF_IPV4"])
  75         capifPort := common29122.Port(myPorts["CAPIF_PORT"])
  76         kongDomain := myEnv["KONG_DOMAIN"]
  77         kongProtocol := myEnv["KONG_PROTOCOL"]
  78         kongIPv4 := common29122.Ipv4Addr(myEnv["KONG_IPV4"])
  79         kongDataPlanePort := common29122.Port(myPorts["KONG_DATA_PLANE_PORT"])
  80         kongControlPlanePort := common29122.Port(myPorts["KONG_CONTROL_PLANE_PORT"])
  81
  82         var group *echo.Group
  83
  84         // Register ProviderManagement
  85         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  86         if err != nil {
  87                 log.Fatalf("error loading ProviderManagement swagger spec\n: %s", err)
  88                 return nil, err
  89         }
  90         providerManagerSwagger.Servers = nil
  91         providerManager := providermanagement.NewProviderManager(capifProtocol, capifIPv4, capifPort)
  92         group = e.Group("/api-provider-management/v1")
  93         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
  94         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
  95
  96         // Register PublishService
  97         publishServiceSwagger, err := publishserviceapi.GetSwagger()
  98         if err != nil {
  99                 log.Fatalf("error loading PublishService swagger spec\n: %s", err)
 100                 return nil, err
 101         }
 102         publishServiceSwagger.Servers = nil
 103         publishService := publishservice.NewPublishService(kongDomain, kongProtocol, kongIPv4, kongDataPlanePort, kongControlPlanePort, capifProtocol, capifIPv4, capifPort)
 104
 105         group = e.Group("/published-apis/v1")
 106         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 107         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 108
 109         // Register InvokerManagement
 110         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 111         if err != nil {
 112                 log.Fatalf("error loading InvokerManagement swagger spec\n: %s", err)
 113                 return nil, err
 114         }
 115         invokerManagerSwagger.Servers = nil
 116         invokerManager := invokermanagement.NewInvokerManager(capifProtocol, capifIPv4, capifPort)
 117         group = e.Group("/api-invoker-management/v1")
 118         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 119         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 120
 121         // Register DiscoverService
 122         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 123         if err != nil {
 124                 log.Fatalf("error loading DiscoverService swagger spec\n: %s", err)
 125                 return nil, err
 126         }
 127
 128         discoverServiceSwagger.Servers = nil
 129         discoverService := discoverservice.NewDiscoverService(capifProtocol, capifIPv4, capifPort)
 130
 131         group = e.Group("/service-apis/v1")
 132         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 133         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 134
 135         e.GET("/", hello)
 136         e.GET("/swagger/:apiName", getSwagger)
 137
 138         return e, err
 139 }
 140
 141 func startWebServer(e *echo.Echo, port int) {
 142         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 143 }
 144
 145 func keepServerAlive() {
 146         forever := make(chan int)
 147         <-forever
 148 }
 149
 150 func hello(c echo.Context) error {
 151         return c.String(http.StatusOK, "Hello, World!")
 152 }
 153
 154 func getSwagger(c echo.Context) error {
 155         var swagger *openapi3.T
 156         var err error
 157         switch api := c.Param("apiName"); api {
 158         case "provider":
 159                 swagger, err = providermanagementapi.GetSwagger()
 160         case "publish":
 161                 swagger, err = publishserviceapi.GetSwagger()
 162         case "invoker":
 163                 swagger, err = invokermanagementapi.GetSwagger()
 164         case "discover":
 165                 swagger, err = discoverserviceapi.GetSwagger()
 166         default:
 167                 return c.JSON(http.StatusBadRequest, getProblemDetails("Invalid API name "+api, http.StatusBadRequest))
 168         }
 169         if err != nil {
 170                 return c.JSON(http.StatusInternalServerError, getProblemDetails("Unable to get swagger for API", http.StatusInternalServerError))
 171         }
 172         return c.JSON(http.StatusOK, swagger)
 173 }
 174
 175 func getProblemDetails(cause string, status int) common29122.ProblemDetails {
 176         return common29122.ProblemDetails{
 177                 Cause:  &cause,
 178                 Status: &status,
 179         }
 180 }