feat: supports container actions for agents (#3164)

Author: amir20

internal/agent/client.go

@@ -368,6 +368,25 @@ func (c *Client) Host() (docker.Host, error) {
 	}, nil
 }
 
+func (c *Client) ContainerAction(containerId string, action docker.ContainerAction) error {
+	var containerAction pb.ContainerAction
+	switch action {
+	case docker.Start:
+		containerAction = pb.ContainerAction_Start
+
+	case docker.Stop:
+		containerAction = pb.ContainerAction_Stop
+
+	case docker.Restart:
+		containerAction = pb.ContainerAction_Restart
+
+	}
+
+	_, err := c.client.ContainerAction(context.Background(), &pb.ContainerActionRequest{ContainerId: containerId, Action: containerAction})
+
+	return err
+}
+
 func (c *Client) Close() error {
 	return c.conn.Close()
 }

internal/agent/pb/rpc.pb.go

@@ -285,6 +285,31 @@ func (s *server) StreamContainerStarted(in *pb.StreamContainerStartedRequest, ou
 	}
 }
 
+func (s *server) ContainerAction(ctx context.Context, in *pb.ContainerActionRequest) (*pb.ContainerActionResponse, error) {
+	var action docker.ContainerAction
+	switch in.Action {
+	case pb.ContainerAction_Start:
+		action = docker.Start
+
+	case pb.ContainerAction_Stop:
+		action = docker.Stop
+
+	case pb.ContainerAction_Restart:
+		action = docker.Restart
+
+	default:
+		return nil, status.Error(codes.InvalidArgument, "invalid action")
+	}
+
+	err := s.client.ContainerActions(action, in.ContainerId)
+
+	if err != nil {
+		return nil, status.Error(codes.Internal, err.Error())
+	}
+
+	return &pb.ContainerActionResponse{}, nil
+}
+
 func NewServer(client docker.Client, certificates tls.Certificate, dozzleVersion string) *grpc.Server {
 	caCertPool := x509.NewCertPool()
 	c, err := x509.ParseCertificate(certificates.Certificate[0])

internal/agent/pb/rpc_grpc.pb.go

@@ -65,5 +65,5 @@ func (d *agentService) SubscribeContainersStarted(ctx context.Context, container
 }
 
 func (a *agentService) ContainerAction(container docker.Container, action docker.ContainerAction) error {
-	panic("not implemented")
+	return a.client.ContainerAction(container.ID, action)
 }

internal/agent/pb/types.pb.go

@@ -107,12 +107,12 @@ func (m *RetriableClientManager) RetryAndList() ([]ClientService, []error) {
 				host.Available = true
 
 				// We don't want to block the subscribers in event.go
-				go func() {
+				go func(host docker.Host) {
 					select {
 					case channel <- host:
 					case <-ctx.Done():
 					}
-				}()
+				}(host)
 
 				return true
 			})

internal/agent/server.go

@@ -19,6 +19,8 @@ service AgentService {
   rpc StreamContainerStarted(StreamContainerStartedRequest)
       returns (stream StreamContainerStartedResponse) {}
   rpc HostInfo(HostInfoRequest) returns (HostInfoResponse) {}
+  rpc ContainerAction(ContainerActionRequest)
+      returns (ContainerActionResponse) {}
 }
 
 message ListContainersRequest {}
@@ -64,3 +66,11 @@ message HostInfoResponse { Host host = 1; }
 
 message StreamContainerStartedRequest {}
 message StreamContainerStartedResponse { Container container = 1; }
+
+message ContainerActionRequest {
+  string containerId = 1;
+  ContainerAction action = 2;
+}
+
+message ContainerActionResponse {
+}

internal/support/docker/agent_service.go

@@ -65,3 +65,9 @@ message Host {
   string agentVersion = 11;
   string dockerVersion = 12;
 }
+
+enum ContainerAction {
+  Start = 0;
+  Stop = 1;
+  Restart = 2;
+}