Add connections_controller and services_controller

1 files changed, 11 insertions, 6 deletions

diff --git a/application_context.go b/application_context.go
index 50a1665..3ac3031 100644
--- a/application_context.go
+++ b/application_context.go
@@ -13,12 +13,14 @@ type Config struct {
 }
 
 type ApplicationContext struct {
-	Storage      Storage
-	Config       Config
-	Accounts     gin.Accounts
-	RulesManager RulesManager
-	PcapImporter *PcapImporter
-	IsConfigured bool
+	Storage               Storage
+	Config                Config
+	Accounts              gin.Accounts
+	RulesManager          RulesManager
+	PcapImporter          *PcapImporter
+	ConnectionsController ConnectionsController
+	ServicesController    *ServicesController
+	IsConfigured          bool
 }
 
 func CreateApplicationContext(storage Storage) (*ApplicationContext, error) {
@@ -88,5 +90,8 @@ func (sm *ApplicationContext) configure() {
 	}
 	sm.RulesManager = rulesManager
 	sm.PcapImporter = NewPcapImporter(sm.Storage, serverIP, sm.RulesManager)
+	sm.ServicesController = NewServicesController(sm.Storage)
+	sm.ConnectionsController = NewConnectionsController(sm.Storage, sm.ServicesController)
 	sm.IsConfigured = true
+
 }