feat(FR-777): use resource limit of resource group

Author: yomybaby

react/data/schema.graphql

@@ -9,6 +9,37 @@ type Queries {
     """The ID of the object"""
     id: ID!
   ): Node
+
+  """Added in 25.6.0."""
+  audit_log_schema: AuditLogSchema
+
+  """Added in 25.6.0."""
+  audit_log_nodes(
+    """
+    Specifies the criteria used to narrow down the query results based on certain conditions.
+    """
+    filter: String
+
+    """Specifies the sorting order of the query result."""
+    order: String
+
+    """Specifies how many items to skip before beginning to return result."""
+    offset: Int
+
+    """If this value is provided, the query will be limited to that value."""
+    before: String
+
+    """Queries the `last` number of results from the query result from last."""
+    after: String
+
+    """
+    Queries the `first` number of results from the query result from first.
+    """
+    first: Int
+
+    """If the given value is provided, the query will start from that value."""
+    last: Int
+  ): AuditLogConnection
   agent(agent_id: String!): Agent
   agent_list(limit: Int!, offset: Int!, filter: String, order: String, scaling_group: String, status: String): AgentList
   agents(scaling_group: String, status: String): [Agent]
@@ -245,6 +276,23 @@ type Queries {
   compute_container_list(limit: Int!, offset: Int!, filter: String, order: String, session_id: ID!, role: String): ComputeContainerList
   legacy_compute_session_list(limit: Int!, offset: Int!, order_key: String, order_asc: Boolean, domain_name: String, group_id: String, access_key: String, status: String): LegacyComputeSessionList
   legacy_compute_session(sess_id: String!, domain_name: String, access_key: String): LegacyComputeSession
+
+  """Added in 25.5.0."""
+  total_resource_slot(
+    """
+    `statuses` argument is an array of session statuses. Only sessions with the specified statuses will be queried to calculate the sum of total resource slots. The argument should be an array of the following valid status values: ['PENDING', 'SCHEDULED', 'PREPARING', 'PULLING', 'PREPARED', 'CREATING', 'RUNNING', 'RESTARTING', 'RUNNING_DEGRADED', 'TERMINATING', 'TERMINATED', 'ERROR', 'CANCELLED'].
+    Default value is null.
+    """
+    statuses: [String] = null
+
+    """
+    `filter` argument is a string that is parsed into query conditions. It works in the same way as the `filter` argument in the `compute_session` query schema, meaning the values are parsed into an identical SQL query expression.
+    Default value is `null`.
+    """
+    filter: String
+    domain_name: String
+    resource_group_name: String
+  ): TotalResourceSlot
   vfolder_host_permissions: PredefinedAtomicPermission
   endpoint(endpoint_id: UUID!): Endpoint
   endpoint_list(limit: Int!, offset: Int!, filter: String, order: String, domain_name: String, group_id: String, user_uuid: String, project: UUID): EndpointList
@@ -290,6 +338,108 @@ interface Node {
   id: ID!
 }
 
+"""
+A schema that contains metadata related to the AuditLogNode.
+It provides a list of values, such as entity_type and status, that can be used in the AuditLog, allowing clients to retrieve them.
+
+Added in 25.6.0.
+"""
+type AuditLogSchema {
+  """
+  Possible values of "AuditLogNode.entity_type"
+  """
+  entity_type_variants: [String]
+
+  """
+  Possible values of "AuditLogNode.status"
+  """
+  status_variants: [String]
+}
+
+"""Added in 25.6.0."""
+type AuditLogConnection {
+  """Pagination data for this connection."""
+  pageInfo: PageInfo!
+
+  """Contains the nodes in this connection."""
+  edges: [AuditLogEdge]!
+
+  """Total count of the GQL nodes of the query."""
+  count: Int
+}
+
+"""
+The Relay compliant `PageInfo` type, containing data necessary to paginate this connection.
+"""
+type PageInfo {
+  """When paginating forwards, are there more items?"""
+  hasNextPage: Boolean!
+
+  """When paginating backwards, are there more items?"""
+  hasPreviousPage: Boolean!
+
+  """When paginating backwards, the cursor to continue."""
+  startCursor: String
+
+  """When paginating forwards, the cursor to continue."""
+  endCursor: String
+}
+
+"""Added in 25.6.0. A Relay edge containing a `AuditLog` and its cursor."""
+type AuditLogEdge {
+  """The item at the end of the edge"""
+  node: AuditLogNode
+
+  """A cursor for use in pagination"""
+  cursor: String!
+}
+
+"""Added in 25.6.0."""
+type AuditLogNode implements Node {
+  """The ID of the object"""
+  id: ID!
+
+  """UUID of the audit log row"""
+  row_id: UUID!
+
+  """Entity ID of the AuditLog"""
+  entity_type: String!
+
+  """Entity type of the AuditLog"""
+  operation: String!
+
+  """Operation type of the AuditLog"""
+  entity_id: String!
+
+  """The time the AuditLog was reported"""
+  created_at: DateTime!
+
+  """RequestID of the AuditLog"""
+  request_id: UUID!
+
+  """Description of the AuditLog"""
+  description: String!
+
+  """Duration taken to perform the operation"""
+  duration: String!
+
+  """Status of the AuditLog"""
+  status: String!
+}
+
+"""
+Leverages the internal Python implementation of UUID (uuid.UUID) to provide native UUID objects
+in fields, resolvers and input.
+"""
+scalar UUID
+
+"""
+The `DateTime` scalar type represents a DateTime
+value as specified by
+[iso8601](https://en.wikipedia.org/wiki/ISO_8601).
+"""
+scalar DateTime
+
 type Agent implements Item {
   id: ID
   status: String
@@ -330,13 +480,6 @@ interface Item {
   id: ID
 }
 
-"""
-The `DateTime` scalar type represents a DateTime
-value as specified by
-[iso8601](https://en.wikipedia.org/wiki/ISO_8601).
-"""
-scalar DateTime
-
 """
 Allows use of a JSON String for input / output from the GraphQL schema.
 
@@ -391,12 +534,6 @@ type ComputeContainer implements Item {
   preopen_ports: [Int]
 }
 
-"""
-Leverages the internal Python implementation of UUID (uuid.UUID) to provide native UUID objects
-in fields, resolvers and input.
-"""
-scalar UUID
-
 type ImageNode implements Node {
   """The ID of the object"""
   id: ID!
@@ -522,23 +659,6 @@ type ScalinGroupConnection {
   count: Int
 }
 
-"""
-The Relay compliant `PageInfo` type, containing data necessary to paginate this connection.
-"""
-type PageInfo {
-  """When paginating forwards, are there more items?"""
-  hasNextPage: Boolean!
-
-  """When paginating backwards, are there more items?"""
-  hasPreviousPage: Boolean!
-
-  """When paginating backwards, the cursor to continue."""
-  startCursor: String
-
-  """When paginating forwards, the cursor to continue."""
-  endCursor: String
-}
-
 """
 Added in 24.12.0. A Relay edge containing a `ScalinGroup` and its cursor.
 """
@@ -836,6 +956,9 @@ type UserNode implements Node {
   Added in 25.2.0. Supplementary group IDs assigned to processes running inside the container.
   """
   container_gids: [Int]
+
+  """Added in 25.5.0."""
+  project_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): GroupConnection
 }
 
 """Added in 24.03.0"""
@@ -1245,6 +1368,9 @@ type ScalingGroup {
     status: String = "ALIVE"
   ): JSONString
 
+  """Added in 25.6.0. The resource slot hard-limit of the resource group."""
+  resource_slot_limit: JSONString
+
   """
   Added in 25.4.0. The sum of occupied slots across compute sessions that occupying agent's resources. Only includes sessions owned by the user.
   """
@@ -1515,6 +1641,13 @@ type LegacyComputeSession implements Item {
   io_cur_scratch_size: BigInt
 }
 
+"""Added in 25.5.0."""
+type TotalResourceSlot implements Item {
+  id: ID
+  occupied_slots: JSONString
+  requested_slots: JSONString
+}
+
 type PredefinedAtomicPermission {
   vfolder_host_permission_list: [String]
 }
@@ -1992,7 +2125,7 @@ type Mutations {
   clear_images(registry: String): ClearImages
 
   """Added in 25.4.0"""
-  purge_images(agent_id: String!, images: [ImageRefType]!): PurgeImages
+  purge_images(keys: [PurgeImagesKey]!, options: PurgeImagesOptions = {force: false, noprune: false}): PurgeImagesPayload
 
   """Added in 24.09.0."""
   modify_compute_session(input: ModifyComputeSessionInput!): ModifyComputeSessionPayload
@@ -2592,17 +2725,34 @@ type ClearImages {
   msg: String
 }
 
-"""Added in 25.4.0."""
-type PurgeImages {
+"""Added in 25.6.0."""
+type PurgeImagesPayload {
   task_id: String
 }
 
+"""Added in 25.6.0."""
+input PurgeImagesKey {
+  agent_id: String!
+  images: [ImageRefType]!
+}
+
 input ImageRefType {
   name: String!
   registry: String
   architecture: String
 }
 
+"""Added in 25.6.0."""
+input PurgeImagesOptions {
+  """
+  Remove the images even if it is being used by stopped containers or has other tags, Added in 25.6.0.
+  """
+  force: Boolean = false
+
+  """Don't delete untagged parent images, Added in 25.6.0."""
+  noprune: Boolean = false
+}
+
 """Added in 24.09.0."""
 type ModifyComputeSessionPayload {
   item: ComputeSessionNode

react/src/components/DynamicUnitInputNumberWithSlider.tsx

@@ -1,4 +1,8 @@
-import { compareNumberWithUnits, convertBinarySizeUnit } from '../helper';
+import {
+  compareNumberWithUnits,
+  convertBinarySizeUnit,
+  toFixedFloorWithoutTrailingZeros,
+} from '../helper';
 import { useUpdatableState } from '../hooks';
 import useControllableState from '../hooks/useControllableState';
 import DynamicUnitInputNumber, {
@@ -222,8 +226,12 @@ const DynamicUnitInputNumberWithSlider: React.FC<
                     maxGiB.number === 0
                       ? maxGiB.number
                       : maxGiB.number >= 1
-                        ? maxGiB.number + 'g'
-                        : maxGiB.number * 1024 + 'm',
+                        ? toFixedFloorWithoutTrailingZeros(maxGiB.number, 2) +
+                          'g'
+                        : toFixedFloorWithoutTrailingZeros(
+                            maxGiB.number * 1024,
+                            2,
+                          ) + 'm',
                 },
               }),
             })}

react/src/components/ResourceAllocationFormItems.tsx

@@ -128,6 +128,7 @@ const ResourceAllocationFormItems: React.FC<
             accelerator_quantum_size
             name
             is_active
+            ...useResourceLimitAndRemainingFragment
           }
         }
       `,
@@ -164,6 +165,7 @@ const ResourceAllocationFormItems: React.FC<
     useResourceLimitAndRemaining({
       currentProjectName: currentProject.name,
       currentResourceGroup: currentResourceGroupInForm || undefined, // global currentResourceGroup can be null
+      currentResourceGroupFrgmt: currentResourceGroupInfo,
       currentImage: currentImage,
     });
 

react/src/hooks/useResourceLimitAndRemaining.tsx

@@ -3,10 +3,13 @@ import { Image } from '../components/ImageEnvironmentSelectFormItems';
 import { AUTOMATIC_DEFAULT_SHMEM } from '../components/ResourceAllocationFormItems';
 import { addNumberWithUnits, convertBinarySizeUnit } from '../helper';
 import { ResourceSlotName, useResourceSlots } from '../hooks/backendai';
+import { useResourceLimitAndRemainingFragment$key } from './__generated__/useResourceLimitAndRemainingFragment.graphql';
 import { useSuspenseTanQuery } from './reactQueryAlias';
 import { useResourceGroupsForCurrentProject } from './useCurrentProject';
+import graphql from 'babel-plugin-relay/macro';
 import _ from 'lodash';
 import { useMemo } from 'react';
+import { useFragment } from 'react-relay';
 
 const maxPerContainerRegex = /^max([A-Za-z0-9]+)PerContainer$/;
 
@@ -96,6 +99,7 @@ interface Props {
   currentProjectName: string;
   currentImage?: Image;
   currentResourceGroup?: string;
+  currentResourceGroupFrgmt?: useResourceLimitAndRemainingFragment$key | null;
   ignorePerContainerConfig?: boolean;
   fetchKey?: string;
 }
@@ -104,6 +108,7 @@ interface Props {
 export const useResourceLimitAndRemaining = ({
   currentImage,
   currentResourceGroup = '',
+  currentResourceGroupFrgmt = null,
   currentProjectName,
   ignorePerContainerConfig = false,
   fetchKey,
@@ -113,6 +118,18 @@ export const useResourceLimitAndRemaining = ({
   const acceleratorSlots = _.omit(resourceSlots, ['cpu', 'mem', 'shmem']);
   const { resourceGroups } = useResourceGroupsForCurrentProject();
 
+  const currentResourceGroupInfo = useFragment(
+    graphql`
+      fragment useResourceLimitAndRemainingFragment on ScalingGroup {
+        name
+        max_available_slots_for_each_resource
+        min_available_slots_for_each_resource
+      }
+    `,
+    currentResourceGroupFrgmt,
+  );
+
+  console.log('#', currentResourceGroupInfo);
   const {
     data: checkPresetInfo,
     refetch,