[RICPLT-2157] Restructure handlers and converters.......
[ric-plt/e2mgr.git] / E2Manager / controllers / controller.go
index 0e13d0d..6c86816 100644 (file)
@@ -22,69 +22,78 @@ import (
        "e2mgr/e2managererrors"
        "e2mgr/logger"
        "e2mgr/models"
-       "e2mgr/providers"
+       "e2mgr/providers/httpmsghandlerprovider"
        "e2mgr/rNibWriter"
        "e2mgr/services"
        "encoding/json"
        "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
-       "github.com/julienschmidt/httprouter"
-       "io"
+       "github.com/gorilla/mux"
        "net/http"
-       "time"
 )
 
 const (
        ParamRanName = "ranName"
 )
+
+type IController interface {
+       ShutdownHandler(writer http.ResponseWriter, r *http.Request)
+       X2ResetHandler(writer http.ResponseWriter, r *http.Request)
+}
+
 type Controller struct {
-       logger         *logger.Logger
-       handlerProvider *providers.IncomingRequestHandlerProvider
+       logger          *logger.Logger
+       handlerProvider *httpmsghandlerprovider.IncomingRequestHandlerProvider
 }
 
 func NewController(logger *logger.Logger, rmrService *services.RmrService, rNibReaderProvider func() reader.RNibReader, rNibWriterProvider func() rNibWriter.RNibWriter,
        config *configuration.Configuration) *Controller {
 
-       provider := providers.NewIncomingRequestHandlerProvider(logger, rmrService, config, rNibWriterProvider, rNibReaderProvider)
+       provider := httpmsghandlerprovider.NewIncomingRequestHandlerProvider(logger, rmrService, config, rNibWriterProvider, rNibReaderProvider)
        return &Controller{
-               logger: logger,
+               logger:          logger,
                handlerProvider: provider,
        }
 }
 
-func (c *Controller)ShutdownHandler(writer http.ResponseWriter, r *http.Request, params httprouter.Params){
-
-       c.handleRequest(writer, &r.Header, providers.ShutdownRequest,nil, false, http.StatusNoContent)
+func (c *Controller) ShutdownHandler(writer http.ResponseWriter, r *http.Request) {
+       c.logger.Infof("[Client -> E2 Manager] #controller.ShutdownHandler - request: %v", prettifyRequest(r))
+       c.handleRequest(writer, &r.Header, httpmsghandlerprovider.ShutdownRequest, nil, false, http.StatusNoContent)
 }
 
-func (c *Controller) X2ResetHandler(writer http.ResponseWriter, r *http.Request, params httprouter.Params){
-       startTime := time.Now()
-       request:= models.ResetRequest{}
-       ranName:= params.ByName(ParamRanName)
-       if !c.extractJsonBody(r.Body, &request, writer){
+func (c *Controller) X2ResetHandler(writer http.ResponseWriter, r *http.Request) {
+       c.logger.Infof("[Client -> E2 Manager] #controller.X2ResetHandler - request: %v", prettifyRequest(r))
+       request := models.ResetRequest{}
+       vars := mux.Vars(r)
+       ranName := vars[ParamRanName]
+
+       if !c.extractJsonBody(r, &request, writer) {
                return
        }
        request.RanName = ranName
-       request.StartTime = startTime
-       c.handleRequest(writer, &r.Header, providers.ResetRequest, request, false, http.StatusNoContent)
+       c.handleRequest(writer, &r.Header, httpmsghandlerprovider.ResetRequest, request, false, http.StatusNoContent)
 }
 
-func (c *Controller) extractJsonBody(body io.Reader, request models.Request, writer http.ResponseWriter) bool{
-       decoder := json.NewDecoder(body)
-       if err:= decoder.Decode(request); err != nil {
-               if err != nil {
-                       c.logger.Errorf("[Client -> E2 Manager] #controller.extractJsonBody - unable to extract json body - error: %s", err)
-                       c.handleErrorResponse(e2managererrors.NewRequestValidationError(), writer)
-                       return false
-               }
+func (c *Controller) extractJsonBody(r *http.Request, request models.Request, writer http.ResponseWriter) bool {
+       if r.ContentLength <= 0 {
+               return true
        }
+
+       decoder := json.NewDecoder(r.Body)
+
+       err := decoder.Decode(request)
+
+       if err != nil {
+               c.logger.Errorf("[Client -> E2 Manager] #controller.extractJsonBody - unable to extract json body - error: %s", err)
+               c.handleErrorResponse(e2managererrors.NewInvalidJsonError(), writer)
+               return false
+       }
+
        return true
 }
 
-func (c *Controller) handleRequest(writer http.ResponseWriter, header *http.Header, requestName providers.IncomingRequest,
+func (c *Controller) handleRequest(writer http.ResponseWriter, header *http.Header, requestName httpmsghandlerprovider.IncomingRequest,
        request models.Request, validateHeader bool, httpStatusResponse int) {
 
-       c.logger.Infof("[Client -> E2 Manager] #controller.handleRequest - request: %v", requestName) //TODO print request if exist
-
        if validateHeader {
 
                err := c.validateRequestHeader(header)
@@ -94,7 +103,8 @@ func (c *Controller) handleRequest(writer http.ResponseWriter, header *http.Head
                }
        }
 
-       handler,err := c.handlerProvider.GetHandler(requestName)
+       handler, err := c.handlerProvider.GetHandler(requestName)
+
        if err != nil {
                c.handleErrorResponse(err, writer)
                return
@@ -111,17 +121,17 @@ func (c *Controller) handleRequest(writer http.ResponseWriter, header *http.Head
        c.logger.Infof("[E2 Manager -> Client] #controller.handleRequest - status response: %v", httpStatusResponse)
 }
 
-func (c *Controller) validateRequestHeader( header *http.Header) error {
+func (c *Controller) validateRequestHeader(header *http.Header) error {
 
-       if header.Get("Content-Type") != "application/json"{
+       if header.Get("Content-Type") != "application/json" {
                c.logger.Errorf("#controller.validateRequestHeader - validation failure, incorrect content type")
 
-               return  e2managererrors.NewHeaderValidationError()
+               return e2managererrors.NewHeaderValidationError()
        }
        return nil
 }
 
-func (c *Controller) handleErrorResponse(err error, writer http.ResponseWriter){
+func (c *Controller) handleErrorResponse(err error, writer http.ResponseWriter) {
 
        var errorResponseDetails models.ErrorResponse
        var httpError int
@@ -130,36 +140,40 @@ func (c *Controller) handleErrorResponse(err error, writer http.ResponseWriter){
                switch err.(type) {
                case *e2managererrors.RnibDbError:
                        e2Error, _ := err.(*e2managererrors.RnibDbError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusInternalServerError
                case *e2managererrors.CommandAlreadyInProgressError:
                        e2Error, _ := err.(*e2managererrors.CommandAlreadyInProgressError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusMethodNotAllowed
                case *e2managererrors.HeaderValidationError:
                        e2Error, _ := err.(*e2managererrors.HeaderValidationError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusUnsupportedMediaType
                case *e2managererrors.WrongStateError:
                        e2Error, _ := err.(*e2managererrors.WrongStateError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusBadRequest
                case *e2managererrors.RequestValidationError:
                        e2Error, _ := err.(*e2managererrors.RequestValidationError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
+                       httpError = http.StatusBadRequest
+               case *e2managererrors.InvalidJsonError:
+                       e2Error, _ := err.(*e2managererrors.InvalidJsonError)
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusBadRequest
                case *e2managererrors.RmrError:
                        e2Error, _ := err.(*e2managererrors.RmrError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusInternalServerError
                case *e2managererrors.ResourceNotFoundError:
                        e2Error, _ := err.(*e2managererrors.ResourceNotFoundError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusNotFound
 
                default:
-                       e2Error, _ := err.(*e2managererrors.InternalError)
-                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Err.Code, Message: e2Error.Err.Message}
+                       e2Error := e2managererrors.NewInternalError()
+                       errorResponseDetails = models.ErrorResponse{Code: e2Error.Code, Message: e2Error.Message}
                        httpError = http.StatusInternalServerError
                }
        }
@@ -174,4 +188,4 @@ func (c *Controller) handleErrorResponse(err error, writer http.ResponseWriter){
        if err != nil {
                c.logger.Errorf("#controller.handleErrorResponse - Cannot send response. writer:%v", writer)
        }
-}
\ No newline at end of file
+}