com.moon.poi.excel.table.CellSetter Maven / Gradle / Ivy
package com.moon.poi.excel.table;
import com.moon.core.util.Table;
import com.moon.core.util.TableImpl;
import com.moon.poi.excel.CellFactory;
import com.moon.poi.excel.annotation.format.DateTimePattern;
import com.moon.poi.excel.annotation.format.NumberPattern;
import org.joda.time.ReadableInstant;
import org.joda.time.ReadablePartial;
import java.lang.annotation.Annotation;
import java.text.*;
import java.time.format.DateTimeFormatter;
import java.time.temporal.TemporalAccessor;
import java.util.*;
import java.util.function.Function;
import java.util.function.Supplier;
import static org.joda.time.format.DateTimeFormat.forPattern;
/**
* @author moonsky
*/
abstract class CellSetter {
private final GetTransfer transformer;
private static boolean typeof(Class expected, Class actual) { return expected.isAssignableFrom(actual); }
protected CellSetter(GetTransfer transformer) { this.transformer = transformer; }
static CellSetter of(Attribute attr) {
Class propertyType = attr.getPropertyType();
DateTimePattern dateFmt = attr.getAnnotation(DateTimePattern.class);
NumberPattern numberFmt = attr.getAnnotation(NumberPattern.class);
if (numberFmt != null && Assert.isNumberType(propertyType)) {
return tryNumberAndCache(numberFmt);
}
if (dateFmt != null) {
CellSetter setter = tryDateOrNull(propertyType, dateFmt);
if (setter != null) {
return setter;
}
}
return basicSetterMap.get(attr.getTransformOrDefault());
}
private final static Map basicSetterMap = new EnumMap(TransferForGet.class);
static {
for (TransferForGet value : TransferForGet.values()) {
basicSetterMap.put(value, new BasicSetter(value));
}
}
/**
* 设置单元格值
*
* @param factory
* @param value
*/
abstract void set(CellFactory factory, Object value);
final GetTransfer getTransfer() { return transformer; }
private static abstract class StringCellSetter extends CellSetter {
StringCellSetter() { super(TransferForGet.STRING); }
}
private final static class NumberSetter extends StringCellSetter {
private final NumberFormat formatter;
private NumberSetter(NumberPattern pattern) {
Locale locale = pattern.locale().getLocale();
DecimalFormatSymbols symbols = DecimalFormatSymbols.getInstance(locale);
DecimalFormat format = new DecimalFormat(pattern.value(), symbols);
format.setRoundingMode(pattern.roundingMode());
boolean grouping = pattern.grouping();
if (grouping) {
format.setGroupingUsed(true);
format.setGroupingSize(pattern.groupingSize());
}
int count = pattern.maxIntDigit();
if (count >= 0) {
format.setMaximumIntegerDigits(count);
}
count = pattern.minIntDigit();
if (count >= 0) {
format.setMinimumIntegerDigits(count);
}
count = pattern.maxFractionDigit();
if (count >= 0) {
format.setMaximumFractionDigits(count);
}
count = pattern.minFractionDigit();
if (count >= 0) {
format.setMinimumFractionDigits(count);
}
this.formatter = format;
}
String format(Object value) { return formatter.format(value); }
@Override
final void set(CellFactory factory, Object value) {
if (value != null) {
getTransfer().transfer(factory, format(value));
}
}
}
private final static Table cached
= TableImpl.newWeakHashTable();
private static synchronized void cache(Class propertyType, Annotation pattern, CellSetter setter) {
cached.putIfAbsent(propertyType, pattern, setter);
}
private static CellSetter getOrBuild(
Class expect, Class propertyType, T pattern, Function builder
) { return getOrBuild(typeof(expect, propertyType), propertyType, pattern, builder); }
private static CellSetter getOrBuild(
boolean matched, Class propertyType, T pattern, Function builder
) {
CellSetter setter = null;
if (matched) {
setter = cached.get(propertyType, pattern);
if (setter == null) {
setter = builder.apply(pattern);
cache(propertyType, pattern, setter);
}
}
return setter;
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* number support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private final static CellSetter tryNumberAndCache(NumberPattern pattern) {
return getOrBuild(Number.class, Number.class, pattern, NumberSetter::new);
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* date support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private final static CellSetter tryDateOrNull(Class type, DateTimePattern pattern) {
CellSetter setter = getOrBuild(TemporalAccessor.class, type, pattern, TemporalAccessorSetter::new);
if (setter != null) {
return setter;
}
if (long.class == type || Long.class == type) {
if (Assert.isImportedJodaTime()) {
return new LongJodaTimeSetter(pattern);
}
return new LongDateSetter(pattern);
}
if (Assert.isImportedJodaTime()) {
setter = tryJodaAndCache(type, pattern);
if (setter != null) {
return setter;
}
}
setter = getOrBuild(Calendar.class, type, pattern, CalendarSetter::new);
if (setter != null) {
return setter;
}
return getOrBuild(Date.class, type, pattern, UtilDateSetter::new);
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* util date time support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private final static class DateFormatBuilder implements Supplier {
private final String format;
private final Locale locale;
DateFormatBuilder(DateTimePattern pattern) {
this.locale = pattern.locale().getLocale();
this.format = pattern.value();
}
@Override
public DateFormat get() { return new SimpleDateFormat(format, locale); }
}
private abstract static class DefaultDateSetter extends StringCellSetter {
private final ThreadLocal formatter;
private final Supplier creator;
protected DefaultDateSetter(DateTimePattern pattern) {
this.creator = new DateFormatBuilder(pattern);
this.formatter = new ThreadLocal<>();
}
final DateFormat getFormatter() {
DateFormat format = formatter.get();
if (format == null) {
format = creator.get();
formatter.set(format);
}
return format;
}
/**
* 转换为 util 日期
*
* @param value
*
* @return
*/
abstract Date toDate(Object value);
@Override
final void set(CellFactory factory, Object value) {
if (value != null) {
String date = getFormatter().format(toDate(value));
getTransfer().transfer(factory, date);
}
}
}
private final static class CalendarSetter extends DefaultDateSetter {
protected CalendarSetter(DateTimePattern pattern) { super(pattern); }
@Override
final Date toDate(Object value) { return ((Calendar) value).getTime(); }
}
private final static class UtilDateSetter extends DefaultDateSetter {
protected UtilDateSetter(DateTimePattern pattern) { super(pattern); }
@Override
final Date toDate(Object value) { return (Date) value; }
}
private final static class LongDateSetter extends DefaultDateSetter {
protected LongDateSetter(DateTimePattern pattern) { super(pattern); }
@Override
final Date toDate(Object value) { return new Date((long) value); }
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* joda datetime support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private static CellSetter tryJodaAndCache(Class type, DateTimePattern pattern) {
CellSetter setter = getOrBuild(ReadableInstant.class, type, pattern, ReadableInstantSetter::new);
if (setter != null) {
return setter;
}
setter = getOrBuild(ReadablePartial.class, type, pattern, ReadablePartialSetter::new);
if (setter != null) {
return setter;
}
return getOrBuild(Number.class, type, pattern, LongJodaTimeSetter::new);
}
private abstract static class JodaDateSetter extends StringCellSetter {
private final org.joda.time.format.DateTimeFormatter formatter;
private JodaDateSetter(DateTimePattern pattern) {
String format = pattern.value();
Locale locale = pattern.locale().getLocale();
this.formatter = forPattern(format).withLocale(locale);
}
final org.joda.time.format.DateTimeFormatter getFormatter() { return formatter; }
/**
* 格式化日期
*
* @param value
*
* @return
*/
abstract String toFormat(Object value);
@Override
final void set(CellFactory factory, Object value) {
if (value != null) {
getTransfer().transfer(factory, toFormat(value));
}
}
}
private final static class LongJodaTimeSetter extends JodaDateSetter {
private LongJodaTimeSetter(DateTimePattern pattern) { super(pattern); }
@Override
String toFormat(Object value) { return getFormatter().print(((Number) value).longValue()); }
}
private final static class ReadableInstantSetter extends JodaDateSetter {
private ReadableInstantSetter(DateTimePattern pattern) { super(pattern); }
@Override
String toFormat(Object value) { return getFormatter().print((ReadableInstant) value); }
}
private final static class ReadablePartialSetter extends JodaDateSetter {
private ReadablePartialSetter(DateTimePattern pattern) { super(pattern); }
@Override
String toFormat(Object value) { return getFormatter().print((ReadablePartial) value); }
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* Java 8 datetime support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private final static class TemporalAccessorSetter extends StringCellSetter {
private final DateTimeFormatter formatter;
protected TemporalAccessorSetter(DateTimePattern pattern) {
Locale locale = pattern.locale().getLocale();
this.formatter = DateTimeFormatter.ofPattern(pattern.value(), locale);
}
final String toFormat(Object value, DateTimeFormatter formatter) {
return formatter.format((TemporalAccessor) value);
}
@Override
final void set(CellFactory factory, Object value) {
if (factory != null) {
String val = toFormat(value, formatter);
getTransfer().transfer(factory, val);
}
}
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* default support
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private final static class BasicSetter extends CellSetter {
BasicSetter(GetTransfer transformer) { super(transformer); }
@Override
void set(CellFactory factory, Object value) { getTransfer().transfer(factory, value); }
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy