/*
* This file is provided 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 com.basho.riak.client.convert.reflect;
import java.util.Collection;
import java.util.Map;
import com.basho.riak.client.RiakLink;
import com.basho.riak.client.cap.VClock;
import com.basho.riak.client.query.indexes.RiakIndexes;
/**
* Singleton that wraps a cache of Class -> AnnotatioInfo and provides
* convenience methods for getting and setting Riak annotated field values
*
* @author russell
*
*/
public class AnnotationHelper {
private static final AnnotationHelper INSTANCE = new AnnotationHelper();
private AnnotationCache annotationCache = new AnnotationCache();
private AnnotationHelper() {}
public static AnnotationHelper getInstance() {
return INSTANCE;
}
public <T> String getRiakKey(T obj) {
String key = null;
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
key = annotationInfo.getRiakKey(obj);
return key;
}
public <T> T setRiakKey(T obj, String key) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
annotationInfo.setRiakKey(obj, key);
return obj;
}
public <T> T setRiakVClock(T obj, VClock vclock) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
if (annotationInfo.hasRiakVClock()) {
annotationInfo.setRiakVClock(obj, vclock);
}
return obj;
}
public <T> VClock getRiakVClock(T obj) {
VClock vclock = null;
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
if (annotationInfo.hasRiakVClock()) {
vclock = annotationInfo.getRiakVClock(obj);
}
return vclock;
}
public <T> T setRiakTombstone(T obj, boolean isTombstone) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
if (annotationInfo.hasRiakTombstone()) {
annotationInfo.setRiakTombstone(obj, isTombstone);
}
return obj;
}
public <T> Boolean getRiakTombstone(T obj) {
Boolean tombstone = null;
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
if (annotationInfo.hasRiakTombstone()) {
tombstone = annotationInfo.getRiakTombstone(obj);
}
return tombstone;
}
public <T> Map<String, String> getUsermetaData(T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
return annotationInfo.getUsermetaData(obj);
}
public <T> T setUsermetaData(Map<String, String> usermetaData, T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
annotationInfo.setUsermetaData(usermetaData, obj);
return obj;
}
public <T> RiakIndexes getIndexes(T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
return annotationInfo.getIndexes(obj);
}
public <T> T setIndexes(RiakIndexes indexes, T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
annotationInfo.setIndexes(indexes, obj);
return obj;
}
public <T> Collection<RiakLink> getLinks(T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
return annotationInfo.getLinks(obj);
}
public <T> T setLinks(Collection<RiakLink> links, T obj) {
final AnnotationInfo annotationInfo = annotationCache.get(obj.getClass());
annotationInfo.setLinks(links, obj);
return obj;
}
}