From e45d77d2355c1bd289314002a774f6d199f59a1f Mon Sep 17 00:00:00 2001
From: Kangyi LI <kangyi.li@datadoghq.com>
Date: Thu, 21 Mar 2024 13:26:39 +0100
Subject: [PATCH] add TypeCollectorECSTask (#296)

---
 process/message.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/process/message.go b/process/message.go
index da75b928..a89b4133 100644
--- a/process/message.go
+++ b/process/message.go
@@ -117,6 +117,7 @@ const (
 	TypeCollectorVerticalPodAutoscaler   = 83
 	TypeCollectorHorizontalPodAutoscaler = 84
 	TypeCollectorNetworkPolicy           = 85
+	TypeCollectorECSTask                 = 200
 )
 
 func (m MessageType) String() string {
@@ -185,6 +186,8 @@ func (m MessageType) String() string {
 		return "horizontal-pod-autoscaler"
 	case TypeCollectorNetworkPolicy:
 		return "network-policy"
+	case TypeCollectorECSTask:
+		return "ecs-task"
 	default:
 		// otherwise convert the type identifier
 		return strconv.Itoa(int(m))
@@ -281,6 +284,8 @@ func DecodeMessage(data []byte) (Message, error) {
 		m = &CollectorHorizontalPodAutoscaler{}
 	case TypeCollectorNetworkPolicy:
 		m = &CollectorNetworkPolicy{}
+	case TypeCollectorECSTask:
+		m = &CollectorECSTask{}
 	default:
 		return Message{}, fmt.Errorf("unhandled message type: %d", header.Type)
 	}
@@ -360,6 +365,8 @@ func DetectMessageType(b MessageBody) (MessageType, error) {
 		t = TypeCollectorHorizontalPodAutoscaler
 	case *CollectorNetworkPolicy:
 		t = TypeCollectorNetworkPolicy
+	case *CollectorECSTask:
+		t = TypeCollectorECSTask
 	default:
 		return 0, fmt.Errorf("unknown message body type: %s", reflect.TypeOf(b))
 	}