将Zookeeper集成到你的应用中
jopen
10年前
在应用中使用zookeeper大多数情况都是使用zookeeper提供的客户端，然后我们在生产环境还需要部署zookeeper的服务端，本文写了一个启动类，用于在程序中启动zookeeper服务。
首先，我们下载zookeeper-3.3.6.tar.gz，最好不要最新版本，新版本在jdk1.5下有问题。解压后在
zookeeper-3.3.6/contrib/fatjar目录下有一个zookeeper-3.3.6-fatjar.jar文件，我们用这个jar来写。
import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.lang.reflect.Method; import java.util.Properties; public class ZKServerStart { /** * 启动zookeeper服务 * @param zoocfg zoo.cfg文件的物理路径 * @param zooDataDir zookeeper的data路径 * @throws Exception */ public static void Run(String zoocfg, String zooDataDir) throws Exception { //加载zoocfg配置文件 Properties prop = loadProperties(zoocfg); //提取本机服务的server编号，这个my.id是默认的zoo.cfg里没有的，需要我们后加上， //它的值就是当前节点的serverNum String serverNum = prop.getProperty("my.id"); //提取zookeeper的客户端IP和端口，把IP和端口提取出来，方便我们的客户端API使用 Global.zkClientIp = prop.getProperty("server." + serverNum).split(":")[0]; Global.zkClientPort = Integer.parseInt(prop.getProperty("clientPort")); //myid文件的路径 String dataDir = zooDataDir + "/ZooData"; //写入myid文件 writeMyid(dataDir, serverNum); prop.setProperty("dataDir", dataDir); //将dataDir保存到zoo.cfg saveConfig(prop, zoocfg); String[] config = {zoocfg}; Class<?> clazz = Class.forName("org.apache.zookeeper.server.quorum.QuorumPeerMain"); Method main = clazz.getMethod("main", String[].class); //启动zookeeper main.invoke(null, (Object)config); } /* * 保存zookeeper的配置文件 */ private static void saveConfig(Properties prop, String configFile) throws IOException { OutputStream out = new FileOutputStream(configFile); try{ prop.store(out, null); } finally { if(out != null) out.close(); } } /* * 将server的编号写入myid文件 */ private static void writeMyid(String dataDir, String serverNum) throws IOException, FileNotFoundException { File dir = new File(dataDir); if(!dir.exists()) dir.mkdirs(); File myid = new File(dataDir + "/myid"); if(!myid.exists()) myid.createNewFile(); OutputStream out = new FileOutputStream(myid); try{ out.write(serverNum.getBytes()); } finally { if(out != null) out.close(); } } /* * 加载zoocfg配置 */ private static Properties loadProperties(String zoocfg) throws FileNotFoundException, IOException { Properties prop = new Properties(); InputStream is = new FileInputStream(zoocfg); try{ prop.load(is); } finally { if(is != null) is.close(); } return prop; } } 注意，使用这个启动类来启动zookeeper的时候要放到线程中。例如，我们在Servlet的init()方法中启动：
import java.io.UnsupportedEncodingException; import java.net.URL; import java.net.URLDecoder; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; public class ZooServlet extends HttpServlet{ @Override public void init() throws ServletException { String zooConfig = this.getInitParameter("ZooConfig"); //找到WEB-INF的物理路径 final String webInfo = getWebInfPath(); //找到zoo.cfg的物理路径 final String configPath = webInfo + zooConfig.substring(1, zooConfig.length()); new Thread(new Runnable(){ public void run() { //启动zookeeper服务 try { ZKServerStart.Run(configPath, webInfo); } catch (Exception e) { e.printStackTrace(); } }}).start(); super.init(); } private String getWebInfPath() { URL url = this.getClass().getProtectionDomain().getCodeSource().getLocation(); String path = url.toString(); int index = path.indexOf("WEB-INF"); if (index == -1) { index = path.indexOf("classes"); } if (index == -1) { index = path.indexOf("bin"); } path = path.substring(0, index); if (path.startsWith("zip")) {// 当class文件在war中时，此时返回zip:D:/...这样的路径 path = path.substring(4); } else if (path.startsWith("file")) {// 当class文件在class文件中时，此时返回file:/D:/...这样的路径 path = path.substring(6); } else if (path.startsWith("jar")) {// 当class文件在jar文件里面时，此时返回jar:file:/D:/...这样的路径 path = path.substring(10); } try { path = URLDecoder.decode(path, "UTF-8"); } catch (UnsupportedEncodingException e) { e.printStackTrace(); } return path; } } web.xml的配置：
<?xml version="1.0" encoding="UTF-8"?> <web-app version="2.4" xmlns="http://java.sun.com/xml/ns/j2ee" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://java.sun.com/xml/ns/j2ee http://java.sun.com/xml/ns/j2ee/web-app_2_4.xsd"> <servlet> <servlet-name>ZooServlet</servlet-name> <servlet-class>ZooServlet</servlet-class> <init-param> <param-name>ZooConfig</param-name> <param-value>/WEB-INF/conf/zoo.cfg</param-value> </init-param> <load-on-startup>1</load-on-startup> </servlet> <servlet-mapping> <servlet-name>ZooServlet</servlet-name> <url-pattern>*.do</url-pattern> </servlet-mapping> <welcome-file-list> <welcome-file>index.jsp</welcome-file> </welcome-file-list> </web-app>
这样，Servlet在初始化的时候就启动了zookeeper，同时将zookeeper的dataDir目录设置到WEB-INF/ZooData/下。同时我们还提取了zookeeper的当前节点的IP和客户端端口，方便在调用客户端API的地方使用。
最后看一下zoo.cfg配置
tickTime=2000 initLimit=10 syncLimit=5 dataDir= clientPort=2181 server.1=192.168.1.1:2888:3888 server.2=192.168.1.2:2888:3888 server.3=192.168.1.3:2888:3888 my.id=1
这个my.id是后加的一个属性，用于记录当前节点的server编号，方便我们写入到myid文件中。
来自：http://my.oschina.net/shenxueliang/blog/228996