-
Notifications
You must be signed in to change notification settings - Fork 119
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add automatic connection to new (previously unknown) cluster nodes
Redirections to existing nodes are handled already.
- Loading branch information
Showing
3 changed files
with
201 additions
and
32 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
139 changes: 139 additions & 0 deletions
139
src/test/java/io/vertx/redis/client/test/RedisClusterMovedToNewNodeTest.java
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,139 @@ | ||
| package io.vertx.redis.client.test; | ||
|
|
||
| import io.vertx.core.Future; | ||
| import io.vertx.core.internal.logging.Logger; | ||
| import io.vertx.core.internal.logging.LoggerFactory; | ||
| import io.vertx.core.net.SocketAddress; | ||
| import io.vertx.ext.unit.Async; | ||
| import io.vertx.ext.unit.TestContext; | ||
| import io.vertx.ext.unit.junit.RunTestOnContext; | ||
| import io.vertx.ext.unit.junit.VertxUnitRunner; | ||
| import io.vertx.redis.client.Command; | ||
| import io.vertx.redis.client.Redis; | ||
| import io.vertx.redis.client.RedisClientType; | ||
| import io.vertx.redis.client.RedisConnection; | ||
| import io.vertx.redis.client.RedisOptions; | ||
| import io.vertx.redis.client.RedisReplicas; | ||
| import io.vertx.redis.client.Request; | ||
| import io.vertx.redis.client.impl.PooledRedisConnection; | ||
| import io.vertx.redis.containers.RedisCluster; | ||
| import org.junit.After; | ||
| import org.junit.Before; | ||
| import org.junit.ClassRule; | ||
| import org.junit.Rule; | ||
| import org.junit.Test; | ||
| import org.junit.runner.RunWith; | ||
|
|
||
| import java.util.concurrent.TimeUnit; | ||
|
|
||
| @RunWith(VertxUnitRunner.class) | ||
| public class RedisClusterMovedToNewNodeTest { | ||
| @ClassRule | ||
| public static final RedisCluster redis = new RedisCluster(); | ||
|
|
||
| @Rule | ||
| public final RunTestOnContext rule = new RunTestOnContext(); | ||
|
|
||
| private final RedisOptions options = new RedisOptions() | ||
| .setType(RedisClientType.CLUSTER) | ||
| .setUseReplicas(RedisReplicas.NEVER) | ||
| .addConnectionString(redis.getRedisNode0Uri()) | ||
| .addConnectionString(redis.getRedisNode1Uri()) | ||
| .addConnectionString(redis.getRedisNode2Uri()) | ||
| .addConnectionString(redis.getRedisNode3Uri()) | ||
| .addConnectionString(redis.getRedisNode4Uri()) | ||
| .addConnectionString(redis.getRedisNode5Uri()); | ||
|
|
||
| private Redis client; | ||
| private ClusterUtils cluster; | ||
|
|
||
| @Before | ||
| public void createClient() { | ||
| client = Redis.createClient(rule.vertx(), options); | ||
| cluster = new ClusterUtils(rule.vertx(), client); | ||
| } | ||
|
|
||
| @After | ||
| public void cleanRedis() { | ||
| client.close(); | ||
| } | ||
|
|
||
| @Test | ||
| public void test(TestContext test) { | ||
| Async async = test.async(); | ||
|
|
||
| // slot number: 16287 | ||
| // keys hashing to the slot: x, exs | ||
| int slot = 16287; | ||
| String key1 = "x"; | ||
| String key2 = "exs"; | ||
|
|
||
| client.connect().compose(clusterConn -> { | ||
| return cluster.connectToMasterThatServesSlot(slot).compose(masterResult -> { | ||
| Redis master = masterResult.redis; | ||
| RedisConnection masterConn = masterResult.conn; | ||
| String masterId = masterResult.id; | ||
| return addNewMaster().compose(newMasterResult -> { | ||
| Redis newMaster = newMasterResult.redis; | ||
| RedisConnection newMasterConn = newMasterResult.conn; | ||
| String newMasterId = newMasterResult.id; | ||
| return clusterConn.send(Request.cmd(Command.SET).arg(key1).arg("fubar")) | ||
| .compose(ignored -> { | ||
| return clusterConn.send(Request.cmd(Command.SET).arg(key2).arg("quux")); | ||
| }) | ||
| .compose(ignored -> { | ||
| return newMasterConn.send(Request.cmd(Command.CLUSTER).arg("SETSLOT").arg(slot).arg("IMPORTING").arg(masterId)); | ||
| }) | ||
| .compose(ignored -> { | ||
| return masterConn.send(Request.cmd(Command.CLUSTER).arg("SETSLOT").arg(slot).arg("MIGRATING").arg(newMasterId)); | ||
| }).compose(ignored -> { | ||
| SocketAddress newMasterAddr = ((PooledRedisConnection) newMasterConn).actual().uri().socketAddress(); | ||
| return masterConn.send(Request.cmd(Command.MIGRATE).arg(newMasterAddr.host()).arg(newMasterAddr.port()) | ||
| .arg("").arg(0).arg(5000).arg("KEYS").arg(key1).arg(key2)); | ||
| }) | ||
| .compose(ignored -> { | ||
| return masterConn.send(Request.cmd(Command.CLUSTER).arg("SETSLOT").arg(slot).arg("NODE").arg(newMasterId)); | ||
| }) | ||
| .compose(ignored -> { | ||
| return newMasterConn.send(Request.cmd(Command.CLUSTER).arg("SETSLOT").arg(slot).arg("NODE").arg(newMasterId)); | ||
| }) | ||
| .compose(ignored -> { | ||
| return clusterConn.send(Request.cmd(Command.GET).arg(key1)); | ||
| }) | ||
| .compose(result -> { | ||
| test.assertEquals("fubar", result.toString()); | ||
| return clusterConn.send(Request.cmd(Command.GET).arg(key2)); | ||
| }) | ||
| .compose(result -> { | ||
| test.assertEquals("quux", result.toString()); | ||
| master.close(); | ||
| newMaster.close(); | ||
| return Future.succeededFuture(); | ||
| }); | ||
| }); | ||
| }); | ||
| }).onComplete(test.asyncAssertSuccess(ignored -> { | ||
| async.complete(); | ||
| })); | ||
| } | ||
|
|
||
| private Future<ClusterUtils.Result> addNewMaster() { | ||
| return rule.vertx() | ||
| .executeBlocking(() -> { | ||
| redis.addMaster(7006); | ||
| return null; | ||
| }) | ||
| .compose(ignored -> { | ||
| Redis client = Redis.createClient(rule.vertx(), "redis://127.0.0.1:7006"); | ||
| return client | ||
| .connect() | ||
| .compose(conn -> { | ||
| return conn.send(Request.cmd(Command.CLUSTER).arg("MYID")) | ||
| .map(resp -> { | ||
| String newId = resp.toString(); | ||
| return new ClusterUtils.Result(client, conn, newId); | ||
| }); | ||
| }); | ||
| }); | ||
| } | ||
| } |