Skip to content

Commit 5ce5f6e

Browse files
author
Arun
committed
fixing test cases
1 parent 85d8c35 commit 5ce5f6e

File tree

3 files changed

+215
-191
lines changed

3 files changed

+215
-191
lines changed

index.js

+7-7
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ var redis = require('redis').createClient;
88
var msgpack = require('notepack.io');
99
var Adapter = require('socket.io-adapter');
1010
var debug = require('debug')('socket.io-redis');
11-
var MersenneTwister = require('mersennetwister');
1211
var crypto = require("crypto")
1312

1413
/**
@@ -86,12 +85,11 @@ function adapter(uri, opts) {
8685
if (!pubs) pubs = [createClient()];
8786
if (!subs) subs = [createClient()];
8887

88+
var pubCount = 0
89+
8990
// this server's key
9091
var uid = uid2(6);
9192

92-
var randNumberSeed = parseInt(crypto.randomBytes(4).toString('hex'), 16);
93-
var mt = new MersenneTwister(randNumberSeed);
94-
9593
function subClientFunction (clients) {
9694
var custom = {}
9795
var commonFunctions = ['quit', 'on', 'subscribe', 'psubscribe']
@@ -110,7 +108,8 @@ function adapter(uri, opts) {
110108
var customFunctions = ['publish']
111109
customFunctions.map(fn => {
112110
custom[fn] = function (...args) {
113-
var randomNumber = Math.floor(mt.rndHiRes() * clients.length);
111+
var randomNumber = pubCount % clients.length;
112+
pubCount = pubCount > 10000000 ? 0 : (pubCount + 1)
114113
return pubs[randomNumber][fn](...args)
115114
};
116115
})
@@ -179,10 +178,11 @@ function adapter(uri, opts) {
179178
pub.on('error', onError);
180179
});
181180

182-
this.mt = mt;
181+
this.pubCount = 0
183182

184183
this.getPub = function () {
185-
var randomNumber = Math.floor(this.mt.rndHiRes() * pubs.length);
184+
var randomNumber = this.pubCount % pubs.length;
185+
this.pubCount = this.pubCount > 10000000 ? 0 : (this.pubCount + 1)
186186
debug(`publishing to pub ${randomNumber}`)
187187
return pubs[randomNumber]
188188
}

0 commit comments

Comments
 (0)