Set flags on MessageData before OnMessage callback

open-telemetry · Jan 14, 2025 · 9a8dabd · 9a8dabd
1 parent cd6ccd7
commit 9a8dabd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/client/internal/receivedprocessor.go b/client/internal/receivedprocessor.go
@@ -75,7 +75,9 @@ func (r *receivedProcessor) ProcessReceivedMessage(ctx context.Context, msg *pro
 		r.logger.Errorf(ctx, "cannot processed received flags:%v", err)
 	}
 
-	msgData := &types.MessageData{}
+	msgData := &types.MessageData{
+		Flags: protobufs.ServerToAgentFlags(msg.Flags),
+	}
 
 	if msg.RemoteConfig != nil {
 		if r.hasCapability(protobufs.AgentCapabilities_AgentCapabilities_AcceptsRemoteConfig) {