X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fcontrollers%2Fnodeb_controller.go;fp=E2Manager%2Fcontrollers%2Fnodeb_controller.go;h=90b9311d8f2da0f52afb2bd8b6c065a9c4012db6;hb=f1035e286c9cdb58fd786daf4d6c5cc9bea45a37;hp=7b8a86fd3852f5f1d2023a74c05bf35cc8c1d51e;hpb=49144464f0450ce157a716010cbbd7c3143d332f;p=ric-plt%2Fe2mgr.git diff --git a/E2Manager/controllers/nodeb_controller.go b/E2Manager/controllers/nodeb_controller.go index 7b8a86f..90b9311 100644 --- a/E2Manager/controllers/nodeb_controller.go +++ b/E2Manager/controllers/nodeb_controller.go @@ -17,7 +17,6 @@ // This source code is part of the near-RT RIC (RAN Intelligent Controller) // platform project (RICP). - package controllers import ( @@ -26,6 +25,9 @@ import ( "e2mgr/models" "e2mgr/providers/httpmsghandlerprovider" "encoding/json" + "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities" + "github.com/golang/protobuf/jsonpb" + "github.com/golang/protobuf/proto" "github.com/gorilla/mux" "io" "io/ioutil" @@ -45,6 +47,7 @@ type INodebController interface { X2Setup(writer http.ResponseWriter, r *http.Request) EndcSetup(writer http.ResponseWriter, r *http.Request) GetNodeb(writer http.ResponseWriter, r *http.Request) + UpdateGnb(writer http.ResponseWriter, r *http.Request) GetNodebIdList(writer http.ResponseWriter, r *http.Request) } @@ -74,6 +77,24 @@ func (c *NodebController) GetNodeb(writer http.ResponseWriter, r *http.Request) c.handleRequest(writer, &r.Header, httpmsghandlerprovider.GetNodebRequest, request, false) } +func (c *NodebController) UpdateGnb(writer http.ResponseWriter, r *http.Request) { + c.logger.Infof("[Client -> E2 Manager] #NodebController.UpdateGnb - request: %v", c.prettifyRequest(r)) + vars := mux.Vars(r) + ranName := vars[ParamRanName] + + request := models.UpdateGnbRequest{} + + gnb := entities.Gnb{} + + if !c.extractRequestBodyToProto(r, &gnb, writer) { + return + } + + request.Gnb = &gnb; + request.RanName = ranName + c.handleRequest(writer, &r.Header, httpmsghandlerprovider.UpdateGnbRequest, request, true) +} + func (c *NodebController) Shutdown(writer http.ResponseWriter, r *http.Request) { c.logger.Infof("[Client -> E2 Manager] #NodebController.Shutdown - request: %v", c.prettifyRequest(r)) c.handleRequest(writer, &r.Header, httpmsghandlerprovider.ShutdownRequest, nil, false) @@ -116,6 +137,20 @@ func (c *NodebController) EndcSetup(writer http.ResponseWriter, r *http.Request) c.handleRequest(writer, &r.Header, httpmsghandlerprovider.EndcSetupRequest, request, true) } +func (c *NodebController) extractRequestBodyToProto(r *http.Request, pb proto.Message , writer http.ResponseWriter) bool { + defer r.Body.Close() + + err := jsonpb.Unmarshal(r.Body, pb) + + if err != nil { + c.logger.Errorf("[Client -> E2 Manager] #NodebController.extractJsonBody - unable to extract json body - error: %s", err) + c.handleErrorResponse(e2managererrors.NewInvalidJsonError(), writer) + return false + } + + return true +} + func (c *NodebController) extractJsonBody(r *http.Request, request models.Request, writer http.ResponseWriter) bool { defer r.Body.Close() body, err := ioutil.ReadAll(io.LimitReader(r.Body, LimitRequest)) @@ -136,9 +171,9 @@ func (c *NodebController) extractJsonBody(r *http.Request, request models.Reques return true } -func (c *NodebController) handleRequest(writer http.ResponseWriter, header *http.Header, requestName httpmsghandlerprovider.IncomingRequest, request models.Request, validateHeader bool) { +func (c *NodebController) handleRequest(writer http.ResponseWriter, header *http.Header, requestName httpmsghandlerprovider.IncomingRequest, request models.Request, validateRequestHeaders bool) { - if validateHeader { + if validateRequestHeaders { err := c.validateRequestHeader(header) if err != nil {