APIGOV-28713 - add dataplane type in root init (#162)

Axway · Sep 6, 2024 · 74248d6 · 74248d6
1 parent ba74c90
commit 74248d6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/discovery/cmd/cmd.go b/pkg/discovery/cmd/cmd.go
@@ -3,6 +3,7 @@ package discovery
 import (
 	"time"
 
+	"github.com/Axway/agent-sdk/pkg/apic"
 	corecmd "github.com/Axway/agent-sdk/pkg/cmd"
 	corecfg "github.com/Axway/agent-sdk/pkg/config"
 	"github.com/Axway/agent-sdk/pkg/util/log"
@@ -25,6 +26,9 @@ func init() {
 		corecfg.DiscoveryAgent,
 	)
 
+	// set the dataplane type that will be added to the agent spec
+	corecfg.AgentDataPlaneType = apic.Kong.String()
+
 	// Get the root command properties and bind the config property in YAML definition
 	rootProps := DiscoveryCmd.GetProperties()
 	config.AddKongProperties(rootProps)

diff --git a/pkg/traceability/cmd/cmd.go b/pkg/traceability/cmd/cmd.go
@@ -4,6 +4,7 @@ import (
 	libcmd "github.com/elastic/beats/v7/libbeat/cmd"
 	"github.com/elastic/beats/v7/libbeat/cmd/instance"
 
+	"github.com/Axway/agent-sdk/pkg/apic"
 	corecmd "github.com/Axway/agent-sdk/pkg/cmd"
 	corecfg "github.com/Axway/agent-sdk/pkg/config"
 
@@ -35,6 +36,9 @@ func init() {
 		corecfg.TraceabilityAgent,
 	)
 
+	// set the dataplane type that will be added to the agent spec
+	corecfg.AgentDataPlaneType = apic.Kong.String()
+
 	rootProps := TraceCmd.GetProperties()
 	config.AddKongProperties(rootProps)
 }