/*
* 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 org.jdbi.v3.sqlobject.config;
import java.lang.annotation.Annotation;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;
import org.jdbi.v3.core.config.ConfigRegistry;
import org.jdbi.v3.core.mapper.RowMappers;
import org.jdbi.v3.core.mapper.reflect.FieldMapper;
@Retention(RetentionPolicy.RUNTIME)
@ConfiguringAnnotation(RegisterFieldMapper.Impl.class)
@Target({ElementType.TYPE, ElementType.METHOD})
public @interface RegisterFieldMapper
{
/**
* The types to map with FieldMapper.
* @return one or more classes.
*/
Class<?>[] value();
/**
* Column name prefix for each mapped type. If omitted, defaults to no prefix. If specified, must have the same
* number of elements as {@link #value()}. Each <code>prefix</code> element is applied to the <code>value</code>
* element at the same index.
*
* @return Column name prefixes corresponding pairwise to the elements in {@link #value()}.
*/
String[] prefix() default {};
class Impl implements Configurer
{
@Override
public void configureForType(ConfigRegistry registry, Annotation annotation, Class<?> sqlObjectType) {
RegisterFieldMapper registerFieldMapper = (RegisterFieldMapper) annotation;
Class<?>[] types = registerFieldMapper.value();
String[] prefixes = registerFieldMapper.prefix();
RowMappers mappers = registry.get(RowMappers.class);
if (prefixes.length == 0) {
for (Class<?> type : types) {
mappers.register(FieldMapper.factory(type));
}
}
else if (prefixes.length == types.length) {
for (int i = 0; i < types.length; i++) {
mappers.register(FieldMapper.factory(types[i], prefixes[i]));
}
}
else {
throw new IllegalStateException("RegisterFieldMapper.prefix() must have the same number of elements as value()");
}
}
@Override
public void configureForMethod(ConfigRegistry registry, Annotation annotation, Class<?> sqlObjectType, Method method)
{
configureForType(registry, annotation, sqlObjectType);
}
}
}