[RICPLT-2527] Fix request model of assoc/dissoc and add dissoc all
[ric-plt/e2mgr.git] / E2Manager / clients / routing_manager_client.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 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 //  platform project (RICP).
19
20 package clients
21
22 import (
23         "bytes"
24         "e2mgr/configuration"
25         "e2mgr/e2managererrors"
26         "e2mgr/logger"
27         "e2mgr/models"
28         "encoding/json"
29         "net/http"
30 )
31
32 const (
33         AddE2TInstanceApiSuffix            = "e2t"
34         AssociateRanToE2TInstanceApiSuffix = "associate-ran-to-e2t"
35         DissociateRanE2TInstanceApiSuffix  = "dissociate-ran"
36 )
37
38 type RoutingManagerClient struct {
39         logger     *logger.Logger
40         config     *configuration.Configuration
41         httpClient HttpClient
42 }
43
44 type IRoutingManagerClient interface {
45         AddE2TInstance(e2tAddress string) error
46         AssociateRanToE2TInstance(e2tAddress string, ranName string) error
47         DissociateRanE2TInstance(e2tAddress string, ranName string) error
48 }
49
50 func NewRoutingManagerClient(logger *logger.Logger, config *configuration.Configuration, httpClient HttpClient) *RoutingManagerClient {
51         return &RoutingManagerClient{
52                 logger:     logger,
53                 config:     config,
54                 httpClient: httpClient,
55         }
56 }
57
58 func (c *RoutingManagerClient) AddE2TInstance(e2tAddress string) error {
59
60         data := models.NewRoutingManagerE2TData(e2tAddress)
61         url := c.config.RoutingManager.BaseUrl + AddE2TInstanceApiSuffix
62
63         return c.PostMessage(data, url)
64 }
65
66 func (c *RoutingManagerClient) AssociateRanToE2TInstance(e2tAddress string, ranName string) error {
67
68         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
69         url := c.config.RoutingManager.BaseUrl + AssociateRanToE2TInstanceApiSuffix
70
71         return c.PostMessage(data, url)
72 }
73
74 func (c *RoutingManagerClient) DissociateRanE2TInstance(e2tAddress string, ranName string) error {
75
76         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
77         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
78
79         return c.PostMessage(data, url)
80 }
81
82 func (c *RoutingManagerClient) DissociateAllRans(e2tAddresses []string) error {
83
84         data := mapE2TAddressesToE2DataList(e2tAddresses)
85         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
86
87         return c.PostMessage(data, url)
88 }
89
90 func (c *RoutingManagerClient) PostMessage(data interface{}, url string) error {
91         marshaled, err := json.Marshal(data)
92
93         if err != nil {
94                 return e2managererrors.NewRoutingManagerError()
95         }
96
97         body := bytes.NewBuffer(marshaled)
98         c.logger.Infof("[E2M -> Routing Manager] #RoutingManagerClient.PostMessage - url: %s, request body: %+v", url, body)
99
100         resp, err := c.httpClient.Post(url, "application/json", body)
101
102         if err != nil {
103                 c.logger.Errorf("#RoutingManagerClient.PostMessage - failed sending request. error: %s", err)
104                 return e2managererrors.NewRoutingManagerError()
105         }
106
107         defer resp.Body.Close()
108
109         if resp.StatusCode >= http.StatusOK && resp.StatusCode < http.StatusMultipleChoices {
110                 c.logger.Infof("[Routing Manager -> E2M] #RoutingManagerClient.PostMessage - success. http status code: %d", resp.StatusCode)
111                 return nil
112         }
113
114         c.logger.Errorf("[Routing Manager -> E2M] #RoutingManagerClient.PostMessage - failure. http status code: %d", resp.StatusCode)
115         return e2managererrors.NewRoutingManagerError()
116 }
117
118 func mapE2TAddressesToE2DataList(e2tAddresses []string) models.RoutingManagerE2TDataList {
119         e2tDataList := make(models.RoutingManagerE2TDataList, len(e2tAddresses))
120
121         for i, v := range e2tAddresses {
122                 e2tDataList[i] = models.NewRoutingManagerE2TData(v)
123         }
124
125         return e2tDataList
126 }