[back] Improve session screen by batching user fetching

OpenCTI-Platform · Nov 18, 2022 · 3f3fdce · 3f3fdce
1 parent 800c4e8
commit 3f3fdce
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/opencti-platform/opencti-graphql/src/generated/graphql.ts b/opencti-platform/opencti-graphql/src/generated/graphql.ts
@@ -4836,6 +4836,7 @@ export enum IdentitiesFilter {
   Name = 'name',
   Revoked = 'revoked',
   UpdatedAt = 'updated_at',
+  XOpenctiAliases = 'x_opencti_aliases',
   XOpenctiWorkflowId = 'x_opencti_workflow_id'
 }
 
@@ -6896,6 +6897,7 @@ export enum LocationType {
 }
 
 export enum LocationsFilter {
+  Aliases = 'aliases',
   Confidence = 'confidence',
   Created = 'created',
   CreatedBy = 'createdBy',

diff --git a/opencti-platform/opencti-graphql/src/resolvers/user.js b/opencti-platform/opencti-graphql/src/resolvers/user.js
@@ -3,10 +3,13 @@ import { withFilter } from 'graphql-subscriptions';
 import {
   addBookmark,
   addUser,
+  assignOrganizationToUser,
   authenticateUser,
   batchGroups,
+  batchOrganizations,
   batchRoleCapabilities,
   batchRoles,
+  batchUsers,
   bookmarks,
   deleteBookmark,
   findAll,
@@ -29,14 +32,12 @@ import {
   userAddRelation,
   userCleanContext,
   userDelete,
+  userDeleteOrganizationRelation,
   userEditContext,
   userEditField,
   userIdDeleteRelation,
   userRenewToken,
   userWithOrigin,
-  batchOrganizations,
-  assignOrganizationToUser,
-  userDeleteOrganizationRelation,
 } from '../domain/user';
 import { BUS_TOPICS, logApp, logAudit } from '../config/conf';
 import passport, { PROVIDERS } from '../config/providers';
@@ -55,6 +56,7 @@ const groupsLoader = batchLoader(batchGroups);
 const organizationsLoader = batchLoader(batchOrganizations);
 const rolesLoader = batchLoader(batchRoles);
 const rolesCapabilitiesLoader = batchLoader(batchRoleCapabilities);
+const usersLoader = batchLoader(batchUsers);
 
 const userResolvers = {
   Query: {
@@ -81,7 +83,7 @@ const userResolvers = {
     userSubscriptions: (current, _, context) => getUserSubscriptions(context, context.user, current.id),
   },
   UserSession: {
-    user: (session, _, context) => findById(context, context.user, session.user_id),
+    user: (session, _, context) => usersLoader.load(session.user_id, context, context.user),
   },
   SessionDetail: {
     ttl: (session) => fetchSessionTtl(session.id),