[RICPLT-2584] Implementation of E2TShutdownManager + Automation fix
[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         DissociateAllRans(e2tAddresses []string) error
49         DeleteE2TInstance(e2tAddress string, ransToBeDissociated []string, e2tToRansAssociations map[string][]string) error
50 }
51
52 func NewRoutingManagerClient(logger *logger.Logger, config *configuration.Configuration, httpClient HttpClient) *RoutingManagerClient {
53         return &RoutingManagerClient{
54                 logger:     logger,
55                 config:     config,
56                 httpClient: httpClient,
57         }
58 }
59
60 func (c *RoutingManagerClient) AddE2TInstance(e2tAddress string) error {
61
62         data := models.NewRoutingManagerE2TData(e2tAddress)
63         url := c.config.RoutingManager.BaseUrl + AddE2TInstanceApiSuffix
64
65         return c.PostMessage(data, url)
66 }
67
68 func (c *RoutingManagerClient) AssociateRanToE2TInstance(e2tAddress string, ranName string) error {
69
70         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
71         url := c.config.RoutingManager.BaseUrl + AssociateRanToE2TInstanceApiSuffix
72
73         return c.PostMessage(data, url)
74 }
75
76 func (c *RoutingManagerClient) DissociateRanE2TInstance(e2tAddress string, ranName string) error {
77
78         data := models.RoutingManagerE2TDataList{models.NewRoutingManagerE2TData(e2tAddress, ranName)}
79         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
80
81         return c.PostMessage(data, url)
82 }
83
84 func (c *RoutingManagerClient) DissociateAllRans(e2tAddresses []string) error {
85
86         data := mapE2TAddressesToE2DataList(e2tAddresses)
87         url := c.config.RoutingManager.BaseUrl + DissociateRanE2TInstanceApiSuffix
88
89         return c.PostMessage(data, url)
90 }
91
92 func (c *RoutingManagerClient) DeleteE2TInstance(e2tAddress string, ransToBeDissociated []string, e2tToRansAssociations map[string][]string) error {
93         //TODO - here should be the call to routing-manager...
94         return nil
95 }
96
97 func (c *RoutingManagerClient) PostMessage(data interface{}, url string) error {
98         marshaled, err := json.Marshal(data)
99
100         if err != nil {
101                 return e2managererrors.NewRoutingManagerError()
102         }
103
104         body := bytes.NewBuffer(marshaled)
105         c.logger.Infof("[E2 Manager -> Routing Manager] #RoutingManagerClient.PostMessage - url: %s, request body: %+v", url, body)
106
107         resp, err := c.httpClient.Post(url, "application/json", body)
108
109         if err != nil {
110                 c.logger.Errorf("#RoutingManagerClient.PostMessage - failed sending request. error: %s", err)
111                 return e2managererrors.NewRoutingManagerError()
112         }
113
114         defer resp.Body.Close()
115
116         if resp.StatusCode >= http.StatusOK && resp.StatusCode < http.StatusMultipleChoices {
117                 c.logger.Infof("[Routing Manager -> E2 Manager] #RoutingManagerClient.PostMessage - success. http status code: %d", resp.StatusCode)
118                 return nil
119         }
120
121         c.logger.Errorf("[Routing Manager -> E2 Manager] #RoutingManagerClient.PostMessage - failure. http status code: %d", resp.StatusCode)
122         return e2managererrors.NewRoutingManagerError()
123 }
124
125 func mapE2TAddressesToE2DataList(e2tAddresses []string) models.RoutingManagerE2TDataList {
126         e2tDataList := make(models.RoutingManagerE2TDataList, len(e2tAddresses))
127
128         for i, v := range e2tAddresses {
129                 e2tDataList[i] = models.NewRoutingManagerE2TData(v)
130         }
131
132         return e2tDataList
133 }