feat: add queueKeys query

Author: Boris

example/src/config.ts

@@ -6,8 +6,8 @@
 //   password: process.env.REDIS_PASSWORD || '',
 // };
 
-const BULL_REDIS_URI =
-  'redis://:uut2tiew5waeli1aefup0Toecaikoque5eepahch5AowaiJ2@10.216.129.127:6379';
+const BULL_REDIS_URI = 'redis://127.0.0.1:6379';
+//'redis://:uut2tiew5waeli1aefup0Toecaikoque5eepahch5AowaiJ2@10.216.129.127:6379';
 
 const BULL_HOST_ID = 'maybe_uuid_and_mac';
 

example/src/schema/mutation/_helpers.ts

@@ -12,6 +12,7 @@ import {
   getMutationStatusEnumTC,
   getMutationErrorCodeEnumTC,
 } from '../types';
+import { createBullConnection } from '../../connectRedis';
 import { MutationError } from './Error';
 
 export function getQueue(queueName: string, context: any): Queue {
@@ -22,6 +23,18 @@ export function getQueue(queueName: string, context: any): Queue {
   return queue;
 }
 
+// export function getQueue(queueName: string, prefix: string): Queue {
+//   const queue = new Queue(queueName, {
+//     prefix,
+//     connection: createBullConnection('queue'),
+//   });
+
+//   if (!queue) {
+//     throw new MutationError('Queue not found!', ErrorCodeEnum.QUEUE_NOT_FOUND);
+//   }
+//   return queue;
+// }
+
 type FieldConfig = Omit<ObjectTypeComposerFieldConfigAsObjectDefinition<any, any>, 'type'> & {
   type: ObjectTypeComposerAsObjectDefinition<any, any>;
 };

example/src/schema/query/index.ts

@@ -1,10 +1,12 @@
 import { createQueuesFC } from './queues';
+import { createQueueKeysFC } from './queueKeys';
 import { createQueueFC } from './queue';
 import { createJobFC } from './job';
 import { SchemaComposer } from 'graphql-compose';
 
 export function createQueryFields(schemaComposer: SchemaComposer<any>): any {
   return {
+    queueKeys: createQueueKeysFC(schemaComposer),
     queues: createQueuesFC(schemaComposer),
     queue: createQueueFC(schemaComposer),
     job: createJobFC(schemaComposer),

example/src/schema/query/queueKeys.ts

@@ -0,0 +1,49 @@
+import { SchemaComposer } from 'graphql-compose';
+import { createBullConnection } from '../../connectRedis';
+
+export function createQueueKeysFC(schemaComposer: SchemaComposer<any>) {
+  return {
+    type: schemaComposer
+      .createObjectTC({
+        name: 'QueueKeysResult',
+        fields: {
+          queueName: 'String!',
+          prefix: 'String!',
+        },
+      })
+      .getTypePlural(),
+    args: {
+      prefixMask: {
+        type: 'String',
+        defaultValue: 'bull*',
+      },
+    },
+    resolve: async (_, { prefixMask }) => {
+      const connection = createBullConnection('custom');
+
+      let prefixMaskNorm = prefixMask;
+      const nameCase = prefixMaskNorm.split(':').length - 1;
+      if (nameCase === 2) {
+        prefixMaskNorm = prefixMaskNorm.endsWith('*')
+          ? prefixMaskNorm.substr(0, prefixMaskNorm.length - 1)
+          : prefixMaskNorm;
+      } else if (nameCase === 1) {
+        prefixMaskNorm += ':';
+      } else {
+        prefixMaskNorm += prefixMaskNorm.endsWith('*') ? ':*:' : '*:*:';
+      }
+
+      prefixMaskNorm += 'meta';
+
+      const keys = await connection.keys(prefixMaskNorm);
+
+      return keys.map((key) => {
+        const parts = key.split(':');
+        return {
+          queueName: parts[0],
+          prefix: parts[1],
+        };
+      });
+    },
+  };
+}

example/src/schema/types/queue/repeatables.ts

@@ -5,19 +5,21 @@ export function createRepeatablesFC(
   schemaComposer: SchemaComposer<any>
 ): ObjectTypeComposerFieldConfigDefinition<any, any> {
   return {
-    type: schemaComposer.createObjectTC({
-      name: 'RepeatableJobInformation',
-      fields: {
-        key: 'String',
-        name: 'String',
-        id: 'String',
-        endDate: 'Date',
-        tz: 'String',
-        cron: 'String',
-        //every: 'Date', //TODO: вроде как должен быть обязательным, проверить - нет в бул-4
-        next: 'Date',
-      },
-    }),
+    type: schemaComposer
+      .createObjectTC({
+        name: 'RepeatableJobInformation',
+        fields: {
+          key: 'String',
+          name: 'String',
+          id: 'String',
+          endDate: 'Date',
+          tz: 'String',
+          cron: 'String',
+          //every: 'Date', //TODO: вроде как должен быть обязательным, проверить - нет в бул-4
+          next: 'Date',
+        },
+      })
+      .getTypePlural(),
     resolve: async (queue: Queue) => {
       return await queue.getRepeatableJobs();
     },