Add cycleio infra platform (#212)

Author: tejaskokje-mw

Dockerfiles/DockerfileKube

@@ -27,4 +27,4 @@ COPY --from=build /app/configyamls-k8s/otel-config-deployment.yaml /app/otel-con
 # A symlink to support existing k8s agent users
 RUN ln -s /usr/bin/mw-agent /usr/bin/api-server
 
-CMD ["mw-agent", "start"]
+CMD ["mw-agent", "start"]

cmd/host-agent/main.go

@@ -323,6 +323,22 @@ func getFlags(execPath string, cfg *agent.HostConfig) []cli.Flag {
 	}
 }
 
+func detectInfraPlatform() agent.InfraPlatform {
+	awsEnv := os.Getenv("AWS_EXECUTION_ENV")
+	if awsEnv == "AWS_ECS_EC2" {
+		return agent.InfraPlatformECSEC2
+	} else if awsEnv == "AWS_ECS_FARGATE" {
+		return agent.InfraPlatformECSFargate
+	}
+
+	cycleIOEnv := os.Getenv("CYCLE_INSTANCE_ID")
+	if cycleIOEnv != "" {
+		return agent.InfraPlatformCycleIO
+	}
+
+	return agent.InfraPlatformInstance
+}
+
 func main() {
 	zapEncoderCfg := zapcore.EncoderConfig{
 		MessageKey: "message",
@@ -407,13 +423,7 @@ func main() {
 						go profiler.StartProfiling("mw-host-agent", cfg.Target, cfg.HostTags)
 					}
 
-					infraPlatform := agent.InfraPlatformInstance
-					awsEnv := os.Getenv("AWS_EXECUTION_ENV")
-					if awsEnv == "AWS_ECS_EC2" {
-						infraPlatform = agent.InfraPlatformECSEC2
-					} else if awsEnv == "AWS_ECS_FARGATE" {
-						infraPlatform = agent.InfraPlatformECSFargate
-					}
+					infraPlatform := detectInfraPlatform()
 
 					hostname, err := os.Hostname()
 					if err != nil {

pkg/agent/definitions.go

@@ -46,6 +46,8 @@ var (
 	InfraPlatformECSEC2 InfraPlatform = 2
 	// InfraPlatformECSFargate is for AWS ECS Fargate platform
 	InfraPlatformECSFargate InfraPlatform = 3
+	// InfraPlatformCycleIO is for Cycle.io platform
+	InfraPlatformCycleIO InfraPlatform = 4
 )
 
 func (p InfraPlatform) String() string {