├── .gitignore ├── LICENSE ├── README.md ├── pom.xml └── src └── main ├── java └── com │ └── brainlounge │ └── zooterrain │ ├── netty │ ├── OutboundConnector.java │ ├── WebSocketServer.java │ ├── WebSocketServerInboundHandler.java │ ├── WebSocketServerIndexPage.java │ ├── WebSocketServerInitializer.java │ └── WebSocketServerJSPage.java │ └── zkclient │ ├── ClientMessage.java │ ├── ClientRequest.java │ ├── ControlMessage.java │ ├── DataMessage.java │ ├── ZNodeMessage.java │ ├── ZNodeState.java │ ├── ZkStateListener.java │ └── ZkStateObserver.java └── resources ├── hexdump.js └── index.html /.gitignore: -------------------------------------------------------------------------------- 1 | *.class 2 | 3 | # Package Files # 4 | *.jar 5 | *.war 6 | *.ear 7 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, 10 | and distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by 13 | the copyright owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all 16 | other entities that control, are controlled by, or are under common 17 | control with that entity. For the purposes of this definition, 18 | "control" means (i) the power, direct or indirect, to cause the 19 | direction or management of such entity, whether by contract or 20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the 21 | outstanding shares, or (iii) beneficial ownership of such entity. 22 | 23 | "You" (or "Your") shall mean an individual or Legal Entity 24 | exercising permissions granted by this License. 25 | 26 | "Source" form shall mean the preferred form for making modifications, 27 | including but not limited to software source code, documentation 28 | source, and configuration files. 29 | 30 | "Object" form shall mean any form resulting from mechanical 31 | transformation or translation of a Source form, including but 32 | not limited to compiled object code, generated documentation, 33 | and conversions to other media types. 34 | 35 | "Work" shall mean the work of authorship, whether in Source or 36 | Object form, made available under the License, as indicated by a 37 | copyright notice that is included in or attached to the work 38 | (an example is provided in the Appendix below). 39 | 40 | "Derivative Works" shall mean any work, whether in Source or Object 41 | form, that is based on (or derived from) the Work and for which the 42 | editorial revisions, annotations, elaborations, or other modifications 43 | represent, as a whole, an original work of authorship. For the purposes 44 | of this License, Derivative Works shall not include works that remain 45 | separable from, or merely link (or bind by name) to the interfaces of, 46 | the Work and Derivative Works thereof. 47 | 48 | "Contribution" shall mean any work of authorship, including 49 | the original version of the Work and any modifications or additions 50 | to that Work or Derivative Works thereof, that is intentionally 51 | submitted to Licensor for inclusion in the Work by the copyright owner 52 | or by an individual or Legal Entity authorized to submit on behalf of 53 | the copyright owner. For the purposes of this definition, "submitted" 54 | means any form of electronic, verbal, or written communication sent 55 | to the Licensor or its representatives, including but not limited to 56 | communication on electronic mailing lists, source code control systems, 57 | and issue tracking systems that are managed by, or on behalf of, the 58 | Licensor for the purpose of discussing and improving the Work, but 59 | excluding communication that is conspicuously marked or otherwise 60 | designated in writing by the copyright owner as "Not a Contribution." 61 | 62 | "Contributor" shall mean Licensor and any individual or Legal Entity 63 | on behalf of whom a Contribution has been received by Licensor and 64 | subsequently incorporated within the Work. 65 | 66 | 2. Grant of Copyright License. Subject to the terms and conditions of 67 | this License, each Contributor hereby grants to You a perpetual, 68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 69 | copyright license to reproduce, prepare Derivative Works of, 70 | publicly display, publicly perform, sublicense, and distribute the 71 | Work and such Derivative Works in Source or Object form. 72 | 73 | 3. Grant of Patent License. Subject to the terms and conditions of 74 | this License, each Contributor hereby grants to You a perpetual, 75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 76 | (except as stated in this section) patent license to make, have made, 77 | use, offer to sell, sell, import, and otherwise transfer the Work, 78 | where such license applies only to those patent claims licensable 79 | by such Contributor that are necessarily infringed by their 80 | Contribution(s) alone or by combination of their Contribution(s) 81 | with the Work to which such Contribution(s) was submitted. If You 82 | institute patent litigation against any entity (including a 83 | cross-claim or counterclaim in a lawsuit) alleging that the Work 84 | or a Contribution incorporated within the Work constitutes direct 85 | or contributory patent infringement, then any patent licenses 86 | granted to You under this License for that Work shall terminate 87 | as of the date such litigation is filed. 88 | 89 | 4. Redistribution. You may reproduce and distribute copies of the 90 | Work or Derivative Works thereof in any medium, with or without 91 | modifications, and in Source or Object form, provided that You 92 | meet the following conditions: 93 | 94 | (a) You must give any other recipients of the Work or 95 | Derivative Works a copy of this License; and 96 | 97 | (b) You must cause any modified files to carry prominent notices 98 | stating that You changed the files; and 99 | 100 | (c) You must retain, in the Source form of any Derivative Works 101 | that You distribute, all copyright, patent, trademark, and 102 | attribution notices from the Source form of the Work, 103 | excluding those notices that do not pertain to any part of 104 | the Derivative Works; and 105 | 106 | (d) If the Work includes a "NOTICE" text file as part of its 107 | distribution, then any Derivative Works that You distribute must 108 | include a readable copy of the attribution notices contained 109 | within such NOTICE file, excluding those notices that do not 110 | pertain to any part of the Derivative Works, in at least one 111 | of the following places: within a NOTICE text file distributed 112 | as part of the Derivative Works; within the Source form or 113 | documentation, if provided along with the Derivative Works; or, 114 | within a display generated by the Derivative Works, if and 115 | wherever such third-party notices normally appear. The contents 116 | of the NOTICE file are for informational purposes only and 117 | do not modify the License. You may add Your own attribution 118 | notices within Derivative Works that You distribute, alongside 119 | or as an addendum to the NOTICE text from the Work, provided 120 | that such additional attribution notices cannot be construed 121 | as modifying the License. 122 | 123 | You may add Your own copyright statement to Your modifications and 124 | may provide additional or different license terms and conditions 125 | for use, reproduction, or distribution of Your modifications, or 126 | for any such Derivative Works as a whole, provided Your use, 127 | reproduction, and distribution of the Work otherwise complies with 128 | the conditions stated in this License. 129 | 130 | 5. Submission of Contributions. Unless You explicitly state otherwise, 131 | any Contribution intentionally submitted for inclusion in the Work 132 | by You to the Licensor shall be under the terms and conditions of 133 | this License, without any additional terms or conditions. 134 | Notwithstanding the above, nothing herein shall supersede or modify 135 | the terms of any separate license agreement you may have executed 136 | with Licensor regarding such Contributions. 137 | 138 | 6. Trademarks. This License does not grant permission to use the trade 139 | names, trademarks, service marks, or product names of the Licensor, 140 | except as required for reasonable and customary use in describing the 141 | origin of the Work and reproducing the content of the NOTICE file. 142 | 143 | 7. Disclaimer of Warranty. Unless required by applicable law or 144 | agreed to in writing, Licensor provides the Work (and each 145 | Contributor provides its Contributions) on an "AS IS" BASIS, 146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 147 | implied, including, without limitation, any warranties or conditions 148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A 149 | PARTICULAR PURPOSE. You are solely responsible for determining the 150 | appropriateness of using or redistributing the Work and assume any 151 | risks associated with Your exercise of permissions under this License. 152 | 153 | 8. Limitation of Liability. In no event and under no legal theory, 154 | whether in tort (including negligence), contract, or otherwise, 155 | unless required by applicable law (such as deliberate and grossly 156 | negligent acts) or agreed to in writing, shall any Contributor be 157 | liable to You for damages, including any direct, indirect, special, 158 | incidental, or consequential damages of any character arising as a 159 | result of this License or out of the use or inability to use the 160 | Work (including but not limited to damages for loss of goodwill, 161 | work stoppage, computer failure or malfunction, or any and all 162 | other commercial damages or losses), even if such Contributor 163 | has been advised of the possibility of such damages. 164 | 165 | 9. Accepting Warranty or Additional Liability. While redistributing 166 | the Work or Derivative Works thereof, You may choose to offer, 167 | and charge a fee for, acceptance of support, warranty, indemnity, 168 | or other liability obligations and/or rights consistent with this 169 | License. However, in accepting such obligations, You may act only 170 | on Your own behalf and on Your sole responsibility, not on behalf 171 | of any other Contributor, and only if You agree to indemnify, 172 | defend, and hold each Contributor harmless for any liability 173 | incurred by, or claims asserted against, such Contributor by reason 174 | of your accepting any such warranty or additional liability. 175 | 176 | END OF TERMS AND CONDITIONS 177 | 178 | APPENDIX: How to apply the Apache License to your work. 179 | 180 | To apply the Apache License to your work, attach the following 181 | boilerplate notice, with the fields enclosed by brackets "{}" 182 | replaced with your own identifying information. (Don't include 183 | the brackets!) The text should be enclosed in the appropriate 184 | comment syntax for the file format. We also recommend that a 185 | file or class name and description of purpose be included on the 186 | same "printed page" as the copyright notice for easier 187 | identification within third-party archives. 188 | 189 | Copyright {yyyy} {name of copyright owner} 190 | 191 | Licensed under the Apache License, Version 2.0 (the "License"); 192 | you may not use this file except in compliance with the License. 193 | You may obtain a copy of the License at 194 | 195 | http://www.apache.org/licenses/LICENSE-2.0 196 | 197 | Unless required by applicable law or agreed to in writing, software 198 | distributed under the License is distributed on an "AS IS" BASIS, 199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 200 | See the License for the specific language governing permissions and 201 | limitations under the License. 202 | 203 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | zooterrain 2 | ========== 3 | 4 | View ZooKeeper znode tree in a browser 5 | 6 | zooterrain is a small server every WebSocket-capable browser is able to connect to. 7 | It will display all the znodes from a running ZooKeeper ensemble. 8 | As znodes are deleted or new ones are created, this will be visualized in the browser. 9 | 10 | ## Features 11 | 12 | 1. Display changes to the tree as they happen (new, removed nodes) 13 | 1. Display changes to the data of each node 14 | 1. Filter nodes by name 15 | 1. Ignore nodes (and their children) 16 | 1. Watch nodes 17 | 1. Display a node's attached data, also as a hex dump 18 | 19 | ## Building 20 | 21 | git clone the repository and run Maven 22 | 23 | mvn install 24 | 25 | ## Running 26 | 27 | In order for zooterrain to connect to a ZooKeeper, you need to know the address of one 28 | ZooKeeper instance you want to track, which typically looks much like this: 29 | 30 | myzookeeper.mycomp.com:2181 31 | 32 | and then start the application the by running 33 | 34 | java -cp target/zooterrain-full.jar -Dzooterrain.conn=${ZOOKEEPER_HOST_PORT} com.brainlounge.zooterrain.netty.WebSocketServer 8080 35 | 36 | Be sure to always replace ${ZOOKEEPER_HOST_PORT} with your ZooKeeper address. 37 | or simply: 38 | 39 | java -Dzooterrain.conn=${ZOOKEEPER_HOST_PORT} -jar target/zooterrain-full.jar 40 | 41 | ## Connecting from Browser 42 | 43 | Open a browser and navigate to http://localhost:8080/ 44 | You can run the server on other ports by changing the last parameter (here: 8080) in the java command line. 45 | 46 | -------------------------------------------------------------------------------- /pom.xml: -------------------------------------------------------------------------------- 1 | 2 | 15 | 17 | 4.0.0 18 | 19 | com.brainlounge.zooterrain 20 | zooterrain-app 21 | jar 22 | 0.0.1-SNAPSHOT 23 | 24 | 25 | 26 | Bernd Fondermann 27 | bernd@brainlounge.de 28 | 29 | 30 | 31 | 32 | 1.7.5 33 | 4.0.14.Beta1 34 | 35 | 36 | 37 | 38 | 39 | 40 | 41 | io.netty 42 | netty-all 43 | ${netty.version} 44 | 45 | 46 | 47 | org.apache.zookeeper 48 | zookeeper 49 | 3.4.5 50 | 51 | 52 | com.sun.jmx 53 | jmxri 54 | 55 | 56 | com.sun.jdmk 57 | jmxtools 58 | 59 | 60 | javax.jms 61 | jms 62 | 63 | 64 | 65 | 66 | 67 | com.google.guava 68 | guava 69 | 13.0.1 70 | 71 | 72 | 73 | commons-io 74 | commons-io 75 | 2.0.1 76 | 77 | 78 | commons-codec 79 | commons-codec 80 | 1.4 81 | 82 | 83 | org.apache.commons 84 | commons-lang3 85 | 3.0.1 86 | 87 | 88 | 89 | com.fasterxml.jackson.core 90 | jackson-core 91 | 2.3.0 92 | 93 | 94 | 95 | com.fasterxml.jackson.core 96 | jackson-databind 97 | 2.3.0 98 | 99 | 100 | 101 | 102 | org.slf4j 103 | slf4j-api 104 | ${org.slf4j.version} 105 | 106 | 107 | org.slf4j 108 | slf4j-simple 109 | ${org.slf4j.version} 110 | 111 | 112 | 113 | 114 | junit 115 | junit 116 | 4.7 117 | test 118 | 119 | 120 | 121 | 122 | 123 | org.apache.maven.wagon 124 | wagon-ssh 125 | 126 | 127 | 128 | 129 | 130 | org.apache.maven.plugins 131 | maven-compiler-plugin 132 | 3.1 133 | 134 | 1.6 135 | 1.6 136 | 137 | 138 | 139 | org.apache.maven.plugins 140 | maven-jar-plugin 141 | 2.4 142 | 143 | 144 | 145 | com.brainlounge.zooterrain.Driver 146 | 147 | 148 | 149 | 150 | 151 | org.apache.maven.plugins 152 | maven-dependency-plugin 153 | 154 | 155 | install 156 | install 157 | 158 | sources 159 | 160 | 161 | 162 | install 163 | 164 | copy-dependencies 165 | 166 | 167 | ${project.build.directory}/lib 168 | 169 | 170 | 171 | 172 | 173 | 174 | maven-assembly-plugin 175 | 2.4 176 | 177 | 178 | 179 | com.brainlounge.zooterrain.netty.WebSocketServer 180 | 181 | 182 | 183 | jar-with-dependencies 184 | 185 | zooterrain-full 186 | false 187 | 188 | 189 | 190 | make-assembly 191 | package 192 | 193 | single 194 | 195 | 196 | 197 | 198 | 199 | 200 | 201 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/OutboundConnector.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.netty; 19 | 20 | import com.brainlounge.zooterrain.zkclient.ClientMessage; 21 | import com.brainlounge.zooterrain.zkclient.ZkStateListener; 22 | import io.netty.channel.ChannelHandlerContext; 23 | import io.netty.handler.codec.http.websocketx.TextWebSocketFrame; 24 | 25 | /** 26 | */ 27 | public class OutboundConnector implements ZkStateListener { 28 | protected ChannelHandlerContext context; 29 | 30 | public OutboundConnector(ChannelHandlerContext context) { 31 | this.context = context; 32 | } 33 | 34 | @Override 35 | public void zkNodeEvent(ClientMessage message) { 36 | try { 37 | context.channel().writeAndFlush(new TextWebSocketFrame(message.toJson()), context.newPromise()); 38 | } catch (Throwable e) { 39 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 40 | } 41 | } 42 | } 43 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/WebSocketServer.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2012 The Netty Project 3 | * 4 | * The Netty Project licenses this file to you under the Apache License, 5 | * version 2.0 (the "License"); you may not use this file except in compliance 6 | * with the License. You may obtain a copy of the License at: 7 | * 8 | * http://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 | * License for the specific language governing permissions and limitations 14 | * under the License. 15 | */ 16 | package com.brainlounge.zooterrain.netty; 17 | 18 | import com.brainlounge.zooterrain.zkclient.ZkStateObserver; 19 | import io.netty.bootstrap.ServerBootstrap; 20 | import io.netty.channel.Channel; 21 | import io.netty.channel.EventLoopGroup; 22 | import io.netty.channel.nio.NioEventLoopGroup; 23 | import io.netty.channel.socket.nio.NioServerSocketChannel; 24 | 25 | /** 26 | * A HTTP server which serves Web Socket requests at: 27 | * 28 | * http://localhost:8080/websocket 29 | * 30 | * Open your browser at http://localhost:8080/, then the demo page will be loaded and a Web Socket connection will be 31 | * made automatically. 32 | * 33 | * This server illustrates support for the different web socket specification versions and will work with: 34 | * 35 | * 43 | */ 44 | public class WebSocketServer { 45 | 46 | protected final int port; 47 | protected final String zkConnection; 48 | 49 | public WebSocketServer(int port, String zkConnection) { 50 | this.port = port; 51 | this.zkConnection = zkConnection; 52 | } 53 | 54 | public void run() throws Exception { 55 | 56 | final ZkStateObserver zkStateObserver = new ZkStateObserver(zkConnection); 57 | 58 | EventLoopGroup bossGroup = new NioEventLoopGroup(); 59 | EventLoopGroup workerGroup = new NioEventLoopGroup(); 60 | try { 61 | ServerBootstrap b = new ServerBootstrap(); 62 | b.group(bossGroup, workerGroup) 63 | .channel(NioServerSocketChannel.class) 64 | .childHandler(new WebSocketServerInitializer(zkStateObserver)); 65 | 66 | Channel ch = b.bind(port).sync().channel(); 67 | System.out.println("Zooterrain server started at port " + port + '.'); 68 | 69 | zkStateObserver.start(); 70 | 71 | ch.closeFuture().sync(); 72 | } finally { 73 | bossGroup.shutdownGracefully(); 74 | workerGroup.shutdownGracefully(); 75 | } 76 | 77 | } 78 | 79 | public static void main(String[] args) throws Exception { 80 | int port; 81 | if (args.length > 0) { 82 | port = Integer.parseInt(args[0]); 83 | } else { 84 | port = 8080; 85 | } 86 | 87 | final String zookeeperConnection = System.getProperty("zooterrain.conn"); 88 | if (zookeeperConnection == null) { 89 | System.err.println("please provide a zookeeper connection string in property 'zooterrain.conn'."); 90 | System.exit(-1); 91 | } 92 | System.out.println("you specified a ZooKeeper at " + zookeeperConnection); 93 | 94 | new WebSocketServer(port, zookeeperConnection).run(); 95 | } 96 | } 97 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/WebSocketServerInboundHandler.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2012 The Netty Project 3 | * 4 | * The Netty Project licenses this file to you under the Apache License, 5 | * version 2.0 (the "License"); you may not use this file except in compliance 6 | * with the License. You may obtain a copy of the License at: 7 | * 8 | * http://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 | * License for the specific language governing permissions and limitations 14 | * under the License. 15 | */ 16 | package com.brainlounge.zooterrain.netty; 17 | 18 | import com.brainlounge.zooterrain.zkclient.ClientMessage; 19 | import com.brainlounge.zooterrain.zkclient.ClientRequest; 20 | import com.brainlounge.zooterrain.zkclient.ControlMessage; 21 | import com.brainlounge.zooterrain.zkclient.DataMessage; 22 | import com.brainlounge.zooterrain.zkclient.ZkStateListener; 23 | import com.brainlounge.zooterrain.zkclient.ZkStateObserver; 24 | import com.fasterxml.jackson.core.JsonParser; 25 | import com.fasterxml.jackson.core.TreeNode; 26 | import com.fasterxml.jackson.databind.ObjectMapper; 27 | import com.fasterxml.jackson.databind.node.TextNode; 28 | import com.google.common.collect.Sets; 29 | import io.netty.buffer.ByteBuf; 30 | import io.netty.buffer.Unpooled; 31 | import io.netty.channel.ChannelFuture; 32 | import io.netty.channel.ChannelFutureListener; 33 | import io.netty.channel.ChannelHandlerContext; 34 | import io.netty.channel.SimpleChannelInboundHandler; 35 | import io.netty.handler.codec.http.DefaultFullHttpResponse; 36 | import io.netty.handler.codec.http.FullHttpRequest; 37 | import io.netty.handler.codec.http.FullHttpResponse; 38 | import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame; 39 | import io.netty.handler.codec.http.websocketx.PingWebSocketFrame; 40 | import io.netty.handler.codec.http.websocketx.PongWebSocketFrame; 41 | import io.netty.handler.codec.http.websocketx.TextWebSocketFrame; 42 | import io.netty.handler.codec.http.websocketx.WebSocketFrame; 43 | import io.netty.handler.codec.http.websocketx.WebSocketServerHandshaker; 44 | import io.netty.handler.codec.http.websocketx.WebSocketServerHandshakerFactory; 45 | import io.netty.util.CharsetUtil; 46 | import org.apache.zookeeper.CreateMode; 47 | 48 | import java.util.logging.Level; 49 | import java.util.logging.Logger; 50 | 51 | import static io.netty.handler.codec.http.HttpHeaders.Names.CONTENT_TYPE; 52 | import static io.netty.handler.codec.http.HttpHeaders.Names.HOST; 53 | import static io.netty.handler.codec.http.HttpHeaders.isKeepAlive; 54 | import static io.netty.handler.codec.http.HttpHeaders.setContentLength; 55 | import static io.netty.handler.codec.http.HttpMethod.GET; 56 | import static io.netty.handler.codec.http.HttpResponseStatus.*; 57 | import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1; 58 | 59 | /** 60 | * Handles handshakes and messages 61 | */ 62 | public class WebSocketServerInboundHandler extends SimpleChannelInboundHandler { 63 | private static final Logger logger = Logger.getLogger(WebSocketServerInboundHandler.class.getName()); 64 | 65 | private static final String WEBSOCKET_PATH = "/firehose"; 66 | 67 | protected WebSocketServerHandshaker handshaker; 68 | protected ZkStateObserver zkStateObserver; 69 | 70 | protected ObjectMapper jsonMapper = new ObjectMapper(); 71 | 72 | public WebSocketServerInboundHandler(ZkStateObserver zkStateObserver) { 73 | this.zkStateObserver = zkStateObserver; 74 | } 75 | 76 | public void messageReceived(ChannelHandlerContext ctx, Object msg) throws Exception { 77 | if (msg instanceof FullHttpRequest) { 78 | handleHttpRequest(ctx, (FullHttpRequest) msg); 79 | } else if (msg instanceof WebSocketFrame) { 80 | handleWebSocketFrame(ctx, (WebSocketFrame) msg); 81 | } 82 | } 83 | 84 | @Override 85 | public void channelReadComplete(ChannelHandlerContext ctx) throws Exception { 86 | ctx.flush(); 87 | } 88 | 89 | private void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest req) throws Exception { 90 | // Handle a bad request. 91 | if (!req.getDecoderResult().isSuccess()) { 92 | sendHttpResponse(ctx, req, new DefaultFullHttpResponse(HTTP_1_1, BAD_REQUEST)); 93 | return; 94 | } 95 | 96 | // Allow only GET methods. 97 | if (req.getMethod() != GET) { 98 | sendHttpResponse(ctx, req, new DefaultFullHttpResponse(HTTP_1_1, FORBIDDEN)); 99 | return; 100 | } 101 | 102 | // Send the demo page and favicon.ico 103 | if ("/favicon.ico".equals(req.getUri())) { 104 | FullHttpResponse res = new DefaultFullHttpResponse(HTTP_1_1, NOT_FOUND); 105 | sendHttpResponse(ctx, req, res); 106 | return; 107 | } else if ("/hexdump.js".equals(req.getUri())) { 108 | final WebSocketServerJSPage webSocketServerJSPage = new WebSocketServerJSPage(); 109 | ByteBuf content = webSocketServerJSPage.getContent(getWebSocketLocation(req)); 110 | sendHttpResponse(ctx, req, content, "text/javascript; charset=UTF-8"); 111 | return; 112 | } else if ("/".equals(req.getUri())) { 113 | WebSocketServerIndexPage webSocketServerIndexPage = new WebSocketServerIndexPage(); 114 | ByteBuf content = webSocketServerIndexPage.getContent(getWebSocketLocation(req)); 115 | sendHttpResponse(ctx, req, content, "text/html; charset=UTF-8"); 116 | return; 117 | } 118 | 119 | // Handshake 120 | WebSocketServerHandshakerFactory wsFactory = new WebSocketServerHandshakerFactory( 121 | getWebSocketLocation(req), null, false); 122 | handshaker = wsFactory.newHandshaker(req); 123 | if (handshaker == null) { 124 | WebSocketServerHandshakerFactory.sendUnsupportedWebSocketVersionResponse(ctx.channel()); 125 | return; 126 | } 127 | 128 | try { 129 | handshaker.handshake(ctx.channel(), req); 130 | } catch (Exception e) { 131 | logger.log(Level.WARNING, "websocket handshake failed", e); 132 | WebSocketServerHandshakerFactory.sendUnsupportedWebSocketVersionResponse(ctx.channel()); 133 | return; 134 | } 135 | zkStateObserver.addListener(new OutboundConnector(ctx)); 136 | } 137 | 138 | private void sendHttpResponse(ChannelHandlerContext ctx, FullHttpRequest req, ByteBuf content, String contentType) { 139 | FullHttpResponse res = new DefaultFullHttpResponse(HTTP_1_1, OK, content); 140 | 141 | res.headers().set(CONTENT_TYPE, contentType); 142 | setContentLength(res, content.readableBytes()); 143 | 144 | sendHttpResponse(ctx, req, res); 145 | } 146 | 147 | private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame) { 148 | 149 | // Check for closing frame 150 | if (frame instanceof CloseWebSocketFrame) { 151 | handshaker.close(ctx.channel(), (CloseWebSocketFrame) frame.retain()); 152 | return; 153 | } 154 | if (frame instanceof PingWebSocketFrame) { 155 | ctx.channel().write(new PongWebSocketFrame(frame.content().retain())); 156 | return; 157 | } 158 | if (!(frame instanceof TextWebSocketFrame)) { 159 | throw new UnsupportedOperationException(String.format("%s frame types not supported", frame.getClass() 160 | .getName())); 161 | } 162 | 163 | String request = ((TextWebSocketFrame) frame).text(); 164 | if (logger.isLoggable(Level.FINE)) { 165 | logger.fine(String.format("%s received %s", ctx.channel(), request)); 166 | } 167 | 168 | // expecting JSON, parse now 169 | TreeNode jsonRequest; 170 | ClientRequest.Type requestType; 171 | try { 172 | final JsonParser parser = jsonMapper.getFactory().createParser(request); 173 | jsonRequest = parser.readValueAsTree(); 174 | requestType = ClientRequest.Type.valueOf(readTextValue(jsonRequest, "r")); 175 | } catch (Exception e) { 176 | logger.info("parsing JSON failed for '" + request + "'"); 177 | // TODO return error to client 178 | return; 179 | } 180 | 181 | if (requestType == null) { 182 | logger.info("parsing JSON failed, no 'r' field"); 183 | return; 184 | } 185 | 186 | switch(requestType) { 187 | case i: 188 | retrieveInitialData(ctx); 189 | break; 190 | case b: 191 | retrieveNodesData(ctx, jsonRequest); 192 | break; 193 | case u: 194 | String znode = readTextValue(jsonRequest, "z"); 195 | String data = readTextValue(jsonRequest, "d"); 196 | setNodeData(znode, data); 197 | break; 198 | default: 199 | System.out.println("unknown, unhandled client request = " + request); 200 | } 201 | } 202 | 203 | private void setNodeData(String znode, String data) { 204 | try { 205 | zkStateObserver.getZk().create(znode, data.getBytes(), null, CreateMode.PERSISTENT); 206 | return; 207 | } catch (Throwable e) { 208 | logger.warning("failed to create new node '" + znode + "': " + e.getMessage()); 209 | } 210 | try { 211 | final DataMessage dataMessage = zkStateObserver.retrieveNodeData(znode); 212 | zkStateObserver.getZk().setData(znode, data.getBytes(), dataMessage.getStat().getVersion()); 213 | } catch (Throwable e) { 214 | logger.warning("failed to update data for " + znode); 215 | } 216 | } 217 | 218 | private String readTextValue(TreeNode jsonRequest, final String node) { 219 | TextNode type = (TextNode)jsonRequest.get(node); 220 | if (type == null) return null; 221 | String value = type.textValue(); 222 | return value; 223 | } 224 | 225 | private void retrieveNodesData(ChannelHandlerContext ctx, TreeNode jsonRequest) { 226 | final String znode; 227 | try { 228 | znode = ((TextNode)jsonRequest.get("z")).textValue(); 229 | } catch (Exception e) { 230 | return; // TODO return error 231 | } 232 | final DataMessage dataMessage = zkStateObserver.retrieveNodeData(znode); 233 | if (dataMessage != null) { 234 | writeClientMessage(ctx, dataMessage); 235 | } 236 | } 237 | 238 | private void retrieveInitialData(ChannelHandlerContext ctx) { 239 | // client requested initial data 240 | 241 | // sending connection string info 242 | final ControlMessage handshakeInfo = new ControlMessage(zkStateObserver.getZkConnection(), ControlMessage.Type.H); 243 | writeClientMessage(ctx, handshakeInfo); 244 | final ControlMessage zkInfo = new ControlMessage(zkStateObserver.getStatus(), ControlMessage.Type.Z); 245 | writeClientMessage(ctx, zkInfo); 246 | 247 | // sending initial znodes 248 | try { 249 | zkStateObserver.initialTree("/", 6, Sets.newHashSet(new OutboundConnector(ctx))); 250 | } catch (InterruptedException e) { 251 | e.printStackTrace(); 252 | } 253 | } 254 | 255 | protected void writeClientMessage(ChannelHandlerContext ctx, ClientMessage clientMessage) { 256 | ctx.channel().writeAndFlush(new TextWebSocketFrame(clientMessage.toJson())); 257 | } 258 | 259 | private static void sendHttpResponse( 260 | ChannelHandlerContext ctx, FullHttpRequest req, FullHttpResponse res) { 261 | // Generate an error page if response getStatus code is not OK (200). 262 | if (res.getStatus().code() != 200) { 263 | ByteBuf buf = Unpooled.copiedBuffer(res.getStatus().toString(), CharsetUtil.UTF_8); 264 | res.content().writeBytes(buf); 265 | buf.release(); 266 | setContentLength(res, res.content().readableBytes()); 267 | } 268 | 269 | // Send the response and close the connection if necessary. 270 | ChannelFuture f = ctx.channel().writeAndFlush(res); 271 | if (!isKeepAlive(req) || res.getStatus().code() != 200) { 272 | f.addListener(ChannelFutureListener.CLOSE); 273 | } 274 | } 275 | 276 | @Override 277 | public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception { 278 | cause.printStackTrace(); 279 | ctx.close(); 280 | } 281 | 282 | private static String getWebSocketLocation(FullHttpRequest req) { 283 | return "ws://" + req.headers().get(HOST) + WEBSOCKET_PATH; 284 | } 285 | 286 | @Override 287 | protected void channelRead0(ChannelHandlerContext channelHandlerContext, Object o) throws Exception { 288 | messageReceived(channelHandlerContext, o); 289 | } 290 | } 291 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/WebSocketServerIndexPage.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2012 The Netty Project 3 | * 4 | * The Netty Project licenses this file to you under the Apache License, 5 | * version 2.0 (the "License"); you may not use this file except in compliance 6 | * with the License. You may obtain a copy of the License at: 7 | * 8 | * http://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 | * License for the specific language governing permissions and limitations 14 | * under the License. 15 | */ 16 | package com.brainlounge.zooterrain.netty; 17 | 18 | import io.netty.buffer.ByteBuf; 19 | import io.netty.buffer.Unpooled; 20 | import io.netty.util.CharsetUtil; 21 | import org.apache.commons.io.IOUtils; 22 | 23 | import java.io.IOException; 24 | import java.io.InputStream; 25 | import java.io.StringReader; 26 | 27 | /** 28 | * Generates the demo HTML page which is served at http://localhost:8080/ 29 | */ 30 | public final class WebSocketServerIndexPage { 31 | 32 | private String content = "error loading page"; 33 | 34 | public WebSocketServerIndexPage() throws IOException { 35 | final InputStream resourceAsStream = WebSocketServerIndexPage.class.getResourceAsStream("/index.html"); 36 | content = IOUtils.toString(resourceAsStream, "UTF-8"); 37 | } 38 | 39 | public ByteBuf getContent(String webSocketLocation) { 40 | String contentReplaced = content.replace("${wsLoc}", webSocketLocation); 41 | return Unpooled.copiedBuffer(contentReplaced, CharsetUtil.US_ASCII); 42 | } 43 | 44 | } 45 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/WebSocketServerInitializer.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2012 The Netty Project 3 | * 4 | * The Netty Project licenses this file to you under the Apache License, 5 | * version 2.0 (the "License"); you may not use this file except in compliance 6 | * with the License. You may obtain a copy of the License at: 7 | * 8 | * http://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 | * License for the specific language governing permissions and limitations 14 | * under the License. 15 | */ 16 | package com.brainlounge.zooterrain.netty; 17 | 18 | import com.brainlounge.zooterrain.zkclient.ZkStateObserver; 19 | import io.netty.channel.ChannelInitializer; 20 | import io.netty.channel.ChannelPipeline; 21 | import io.netty.channel.socket.SocketChannel; 22 | import io.netty.handler.codec.http.HttpObjectAggregator; 23 | import io.netty.handler.codec.http.HttpServerCodec; 24 | 25 | /** 26 | */ 27 | public class WebSocketServerInitializer extends ChannelInitializer { 28 | 29 | protected ZkStateObserver zkStateObserver; 30 | 31 | public WebSocketServerInitializer(ZkStateObserver zkStateObserver) { 32 | this.zkStateObserver = zkStateObserver; 33 | } 34 | 35 | @Override 36 | public void initChannel(SocketChannel ch) throws Exception { 37 | 38 | ChannelPipeline pipeline = ch.pipeline(); 39 | pipeline.addLast("codec-http", new HttpServerCodec()); 40 | pipeline.addLast("aggregator", new HttpObjectAggregator(65536)); 41 | pipeline.addLast("inboundhandler", new WebSocketServerInboundHandler(zkStateObserver)); 42 | } 43 | } 44 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/netty/WebSocketServerJSPage.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2012 The Netty Project 3 | * 4 | * The Netty Project licenses this file to you under the Apache License, 5 | * version 2.0 (the "License"); you may not use this file except in compliance 6 | * with the License. You may obtain a copy of the License at: 7 | * 8 | * http://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 12 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 13 | * License for the specific language governing permissions and limitations 14 | * under the License. 15 | */ 16 | package com.brainlounge.zooterrain.netty; 17 | 18 | import io.netty.buffer.ByteBuf; 19 | import io.netty.buffer.Unpooled; 20 | import io.netty.util.CharsetUtil; 21 | import org.apache.commons.io.IOUtils; 22 | 23 | import java.io.IOException; 24 | import java.io.InputStream; 25 | 26 | /** 27 | * Generates the demo HTML page which is served at http://localhost:8080/ 28 | */ 29 | public final class WebSocketServerJSPage { 30 | 31 | private String content = "error loading page"; 32 | 33 | public WebSocketServerJSPage() throws IOException { 34 | final InputStream resourceAsStream = WebSocketServerJSPage.class.getResourceAsStream("/hexdump.js"); 35 | content = IOUtils.toString(resourceAsStream, "UTF-8"); 36 | } 37 | 38 | public ByteBuf getContent(String webSocketLocation) { 39 | String contentReplaced = content.replace("${wsLoc}", webSocketLocation); 40 | return Unpooled.copiedBuffer(contentReplaced, CharsetUtil.US_ASCII); 41 | } 42 | 43 | } 44 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ClientMessage.java: -------------------------------------------------------------------------------- 1 | package com.brainlounge.zooterrain.zkclient; 2 | 3 | /** 4 | */ 5 | public abstract class ClientMessage { 6 | public abstract String toJson(); 7 | 8 | protected String quoted(String notQuoted) { 9 | return "\"" + notQuoted + "\""; 10 | } 11 | } 12 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ClientRequest.java: -------------------------------------------------------------------------------- 1 | package com.brainlounge.zooterrain.zkclient; 2 | 3 | /** 4 | */ 5 | public interface ClientRequest { 6 | 7 | enum Type { 8 | i, // retrieve all initial data 9 | b, // retrieve node data 10 | u, // update node with new data, create node if not already existant 11 | } 12 | } 13 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ControlMessage.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.zkclient; 19 | 20 | /** 21 | */ 22 | public class ControlMessage extends ClientMessage { 23 | public enum Type { H, Z, I } 24 | 25 | protected Type type; 26 | protected String message; 27 | 28 | public ControlMessage(String message, Type type) { 29 | this.message = message; 30 | this.type = type; 31 | } 32 | 33 | @Override 34 | public String toJson() { 35 | String json = new StringBuilder(). 36 | append("{"). 37 | append("\"type\":").append(quoted(type.toString())).append(","). 38 | append("\"msg\":").append(quoted(message)). 39 | append("}").toString(); 40 | return json; 41 | } 42 | } 43 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/DataMessage.java: -------------------------------------------------------------------------------- 1 | package com.brainlounge.zooterrain.zkclient; 2 | 3 | import org.apache.commons.codec.binary.Base64; 4 | import org.apache.commons.codec.binary.StringUtils; 5 | import org.apache.zookeeper.data.Stat; 6 | 7 | /** 8 | */ 9 | public class DataMessage extends ClientMessage { 10 | protected final Stat stat; 11 | protected String znode; 12 | protected byte[] rawData; 13 | 14 | public DataMessage(String znode, byte[] data, Stat stat) { 15 | this.znode = znode; 16 | rawData = data; 17 | this.stat = stat; 18 | } 19 | 20 | public Stat getStat() { 21 | return stat; 22 | } 23 | 24 | public String getZnode() { 25 | return znode; 26 | } 27 | 28 | public byte[] getRawData() { 29 | return rawData; 30 | } 31 | 32 | public String getDataBase64Encoded() { 33 | return StringUtils.newStringUtf8(Base64.encodeBase64(rawData, false)); 34 | } 35 | 36 | @Override 37 | public String toJson() { 38 | StringBuilder builder = new StringBuilder(). 39 | append("{"). 40 | append("\"type\":").append(quoted("B")).append(","). 41 | append("\"znode\":").append(quoted(znode)).append(","). 42 | append("\"data\":").append(quoted(getDataBase64Encoded())); 43 | if (stat != null) { 44 | builder.append(","). 45 | append("\"size\":").append(stat.getDataLength()).append(","). 46 | append("\"version\":").append(stat.getVersion()); 47 | } 48 | builder.append("}"); 49 | return builder.toString(); 50 | } 51 | } 52 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ZNodeMessage.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.zkclient; 19 | 20 | import org.apache.zookeeper.data.Stat; 21 | 22 | /** 23 | */ 24 | public class ZNodeMessage extends ClientMessage { 25 | public enum Type { C, U, D } 26 | 27 | protected Type type; 28 | protected String fqZNodeName; 29 | protected Stat stat; 30 | 31 | public ZNodeMessage(String fqZNodeName, Type type, Stat stat) { 32 | this.fqZNodeName = fqZNodeName; 33 | this.type = type; 34 | this.stat = stat; 35 | } 36 | 37 | @Override 38 | public String toJson() { 39 | final StringBuilder stringBuilder = new StringBuilder(); 40 | stringBuilder. 41 | append("{"). 42 | append("\"type\":").append(quoted(type.toString())).append(","). 43 | append("\"znode\":").append(quoted(fqZNodeName)); 44 | if (stat != null) { 45 | stringBuilder.append(","); 46 | stringBuilder.append("\"eph\":").append(stat.getEphemeralOwner() != 0).append(","); 47 | stringBuilder.append("\"dl\":").append(stat.getDataLength()).append(","); 48 | stringBuilder.append("\"dv\":").append(stat.getVersion()).append(","); 49 | stringBuilder.append("\"ct\":").append(stat.getCtime()).append(","); 50 | stringBuilder.append("\"mt\":").append(stat.getMtime()).append(","); 51 | stringBuilder.append("\"px\":").append(stat.getPzxid()); 52 | } 53 | String json = stringBuilder.append("}").toString(); 54 | return json; 55 | } 56 | 57 | } 58 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ZNodeState.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.zkclient; 19 | 20 | import org.apache.zookeeper.data.Stat; 21 | 22 | import java.util.Collections; 23 | import java.util.Set; 24 | import java.util.concurrent.Semaphore; 25 | 26 | /** 27 | */ 28 | public class ZNodeState { 29 | 30 | private final Semaphore access = new Semaphore(1); 31 | 32 | protected Set children = Collections.emptySet(); 33 | protected Stat stat = null; 34 | 35 | public ZNodeState() { 36 | } 37 | 38 | public void acquire() throws InterruptedException { 39 | access.acquire(); 40 | } 41 | 42 | public void release() { 43 | access.release(); 44 | } 45 | 46 | public Set getChildren() { 47 | return children; 48 | } 49 | 50 | public void setChildren(Set newChildren) { 51 | children = newChildren; 52 | } 53 | 54 | public Stat getStat() { 55 | return stat; 56 | } 57 | 58 | public void setStat(Stat stat) { 59 | this.stat = stat; 60 | } 61 | } 62 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ZkStateListener.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.zkclient; 19 | 20 | /** 21 | */ 22 | public interface ZkStateListener { 23 | void zkNodeEvent(ClientMessage message); 24 | } 25 | -------------------------------------------------------------------------------- /src/main/java/com/brainlounge/zooterrain/zkclient/ZkStateObserver.java: -------------------------------------------------------------------------------- 1 | /** 2 | * Licensed to the Apache Software Foundation (ASF) under one 3 | * or more contributor license agreements. See the NOTICE file 4 | * distributed with this work for additional information 5 | * regarding copyright ownership. The ASF licenses this file 6 | * to you under the Apache License, Version 2.0 (the 7 | * "License"); you may not use this file except in compliance 8 | * with the License. You may obtain a copy of the License at 9 | * 10 | * http://www.apache.org/licenses/LICENSE-2.0 11 | * 12 | * Unless required by applicable law or agreed to in writing, software 13 | * distributed under the License is distributed on an "AS IS" BASIS, 14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | * See the License for the specific language governing permissions and 16 | * limitations under the License. 17 | */ 18 | package com.brainlounge.zooterrain.zkclient; 19 | 20 | import com.google.common.collect.Sets; 21 | import org.apache.zookeeper.KeeperException; 22 | import org.apache.zookeeper.WatchedEvent; 23 | import org.apache.zookeeper.Watcher; 24 | import org.apache.zookeeper.ZooKeeper; 25 | import org.apache.zookeeper.data.Stat; 26 | 27 | import java.io.IOException; 28 | import java.util.HashMap; 29 | import java.util.HashSet; 30 | import java.util.List; 31 | import java.util.Map; 32 | import java.util.Set; 33 | import java.util.concurrent.atomic.AtomicReference; 34 | import java.util.logging.Logger; 35 | 36 | /** 37 | */ 38 | public class ZkStateObserver implements Watcher { 39 | 40 | private static final Logger logger = Logger.getLogger(ZkStateObserver.class.getName()); 41 | 42 | private String zkConnection; 43 | protected ZooKeeper zk; 44 | protected Set listeners = new HashSet(); 45 | protected final Map znodeStateCache = new HashMap(); 46 | 47 | protected AtomicReference status = new AtomicReference("unknown"); 48 | 49 | public ZkStateObserver(String zkConnection) { 50 | this.zkConnection = zkConnection; 51 | } 52 | 53 | public String getZkConnection() { 54 | return zkConnection; 55 | } 56 | 57 | public ZooKeeper getZk() { 58 | return zk; 59 | } 60 | 61 | public String getStatus() { 62 | return status.get(); 63 | } 64 | 65 | public void addListener(ZkStateListener listener) { 66 | listeners.add(listener); 67 | } 68 | 69 | public boolean removeListener(ZkStateListener listener) { 70 | return listeners.remove(listener); 71 | } 72 | 73 | public void start() { 74 | try { 75 | logger.info("trying reaching ZK at " + zkConnection); 76 | zk = new ZooKeeper(zkConnection, 30 * 1000, null, false); 77 | } catch (IOException e) { 78 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 79 | throw new RuntimeException("zk connection failed"); 80 | } 81 | zk.register(this); 82 | try { 83 | zk.getChildren("/", this); 84 | } catch (KeeperException e) { 85 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 86 | } catch (InterruptedException e) { 87 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 88 | } 89 | } 90 | 91 | public void stop() { 92 | ZooKeeper zkLocal = zk; 93 | zk = null; 94 | if (zkLocal != null) { 95 | try { 96 | zkLocal.close(); 97 | } catch (Exception e) { 98 | ; 99 | } 100 | } 101 | } 102 | 103 | public void initialTree(String subtree, int depth, Set receivers) throws InterruptedException { 104 | if (depth <= 0) return; 105 | if (subtree == null) subtree = "/"; 106 | 107 | ZNodeState zNodeState = retrieveOrCreateZNodeState(subtree); 108 | zNodeState.acquire(); 109 | Set currentChildNames = new HashSet(); 110 | try { 111 | final List children = zk.getChildren(subtree, true); 112 | 113 | for (String child : children) { 114 | currentChildNames.add(child); 115 | } 116 | zNodeState.setChildren(currentChildNames); 117 | } catch (KeeperException e) { 118 | // znode has gone away 119 | znodeStateCache.remove(subtree); 120 | } finally { 121 | zNodeState.release(); 122 | } 123 | 124 | for (String child : currentChildNames) { 125 | final String childFQPath = subtree.equals("/") ? ("/" + child) : (subtree + "/" + child); 126 | Stat childStat; 127 | try { 128 | childStat = zk.exists(childFQPath, true); 129 | } catch (KeeperException e) { 130 | e.printStackTrace(); 131 | continue; 132 | } 133 | propagateToListeners(new ZNodeMessage(childFQPath, ZNodeMessage.Type.U, childStat), receivers); 134 | initialTree(childFQPath, depth - 1, receivers); 135 | } 136 | } 137 | 138 | public DataMessage retrieveNodeData(String znode) { 139 | Stat stat = new Stat(); 140 | try { 141 | byte[] data = zk.getData(znode, true, stat); 142 | if (data != null && data.length > 500) { 143 | byte[] dataShortend = new byte[500]; 144 | System.arraycopy(data, 0, dataShortend, 0, 500); 145 | data = dataShortend; 146 | } 147 | return new DataMessage(znode, data, stat); 148 | } catch (Exception e) { 149 | e.printStackTrace(); 150 | } 151 | return null; 152 | } 153 | 154 | @Override 155 | public void process(WatchedEvent event) { 156 | try { 157 | final String eventPath = event.getPath(); 158 | switch (event.getType()) { 159 | 160 | case None: 161 | final Event.KeeperState state = event.getState(); 162 | logger.info("new ZK connection state: " + state.toString()); 163 | 164 | if (state == Event.KeeperState.Expired) { 165 | logger.info("trying to reestablished expired connection"); 166 | stop(); 167 | start(); 168 | return; 169 | } 170 | 171 | status.set(state.toString()); 172 | propagateToListeners(new ControlMessage(state.toString(), ControlMessage.Type.Z)); 173 | break; 174 | case NodeCreated: 175 | //propagateToListeners(new ZNodeMessage(eventPath, ZNodeMessage.Type.C)); 176 | break; 177 | case NodeDeleted: 178 | //propagateToListeners(new ZNodeMessage(eventPath, ZNodeMessage.Type.D)); 179 | break; 180 | case NodeDataChanged: 181 | handleDataChanged(eventPath); 182 | 183 | case NodeChildrenChanged: 184 | handleChildrenChanged(eventPath); 185 | break; 186 | } 187 | } catch (Exception e) { 188 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 189 | } 190 | } 191 | 192 | private void handleDataChanged(String eventPath) { 193 | Stat stat = null; 194 | try { 195 | stat = zk.exists(eventPath, true); 196 | } catch (KeeperException e) { 197 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 198 | } catch (InterruptedException e) { 199 | e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates. 200 | } 201 | propagateToListeners(new ZNodeMessage(eventPath, ZNodeMessage.Type.U, stat)); 202 | } 203 | 204 | private void handleChildrenChanged(String eventPath) throws InterruptedException, KeeperException { 205 | ZNodeState zNodeState = retrieveOrCreateZNodeState(eventPath); 206 | zNodeState.acquire(); 207 | try { 208 | final Set lastTimeChildren = zNodeState.getChildren(); 209 | Set newChildNames = new HashSet(); 210 | 211 | final List children; 212 | try { 213 | children = zk.getChildren(eventPath, true); 214 | } catch (KeeperException e) { 215 | znodeStateCache.remove(eventPath); 216 | return; 217 | } 218 | 219 | for (String child : children) { 220 | newChildNames.add(child); 221 | } 222 | zNodeState.setChildren(newChildNames); 223 | 224 | final Sets.SetView removedChilds = Sets.difference(lastTimeChildren, newChildNames); 225 | final Sets.SetView newChilds = Sets.difference(newChildNames, lastTimeChildren); 226 | 227 | for (String removedChild : removedChilds) { 228 | final String childFQPath = eventPath.equals("/") ? ("/" + removedChild) : (eventPath + "/" + removedChild); 229 | propagateToListeners(new ZNodeMessage(childFQPath, ZNodeMessage.Type.D, null)); 230 | } 231 | 232 | for (String newChild : newChilds) { 233 | final String childFQPath = eventPath.equals("/") ? ("/" + newChild) : (eventPath + "/" + newChild); 234 | 235 | Stat childStat; 236 | try { 237 | childStat = zk.exists(childFQPath, true); 238 | } catch (KeeperException e) { 239 | continue; 240 | } 241 | 242 | propagateToListeners(new ZNodeMessage(childFQPath, ZNodeMessage.Type.C, childStat)); 243 | zk.getChildren(childFQPath, true); 244 | } 245 | } finally { 246 | zNodeState.release(); 247 | } 248 | } 249 | 250 | private ZNodeState retrieveOrCreateZNodeState(String path) { 251 | ZNodeState zNodeState = znodeStateCache.get(path); 252 | if (zNodeState == null) { 253 | synchronized (znodeStateCache) { 254 | ZNodeState zNodeStateAgain = znodeStateCache.get(path); 255 | if (zNodeStateAgain == null) { 256 | zNodeState = new ZNodeState(); 257 | znodeStateCache.put(path, zNodeState); 258 | } 259 | } 260 | } 261 | return zNodeState; 262 | } 263 | 264 | private void propagateToListeners(ClientMessage message) { 265 | propagateToListeners(message, listeners); 266 | } 267 | 268 | private void propagateToListeners(ClientMessage message, Set stateListeners) { 269 | for (ZkStateListener listener : stateListeners) { 270 | listener.zkNodeEvent(message); 271 | } 272 | } 273 | } 274 | -------------------------------------------------------------------------------- /src/main/resources/hexdump.js: -------------------------------------------------------------------------------- 1 | // Hexdump.js 0.1.1 2 | // (c) 2011 Dustin Willis Webber 3 | // Hexdump is freely distributable under the MIT license. 4 | // For all details and documentation: 5 | // http://github.com/mephux/hexdump.js 6 | 7 | var Hexdump; 8 | 9 | Hexdump = (function() { 10 | 11 | // Hexdump Initializer 12 | // data => The string payload. 13 | // options => hexdump configurations 14 | function Hexdump(data, options) { 15 | var self = this; 16 | self.hexdump = []; 17 | self.hex = false; 18 | self.options = { 19 | container: options.container || '' 20 | , width: options.width || 16 21 | , byteGrouping: options.byteGrouping || 0 22 | , ascii: options.ascii 23 | , lineNumber: options.lineNumber 24 | , endian: options.endian || 'big' 25 | , html: options.html 26 | , base: options.base || 'hexadecimal' 27 | , nonPrintable: options.nonPrintable || '.' 28 | , style: { 29 | lineNumberLeft: options.style.lineNumberLeft || '' 30 | , lineNumberRight: options.style.lineNumberRight || ':' 31 | , stringLeft: options.style.stringLeft || '|' 32 | , stringRight: options.style.stringRight || '|' 33 | , hexLeft: options.style.hexLeft || '' 34 | , hexRight: options.style.hexRight || '' 35 | , hexNull: options.style.hexNull || '.' 36 | , stringNull: options.style.stringNull || ' ' 37 | } 38 | }; 39 | 40 | if (self.options.base == 'hex') { 41 | self.hex = true; 42 | } else if (self.options.base == 'hexadecimal') { 43 | self.hex = true; 44 | }; 45 | 46 | // Check for the line number option and turn it off 47 | // if not set unless it has been explicitly turned 48 | // off by the user. 49 | var ln = self.options.lineNumber; 50 | if (typeof ln == "undefined" || ln == null) { 51 | self.options.lineNumber = true; 52 | }; 53 | 54 | var askey = self.options.ascii; 55 | if (typeof askey == "undefined" || askey == null) { 56 | self.options.ascii = false; 57 | }; 58 | 59 | 60 | var html = self.options.html; 61 | if (typeof html == "undefined" || html == null) { 62 | self.options.html = true; 63 | }; 64 | 65 | if (self.endian != ('little' || 'big')) { 66 | self.endian = 'big'; 67 | }; 68 | 69 | // Make sure spacing is within proper range. 70 | if (self.options.byteGrouping > data.length) { 71 | self.options.byteGrouping = data.length; 72 | }; 73 | self.options.byteGrouping--; 74 | 75 | // Make sure width is within proper range. 76 | if (self.options.width > data.length) { 77 | self.options.width = data.length; 78 | }; 79 | 80 | // Base padding 81 | self.padding = { 82 | hex: 4, 83 | dec: 5, 84 | bin: 8 85 | }; 86 | 87 | // Base conversion logic 88 | switch(self.options.base) { 89 | case 'hexadecimal': case 'hex': case 16: 90 | self.setNullPadding(self.padding.hex); 91 | self.baseConvert = function(characters) { 92 | 93 | for (var i=0; i < characters.length; i++) { 94 | return self.addPadding(characters[i].charCodeAt(0).toString(16), self.padding.hex); 95 | }; 96 | 97 | }; break; 98 | case 'decimal': case 'dec': case 10: 99 | self.setNullPadding(self.padding.dec); 100 | self.baseConvert = function(characters) { 101 | 102 | for (var i=0; i < characters.length; i++) { 103 | return self.addPadding(characters[i].charCodeAt(0), self.padding.dec); 104 | }; 105 | 106 | }; break; 107 | case 'binary': case 'bin': case 2: 108 | self.setNullPadding(self.padding.bin); 109 | self.baseConvert = function(characters) { 110 | for (var i=0; i < characters.length; i++) { 111 | var ddx = characters[i].charCodeAt(0), r = ""; 112 | 113 | for (var bbx = 0; bbx < 8; bbx++) { 114 | r = (ddx%2) + r; ddx = Math.floor(ddx/2); 115 | }; 116 | 117 | return self.addPadding(r, self.padding.bin); 118 | }; 119 | }; break; 120 | default: 121 | self.options.base = 'hex'; 122 | self.hex = true; 123 | 124 | self.setNullPadding(self.padding.hex); 125 | self.baseConvert = function(characters) { 126 | 127 | for (var i=0; i < characters.length; i++) { 128 | return self.addPadding(characters[i].charCodeAt(0).toString(16), self.padding.hex); 129 | }; 130 | 131 | }; 132 | }; 133 | 134 | var regex = new RegExp('.{1,' + this.options.width + '}', 'g'); 135 | 136 | self.data = data.match(regex); 137 | 138 | self.nullCount = (self.options.width - self.data[self.data.length - 1].length); 139 | 140 | self.hexCounter = 0; 141 | 142 | self.stringCounter = 0; 143 | 144 | for (var i=0; i < self.data.length; i++) { 145 | var tempData = self.process(self.data[i]); 146 | 147 | self.hexdump.push({ 148 | data: tempData.data, 149 | string: tempData.string, 150 | length: self.data[i].length, 151 | missing: (self.options.width - self.data[i].length) 152 | }); 153 | }; 154 | 155 | self.dump(); 156 | } 157 | 158 | Hexdump.prototype.dump = function() { 159 | var self = this; 160 | 161 | self.output = ''; 162 | for (var i=0; i < self.hexdump.length; i++) { 163 | 164 | if (self.options.lineNumber) { 165 | var tempLineNumberStyle = ''; 166 | tempLineNumberStyle += self.options.style.lineNumberLeft; 167 | 168 | var currentLineCount = (i * self.options.width); //.toString(16); 169 | var temLineCount = 8 - currentLineCount.toString().length; 170 | for (var l=0; l < temLineCount; l++) { 171 | tempLineNumberStyle += '0'; 172 | }; 173 | 174 | tempLineNumberStyle += currentLineCount; 175 | tempLineNumberStyle += self.options.style.lineNumberRight + ' '; 176 | 177 | if (self.options.html) { 178 | self.output += ''+tempLineNumberStyle+''; 179 | } else { 180 | self.output += tempLineNumberStyle; 181 | }; 182 | }; 183 | 184 | var spacingCount = 0; 185 | var breakPoint = Math.floor(self.options.width / 2); 186 | 187 | self.output += self.options.style.hexLeft; 188 | 189 | for (var x=0; x < self.hexdump[i].data.length; x++) { 190 | 191 | if (spacingCount == self.options.byteGrouping) { 192 | if (x == self.hexdump[i].data.length - 1) { 193 | self.output += self.hexdump[i].data[x]; 194 | } else { 195 | self.output += self.hexdump[i].data[x] + ' '; 196 | }; 197 | spacingCount = 0; 198 | } else { 199 | self.output += self.hexdump[i].data[x]; 200 | spacingCount++; 201 | }; 202 | }; 203 | 204 | self.output += self.options.style.hexRight; 205 | 206 | self.appendString(self.hexdump[i]); 207 | self.output += "\n"; 208 | }; 209 | 210 | var hexdump_container = document.getElementById(this.options.container); 211 | hexdump_container.innerHTML = this.output; 212 | }; 213 | 214 | Hexdump.prototype.appendString = function(data) { 215 | var self = this; 216 | self.output += ' ' + self.options.style.stringLeft; 217 | self.output += data.string; 218 | self.output += self.options.style.stringRight; 219 | }; 220 | 221 | Hexdump.prototype.splitNulls = function(code) { 222 | var split = []; 223 | var buffer = ""; 224 | 225 | if (code && code.length > 2) { 226 | for (var cc = 0; cc < code.length; cc++) { 227 | var tempi = cc + 1; 228 | 229 | if (tempi % 2 == 0) { 230 | 231 | buffer += code[cc].toString(); 232 | split.push(buffer); 233 | 234 | buffer = ""; 235 | 236 | } else { 237 | 238 | buffer += code[cc].toString(); 239 | 240 | }; 241 | 242 | }; 243 | }; 244 | 245 | return split; 246 | }; 247 | 248 | Hexdump.prototype.process = function(data) { 249 | var self = this; 250 | var stringArray = []; 251 | var hexArray = []; 252 | 253 | for (var i=0; i < data.length; i++) { 254 | if (self.options.html) { 255 | 256 | var code = self.baseConvert(data[i]); 257 | 258 | if (self.hex) { 259 | var split = self.splitNulls(code); 260 | 261 | for (var y = 0; y < split.length; y++) { 262 | hexArray.push('' + 263 | split[y] + ''); 264 | }; 265 | 266 | } else { 267 | 268 | hexArray.push('' + 269 | code + ''); 270 | 271 | }; 272 | 273 | stringArray.push('' + 274 | self.checkForNonPrintable(data[i]) + ''); 275 | 276 | } else { 277 | 278 | var code = self.baseConvert(data[i]); 279 | 280 | if (self.hex) { 281 | var split = self.splitNulls(code); 282 | 283 | for (var y = 0; y < split.length; y++) { 284 | hexArray.push(split[y]); 285 | }; 286 | 287 | } else { 288 | hexArray.push(code); 289 | }; 290 | 291 | stringArray.push(self.checkForNonPrintable(data[i])); 292 | 293 | }; 294 | 295 | self.hexCounter++; 296 | }; 297 | 298 | if (self.hex) { 299 | var splitHexWidth = self.options.width * 2; 300 | } else { 301 | var splitHexWidth = self.options.width; 302 | }; 303 | 304 | if (hexArray.length < splitHexWidth) { 305 | var amount = (splitHexWidth - hexArray.length); 306 | 307 | for (var i=0; i < amount; i++) { 308 | var nullHex = ''; 309 | 310 | if (self.options.html) { 311 | nullHex = '' + self.options.style.hexNull + ''; 312 | } else { 313 | nullHex = self.options.style.hexNull; 314 | }; 315 | 316 | hexArray.push(nullHex); 317 | }; 318 | }; 319 | 320 | if (stringArray.length < self.options.width) { 321 | var stringAmount = self.options.width - stringArray.length; 322 | for (var i=0; i < stringAmount; i++) { 323 | var nullString = ''; 324 | 325 | if (self.options.html) { 326 | nullString = '' + self.options.style.stringNull + ''; 327 | } else { 328 | nullString = self.options.style.stringNull; 329 | }; 330 | 331 | 332 | stringArray.push(nullString); 333 | }; 334 | }; 335 | 336 | return { data: hexArray, string: stringArray.join('') }; 337 | }; 338 | 339 | Hexdump.prototype.setNullPadding = function(padding) { 340 | var self = this; 341 | 342 | var hexNull = self.options.style.hexNull[0] 343 | self.options.style.hexNull = ""; 344 | 345 | if (self.hex) { 346 | padding = padding / 2; 347 | }; 348 | 349 | for (var p=0; p < padding; p++) { 350 | self.options.style.hexNull += hexNull; 351 | }; 352 | }; 353 | 354 | Hexdump.prototype.addPadding = function(ch, padding) { 355 | var self = this, length = ch.toString().length, pad = ''; 356 | 357 | for (var i=0; i < (padding - length); i++) { 358 | pad += '0' 359 | }; 360 | 361 | if (self.options.endian == 'big') { 362 | return pad + ch; 363 | } else { 364 | return ch + pad; 365 | }; 366 | }; 367 | 368 | Hexdump.prototype.checkForNonPrintable = function(character) { 369 | var self = this; 370 | var c = character.charCodeAt(0).toString(16); 371 | 372 | if (c == 0x9) { 373 | return '.' 374 | } else if (c == 0x7F) { 375 | return '.' 376 | } else if (c.length > 2 && self.options.ascii) { 377 | return '.' 378 | } else { 379 | return character; 380 | }; 381 | 382 | }; 383 | 384 | return Hexdump; 385 | })(); 386 | 387 | -------------------------------------------------------------------------------- /src/main/resources/index.html: -------------------------------------------------------------------------------- 1 | 14 | 15 | 16 | ZooTerrain 17 | 18 | 19 | 20 | 21 | 22 | 79 | 80 | 81 | 416 | 417 |
418 |
419 |
420 |

ZooTerrain Insights into running ZooKeeper clusters

421 |
422 |
423 |
424 |
425 |
Browser -> ZooTerrain 426 | 427 | 428 | 429 |
430 |
ZooTerrain -> Zookeeper 431 |
432 |
433 | 434 |
435 |
436 |
437 |
438 | 439 |
440 |
441 |
442 | 443 |
444 |
445 | 446 |
447 |
448 | 449 |
450 |
451 |
452 | 453 |
454 |
455 | 456 | 457 | 458 | 459 |
460 |
461 | 462 | 463 | 464 | 465 | 466 | 467 |
 znode ? data lengthdata version
No znode entries.
468 |
469 |
470 |
471 | 484 |
485 |

© brainlounge 2014

486 |
487 |
488 | 489 | 490 | 493 | 494 | 495 | 496 | 497 | 498 | --------------------------------------------------------------------------------