package dev.kkorolyov.sqlob.column.factory;

import dev.kkorolyov.simplefuncs.throwing.ThrowingBiFunction;
import dev.kkorolyov.sqlob.column.Column;

import java.lang.reflect.Field;
import java.sql.Date;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.function.Function;

/**
 * Handles fields mappable directly to SQL types.
 */
public class PrimitiveColumnFactory extends BaseColumnFactory {
	private final Map<Class<?>, String> types = new HashMap<>();
	private final Map<Class<?>, Function<?, ?>> converters = new HashMap<>();
	private final Map<Class<?>, ThrowingBiFunction<ResultSet, String, ?, SQLException>> extractors = new HashMap<>();

	/**
	 * Constructs a new primitive-type field handler.
	 */
	public PrimitiveColumnFactory() {
		put("TINYINT", ResultSet::getByte,
				Byte.TYPE, Byte.class);
		put("SMALLINT", ResultSet::getShort,
				Short.TYPE, Short.class);
		put("INTEGER", ResultSet::getInt,
				Integer.TYPE, Integer.class);
		put("BIGINT", ResultSet::getLong,
				Long.TYPE, Long.class);
		put("REAL", ResultSet::getFloat,
				Float.TYPE, Float.class);
		put("DOUBLE", ResultSet::getDouble,
				Double.TYPE, Double.class);

		put("BOOLEAN", ResultSet::getBoolean,
				Boolean.TYPE, Boolean.class);

		put("CHAR(1)", (rs, column) -> {
			String string = rs.getString(column);
			return string == null ? null : string.charAt(0);
		}, Character.TYPE, Character.class);

		put("VARCHAR(1024)", ResultSet::getString,
				String.class);

		put("VARBINARY(1024)", ResultSet::getBytes,
				byte[].class);

		put("DATE", ResultSet::getDate,
				Date.class);
		put("TIME(6)", ResultSet::getTime,
				Time.class);
		put("TIMESTAMP(6)", ResultSet::getTimestamp,
				Timestamp.class);

		put("CHAR(16)", UUID::toString, (rs, column) -> UUID.fromString(rs.getString(column)),
				UUID.class);

		addAll(types.keySet());
	}
	private <T> void put(String sqlType, ThrowingBiFunction<ResultSet, String, T, SQLException> extractor, Class<T>... classes) {
		put(sqlType, Function.identity(), extractor, classes);
	}
	private <T> void put(String sqlType, Function<T, ?> converter, ThrowingBiFunction<ResultSet, String, T, SQLException> extractor, Class<T>... classes) {
		for (Class<?> c : classes) {
			types.put(c, sqlType);
			converters.put(c, converter);
			extractors.put(c, extractor);
		}
	}

	@Override
	public Column<?> get(Field f) {
		return new Column<>(f, types.get(f.getType()), extractors.get(f.getType()));
	}
}

@@ -75,16 +75,16 @@
 ncs.
-throwing
+function
 .Thr
@@ -175,16 +175,45 @@
 .Field;%0A
+import java.math.BigDecimal;%0A
 import j
@@ -1305,32 +1305,98 @@
 , Double.class);
+%0A%09%09put(%22NUMERIC%22, ResultSet::getBigDecimal,%0A%09%09%09%09BigDecimal.class);
 %0A%0A%09%09put(%22BOOLEAN

Handle collections: Add missing numeric type handler