feat: configurable mocktr181 service name

xmidt-org · Apr 8, 2024 · 4800736 · 4800736
1 parent 7837ebc
commit 4800736
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/cmd/xmidt-agent/config.go b/cmd/xmidt-agent/config.go
@@ -186,8 +186,9 @@ type Storage struct {
 }
 
 type MockTr181 struct {
-	FilePath string
-	Enabled  bool
+	FilePath    string
+	Enabled     bool
+	ServiceName string
 }
 
 // Collect and process the configuration files and env vars and
@@ -346,6 +347,7 @@ var defaultConfig = Config{
 		},
 	},
 	MockTr181: MockTr181{
-		FilePath: "./mock_tr181.json",
+		FilePath:    "./mock_tr181.json",
+		ServiceName: "config",
 	},
 }
diff --git a/cmd/xmidt-agent/wrphandlers.go b/cmd/xmidt-agent/wrphandlers.go
@@ -142,7 +142,7 @@ func providePubSubHandler(in pubsubIn) (pubsubOut, error) {
 			return pubsubOut{}, errors.Join(ErrWRPHandlerConfig, err)
 		}
 
-		mocktr, err := ps.SubscribeService("mocktr181", mocktr181Handler)
+		mocktr, err := ps.SubscribeService(in.MockTr181.ServiceName, mocktr181Handler)
 		if err != nil {
 			return pubsubOut{}, errors.Join(ErrWRPHandlerConfig, err)
 		}

diff --git a/cmd/xmidt-agent/xmidt_agent.yaml b/cmd/xmidt-agent/xmidt_agent.yaml
@@ -27,4 +27,7 @@ operational_state:
 # Optional
 # storage:
 #   temporary: ~/local-rdk-testing/temporary
-#   durable: ~/local-rdk-testing/durable
+#   durable: ~/local-rdk-testing/durable
+# mock_tr_181:
+#   enabled: true
+#   file_path: xmidt-agent/internal/wrphandlers/mocktr181/mock_tr181.json