Java tutorial
/* * * Copyright 2011 Netflix, 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.netflix.curator.framework.imps; import com.netflix.curator.RetryLoop; import com.netflix.curator.TimeTrace; import com.netflix.curator.framework.api.BackgroundCallback; import com.netflix.curator.framework.api.BackgroundPathable; import com.netflix.curator.framework.api.CuratorEvent; import com.netflix.curator.framework.api.CuratorEventType; import com.netflix.curator.framework.api.CuratorWatcher; import com.netflix.curator.framework.api.GetDataBuilder; import com.netflix.curator.framework.api.GetDataWatchBackgroundStatable; import com.netflix.curator.framework.api.Pathable; import com.netflix.curator.framework.api.WatchPathable; import org.apache.zookeeper.AsyncCallback; import org.apache.zookeeper.KeeperException; import org.apache.zookeeper.Watcher; import org.apache.zookeeper.data.Stat; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.util.concurrent.Callable; import java.util.concurrent.Executor; class GetDataBuilderImpl implements GetDataBuilder, BackgroundOperation<String> { private final Logger log = LoggerFactory.getLogger(getClass()); private final CuratorFrameworkImpl client; private Stat responseStat; private Watching watching; private Backgrounding backgrounding; private boolean decompress; GetDataBuilderImpl(CuratorFrameworkImpl client) { this.client = client; responseStat = null; watching = new Watching(); backgrounding = new Backgrounding(); decompress = false; } @Override public GetDataWatchBackgroundStatable decompressed() { decompress = true; return new GetDataWatchBackgroundStatable() { @Override public Pathable<byte[]> inBackground() { return GetDataBuilderImpl.this.inBackground(); } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Object context) { return GetDataBuilderImpl.this.inBackground(callback, context); } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Object context, Executor executor) { return GetDataBuilderImpl.this.inBackground(callback, context, executor); } @Override public Pathable<byte[]> inBackground(Object context) { return GetDataBuilderImpl.this.inBackground(context); } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback) { return GetDataBuilderImpl.this.inBackground(callback); } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Executor executor) { return GetDataBuilderImpl.this.inBackground(callback, executor); } @Override public byte[] forPath(String path) throws Exception { return GetDataBuilderImpl.this.forPath(path); } @Override public WatchPathable<byte[]> storingStatIn(Stat stat) { return GetDataBuilderImpl.this.storingStatIn(stat); } @Override public BackgroundPathable<byte[]> watched() { return GetDataBuilderImpl.this.watched(); } @Override public BackgroundPathable<byte[]> usingWatcher(Watcher watcher) { return GetDataBuilderImpl.this.usingWatcher(watcher); } @Override public BackgroundPathable<byte[]> usingWatcher(CuratorWatcher watcher) { return GetDataBuilderImpl.this.usingWatcher(watcher); } }; } @Override public WatchPathable<byte[]> storingStatIn(Stat stat) { this.responseStat = stat; return new WatchPathable<byte[]>() { @Override public byte[] forPath(String path) throws Exception { return GetDataBuilderImpl.this.forPath(path); } @Override public Pathable<byte[]> watched() { GetDataBuilderImpl.this.watched(); return GetDataBuilderImpl.this; } @Override public Pathable<byte[]> usingWatcher(Watcher watcher) { GetDataBuilderImpl.this.usingWatcher(watcher); return GetDataBuilderImpl.this; } @Override public Pathable<byte[]> usingWatcher(CuratorWatcher watcher) { GetDataBuilderImpl.this.usingWatcher(watcher); return GetDataBuilderImpl.this; } }; } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Object context) { backgrounding = new Backgrounding(callback, context); return this; } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Object context, Executor executor) { backgrounding = new Backgrounding(client, callback, context, executor); return this; } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback) { backgrounding = new Backgrounding(callback); return this; } @Override public Pathable<byte[]> inBackground(BackgroundCallback callback, Executor executor) { backgrounding = new Backgrounding(client, callback, executor); return this; } @Override public Pathable<byte[]> inBackground() { backgrounding = new Backgrounding(true); return this; } @Override public Pathable<byte[]> inBackground(Object context) { backgrounding = new Backgrounding(context); return this; } @Override public BackgroundPathable<byte[]> watched() { watching = new Watching(true); return this; } @Override public BackgroundPathable<byte[]> usingWatcher(Watcher watcher) { watching = new Watching(client, watcher); return this; } @Override public BackgroundPathable<byte[]> usingWatcher(CuratorWatcher watcher) { watching = new Watching(client, watcher); return this; } @Override public void performBackgroundOperation(final OperationAndData<String> operationAndData) throws Exception { final TimeTrace trace = client.getZookeeperClient().startTracer("GetDataBuilderImpl-Background"); AsyncCallback.DataCallback callback = new AsyncCallback.DataCallback() { @Override public void processResult(int rc, String path, Object ctx, byte[] data, Stat stat) { trace.commit(); if (decompress && (data != null)) { try { data = client.getCompressionProvider().decompress(path, data); } catch (Exception e) { log.error("Decompressing for path: " + path, e); rc = KeeperException.Code.DATAINCONSISTENCY.intValue(); } } CuratorEvent event = new CuratorEventImpl(client, CuratorEventType.GET_DATA, rc, path, null, ctx, stat, data, null, null, null); client.processBackgroundOperation(operationAndData, event); } }; if (watching.isWatched()) { client.getZooKeeper().getData(operationAndData.getData(), true, callback, backgrounding.getContext()); } else { client.getZooKeeper().getData(operationAndData.getData(), watching.getWatcher(), callback, backgrounding.getContext()); } } @Override public byte[] forPath(String path) throws Exception { path = client.fixForNamespace(path); byte[] responseData = null; if (backgrounding.inBackground()) { client.processBackgroundOperation( new OperationAndData<String>(this, path, backgrounding.getCallback(), null), null); } else { responseData = pathInForeground(path); } return responseData; } private byte[] pathInForeground(final String path) throws Exception { TimeTrace trace = client.getZookeeperClient().startTracer("GetDataBuilderImpl-Foreground"); byte[] responseData = RetryLoop.callWithRetry(client.getZookeeperClient(), new Callable<byte[]>() { @Override public byte[] call() throws Exception { byte[] responseData; if (watching.isWatched()) { responseData = client.getZooKeeper().getData(path, true, responseStat); } else { responseData = client.getZooKeeper().getData(path, watching.getWatcher(), responseStat); } return responseData; } }); trace.commit(); return decompress ? client.getCompressionProvider().decompress(path, responseData) : responseData; } }