E2Manager/providers/request_handler_provider.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   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
  18 package providers
  19
  20 import (
  21         "e2mgr/handlers"
  22         "e2mgr/logger"
  23         "e2mgr/rNibWriter"
  24         "errors"
  25         "fmt"
  26 )
  27
  28 var requestMap map[string]handlers.Handler
  29
  30 type RequestHandlerProvider struct{}
  31
  32 func NewRequestHandlerProvider(rnibWriterProvider func() rNibWriter.RNibWriter) *RequestHandlerProvider {
  33         requestMap = initRequestMap(rnibWriterProvider)
  34         return &RequestHandlerProvider{}
  35 }
  36
  37 func initRequestMap(rnibWriterProvider func() rNibWriter.RNibWriter) map[string]handlers.Handler {
  38         return map[string]handlers.Handler{
  39                 "x2-setup":   handlers.NewSetupRequestHandler(rnibWriterProvider),
  40                 "endc-setup": handlers.NewEndcSetupRequestHandler(rnibWriterProvider),
  41         }
  42 }
  43
  44 func (provider RequestHandlerProvider) GetHandler(logger *logger.Logger, requestType string) (handlers.Handler, error) {
  45         handler, ok := requestMap[requestType]
  46
  47         if !ok {
  48                 errorMessage := fmt.Sprintf("#request_handler_provider.GetHandler - Cannot find handler for request type: %s", requestType)
  49                 logger.Errorf(errorMessage)
  50                 return nil, errors.New(errorMessage)
  51         }
  52
  53         return handler, nil
  54 }