Merge 1fbe219 into 2b48b4c

mesos · Jun 13, 2019 · 2467745 · 2467745
2 parents 2b48b4c + 1fbe219
commit 2467745
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/api/v1/lib/agent/calls/calls.go b/api/v1/lib/agent/calls/calls.go
@@ -220,6 +220,22 @@ func AttachContainerInputTTY(t *mesos.TTYInfo) *agent.Call {
 	}
 }
 
+func AttachContainerInputHeartbeat(heartbeat *agent.ProcessIO_Control_Heartbeat) *agent.Call {
+	return &agent.Call{
+		Type: agent.Call_ATTACH_CONTAINER_INPUT,
+		AttachContainerInput: &agent.Call_AttachContainerInput{
+			Type: agent.Call_AttachContainerInput_PROCESS_IO,
+			ProcessIO: &agent.ProcessIO{
+				Type: agent.ProcessIO_CONTROL,
+				Control: &agent.ProcessIO_Control{
+					Type:      agent.ProcessIO_Control_HEARTBEAT,
+					Heartbeat: heartbeat,
+				},
+			},
+		},
+	}
+}
+
 func AddResourceProviderConfig(rpi mesos.ResourceProviderInfo) *agent.Call {
 	return &agent.Call{
 		Type: agent.Call_ADD_RESOURCE_PROVIDER_CONFIG,

diff --git a/api/v1/lib/agent/calls/calls_test.go b/api/v1/lib/agent/calls/calls_test.go
@@ -75,6 +75,7 @@ func Example() {
 		AttachContainerInput(mesos.ContainerID{}),
 		AttachContainerInputTTY(nil),
 		AttachContainerInputData(nil),
+		AttachContainerInputHeartbeat(nil),
 	)))
 
 	// Output: