[ https://issues.apache.org/jira/browse/STORM-1273?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15159438#comment-15159438 ]
ASF GitHub Bot commented on STORM-1273: --------------------------------------- Github user revans2 commented on a diff in the pull request: https://github.com/apache/storm/pull/1081#discussion_r53830934 --- Diff: storm-core/src/jvm/org/apache/storm/cluster/ZKStateStorage.java --- @@ -0,0 +1,270 @@ +/** + * 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.storm.cluster; + +import org.apache.curator.framework.CuratorFramework; +import org.apache.curator.framework.state.*; +import org.apache.curator.framework.state.ConnectionState; +import org.apache.storm.Config; +import org.apache.storm.callback.DefaultWatcherCallBack; +import org.apache.storm.callback.WatcherCallBack; +import org.apache.storm.callback.ZKStateChangedCallback; +import org.apache.storm.utils.Utils; +import org.apache.storm.zookeeper.Zookeeper; +import org.apache.zookeeper.CreateMode; +import org.apache.zookeeper.KeeperException; +import org.apache.zookeeper.Watcher; +import org.apache.zookeeper.data.ACL; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import java.io.IOException; +import java.util.List; +import java.util.Map; +import java.util.UUID; +import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.atomic.AtomicBoolean; + +public class ZKStateStorage implements IStateStorage { + + private static Logger LOG = LoggerFactory.getLogger(ZKStateStorage.class); + + private ConcurrentHashMap<String, ZKStateChangedCallback> callbacks = new ConcurrentHashMap<String, ZKStateChangedCallback>(); + private CuratorFramework zkWriter; + private CuratorFramework zkReader; + private AtomicBoolean active; + + private boolean isNimbus; + private Map authConf; + private Map<Object, Object> conf; + + public ZKStateStorage(Map<Object, Object> conf, Map authConf, List<ACL> acls, ClusterStateContext context) throws Exception { + this.conf = conf; + this.authConf = authConf; + if (context.getDaemonType().equals(DaemonType.NIMBUS)) + this.isNimbus = true; + + // just mkdir STORM_ZOOKEEPER_ROOT dir + CuratorFramework zkTemp = mkZk(); + String rootPath = String.valueOf(conf.get(Config.STORM_ZOOKEEPER_ROOT)); + Zookeeper.mkdirs(zkTemp, rootPath, acls); + zkTemp.close(); + + active = new AtomicBoolean(true); + zkWriter = mkZk(new WatcherCallBack() { + @Override + public void execute(Watcher.Event.KeeperState state, Watcher.Event.EventType type, String path) { + if (active.get()) { + if (!(state.equals(Watcher.Event.KeeperState.SyncConnected))) { + LOG.warn("Received event {} : {}: {} with disconnected Zookeeper.", state, type, path); + } else { + LOG.info("Received event {} : {} : {}", state, type, path); + } + + if (!type.equals(Watcher.Event.EventType.None)) { + for (Map.Entry<String, ZKStateChangedCallback> e : callbacks.entrySet()) { + ZKStateChangedCallback fn = e.getValue(); + fn.changed(type, path); + } + } + } + } + }); + if (isNimbus) { + zkReader = mkZk(new WatcherCallBack() { --- End diff -- It might be nice to reuse the same watcher callback instance, just so we don't have two identical anonymous classes. > port backtype.storm.cluster to java > ----------------------------------- > > Key: STORM-1273 > URL: https://issues.apache.org/jira/browse/STORM-1273 > Project: Apache Storm > Issue Type: New Feature > Components: storm-core > Reporter: Robert Joseph Evans > Assignee: John Fang > Labels: java-migration, jstorm-merger > > current state of the cluster (Some of this moves to java as a part of > heartbeat server) > https://github.com/apache/storm/tree/jstorm-import/jstorm-core/src/main/java/com/alibaba/jstorm/cluster > as an example -- This message was sent by Atlassian JIRA (v6.3.4#6332)