support redis cluster, redis sentinel

nest-cloud · Jun 11, 2019 · 6958380 · 6958380
1 parent db301d7
commit 6958380
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 13 deletions.
diff --git a/packages/redis/interfaces/redis-options.interface.ts b/packages/redis/interfaces/redis-options.interface.ts
@@ -1,6 +1,11 @@
-import { RedisOptions } from 'ioredis';
+import { ClusterNode, ClusterOptions, RedisOptions } from 'ioredis';
 
-export interface RedisModuleOptions extends RedisOptions {
+export interface RedisModuleOptions {
     name?: string;
     dependencies?: string[];
+    cluster: {
+        nodes: ClusterNode[];
+        options?: ClusterOptions;
+    };
+    redisOptions: RedisOptions;
 }
diff --git a/packages/redis/redis.module.ts b/packages/redis/redis.module.ts
@@ -1,9 +1,8 @@
 import { DynamicModule, Global, Module } from '@nestjs/common';
-import { createClient, createOptions } from "./redis.provider";
-import { RedisModuleOptions } from "./interfaces/redis-options.interface";
+import { create, createOptions } from './redis.provider';
+import { RedisModuleOptions } from './interfaces/redis-options.interface';
 import { NEST_BOOT_PROVIDER, NEST_BOOT, NEST_CONSUL_CONFIG_PROVIDER, NEST_CONSUL_CONFIG } from '@nestcloud/common';
 
-
 @Global()
 @Module({})
 export class RedisModule {
@@ -20,7 +19,7 @@ export class RedisModule {
             }
         }
 
-        const redisProvider = createClient(options.name, inject);
+        const redisProvider = create(options.name, inject);
         const optionsProvider = createOptions(options.name, inject, options);
 
         return {

diff --git a/packages/redis/redis.provider.ts b/packages/redis/redis.provider.ts
@@ -1,22 +1,38 @@
 import * as Redis from 'ioredis';
+import { ClusterNode, ClusterOptions, RedisOptions } from 'ioredis';
 import { IBoot, IConsulConfig } from '@nestcloud/common';
 
-import { REDIS_CLIENT, REDIS_MODULE_OPTIONS } from "./constants";
-import { RedisModuleOptions } from "./interfaces/redis-options.interface";
+import { REDIS_CLIENT, REDIS_MODULE_OPTIONS } from './constants';
+import { RedisModuleOptions } from './interfaces/redis-options.interface';
 
-export const createClient = (name: string = 'default', inject?: string[]) => ({
+export const create = (name: string = 'default', inject?: string[]) => ({
     provide: REDIS_CLIENT + name,
-    useFactory: (options: RedisModuleOptions, config: IBoot | IConsulConfig): Redis.Redis => {
+    useFactory: (options: RedisModuleOptions, config: IBoot | IConsulConfig): Redis.Redis | Redis.Cluster => {
         if (config) {
-            options = (config as IBoot).get<RedisModuleOptions>(`redis.${ name }`, options);
+            options = (config as IBoot).get<RedisModuleOptions>(`redis.${name}`, options);
+        }
+        if (options.cluster) {
+            const nodes: ClusterNode[] = options.cluster.nodes;
+            const clusterOptions: ClusterOptions = options.cluster.options;
+            return createCluster(nodes, clusterOptions);
+        } else {
+            const redisOptions = options.redisOptions;
+            return createClient(redisOptions);
         }
-        return new Redis(options);
     },
     inject: [REDIS_MODULE_OPTIONS + name, ...inject],
 });
 
 export const createOptions = (name: string = 'default', inject?: string[], options?: RedisModuleOptions) => ({
     provide: REDIS_MODULE_OPTIONS + name,
     useValue: options,
-    inject
+    inject,
 });
+
+const createClient = (options: RedisOptions): Redis.Redis => {
+    return new Redis(options);
+};
+
+const createCluster = (nodes: ClusterNode[], options: ClusterOptions): Redis.Cluster => {
+    return new Redis.Cluster(nodes, options);
+};