From ecc91fb09c65fdbf617628110ad4cb456d7c6f70 Mon Sep 17 00:00:00 2001 From: rfeng Date: Tue, 10 Nov 2009 22:57:41 +0000 Subject: Add a standalone server and test case git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@834701 13f79535-47bb-0310-9956-ffa450edef68 --- .../endpoint/zookeeper/LocalZooKeeperServer.java | 148 +++++++++++++++++++++ .../endpoint/zookeeper/DistributedMapTestCase.java | 45 +++++++ .../zookeeper/LocalZooKeeperServerTestCase.java | 85 ++++++++++++ 3 files changed, 278 insertions(+) create mode 100644 java/sca/modules/endpoint-zookeeper/src/main/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServer.java create mode 100644 java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/DistributedMapTestCase.java create mode 100644 java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServerTestCase.java (limited to 'java') diff --git a/java/sca/modules/endpoint-zookeeper/src/main/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServer.java b/java/sca/modules/endpoint-zookeeper/src/main/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServer.java new file mode 100644 index 0000000000..49bb46f983 --- /dev/null +++ b/java/sca/modules/endpoint-zookeeper/src/main/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServer.java @@ -0,0 +1,148 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.endpoint.zookeeper; + +import java.io.File; +import java.io.IOException; +import java.util.logging.Level; +import java.util.logging.Logger; + +import org.apache.zookeeper.server.NIOServerCnxn; +import org.apache.zookeeper.server.ServerConfig; +import org.apache.zookeeper.server.ZooKeeperServer; +import org.apache.zookeeper.server.persistence.FileTxnSnapLog; +import org.apache.zookeeper.server.quorum.QuorumPeerConfig.ConfigException; + +/** + * This class starts and runs a standalone ZooKeeperServer. + */ +public class LocalZooKeeperServer { + private static final Logger logger = Logger.getLogger(LocalZooKeeperServer.class.getName()); + private static final String USAGE = "Usage: LocalZooKeeperServer configfile | port datadir [ticktime] [maxcnxns]"; + private NIOServerCnxn.Factory cnxnFactory; + private ZooKeeperServer zkServer; + + /** + * + */ + public LocalZooKeeperServer() { + super(); + } + + public Thread folk(final ServerConfig config) { + Thread thread = new Thread() { + public void run() { + try { + LocalZooKeeperServer.this.run(config); + } catch (IOException e) { + throw new IllegalStateException(e); + } + } + }; + thread.start(); + return thread; + } + + public Thread folk(final String[] args) { + Thread thread = new Thread() { + public void run() { + try { + LocalZooKeeperServer.this.run(args); + } catch (Exception e) { + throw new IllegalStateException(e); + } + } + }; + thread.start(); + return thread; + } + + /** + * Run from a ServerConfig. + * @param config ServerConfig to use. + * @throws IOException + */ + public void run(ServerConfig config) throws IOException { + logger.info("Starting ZooKeeper server"); + try { + // Note that this thread isn't going to be doing anything else, + // so rather than spawning another thread, we will just call + // run() in this thread. + // create a file logger url from the command line args + zkServer = new ZooKeeperServer(); + + FileTxnSnapLog ftxn = new FileTxnSnapLog(new File(config.getDataLogDir()), new File(config.getDataDir())); + zkServer.setTxnLogFactory(ftxn); + zkServer.setTickTime(config.getTickTime()); + cnxnFactory = new NIOServerCnxn.Factory(config.getClientPort(), config.getMaxClientCnxns()); + cnxnFactory.startup(zkServer); + cnxnFactory.join(); + if (zkServer.isRunning()) { + zkServer.shutdown(); + } + } catch (InterruptedException e) { + // warn, but generally this is ok + logger.log(Level.WARNING, "Server interrupted", e); + } + } + + public void run(String[] args) throws ConfigException, IOException { + ServerConfig config = new ServerConfig(); + if (args.length == 1) { + config.parse(args[0]); + } else { + config.parse(args); + } + run(config); + } + + /** + * Shutdown the serving instance + */ + public void shutdown() { + cnxnFactory.shutdown(); + } + + /* + * Start up the ZooKeeper server. + * + * @param args the configfile or the port datadir [ticktime] + */ + public static void main(String[] args) { + LocalZooKeeperServer main = new LocalZooKeeperServer(); + try { + main.run(args); + } catch (IllegalArgumentException e) { + logger.log(Level.SEVERE, "Invalid arguments, exiting abnormally", e); + logger.info(USAGE); + System.err.println(USAGE); + System.exit(2); + } catch (ConfigException e) { + logger.log(Level.SEVERE, "Invalid config, exiting abnormally", e); + System.err.println("Invalid config, exiting abnormally"); + System.exit(2); + } catch (Exception e) { + logger.log(Level.SEVERE, "Unexpected exception, exiting abnormally", e); + System.exit(1); + } + logger.info("Exiting normally"); + System.exit(0); + } +} diff --git a/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/DistributedMapTestCase.java b/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/DistributedMapTestCase.java new file mode 100644 index 0000000000..5489f20098 --- /dev/null +++ b/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/DistributedMapTestCase.java @@ -0,0 +1,45 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.endpoint.zookeeper; + + +import org.junit.AfterClass; +import org.junit.BeforeClass; + +/** + * + */ +public class DistributedMapTestCase { + + /** + * @throws java.lang.Exception + */ + @BeforeClass + public static void setUpBeforeClass() throws Exception { + } + + /** + * @throws java.lang.Exception + */ + @AfterClass + public static void tearDownAfterClass() throws Exception { + } + +} diff --git a/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServerTestCase.java b/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServerTestCase.java new file mode 100644 index 0000000000..51a4d7455e --- /dev/null +++ b/java/sca/modules/endpoint-zookeeper/src/test/java/org/apache/tuscany/sca/endpoint/zookeeper/LocalZooKeeperServerTestCase.java @@ -0,0 +1,85 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.endpoint.zookeeper; + +import java.io.File; + +import org.apache.zookeeper.CreateMode; +import org.apache.zookeeper.WatchedEvent; +import org.apache.zookeeper.Watcher; +import org.apache.zookeeper.ZooKeeper; +import org.apache.zookeeper.Watcher.Event.KeeperState; +import org.apache.zookeeper.ZooDefs.Ids; +import org.apache.zookeeper.server.PurgeTxnLog; +import org.junit.AfterClass; +import org.junit.Assert; +import org.junit.BeforeClass; +import org.junit.Test; + +/** + * + */ +public class LocalZooKeeperServerTestCase implements Watcher { + private static LocalZooKeeperServer server; + + /** + * @throws java.lang.Exception + */ + @BeforeClass + public static void setUpBeforeClass() throws Exception { + server = new LocalZooKeeperServer(); + } + + @Test + public void testServer() throws Exception { + String[] args = new String[] {"8085", "target/zookeeper"}; + Thread thread = server.folk(args); + ZooKeeper client = new ZooKeeper("localhost:8085", 500, this); + synchronized (this) { + wait(10000); + } + client.create("/test", "123".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL); + byte[] data = client.getData("/test", false, null); + Assert.assertEquals("123", new String(data)); + client.close(); + } + + /** + * @throws java.lang.Exception + */ + @AfterClass + public static void tearDownAfterClass() throws Exception { + if (server != null) { + server.shutdown(); + File dir = new File("target/zookeeper"); + PurgeTxnLog.purge(dir, dir, 3); + } + } + + public void process(WatchedEvent event) { + System.out.println(event); + if (event.getPath() == null && event.getState() == KeeperState.SyncConnected) { + synchronized (this) { + notifyAll(); + } + } + } + +} -- cgit v1.2.3