Copy latest code to master
[ric-plt/resource-status-manager.git] / RSM / providers / rmrmsghandlerprovider / message_handler_provider.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
21 package rmrmsghandlerprovider
22
23 import (
24         "fmt"
25         "github.com/pkg/errors"
26         "rsm/configuration"
27         "rsm/converters"
28         "rsm/handlers/rmrmsghandlers"
29         "rsm/logger"
30         "rsm/rmrcgo"
31         "rsm/services"
32         "rsm/services/rmrsender"
33 )
34
35 const (
36         RanConnected string = "Ran connected"
37         RanRestarted string = "Ran restarted"
38 )
39
40 type MessageHandlerProvider struct {
41         msgHandlers map[int]rmrmsghandlers.RmrMessageHandler
42 }
43
44 func NewMessageHandlerProvider(logger *logger.Logger, config *configuration.Configuration, rnibDataService services.RNibDataService, rmrSender *rmrsender.RmrSender, resourceStatusService *services.ResourceStatusService, rsConverter converters.ResourceStatusResponseConverter, rsFailureConverter converters.ResourceStatusFailureConverter) *MessageHandlerProvider {
45         return &MessageHandlerProvider{
46                 msgHandlers: initMessageHandlersMap(logger, config, rnibDataService, rmrSender, resourceStatusService, rsConverter, rsFailureConverter),
47         }
48 }
49
50 func initMessageHandlersMap(logger *logger.Logger, config *configuration.Configuration, rnibDataService services.RNibDataService, rmrSender *rmrsender.RmrSender, resourceStatusService *services.ResourceStatusService, rsConverter converters.ResourceStatusResponseConverter, rsFailureConverter converters.ResourceStatusFailureConverter) map[int]rmrmsghandlers.RmrMessageHandler {
51         return map[int]rmrmsghandlers.RmrMessageHandler{
52                 rmrcgo.RanConnected:        rmrmsghandlers.NewResourceStatusInitiateNotificationHandler(logger, rnibDataService, resourceStatusService, RanConnected),
53                 rmrcgo.RanRestarted:        rmrmsghandlers.NewResourceStatusInitiateNotificationHandler(logger, rnibDataService, resourceStatusService, RanRestarted),
54                 rmrcgo.RicResStatusFailure: rmrmsghandlers.NewResourceStatusFailureHandler(logger, rsFailureConverter),
55                 rmrcgo.RicResStatusResp:    rmrmsghandlers.NewResourceStatusResponseHandler(logger, rsConverter, rnibDataService),
56         }
57 }
58
59 func (provider MessageHandlerProvider) GetMessageHandler(messageType int) (rmrmsghandlers.RmrMessageHandler, error) {
60         handler, ok := provider.msgHandlers[messageType]
61
62         if !ok {
63                 msg := fmt.Sprintf("#MessageHandlerProvider.GetMessageHandler - notification handler not found for message %d", messageType)
64                 return nil, errors.New(msg)
65         }
66
67         return handler, nil
68
69 }