org.simpleflatmapper.csv.impl.CellValueReaderFactoryImpl Maven / Gradle / Ivy

Go to download
package org.simpleflatmapper.csv.impl;

import org.simpleflatmapper.converter.Converter;
import org.simpleflatmapper.converter.ConverterService;
import org.simpleflatmapper.csv.CellValueReader;
import org.simpleflatmapper.csv.CellValueReaderFactory;
import org.simpleflatmapper.csv.CsvColumnDefinition;
import org.simpleflatmapper.csv.ParsingContext;
import org.simpleflatmapper.csv.impl.cellreader.*;
import org.simpleflatmapper.csv.ParsingContextFactoryBuilder;

import java.lang.reflect.Type;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

import org.simpleflatmapper.util.TypeHelper;

public final class CellValueReaderFactoryImpl implements CellValueReaderFactory {

	private static final Map, CellValueReader> READERS = new HashMap, CellValueReader>();

	static {
		READERS.put(boolean.class, new BooleanCellValueReaderImpl());
		READERS.put(byte.class, new ByteCellValueReaderImpl());
		READERS.put(char.class, new CharCellValueReaderImpl());
		READERS.put(short.class, new ShortCellValueReaderImpl());
		READERS.put(int.class, new IntegerCellValueReaderImpl());
		READERS.put(long.class, new LongCellValueReaderImpl());
		READERS.put(float.class, new FloatCellValueReaderImpl());
		READERS.put(double.class, new DoubleCellValueReaderImpl());
		READERS.put(Boolean.class, new BooleanCellValueReaderImpl());
		READERS.put(Byte.class, new ByteCellValueReaderImpl());
		READERS.put(Character.class, new CharCellValueReaderImpl());
		READERS.put(Short.class, new ShortCellValueReaderImpl());
		READERS.put(Integer.class, new IntegerCellValueReaderImpl());
		READERS.put(Long.class, new LongCellValueReaderImpl());
		READERS.put(Float.class, new FloatCellValueReaderImpl());
		READERS.put(Double.class, new DoubleCellValueReaderImpl());
		READERS.put(String.class,    new StringCellValueReader());
		READERS.put(CharSequence.class,    new StringCellValueReader());
		READERS.put(Object.class,    new StringCellValueReader());
	}

	private ConverterService converterService = ConverterService.getInstance();

	@Override
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public  CellValueReader
 getReader(Type propertyType, int index, CsvColumnDefinition columnDefinition, ParsingContextFactoryBuilder parsingContextFactoryBuilder) {
		Class propertyClass =  TypeHelper.toClass(propertyType);

		CellValueReader
 reader = null;

		if (propertyClass.equals(Date.class)) {
			String[] patterns = columnDefinition.dateFormats();
			if (patterns.length == 1) {
				DateCellValueReader dateCellValueReader = new DateCellValueReader(index, patterns[0], columnDefinition.getTimeZone());
				reader = (CellValueReader
) dateCellValueReader;
				parsingContextFactoryBuilder.addParsingContextProvider(index, dateCellValueReader);
			} else {
				DateMultiFormatCellValueReader dateCellValueReader = new DateMultiFormatCellValueReader(index, patterns, columnDefinition.getTimeZone());
				reader = (CellValueReader
) dateCellValueReader;
				parsingContextFactoryBuilder.addParsingContextProvider(index, dateCellValueReader);
			}
		} else if (Calendar.class.equals(propertyClass)) {
			reader = (CellValueReader
) new CalendarCellValueReader(this.getReader(Date.class, index, columnDefinition, parsingContextFactoryBuilder));
		}  else if (Enum.class.isAssignableFrom(propertyClass)) {
			reader = new EnumCellValueReader(propertyClass);
		} else if (UUID.class.equals(propertyClass)) {
			reader = (CellValueReader
) new UUIDCellValueReader();
		}

		if (reader == null) {
			reader = getCellValueTransformer(propertyClass);
		}


		if (reader == null) {
			Converter converter = converterService.findConverter(CharSequence.class, propertyType, columnDefinition != null ? columnDefinition.properties() : new Object[0]);
			if (converter != null) {
				return new CharSequenceConverterCellValueReader
(converter);
			}
		}

		return reader;
	}

	@SuppressWarnings("unchecked")
	private 
 CellValueReader
 getCellValueTransformer(Class propertyType) {
		return (CellValueReader) READERS.get(propertyType);
	}


}