com.impossibl.postgres.system.procs.Bools Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of pgjdbc-ng Show documentation
Show all versions of pgjdbc-ng Show documentation
PostgreSQL JDBC - NG - Driver
/**
* Copyright (c) 2013, impossibl.com
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* * Neither the name of impossibl.com nor the names of its contributors may
* be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package com.impossibl.postgres.system.procs;
import com.impossibl.postgres.system.Context;
import com.impossibl.postgres.system.ConversionException;
import com.impossibl.postgres.types.Type;
import java.io.IOException;
import java.math.BigDecimal;
import java.math.BigInteger;
import static java.lang.Boolean.FALSE;
import static java.lang.Boolean.TRUE;
import io.netty.buffer.ByteBuf;
public class Bools extends SimpleProcProvider {
public Bools() {
super(new TxtEncoder(), new TxtDecoder(), new BinEncoder(), new BinDecoder(), "bool");
}
@SuppressWarnings("unused")
private static Boolean convertInput(Context context, Object source, Object sourceContext) throws ConversionException {
if (source instanceof Boolean) {
return (Boolean) source;
}
if (source instanceof Byte) {
return (Byte) source != 0 ? TRUE : FALSE;
}
if (source instanceof Short) {
return (Short) source != 0 ? TRUE : FALSE;
}
if (source instanceof Integer) {
return (Integer) source != 0 ? TRUE : FALSE;
}
if (source instanceof Long) {
return (Long) source != 0 ? TRUE : FALSE;
}
if (source instanceof BigInteger) {
return ((BigInteger) source).compareTo(BigInteger.ZERO) != 0 ? TRUE : FALSE;
}
if (source instanceof Float) {
return (Float) source != 0 ? TRUE : FALSE;
}
if (source instanceof Double) {
return (Double) source != 0 ? TRUE : FALSE;
}
if (source instanceof BigDecimal) {
return ((BigDecimal) source).compareTo(BigDecimal.ZERO) != 0 ? TRUE : FALSE;
}
if (source instanceof Character) {
switch ((Character) source) {
case 'f':
case 'F':
case 'n':
case 'N':
case '0':
return false;
case 't':
case 'T':
case 'y':
case 'Y':
case '1':
return true;
default:
throw new ConversionException("Cannot convert value '" + source + "' to bool");
}
}
if (source instanceof String) {
String val = (String) source;
if (val.equalsIgnoreCase("f") ||
val.equalsIgnoreCase("false") ||
val.equalsIgnoreCase("n") ||
val.equalsIgnoreCase("no") ||
val.equalsIgnoreCase("off") ||
val.equalsIgnoreCase("0"))
return false;
if (val.equalsIgnoreCase("t") ||
val.equalsIgnoreCase("true") ||
val.equalsIgnoreCase("y") ||
val.equalsIgnoreCase("yes") ||
val.equalsIgnoreCase("on") ||
val.equalsIgnoreCase("1"))
return true;
throw new ConversionException("Cannot convert value \"" + val + "\" to bool");
}
return null;
}
@SuppressWarnings("unused")
private static Object convertOutput(Context context, Boolean decoded, Class targetClass, Object targetContext) {
if (targetClass == Boolean.class || targetClass == boolean.class) {
return decoded;
}
if (targetClass == Byte.class || targetClass == byte.class) {
return decoded ? (byte)1 : (byte)0;
}
if (targetClass == Short.class || targetClass == short.class) {
return decoded ? (short)1 : (short)0;
}
if (targetClass == Integer.class || targetClass == int.class) {
return decoded ? 1 : 0;
}
if (targetClass == Long.class || targetClass == long.class) {
return decoded ? (long)1 : (long)0;
}
if (targetClass == BigInteger.class) {
return decoded ? BigInteger.ONE : BigInteger.ZERO;
}
if (targetClass == Float.class || targetClass == float.class) {
return decoded ? (float)1.0 : (float)0.0;
}
if (targetClass == Double.class || targetClass == double.class) {
return decoded ? 1.0 : 0.0;
}
if (targetClass == BigDecimal.class) {
return decoded ? BigDecimal.ONE : BigDecimal.ZERO;
}
if (targetClass == String.class) {
return decoded ? "t" : "f";
}
return null;
}
static class BinDecoder extends AutoConvertingBinaryDecoder {
BinDecoder() {
super(1, Bools::convertOutput);
}
@Override
public Class getDefaultClass() {
return Boolean.class;
}
@Override
protected Boolean decodeNativeValue(Context context, Type type, Short typeLength, Integer typeModifier, ByteBuf buffer, Class targetClass, Object targetContext) throws IOException {
return buffer.readByte() != 0;
}
}
static class BinEncoder extends AutoConvertingBinaryEncoder {
BinEncoder() {
super(1, Bools::convertInput);
}
@Override
public Class getDefaultClass() {
return Boolean.class;
}
@Override
protected void encodeNativeValue(Context context, Type type, Boolean value, Object sourceContext, ByteBuf buffer) throws IOException {
buffer.writeByte(value ? 1 : 0);
}
}
static class TxtDecoder extends AutoConvertingTextDecoder {
TxtDecoder() {
super(Bools::convertOutput);
}
@Override
public Class getDefaultClass() {
return Boolean.class;
}
@Override
protected Boolean decodeNativeValue(Context context, Type type, Short typeLength, Integer typeModifier, CharSequence buffer, Class targetClass, Object targetContext) throws IOException {
switch (buffer.toString().toLowerCase()) {
case "t":
case "true":
case "y":
case "yes":
case "on":
case "1":
return true;
case "f":
case "false":
case "n":
case "no":
case "off":
case "0":
return false;
}
throw new ConversionException("Invalid format for boolean");
}
}
static class TxtEncoder extends AutoConvertingTextEncoder {
TxtEncoder() {
super(Bools::convertInput);
}
@Override
public Class getDefaultClass() {
return Boolean.class;
}
@Override
protected void encodeNativeValue(Context context, Type type, Boolean value, Object sourceContext, StringBuilder buffer) throws IOException {
buffer.append(value ? "t" : "f");
}
}
}