feat: add redis memory info query (#16)

Co-authored-by: Boris Dorofeev <[email protected]>

Author: intpro

src/composeBull.ts

@@ -1,7 +1,13 @@
 import { schemaComposer, SchemaComposer } from 'graphql-compose';
 import { Options } from './definitions';
 import { getQueueTC, getJobTC } from './types';
-import { createQueuesFC, createQueueKeysFC, createQueueFC, createJobFC } from './query';
+import {
+  createQueuesFC,
+  createQueueKeysFC,
+  createQueueFC,
+  createJobFC,
+  createInfoFC,
+} from './query';
 import {
   createQueueCleanFC,
   createQueueDrainFC,
@@ -39,6 +45,7 @@ export function composeBull(opts: Options & { schemaComposer?: SchemaComposer<an
       queues: wrapQuery(createQueuesFC),
       queue: wrapQuery(createQueueFC),
       job: wrapQuery(createJobFC),
+      info: wrapQuery(createInfoFC),
     },
     mutationFields: {
       queueClean: wrapMutation(createQueueCleanFC),

src/query/index.ts

@@ -2,3 +2,4 @@ export { createQueuesFC } from './queues';
 export { createQueueKeysFC } from './queueKeys';
 export { createQueueFC } from './queue';
 export { createJobFC } from './job';
+export { createInfoFC } from './info';

src/query/info.ts

@@ -0,0 +1,15 @@
+import { SchemaComposer, ObjectTypeComposerFieldConfigAsObjectDefinition } from 'graphql-compose';
+import { getInfoTC } from '../types/info/Info';
+import { Options } from '../definitions';
+
+export function createInfoFC(
+  sc: SchemaComposer<any>,
+  opts: Options
+): ObjectTypeComposerFieldConfigAsObjectDefinition<any, any> {
+  return {
+    type: getInfoTC(sc, opts),
+    resolve: async () => {
+      return {};
+    },
+  };
+}

src/types/info/Info.memory.ts

@@ -0,0 +1,67 @@
+import { SchemaComposer, ObjectTypeComposerFieldConfigDefinition } from 'graphql-compose';
+import { Options } from '../../definitions';
+import { getBullConnection } from '../../helpers/getBullConnection';
+
+export function createMemoryFC(
+  sc: SchemaComposer<any>,
+  opts: Options
+): ObjectTypeComposerFieldConfigDefinition<any, any> {
+  const { typePrefix } = opts;
+
+  return {
+    type: sc.createObjectTC({
+      name: `${typePrefix}InfoMemory`,
+      fields: {
+        used_memory: 'String',
+        used_memory_human: 'String',
+        used_memory_rss: 'String',
+        used_memory_rss_human: 'String',
+        used_memory_peak: 'String',
+        used_memory_peak_human: 'String',
+        used_memory_peak_perc: 'String',
+        used_memory_overhead: 'String',
+        used_memory_startup: 'String',
+        used_memory_dataset: 'String',
+        used_memory_dataset_perc: 'String',
+        allocator_allocated: 'String',
+        allocator_active: 'String',
+        allocator_resident: 'String',
+        total_system_memory: 'String',
+        total_system_memory_human: 'String',
+        used_memory_lua: 'String',
+        used_memory_lua_human: 'String',
+        used_memory_scripts: 'String',
+        used_memory_scripts_human: 'String',
+        number_of_cached_scripts: 'String',
+        maxmemory: 'String',
+        maxmemory_human: 'String',
+        maxmemory_policy: 'String',
+        allocator_frag_ratio: 'String',
+        allocator_frag_bytes: 'String',
+        allocator_rss_ratio: 'String',
+        allocator_rss_bytes: 'String',
+        rss_overhead_ratio: 'String',
+        rss_overhead_bytes: 'String',
+        mem_fragmentation_ratio: 'String',
+        mem_fragmentation_bytes: 'String',
+        mem_not_counted_for_evict: 'String',
+        mem_replication_backlog: 'String',
+        mem_clients_slaves: 'String',
+        mem_clients_normal: 'String',
+        mem_aof_buffer: 'String',
+        mem_allocator: 'String',
+        active_defrag_running: 'String',
+        lazyfree_pending_objects: 'String',
+      },
+    }),
+    resolve: async () => {
+      const redis = getBullConnection(opts);
+      const memoryInfo = await redis.info('memory');
+      return memoryInfo
+        .split('\r\n')
+        .filter((item) => item.includes(':'))
+        .map((item) => item.split(':'))
+        .reduce((acc, metric) => ({ ...acc, [metric[0]]: metric[1] }), {});
+    },
+  };
+}

src/types/info/Info.ts

@@ -0,0 +1,13 @@
+import { createMemoryFC } from './Info.memory';
+import { SchemaComposer } from 'graphql-compose';
+import { Options } from '../../definitions';
+
+export function getInfoTC(sc: SchemaComposer<any>, opts: Options) {
+  const { typePrefix } = opts;
+
+  return sc.getOrCreateOTC(`${typePrefix}Info`, (etc) => {
+    etc.addFields({
+      memory: createMemoryFC(sc, opts),
+    });
+  });
+}