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     realConfigReader := &envreader.RealConfigReader{}
  50     myEnv, myPorts, err := realConfigReader.ReadDotEnv()
  51     if err != nil {
  52                 log.Fatal("error loading environment file")
  53         return
  54     }
  55
  56         eServiceManager := echo.New()
  57         err = registerHandlers(eServiceManager, myEnv, myPorts)
  58
  59         if err != nil {
  60                 log.Fatal("registerHandlers fatal error")
  61                 return
  62         }
  63
  64         port := myPorts["SERVICE_MANAGER_PORT"]
  65
  66         go startWebServer(eServiceManager, port)
  67         log.Info("server started and listening on port: ", port)
  68         keepServerAlive()
  69 }
  70
  71 func registerHandlers(e *echo.Echo, myEnv map[string]string, myPorts map[string]int) (err error) {
  72         // Log all requests
  73         e.Use(echomiddleware.Logger())
  74
  75         capifProtocol := myEnv["CAPIF_PROTOCOL"]
  76         capifIPv4 := common29122.Ipv4Addr(myEnv["CAPIF_IPV4"])
  77         capifPort := common29122.Port(myPorts["CAPIF_PORT"])
  78         kongDomain := myEnv["KONG_DOMAIN"]
  79         kongProtocol := myEnv["KONG_PROTOCOL"]
  80         kongIPv4 := common29122.Ipv4Addr(myEnv["KONG_IPV4"])
  81         kongDataPlanePort := common29122.Port(myPorts["KONG_DATA_PLANE_PORT"])
  82         kongControlPlanePort := common29122.Port(myPorts["KONG_CONTROL_PLANE_PORT"])
  83
  84         var group *echo.Group
  85
  86         // Register ProviderManagement
  87         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  88         if err != nil {
  89                 log.Fatalf("error loading ProviderManagement swagger spec\n: %s", err)
  90                 return err
  91         }
  92         providerManagerSwagger.Servers = nil
  93         providerManager := providermanagement.NewProviderManager(capifProtocol, capifIPv4, capifPort)
  94         group = e.Group("/api-provider-management/v1")
  95         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
  96         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
  97
  98         // Register PublishService
  99         publishServiceSwagger, err := publishserviceapi.GetSwagger()
 100         if err != nil {
 101                 log.Fatalf("error loading PublishService swagger spec\n: %s", err)
 102                 return err
 103         }
 104         publishServiceSwagger.Servers = nil
 105         publishService := publishservice.NewPublishService(kongDomain, kongProtocol, kongIPv4, kongDataPlanePort, kongControlPlanePort, capifProtocol, capifIPv4, capifPort)
 106
 107         group = e.Group("/published-apis/v1")
 108         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 109         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 110
 111         // Register InvokerManagement
 112         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 113         if err != nil {
 114                 log.Fatalf("error loading InvokerManagement swagger spec\n: %s", err)
 115                 return err
 116         }
 117         invokerManagerSwagger.Servers = nil
 118         invokerManager := invokermanagement.NewInvokerManager(capifProtocol, capifIPv4, capifPort)
 119         group = e.Group("/api-invoker-management/v1")
 120         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 121         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 122
 123         // Register DiscoverService
 124         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 125         if err != nil {
 126                 log.Fatalf("error loading DiscoverService swagger spec\n: %s", err)
 127                 return err
 128         }
 129
 130         discoverServiceSwagger.Servers = nil
 131         discoverService := discoverservice.NewDiscoverService(capifProtocol, capifIPv4, capifPort)
 132
 133         group = e.Group("/service-apis/v1")
 134         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 135         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 136
 137         e.GET("/", hello)
 138         e.GET("/swagger/:apiName", getSwagger)
 139
 140         return err
 141 }
 142
 143 func startWebServer(e *echo.Echo, port int) {
 144         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 145 }
 146
 147 func keepServerAlive() {
 148         forever := make(chan int)
 149         <-forever
 150 }
 151
 152 func hello(c echo.Context) error {
 153         return c.String(http.StatusOK, "Hello, World!")
 154 }
 155
 156 func getSwagger(c echo.Context) error {
 157         var swagger *openapi3.T
 158         var err error
 159         switch api := c.Param("apiName"); api {
 160         case "provider":
 161                 swagger, err = providermanagementapi.GetSwagger()
 162         case "publish":
 163                 swagger, err = publishserviceapi.GetSwagger()
 164         case "invoker":
 165                 swagger, err = invokermanagementapi.GetSwagger()
 166         case "discover":
 167                 swagger, err = discoverserviceapi.GetSwagger()
 168         default:
 169                 return c.JSON(http.StatusBadRequest, getProblemDetails("Invalid API name "+api, http.StatusBadRequest))
 170         }
 171         if err != nil {
 172                 return c.JSON(http.StatusInternalServerError, getProblemDetails("Unable to get swagger for API", http.StatusInternalServerError))
 173         }
 174         return c.JSON(http.StatusOK, swagger)
 175 }
 176
 177 func getProblemDetails(cause string, status int) common29122.ProblemDetails {
 178         return common29122.ProblemDetails{
 179                 Cause:  &cause,
 180                 Status: &status,
 181         }
 182 }