是否可以在进程中启动zookeeper服务器实例,比如单元测试？

Question

调用org.apache.zookeeper.server.quorum.QuorumPeerMain.main()不起作用.

Answer 1

Netfix opensourced 策展人使用Zookeeper的框架更加方便.它内置了测试服务器类.只需将此测试依赖项添加到您的项目描述符,无论是maven,gradle还是:

org.apache.curator:curator-framework:4.0.1
org.apache.curator:curator-test:4.0.1

以下是测试要点.

TestingServer zkTestServer;
CuratorFramework cli;

@Before
public void startZookeeper() throws Exception {
    zkTestServer = new TestingServer(2181);
    cli = CuratorFrameworkFactory.newClient(zkTestServer.getConnectString(), new RetryOneTime(2000));
    cli.start();
}

@After
public void stopZookeeper() throws IOException {
    cli.close();
    zkTestServer.stop();
}

随着cli创建任何测试数据是很容易的(需要curator-framework依赖).

cli.create()
   .creatingParentsIfNeeded()
   .forPath("/a1", "testvalue".getBytes("UTF-8"));

Answer 2

要开始ZooKeeper你必须执行ZooKeeperServerMain课程.

您可以使用以下代码以ZooKeeper嵌入模式启动.

Properties startupProperties = ...

QuorumPeerConfig quorumConfiguration = new QuorumPeerConfig();
try {
    quorumConfiguration.parseProperties(startupProperties);
} catch(Exception e) {
    throw new RuntimeException(e);
}

zooKeeperServer = new ZooKeeperServerMain();
final ServerConfig configuration = new ServerConfig();
configuration.readFrom(quorumConfiguration);

new Thread() {
    public void run() {
        try {
            zooKeeperServer.runFromConfig(configuration);
        } catch (IOException e) {
            log.error("ZooKeeper Failed", e);
        }
    }
}.start();

Answer 3

你可以使用这样的东西.

int clientPort = 21818; // none-standard
int numConnections = 5000;
int tickTime = 2000;
String dataDirectory = System.getProperty("java.io.tmpdir");

File dir = new File(dataDirectory, "zookeeper").getAbsoluteFile();

ZooKeeperServer server = new ZooKeeperServer(dir, dir, tickTime);
NIOServerCnxn.Factory standaloneServerFactory = new NIOServerCnxn.Factory(new InetSocketAddress(clientPort), numConnections);

standaloneServerFactory.startup(server); // start the server.

并关闭它只是打电话 standaloneServerFactory.shutdown()