/** Append the given direct object to the buffer. */
public void appendPObject(Object value, ServerType type, boolean binary) throws IOException {
if (type.getType().typeClass() instanceof TString && value instanceof String) {
// Optimize the common case of directly encoding a string.
printWriter.write((String) value);
return;
}
ValueSource source = valuefromObject(value, type);
appendValue(source, type, binary);
}
public ValueSource valuefromObject(Object value, ServerType type) {
if (value instanceof Date) {
TInstance dateType = javaDateTInstance(value);
Value dateValue = new Value(dateType);
typesTranslator.setTimestampMillisValue(
dateValue,
((Date) value).getTime(),
(value instanceof java.sql.Timestamp) ? ((java.sql.Timestamp) value).getNanos() : 0);
TInstance targetType = type.getType();
if (dateType.equals(targetType)) return dateValue;
TExecutionContext context =
new TExecutionContext(Collections.singletonList(dateType), targetType, null);
Value result = new Value(targetType);
targetType.typeClass().fromObject(context, dateValue, result);
return result;
} else {
// TODO this is inefficient, but I want to get it working.
return ValueSources.valuefromObject(value, type.getType());
}
}
/**
* Encode the given value into a stream that can then be passed to <code>writeByteStream</code>.
*/
public ByteArrayOutputStream encodeValue(ValueSource value, ServerType type, boolean binary)
throws IOException {
if (value.isNull()) return null;
if ((zeroDateTimeBehavior != ZeroDateTimeBehavior.NONE)
&& (((type.getType().typeClass() == MDateAndTime.DATE) && (value.getInt32() == 0))
|| ((type.getType().typeClass() == MDateAndTime.DATETIME)
&& (value.getInt64() == 0)))) {
switch (zeroDateTimeBehavior) {
case EXCEPTION:
throw new ZeroDateTimeException();
case ROUND:
value =
(type.getType().typeClass() == MDateAndTime.DATETIME)
? ROUND_ZERO_DATETIME_SOURCE
: ROUND_ZERO_DATE_SOURCE;
break;
case CONVERT_TO_NULL:
return null;
}
}
reset();
appendValue(value, type, binary);
return getByteStream();
}
/** Append the given value to the buffer. */
public void appendValue(ValueSource value, ServerType type, boolean binary) throws IOException {
if (!binary) {
// Handle unusual text encoding of binary types.
switch (type.getBinaryEncoding()) {
case BINARY_OCTAL_TEXT:
processBinaryText(value);
break;
default:
type.getType().format(value, appender);
break;
}
} else {
switch (type.getBinaryEncoding()) {
case BINARY_OCTAL_TEXT:
getByteStream().write(value.getBytes());
break;
case INT_8:
getDataStream().write((byte) typesTranslator.getIntegerValue(value));
break;
case INT_16:
getDataStream().writeShort((short) typesTranslator.getIntegerValue(value));
break;
case INT_32:
getDataStream().writeInt((int) typesTranslator.getIntegerValue(value));
break;
case INT_64:
getDataStream().writeLong(typesTranslator.getIntegerValue(value));
break;
case FLOAT_32:
getDataStream().writeFloat(value.getFloat());
break;
case FLOAT_64:
getDataStream().writeDouble(value.getDouble());
break;
case STRING_BYTES:
getByteStream().write(value.getString().getBytes(encoding));
break;
case BOOLEAN_C:
getDataStream().write(value.getBoolean() ? 1 : 0);
break;
case TIMESTAMP_FLOAT64_SECS_2000_NOTZ:
getDataStream()
.writeDouble(
seconds2000NoTZ(typesTranslator.getTimestampMillisValue(value))
+ typesTranslator.getTimestampNanosValue(value) / 1.0e9);
break;
case TIMESTAMP_INT64_MICROS_2000_NOTZ:
getDataStream()
.writeLong(
seconds2000NoTZ(typesTranslator.getTimestampMillisValue(value)) * 1000000L
+ typesTranslator.getTimestampNanosValue(value) / 1000);
break;
case DAYS_2000:
getDataStream().writeInt(days2000(typesTranslator.getTimestampMillisValue(value)));
break;
case TIME_FLOAT64_SECS_NOTZ:
getDataStream().writeDouble(timeSecsNoTZ(typesTranslator.getTimestampMillisValue(value)));
break;
case TIME_INT64_MICROS_NOTZ:
getDataStream()
.writeLong(timeSecsNoTZ(typesTranslator.getTimestampMillisValue(value)) * 1000000L);
break;
case DECIMAL_PG_NUMERIC_VAR:
for (short d : pgNumericVar(typesTranslator.getDecimalValue(value))) {
getDataStream().writeShort(d);
}
break;
case UUID:
getDataStream().write(AkGUID.uuidToBytes((java.util.UUID) value.getObject()));
break;
case NONE:
default:
throw new UnsupportedOperationException("No binary encoding for " + type);
}
}
}