feat: Add RPC and statuses for session lifecycle management (#467)

Protos/V1/session_status.proto

@@ -14,4 +14,7 @@ enum SessionStatus {
   SESSION_STATUS_UNSPECIFIED = 0; /** Session is in an unknown state. */
   SESSION_STATUS_RUNNING = 1; /** Session is open and accepting tasks for execution. */
   SESSION_STATUS_CANCELLED = 2; /** Session is cancelled. No more tasks can be submitted and no more tasks will be executed. */
+  SESSION_STATUS_PAUSED = 3; /** Session is paused. Tasks can be submitted but no more new tasks will be executed. Already running tasks will continue until they finish. */
+  SESSION_STATUS_CLOSED = 4; /** Session is closed. No more tasks can be submitted and executed. Results data will be deleted. */
+  SESSION_STATUS_DELETED = 5; /** Session is deleted. Sessions, tasks and results metadata associated to the session will be deleted. */
 }

Protos/V1/sessions_common.proto

@@ -122,3 +122,67 @@ message CreateSessionRequest {
 message CreateSessionReply {
   string session_id = 1; /** Session id of the created session if successful */
 }
+
+/**
+ * Request for pausing a single session.
+ */
+message PauseSessionRequest {
+  string session_id = 1; /** The session ID. */
+}
+
+/**
+ * Response for pausing a single session.
+ *
+ * Return a raw session.
+ */
+message PauseSessionResponse {
+  SessionRaw session = 1; /** The session. */
+}
+
+/**
+ * Request for resuming a single session.
+ */
+message ResumeSessionRequest {
+  string session_id = 1; /** The session ID. */
+}
+
+/**
+ * Response for resuming a single session.
+ *
+ * Return a raw session.
+ */
+message ResumeSessionResponse {
+  SessionRaw session = 1; /** The session. */
+}
+
+/**
+ * Request for closing a single session.
+ */
+message CloseSessionRequest {
+  string session_id = 1; /** The session ID. */
+}
+
+/**
+ * Response for closing a single session.
+ *
+ * Return a raw session.
+ */
+message CloseSessionResponse {
+  SessionRaw session = 1; /** The session. */
+}
+
+/**
+ * Request for deleting a single session.
+ */
+message DeleteSessionRequest {
+  string session_id = 1; /** The session ID. */
+}
+
+/**
+ * Response for deleting a single session.
+ *
+ * Return a raw session.
+ */
+message DeleteSessionResponse {
+  SessionRaw session = 1; /** The session. */
+}

Protos/V1/sessions_service.proto

@@ -32,4 +32,24 @@ service Sessions {
    * Create a session
    */
   rpc CreateSession(CreateSessionRequest) returns (CreateSessionReply);
+
+  /**
+   * Pause a session by its id.
+   */
+  rpc PauseSession(PauseSessionRequest) returns (PauseSessionResponse);
+
+  /**
+   * Resume a paused session by its id.
+   */
+  rpc ResumeSession(ResumeSessionRequest) returns (ResumeSessionResponse);
+
+  /**
+   * Close a session by its id. Removes Results data.
+   */
+  rpc CloseSession(CloseSessionRequest) returns (CloseSessionResponse);
+
+  /**
+   * Delete a session by its id. Removes metadata from Results, Sessions and Tasks associated to the session.
+   */
+  rpc DeleteSession(DeleteSessionRequest) returns (DeleteSessionResponse);
 }

packages/csharp/ArmoniK.Api.Mock/Services/Sessions.cs

@@ -69,4 +69,40 @@ public override Task<ListSessionsResponse> ListSessions(ListSessionsRequest requ
                          PageSize = request.PageSize,
                          Total    = 0,
                        });
+
+  /// <inheritdoc />
+  [Count]
+  public override Task<CloseSessionResponse> CloseSession(CloseSessionRequest request,
+                                                          ServerCallContext   context)
+    => Task.FromResult(new CloseSessionResponse
+                       {
+                         Session = MockSession,
+                       });
+
+  /// <inheritdoc />
+  [Count]
+  public override Task<DeleteSessionResponse> DeleteSession(DeleteSessionRequest request,
+                                                            ServerCallContext    context)
+    => Task.FromResult(new DeleteSessionResponse
+                       {
+                         Session = MockSession,
+                       });
+
+  /// <inheritdoc />
+  [Count]
+  public override Task<PauseSessionResponse> PauseSession(PauseSessionRequest request,
+                                                          ServerCallContext   context)
+    => Task.FromResult(new PauseSessionResponse
+                       {
+                         Session = MockSession,
+                       });
+
+  /// <inheritdoc />
+  [Count]
+  public override Task<ResumeSessionResponse> ResumeSession(ResumeSessionRequest request,
+                                                            ServerCallContext    context)
+    => Task.FromResult(new ResumeSessionResponse
+                       {
+                         Session = MockSession,
+                       });
 }