Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.jackrabbit.oak.plugins.document.util.LoggingDocumentStoreWrapper Maven / Gradle / Ivy
/*
* 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.jackrabbit.oak.plugins.document.util;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import javax.annotation.Nonnull;
import org.apache.jackrabbit.oak.cache.CacheStats;
import org.apache.jackrabbit.oak.commons.json.JsopBuilder;
import org.apache.jackrabbit.oak.plugins.document.Collection;
import org.apache.jackrabbit.oak.plugins.document.Document;
import org.apache.jackrabbit.oak.plugins.document.DocumentStore;
import org.apache.jackrabbit.oak.plugins.document.DocumentStoreException;
import org.apache.jackrabbit.oak.plugins.document.RevisionListener;
import org.apache.jackrabbit.oak.plugins.document.RevisionVector;
import org.apache.jackrabbit.oak.plugins.document.UpdateOp;
import org.apache.jackrabbit.oak.plugins.document.cache.CacheInvalidationStats;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Implements a DocumentStore wrapper and logs all calls.
*/
public class LoggingDocumentStoreWrapper implements DocumentStore, RevisionListener {
private static final Logger LOG = LoggerFactory.getLogger(LoggingDocumentStoreWrapper.class);
private static final boolean DEBUG = Boolean.parseBoolean(System.getProperty("ds.debug", "true"));
final DocumentStore store;
private boolean logThread;
public LoggingDocumentStoreWrapper(DocumentStore store) {
this.store = store;
}
public LoggingDocumentStoreWrapper withThreadNameLogging() {
this.logThread = true;
return this;
}
@Override
public T find(final Collection collection,
final String key) {
try {
logMethod("find", collection, key);
return logResult(new Callable() {
@Override
public T call() throws Exception {
return store.find(collection, key);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public T find(final Collection collection,
final String key,
final int maxCacheAge) {
try {
logMethod("find", collection, key, maxCacheAge);
return logResult(new Callable() {
@Override
public T call() throws Exception {
return store.find(collection, key, maxCacheAge);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Nonnull
@Override
public List query(final Collection collection,
final String fromKey,
final String toKey,
final int limit) {
try {
logMethod("query", collection, fromKey, toKey, limit);
return logResult(new Callable>() {
@Override
public List call() throws Exception {
return store.query(collection, fromKey, toKey, limit);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
@Nonnull
public List query(final Collection collection,
final String fromKey,
final String toKey,
final String indexedProperty,
final long startValue,
final int limit) {
try {
logMethod("query", collection, fromKey, toKey, indexedProperty, startValue, limit);
return logResult(new Callable>() {
@Override
public List call() throws Exception {
return store.query(collection, fromKey, toKey, indexedProperty, startValue, limit);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void remove(Collection collection, String key) {
try {
logMethod("remove", collection, key);
store.remove(collection, key);
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void remove(Collection collection, List keys) {
try {
logMethod("remove", collection, keys);
store.remove(collection, keys);
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public int remove(final Collection collection,
final Map> toRemove) {
try {
logMethod("remove", collection, toRemove);
return logResult(new Callable() {
@Override
public Integer call() throws Exception {
return store.remove(collection, toRemove);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public int remove(final Collection collection,
final String indexedProperty, final long startValue, final long endValue)
throws DocumentStoreException {
try {
logMethod("remove", collection, indexedProperty, startValue, endValue);
return logResult(new Callable() {
@Override
public Integer call() throws Exception {
return store.remove(collection, indexedProperty, startValue, endValue);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public boolean create(final Collection collection,
final List updateOps) {
try {
logMethod("create", collection, updateOps);
return logResult(new Callable() {
@Override
public Boolean call() throws Exception {
return store.create(collection, updateOps);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void update(final Collection collection,
final List keys,
final UpdateOp updateOp) {
try {
logMethod("update", collection, keys, updateOp);
logResult(new Callable() {
@Override
public Void call() throws Exception {
store.update(collection, keys, updateOp);
return null;
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Nonnull
@Override
public T createOrUpdate(final Collection collection,
final UpdateOp update) {
try {
logMethod("createOrUpdate", collection, update);
return logResult(new Callable() {
@Override
public T call() throws Exception {
return store.createOrUpdate(collection, update);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public List createOrUpdate(final Collection collection,
final List updateOps) {
try {
logMethod("createOrUpdate", collection, updateOps);
return logResult(new Callable>() {
@Override
public List call() throws Exception {
return store.createOrUpdate(collection, updateOps);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public T findAndUpdate(final Collection collection,
final UpdateOp update) {
try {
logMethod("findAndUpdate", collection, update);
return logResult(new Callable() {
@Override
public T call() throws Exception {
return store.findAndUpdate(collection, update);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public CacheInvalidationStats invalidateCache() {
try {
logMethod("invalidateCache");
return store.invalidateCache();
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public CacheInvalidationStats invalidateCache(Iterable keys) {
try {
logMethod("invalidateCache", keys);
return store.invalidateCache(keys);
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void invalidateCache(Collection collection, String key) {
try {
logMethod("invalidateCache", collection, key);
store.invalidateCache(collection, key);
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void dispose() {
try {
logMethod("dispose");
store.dispose();
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public T getIfCached(final Collection collection,
final String key) {
try {
logMethod("getIfCached", collection, key);
return logResult(new Callable() {
@Override
public T call() throws Exception {
return store.getIfCached(collection, key);
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public void setReadWriteMode(String readWriteMode) {
try {
logMethod("setReadWriteMode", readWriteMode);
store.setReadWriteMode(readWriteMode);
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public Iterable getCacheStats() {
try {
logMethod("getCacheStats");
return logResult(new Callable>() {
@Override
public Iterable call() throws Exception {
return store.getCacheStats();
}
});
} catch (Exception e) {
logException(e);
throw convert(e);
}
}
@Override
public Map getMetadata() {
return store.getMetadata();
}
@Override
public long determineServerTimeDifferenceMillis() {
logMethod("determineServerTimeDifferenceMillis", "start");
long result = store.determineServerTimeDifferenceMillis();
logMethod("determineServerTimeDifferenceMillis", "end", result);
return result;
}
private void logMethod(String methodName, Object... args) {
StringBuilder buff = new StringBuilder("ds");
buff.append('.').append(methodName).append('(');
for (int i = 0; i < args.length; i++) {
if (i > 0) {
buff.append(", ");
}
buff.append(quote(args[i]));
}
buff.append(");");
log(buff.toString());
}
public static String quote(Object o) {
if (o == null) {
return "null";
} else if (o instanceof String) {
return JsopBuilder.encode((String) o);
}
return o.toString();
}
private RuntimeException convert(Exception e) {
if (e instanceof RuntimeException) {
return (RuntimeException) e;
}
log("// unexpected exception type: " + e.getClass().getName());
return new DocumentStoreException("Unexpected exception: " + e.toString(), e);
}
private void logException(Exception e) {
log("// exception: " + e.toString());
}
private T logResult(Callable callable) throws Exception {
long time = System.nanoTime();
T result = callable.call();
time = System.nanoTime() - time;
log("// " + (time / 1000) + " us\t" + quote(result));
return result;
}
private void log(String message) {
String out = this.logThread ? (Thread.currentThread() + " " + message) : message;
if (DEBUG) {
System.out.println(out);
}
LOG.info(out);
}
@Override
public void updateAccessedRevision(RevisionVector revision, int currentClusterId) {
logMethod("updateAccessedRevision", revision);
if (store instanceof RevisionListener) {
((RevisionListener) store).updateAccessedRevision(revision, currentClusterId);
}
}
}
