/*
* 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;
import java.lang.reflect.Field;
import java.util.LinkedList;
import java.util.List;
import org.codehaus.jackson.map.SerializationConfig;
import org.codehaus.jackson.map.introspect.AnnotatedField;
import org.codehaus.jackson.map.introspect.AnnotatedMember;
import org.codehaus.jackson.map.introspect.BasicBeanDescription;
import org.codehaus.jackson.map.ser.BeanPropertyWriter;
import org.codehaus.jackson.map.ser.BeanSerializerModifier;
/**
* {@link BeanSerializerModifier} that drops {@link RiakKey} and
* {@link RiakUsermeta} annotated fields from the the set of fields that Jackson
* will serialize
*
* @author russell
*
*/
public class RiakBeanSerializerModifier extends BeanSerializerModifier {
private static final RiakBeanSerializerModifier INSTANCE = new RiakBeanSerializerModifier();
/**
*
*/
private RiakBeanSerializerModifier() {}
public static RiakBeanSerializerModifier getInstance() {
return INSTANCE;
}
/*
* (non-Javadoc)
*
* @see
* org.codehaus.jackson.map.ser.BeanSerializerModifier#changeProperties(
* org.codehaus.jackson.map.SerializationConfig,
* org.codehaus.jackson.map.introspect.BasicBeanDescription, java.util.List)
*/
@Override public List<BeanPropertyWriter> changeProperties(SerializationConfig config,
BasicBeanDescription beanDesc,
List<BeanPropertyWriter> beanProperties) {
List<BeanPropertyWriter> keptProperties = new LinkedList<BeanPropertyWriter>();
for (BeanPropertyWriter w : beanProperties) {
if (keepProperty(w)) {
keptProperties.add(w);
}
}
return keptProperties;
}
/**
* Checks if the property has any of the Riak annotations on it, if so it
* returns false, otherwise true
*
* @param beanPropertyWriter
* a {@link BeanPropertyWriter} to check for Riak* annotations
* @return true if the property is not Riak annotated, false otherwise
*/
private boolean keepProperty(BeanPropertyWriter beanPropertyWriter) {
RiakKey key = null;
RiakUsermeta usermeta = null;
RiakLinks links = null;
AnnotatedMember member = beanPropertyWriter.getMember();
if (member instanceof AnnotatedField) {
key = beanPropertyWriter.getAnnotation(RiakKey.class);
usermeta = beanPropertyWriter.getAnnotation(RiakUsermeta.class);
links = beanPropertyWriter.getAnnotation(RiakLinks.class);
} else {
@SuppressWarnings("rawtypes") Class clazz = member.getDeclaringClass();
Field field;
try {
field = clazz.getDeclaredField(beanPropertyWriter.getName());
key = field.getAnnotation(RiakKey.class);
usermeta = field.getAnnotation(RiakUsermeta.class);
links = field.getAnnotation(RiakLinks.class);
} catch (SecurityException e) {
throw new RuntimeException(e);
} catch (NoSuchFieldException e) {
// ignore, not a field means not a Riak annotated field.
}
}
return key == null && usermeta == null && links == null;
}
}