org.onetwo.dbm.mapping.CompositedFieldValueConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of onetwo-dbm Show documentation
onetwo-dbm
The newest version!
package org.onetwo.dbm.mapping;

import java.util.Arrays;
import java.util.List;

import org.onetwo.common.utils.LangUtils;
import org.springframework.core.annotation.AnnotationAwareOrderComparator;

import com.google.common.collect.Lists;

/**
 * @author wayshall
 * 

 */
public class CompositedFieldValueConverter implements DbmFieldValueConverter {
	
	final public static DbmFieldValueConverter ENUM_CONVERTER = new DbmFieldValueConverter() {
		public Object forJava(DbmMappedField field, Object value) {
			return field.getEnumType().forJava(field, value);
		}
		public Object forStore(DbmMappedField field, Object value) {
			return field.getEnumType().forStore(field, value);
		}
	};
	/*final public static DbmFieldValueConverter ENUM_CONVERTER = new DbmFieldValueConverter() {
		
		@Override
		public Object forJava(DbmMappedField field, Object value) {
			if(value==null){
				return value;
			}
			Object actualValue = null;
			DbmEnumType etype = field.getEnumType();
			if(etype==DbmEnumType.ORDINAL){
//				actualValue = Types.convertValue(field.getColumnType().cast(value), field.getPropertyInfo().getType());
				actualValue = Types.convertValue((Integer)value, field.getPropertyInfo().getType());
			}else if(etype==DbmEnumType.STRING){
				actualValue = Types.convertValue(value.toString(), field.getPropertyInfo().getType());
			}else if(etype == DbmEnumType.MAPPING) {
				DbmEnumValueMapping[] values = (DbmEnumValueMapping[]) field.getPropertyInfo().getType().getEnumConstants();
				DbmEnumValueMapping valueMapping = Stream.of(values)
//														.filter(dvm->Integer.valueOf(dvm.getMappingValue()).equals(value))
														.filter(dvm->dvm.getEnumMappingValue().equals(value))
														.findFirst()
														.orElseThrow(()-> {
															return new DbmException("error enum mapping value: " + value);
														});
				actualValue = valueMapping;
			}else{
				throw new DbmException("error enum type: " + etype);
			}
			return actualValue;
		}
		
		@Override
		public Object forStore(DbmMappedField field, Object value) {
			if(value==null){
				return value;
			}
			Object actualValue = null;
			DbmEnumType etype = field.getEnumType();
			Enum enumValue = (Enum) value;
			if(etype==DbmEnumType.ORDINAL){
				actualValue = enumValue.ordinal();
			}else if(etype==DbmEnumType.STRING){
				actualValue = enumValue.name();
			}else if(etype == DbmEnumType.MAPPING) {
				DbmEnumIntMapping mapping = (DbmEnumIntMapping)value;
				actualValue = mapping.getMappingValue();
			}else{
				throw new DbmException("error enum type: " + etype);
			}
			return actualValue;
		}
	};*/

	public static CompositedFieldValueConverter composited(DbmFieldValueConverter... converters) {
		List list = Lists.newArrayList();
//		list.add(ENUM_CONVERTER);
		if(!LangUtils.isEmpty(converters)){
			list.addAll(Arrays.asList(converters));
		}
		return new CompositedFieldValueConverter(list);
	}
	
	private List converters;
	
	private CompositedFieldValueConverter(List converters) {
		super();
		this.converters = converters;
	}
	
	CompositedFieldValueConverter addFieldValueConverter(DbmFieldValueConverter converter){
		this.converters.add(converter);
		return this;
	}

	@Override
	public Object forJava(DbmMappedField field, Object fieldValue) {
		List converters = this.converters;
		if(converters.isEmpty()){
			return fieldValue;
		}
		Object newValue = fieldValue;
		for(DbmFieldValueConverter converter : converters){
			newValue = converter.forJava(field, newValue);
		}
		return newValue;
	}

	@Override
	public Object forStore(DbmMappedField field, Object fieldValue) {
		List converters = this.converters;
		if(converters.isEmpty()){
			return fieldValue;
		}
		Object newValue = fieldValue;
		for(DbmFieldValueConverter converter : converters){
			newValue = converter.forStore(field, newValue);
		}
		return newValue;
	}
	
	public void sort() {
		AnnotationAwareOrderComparator.sort(converters);
	}
	

}