318b7cff4f119273d1b27858d365f3a80ebabb2e
[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         DeleteE2TInstanceApiSuffix         = "e2t"
37 )
38
39 type RoutingManagerClient struct {
40         logger     *logger.Logger
41         config     *configuration.Configuration
42         httpClient IHttpClient
43 }
44
45 type IRoutingManagerClient interface {
46         AddE2TInstance(e2tAddress string) error
47         AssociateRanToE2TInstance(e2tAddress string, ranName string) error
48         DissociateRanE2TInstance(e2tAddress string, ranName string) error
49         DissociateAllRans(e2tAddresses []string) error
50         DeleteE2TInstance(e2tAddress string, ransToBeDissociated []string, e2tToRansAssociations map[string][]string) error
51 }
52
53 func NewRoutingManagerClient(logger *logger.Logger, config *configuration.Configuration, httpClient IHttpClient) *RoutingManagerClient {
54         return &RoutingManagerClient{
55                 logger:     logger,
56                 config:     config,
57                 httpClient: httpClient,
58         }
59 }
60
61 func (c *RoutingManagerClient) AddE2TInstance(e2tAddress string) error {
62
63         data := models.NewRoutingManagerE2TData(e2tAddress)
64         url := c.config.RoutingManager.BaseUrl + AddE2TInstanceApiSuffix
65
66         return c.PostMessage(url, data)
67 }
68
69 func (c *RoutingManagerClient) AssociateRanToE2TInstance(e2tAddress string, ranName string) error {
70
71         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
72         url := c.config.RoutingManager.BaseUrl + AssociateRanToE2TInstanceApiSuffix
73
74         return c.PostMessage(url, data)
75 }
76
77 func (c *RoutingManagerClient) DissociateRanE2TInstance(e2tAddress string, ranName string) error {
78
79         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
80         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
81
82         return c.PostMessage(url, data)
83 }
84
85 func (c *RoutingManagerClient) DissociateAllRans(e2tAddresses []string) error {
86
87         data := mapE2TAddressesToE2DataList(e2tAddresses)
88         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
89
90         return c.PostMessage(url, data)
91 }
92
93 func (c *RoutingManagerClient) DeleteE2TInstance(e2tAddress string, ransTobeDissociated []string, e2tToRansAssociations map[string][]string) error {
94         e2tDataList := convertE2TToRansAssociationsMapToE2TDataList(e2tToRansAssociations)
95         data := models.NewRoutingManagerDeleteRequestModel(e2tAddress, ransTobeDissociated, e2tDataList)
96         url := c.config.RoutingManager.BaseUrl + DeleteE2TInstanceApiSuffix
97         return c.DeleteMessage(url, data)
98 }
99
100 func (c *RoutingManagerClient) sendMessage(method string, url string, data interface{}) error {
101         marshaled, err := json.Marshal(data)
102
103         if err != nil {
104                 return e2managererrors.NewRoutingManagerError()
105         }
106
107         body := bytes.NewBuffer(marshaled)
108         c.logger.Infof("[E2 Manager -> Routing Manager] #RoutingManagerClient.sendMessage - %s url: %s, request body: %+v", method, url, body)
109
110         var resp *http.Response
111
112         if method == http.MethodPost {
113                 resp, err = c.httpClient.Post(url, "application/json", body)
114         } else if method == http.MethodDelete {
115                 resp, err = c.httpClient.Delete(url, "application/json", body)
116         }
117
118         if err != nil {
119                 c.logger.Errorf("#RoutingManagerClient.sendMessage - failed sending request. error: %s", err)
120                 return e2managererrors.NewRoutingManagerError()
121         }
122
123         defer resp.Body.Close()
124
125         if resp.StatusCode >= http.StatusOK && resp.StatusCode < http.StatusMultipleChoices {
126                 c.logger.Infof("[Routing Manager -> E2 Manager] #RoutingManagerClient.sendMessage - success. http status code: %d", resp.StatusCode)
127                 return nil
128         }
129
130         c.logger.Errorf("[Routing Manager -> E2 Manager] #RoutingManagerClient.sendMessage - failure. http status code: %d", resp.StatusCode)
131         return e2managererrors.NewRoutingManagerError()
132 }
133
134 func (c *RoutingManagerClient) DeleteMessage(url string, data interface{}) error {
135         return c.sendMessage(http.MethodDelete, url, data)
136 }
137
138 func (c *RoutingManagerClient) PostMessage(url string, data interface{}) error {
139         return c.sendMessage(http.MethodPost, url, data)
140 }
141
142 func mapE2TAddressesToE2DataList(e2tAddresses []string) models.RoutingManagerE2TDataList {
143         e2tDataList := make(models.RoutingManagerE2TDataList, len(e2tAddresses))
144
145         for i, v := range e2tAddresses {
146                 e2tDataList[i] = models.NewRoutingManagerE2TData(v)
147         }
148
149         return e2tDataList
150 }
151
152 func convertE2TToRansAssociationsMapToE2TDataList(e2tToRansAssociations map[string][]string) models.RoutingManagerE2TDataList {
153         e2tDataList := make(models.RoutingManagerE2TDataList, len(e2tToRansAssociations))
154         i := 0
155         for k, v := range e2tToRansAssociations {
156                 e2tDataList[i] = models.NewRoutingManagerE2TData(k, v...)
157                 i++
158         }
159
160         return e2tDataList
161 }