@Before
public void setup() throws Exception {
zkServer = new TestingServer(2222);
zkClient =
CuratorFrameworkFactory.newClient(
zkServer.getConnectString(), new ExponentialBackoffRetry(300, 5));
dragonZK = new DragonZooKeeper(zkClient, "/dragon");
}
private ZkTestSystem() {
LOG.info("~~~~~~~~~~~~~~~ starting zk system ~~~~~~~~~~~~~~~");
try {
_zkServer = new TestingServer(PORT);
_zkClient = ZkTestSystem.createZkClient(_zkServer.getConnectString());
} catch (Exception e) {
throw new RuntimeException(e);
}
LOG.info("~~~~~~~~~~~~~~~ zk system started ~~~~~~~~~~~~~~~");
}
@Before
public void setUp() throws Exception {
server = new TestingServer();
String connectionString = server.getConnectString();
Map conf = new HashMap();
conf.put(Config.STORM_ZOOKEEPER_SESSION_TIMEOUT, 1000);
conf.put(Config.STORM_ZOOKEEPER_RETRY_TIMES, 4);
conf.put(Config.STORM_ZOOKEEPER_RETRY_INTERVAL, 5);
ExponentialBackoffRetry retryPolicy = new ExponentialBackoffRetry(1000, 3);
zookeeper = CuratorFrameworkFactory.newClient(connectionString, retryPolicy);
dynamicBrokersReader = new DynamicBrokersReader(conf, connectionString, masterPath, topic);
zookeeper.start();
}
public KafkaTestBroker() {
try {
server = new TestingServer();
String zookeeperConnectionString = server.getConnectString();
ExponentialBackoffRetry retryPolicy = new ExponentialBackoffRetry(1000, 3);
zookeeper = CuratorFrameworkFactory.newClient(zookeeperConnectionString, retryPolicy);
zookeeper.start();
port = InstanceSpec.getRandomPort();
kafka.server.KafkaConfig config = buildKafkaConfig(zookeeperConnectionString);
kafka = new KafkaServerStartable(config);
kafka.startup();
} catch (Exception ex) {
throw new RuntimeException("Could not start test broker", ex);
}
}
