feat: Support new status PREPARED of session and kernel (#2848)

related to lablup/backend.ai#2647 We decided to add a new session and kernel status called 'PREPARED' to Backend.AI. This status indicates that the session is ready to be created and doesn't require any miscellaneous tasks such as pulling images. **Checklist:** (if applicable) - [ ] Mention to the original issue - [ ] Documentation - [x] Minimum required manager version: 24.12.0 - [ ] Specific setting for review (eg., KB link, endpoint or how to setup) - [ ] Minimum requirements to check during review - [ ] Test case(s) to demonstrate the difference of before/after
lablup · Nov 14, 2024 · 7265dfb · 7265dfb
1 parent 0479273
commit 7265dfb
Show file tree

Hide file tree

Showing 9 changed files with 19 additions and 2 deletions.
diff --git a/react/src/components/ComputeSessionNodeItems/SessionStatusTag.tsx b/react/src/components/ComputeSessionNodeItems/SessionStatusTag.tsx
@@ -16,6 +16,7 @@ interface SessionStatusTagProps {
 const statusTagColor = {
   //prepare
   RESTARTING: 'blue',
+  PREPARED: 'blue',
   PREPARING: 'blue',
   PULLING: 'blue',
   //running

diff --git a/react/src/components/SessionList.tsx b/react/src/components/SessionList.tsx
@@ -115,6 +115,7 @@ const SessionList: React.FC<SessionListProps> = ({
   const statusTagColor = {
     //prepare
     RESTARTING: 'blue',
+    PREPARED: 'blue',
     PREPARING: 'blue',
     PULLING: 'blue',
     //running

diff --git a/react/src/components/SessionListColums/SessionInfoCell.tsx b/react/src/components/SessionListColums/SessionInfoCell.tsx
@@ -20,13 +20,14 @@ const isRunningStatus = (status: string = '') => {
     'TERMINATING',
     'PENDING',
     'SCHEDULED',
+    'PREPARED',
     'PREPARING',
     'PULLING',
   ].includes(status);
 };
 
 const isPreparing = (status: string = '') => {
-  return ['RESTARTING', 'PREPARING', 'PULLING'].includes(status);
+  return ['RESTARTING', 'PREPARED', 'PREPARING', 'PULLING'].includes(status);
 };
 
 const SessionInfoCell: React.FC<{

diff --git a/react/src/pages/SessionListPage.tsx b/react/src/pages/SessionListPage.tsx
@@ -15,6 +15,7 @@ const RUNNINGS = [
   'TERMINATING',
   'PENDING',
   'SCHEDULED',
+  'PREPARED',
   'PREPARING',
   'PULLING',
 ];

diff --git a/src/components/backend-ai-edu-applauncher.ts b/src/components/backend-ai-edu-applauncher.ts
@@ -258,6 +258,7 @@ export default class BackendAiEduApplauncher extends BackendAIPage {
         'TERMINATING',
         'PENDING',
         'SCHEDULED',
+        'PREPARED',
         'PREPARING',
         'PULLING',
       ].join(',');
@@ -267,6 +268,7 @@ export default class BackendAiEduApplauncher extends BackendAIPage {
         'RESTARTING',
         'TERMINATING',
         'PENDING',
+        'PREPARED',
         'PREPARING',
         'PULLING',
       ].join(',');

diff --git a/src/components/backend-ai-session-list.ts b/src/components/backend-ai-session-list.ts
@@ -578,7 +578,12 @@ export default class BackendAISessionList extends BackendAIPage {
   }
 
   _isPreparing(status) {
-    const preparingStatuses = ['RESTARTING', 'PREPARING', 'PULLING'];
+    const preparingStatuses = [
+      'RESTARTING',
+      'PREPARED',
+      'PREPARING',
+      'PULLING',
+    ];
     if (preparingStatuses.indexOf(status) === -1) {
       return false;
     }
@@ -760,6 +765,7 @@ export default class BackendAISessionList extends BackendAIPage {
           'TERMINATING',
           'PENDING',
           'SCHEDULED',
+          'PREPARED',
           'PREPARING',
           'PULLING',
           'ERROR',
@@ -775,6 +781,7 @@ export default class BackendAISessionList extends BackendAIPage {
           'TERMINATING',
           'PENDING',
           'SCHEDULED',
+          'PREPARED',
           'PREPARING',
           'PULLING',
         ];

diff --git a/src/components/backend-ai-session-view.ts b/src/components/backend-ai-session-view.ts
@@ -334,6 +334,7 @@ export default class BackendAISessionView extends BackendAIPage {
         'TERMINATING',
         'PENDING',
         'SCHEDULED',
+        'PREPARED',
         'PREPARING',
         'PULLING',
         'TERMINATED',
@@ -346,6 +347,7 @@ export default class BackendAISessionView extends BackendAIPage {
         'RESTARTING',
         'TERMINATING',
         'PENDING',
+        'PREPARED',
         'PREPARING',
         'PULLING',
         'TERMINATED',

diff --git a/src/pipeline/components/pipeline-job-list.ts b/src/pipeline/components/pipeline-job-list.ts
@@ -298,6 +298,7 @@ export default class PipelineJobList extends BackendAIPage {
             [
               'PENDING',
               'SCHEDULED',
+              'PREPARED',
               'PREPARING',
               'BUILDING',
               'PULLING',

diff --git a/src/pipeline/lib/pipeline-utils.ts b/src/pipeline/lib/pipeline-utils.ts
@@ -106,6 +106,7 @@ export default class PipelineUtils extends LitElement {
       case 'BUILDING':
       case 'PULLING':
       case 'SCHEDULED':
+      case 'PREPARED':
       case 'PREPARING':
       case 'PENDING':
         color = 'yellow';