com.alibaba.hologres.client.impl.handler.jdbc.JdbcBooleanColumnValues Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of holo-client Show documentation
client for Hologres
There is a newer version: 2.5.5
/*
 * Copyright (c) 2022. Alibaba Group Holding Limited
 */

package com.alibaba.hologres.client.impl.handler.jdbc;

import org.postgresql.jdbc.TimestampUtils;
import org.postgresql.util.PSQLException;
import org.postgresql.util.PSQLState;

import java.sql.SQLException;

/**
 * bool 列存类.
 */
public class JdbcBooleanColumnValues extends JdbcColumnValues {

	Boolean[] array;

	public JdbcBooleanColumnValues(TimestampUtils timestampUtils, int rowCount) {
		super(timestampUtils, rowCount);
		array = new Boolean[rowCount];
	}

	@Override
	public void doSet(int row, Object obj) throws SQLException {
		array[row] = castToBoolean(obj);
	}

	/**
	 * Cast an Object value to the corresponding boolean value.
	 *
	 * @param in Object to cast into boolean
	 * @return boolean value corresponding to the cast of the object
	 * @throws PSQLException PSQLState.CANNOT_COERCE
	 */
	static boolean castToBoolean(final Object in) throws PSQLException {
		if (in instanceof Boolean) {
			return (Boolean) in;
		}
		if (in instanceof String) {
			return fromString((String) in);
		}
		if (in instanceof Character) {
			return fromCharacter((Character) in);
		}
		if (in instanceof Number) {
			return fromNumber((Number) in);
		}
		throw new PSQLException("Cannot cast to boolean", PSQLState.CANNOT_COERCE);
	}

	static boolean fromString(final String strval) throws PSQLException {
		// Leading or trailing whitespace is ignored, and case does not matter.
		final String val = strval.trim();
		if ("1".equals(val) || "true".equalsIgnoreCase(val)
				|| "t".equalsIgnoreCase(val) || "yes".equalsIgnoreCase(val)
				|| "y".equalsIgnoreCase(val) || "on".equalsIgnoreCase(val)) {
			return true;
		}
		if ("0".equals(val) || "false".equalsIgnoreCase(val)
				|| "f".equalsIgnoreCase(val) || "no".equalsIgnoreCase(val)
				|| "n".equalsIgnoreCase(val) || "off".equalsIgnoreCase(val)) {
			return false;
		}
		throw cannotCoerceException(strval);
	}

	private static boolean fromCharacter(final Character charval) throws PSQLException {
		if ('1' == charval || 't' == charval || 'T' == charval
				|| 'y' == charval || 'Y' == charval) {
			return true;
		}
		if ('0' == charval || 'f' == charval || 'F' == charval
				|| 'n' == charval || 'N' == charval) {
			return false;
		}
		throw cannotCoerceException(charval);
	}

	private static boolean fromNumber(final Number numval) throws PSQLException {
		// Handles BigDecimal, Byte, Short, Integer, Long Float, Double
		// based on the widening primitive conversions.
		final double value = numval.doubleValue();
		if (value == 1.0d) {
			return true;
		}
		if (value == 0.0d) {
			return false;
		}
		throw cannotCoerceException(numval);
	}

	@Override
	public Object[] getArray() {
		return array;
	}
}