[RICPLT-1705] Reset Request received from E2T (Remote RAN)

[ric-plt/e2mgr.git] / E2Manager / managers / notification_manager.go

diff --git a/E2Manager/managers/notification_manager.go b/E2Manager/managers/notification_manager.go
index 9e9673d..5533c52 100644 (file)
--- a/E2Manager/managers/notification_manager.go
+++ b/E2Manager/managers/notification_manager.go
@@ -30,22 +30,21 @@ import (
 )
 
 type NotificationManager struct {
-       rnibReaderProvider func() reader.RNibReader
-       rnibWriterProvider func() rNibWriter.RNibWriter
+       notificationHandlerProvider *providers.NotificationHandlerProvider
 }
 
 func NewNotificationManager(rnibReaderProvider func() reader.RNibReader, rnibWriterProvider func() rNibWriter.RNibWriter) *NotificationManager {
+       notificationHandlerProvider := providers.NewNotificationHandlerProvider(rnibReaderProvider, rnibWriterProvider)
+
        return &NotificationManager{
-               rnibReaderProvider: rnibReaderProvider,
-               rnibWriterProvider: rnibWriterProvider,
+               notificationHandlerProvider: notificationHandlerProvider,
        }
 }
 
 //TODO add NEWHandler with log
 func (m NotificationManager) HandleMessage(logger *logger.Logger, e2Sessions sessions.E2Sessions, mbuf *rmrCgo.MBuf, responseChannel chan<- *models.NotificationResponse) {
 
-       provider := providers.NewNotificationHandlerProvider(m.rnibReaderProvider, m.rnibWriterProvider)
-       notificationHandler, err := provider.GetNotificationHandler(mbuf.MType)
+       notificationHandler, err := m.notificationHandlerProvider.GetNotificationHandler(mbuf.MType)
 
        if err != nil {
                logger.Errorf(fmt.Sprintf("%s", err))