-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(cluster): support scaling reads to slaves
The new option scaleReads is used to specify where to send the reads.
Add two new events:
1. "+node": a new node is discovered.
2. "-node": a node is disconnected.
BREAKING CHANGE:
1. Cluster#masterNodes and Cluster#nodes is removed. Use Cluster#nodes('masters') and Cluster#nodes('all') instead.
2. Cluster#to() is removed. Use
Promise.all(Cluster#nodes().map(function (node) {})) instead.
Closes #170.- Loading branch information
Showing
7 changed files
with
333 additions
and
395 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,89 @@ | ||
| 'use strict'; | ||
|
|
||
| var util = require('util'); | ||
| var EventEmitter = require('events').EventEmitter; | ||
| var _ = require('lodash'); | ||
| var Redis = require('../redis'); | ||
|
|
||
| function ConnectionPool(redisOptions) { | ||
| EventEmitter.call(this); | ||
| this.redisOptions = redisOptions; | ||
|
|
||
| // this.masters + this.slaves = this.nodes | ||
| this.nodes = {}; | ||
| this.masters = {}; | ||
| this.slaves = {}; | ||
|
|
||
| this.specifiedOptions = {}; | ||
| } | ||
|
|
||
| util.inherits(ConnectionPool, EventEmitter); | ||
|
|
||
| ConnectionPool.prototype.findOrCreate = function (node, readOnly) { | ||
| node.port = node.port || 6379; | ||
| node.host = node.host || '127.0.0.1'; | ||
| node.key = node.key || node.host + ':' + node.port; | ||
|
|
||
| if (this.specifiedOptions[node.key]) { | ||
| _.assign(node, this.specifiedOptions[node.key]); | ||
| } else { | ||
| this.specifiedOptions[node.key] = node; | ||
| } | ||
|
|
||
| if (this.nodes[node.key] && this.nodes[node.key].options.readOnly !== readOnly) { | ||
| this.remove(node.key); | ||
| } | ||
|
|
||
| if (!this.nodes[node.key]) { | ||
| var redis = this.nodes[node.key] = new Redis(_.defaults({ | ||
| retryStrategy: null, | ||
| enableOfflineQueue: true, | ||
| readOnly: readOnly | ||
| }, node, this.redisOptions, { lazyConnect: true })); | ||
| this[readOnly ? 'slaves' : 'masters'][node.key] = redis; | ||
|
|
||
| var _this = this; | ||
| redis.once('end', function () { | ||
| delete _this.nodes[node.key]; | ||
| delete _this.masters[node.key]; | ||
| delete _this.slaves[node.key]; | ||
| _this.emit('-node', redis); | ||
| if (!Object.keys(_this.nodes).length) { | ||
| _this.emit('drain'); | ||
| } | ||
| }); | ||
|
|
||
| this.emit('+node', redis); | ||
| } | ||
|
|
||
| return this.nodes[node.key]; | ||
| }; | ||
|
|
||
| ConnectionPool.prototype.remove = function (key) { | ||
| if (this.nodes[key]) { | ||
| this.nodes[key].disconnect(); | ||
| delete this.nodes[key]; | ||
| delete this.masters[key]; | ||
| delete this.slaves[key]; | ||
| } | ||
| }; | ||
|
|
||
| ConnectionPool.prototype.reset = function (nodes) { | ||
| var newNodes = {}; | ||
| for (var i = 0; i < nodes.length; i++) { | ||
| var node = nodes[i]; | ||
| node.key = node.host + ':' + node.port; | ||
| newNodes[node.key] = node; | ||
| } | ||
| var _this = this; | ||
| Object.keys(this.nodes).forEach(function (key) { | ||
| if (!newNodes[key]) { | ||
| _this.remove(key); | ||
| } | ||
| }); | ||
| Object.keys(newNodes).forEach(function (key) { | ||
| _this.findOrCreate(newNodes[key], newNodes[key].readOnly); | ||
| }); | ||
| }; | ||
|
|
||
| module.exports = ConnectionPool; |
Oops, something went wrong.