/* * Copyright (C) 2012 Facebook, Inc. * * Licensed 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 com.facebook.zookeeper.mock; import com.facebook.zookeeper.ZooKeeperIface; import org.apache.zookeeper.CreateMode; import org.apache.zookeeper.KeeperException; import org.apache.zookeeper.WatchedEvent; import org.apache.zookeeper.Watcher; import org.apache.zookeeper.Watcher.Event.EventType; import org.apache.zookeeper.Watcher.Event.KeeperState; import org.apache.zookeeper.ZooKeeper; import org.apache.zookeeper.data.ACL; import org.apache.zookeeper.data.Stat; import java.util.List; public class MockZooKeeper implements ZooKeeperIface { private final MockZooKeeperDataStore dataStore; private final long sessionId; private final Watcher defaultWatcher; private ZooKeeper.States state = ZooKeeper.States.CONNECTING; public MockZooKeeper(Watcher watcher, MockZooKeeperDataStore dataStore) { this.dataStore = dataStore; this.sessionId = dataStore.getUniqueSessionId(); this.defaultWatcher = watcher; } public synchronized void triggerConnect() { state = ZooKeeper.States.CONNECTED; WatchedEvent watchedEvent = new WatchedEvent(EventType.None, KeeperState.SyncConnected, null); dataStore.signalSessionEvent(sessionId, watchedEvent); defaultWatcher.process(watchedEvent); } public synchronized void triggerDisconnect() { state = ZooKeeper.States.CONNECTING; WatchedEvent watchedEvent = new WatchedEvent(EventType.None, KeeperState.Disconnected, null); dataStore.signalSessionEvent(sessionId, watchedEvent); defaultWatcher.process(watchedEvent); } public synchronized void triggerSessionExpiration() { state = ZooKeeper.States.CLOSED; WatchedEvent watchedEvent = new WatchedEvent(EventType.None, KeeperState.Expired, null); dataStore.signalSessionEvent(sessionId, watchedEvent); defaultWatcher.process(watchedEvent); dataStore.clearSession(sessionId); } @Override public long getSessionId() { return sessionId; } @Override public synchronized void close() throws InterruptedException { state = ZooKeeper.States.CLOSED; dataStore.clearSession(sessionId); } private void verifyConnected() throws KeeperException { if (state == ZooKeeper.States.CLOSED) { throw new KeeperException.SessionExpiredException(); } if (state != ZooKeeper.States.CONNECTED) { throw new KeeperException.ConnectionLossException(); } } @Override public synchronized String create(String path, byte[] data, List<ACL> acl, CreateMode createMode) throws KeeperException, InterruptedException { verifyConnected(); return dataStore.create(sessionId, path, data, acl, createMode); } @Override public synchronized void delete(String path, int expectedVersion) throws InterruptedException, KeeperException { verifyConnected(); dataStore.delete(path, expectedVersion); } @Override public synchronized Stat exists(String path, Watcher watcher) throws KeeperException, InterruptedException { verifyConnected(); return dataStore.exists(sessionId, path, watcher); } @Override public synchronized Stat exists(String path, boolean watch) throws KeeperException, InterruptedException { return exists(path, watch ? defaultWatcher : null); } @Override public synchronized byte[] getData(String path, Watcher watcher, Stat stat) throws KeeperException, InterruptedException { verifyConnected(); return dataStore.getData(sessionId, path, watcher, stat); } @Override public synchronized byte[] getData(String path, boolean watch, Stat stat) throws KeeperException, InterruptedException { return getData(path, watch ? defaultWatcher : null, stat); } @Override public synchronized Stat setData(String path, byte[] data, int expectedVersion) throws KeeperException, InterruptedException { verifyConnected(); return dataStore.setData(path, data, expectedVersion); } @Override public synchronized List<String> getChildren(String path, Watcher watcher) throws KeeperException, InterruptedException { verifyConnected(); return dataStore.getChildren(sessionId, path, watcher); } @Override public synchronized List<String> getChildren(String path, boolean watch) throws KeeperException, InterruptedException { return getChildren(path, watch ? defaultWatcher : null); } @Override public synchronized ZooKeeper.States getState() { return state; } }