org.mariadb.jdbc.plugin.codec.LocalTimeCodec Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mariadb-java-client Show documentation
Show all versions of mariadb-java-client Show documentation
JDBC driver for MariaDB and MySQL
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (c) 2012-2014 Monty Program Ab
// Copyright (c) 2015-2023 MariaDB Corporation Ab
package org.mariadb.jdbc.plugin.codec;
import static org.mariadb.jdbc.client.result.Result.NULL_LENGTH;
import java.io.IOException;
import java.sql.SQLDataException;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.ZoneId;
import java.time.format.DateTimeParseException;
import java.util.Calendar;
import java.util.EnumSet;
import java.util.TimeZone;
import org.mariadb.jdbc.client.*;
import org.mariadb.jdbc.client.socket.Writer;
import org.mariadb.jdbc.client.util.MutableInt;
import org.mariadb.jdbc.plugin.Codec;
/** LocalTime codec */
public class LocalTimeCodec implements Codec {
/** default instance */
public static final LocalTimeCodec INSTANCE = new LocalTimeCodec();
private static final EnumSet COMPATIBLE_TYPES =
EnumSet.of(
DataType.TIME,
DataType.DATETIME,
DataType.TIMESTAMP,
DataType.VARSTRING,
DataType.VARCHAR,
DataType.STRING,
DataType.BLOB,
DataType.TINYBLOB,
DataType.MEDIUMBLOB,
DataType.LONGBLOB);
/**
* Parse text value into hour/minutes/seconds/microseconds array
*
* @param buf packet buffer
* @param length data length
* @param column column metadata
* @return hour/minutes/seconds/microseconds array
* @throws SQLDataException if parsing error occurs
*/
public static int[] parseTime(ReadableByteBuf buf, MutableInt length, ColumnDecoder column)
throws SQLDataException {
int initialPos = buf.pos();
int[] parts = new int[5];
parts[0] = 1;
int idx = 1;
int partLength = 0;
byte b;
int i = 0;
if (length.get() > 0 && buf.getByte() == '-') {
buf.skip();
i++;
parts[0] = -1;
}
for (; i < length.get(); i++) {
b = buf.readByte();
if (b == ':' || b == '.') {
idx++;
partLength = 0;
continue;
}
if (b < '0' || b > '9') {
buf.pos(initialPos);
String val = buf.readString(length.get());
throw new SQLDataException(
String.format("%s value '%s' cannot be decoded as Time", column.getType(), val));
}
partLength++;
parts[idx] = parts[idx] * 10 + (b - '0');
}
if (idx < 2) {
buf.pos(initialPos);
String val = buf.readString(length.get());
throw new SQLDataException(
String.format("%s value '%s' cannot be decoded as Time", column.getType(), val));
}
// set nano real value
if (idx == 4) {
for (i = 0; i < 9 - partLength; i++) {
parts[4] = parts[4] * 10;
}
}
return parts;
}
public String className() {
return LocalTime.class.getName();
}
public boolean canDecode(ColumnDecoder column, Class type) {
return COMPATIBLE_TYPES.contains(column.getType()) && type.isAssignableFrom(LocalTime.class);
}
public boolean canEncode(Object value) {
return value instanceof LocalTime;
}
@Override
@SuppressWarnings("fallthrough")
public LocalTime decodeText(
ReadableByteBuf buf, MutableInt length, ColumnDecoder column, Calendar cal)
throws SQLDataException {
int[] parts;
switch (column.getType()) {
case TIMESTAMP:
case DATETIME:
parts = LocalDateTimeCodec.parseTimestamp(buf.readString(length.get()));
if (parts == null) {
length.set(NULL_LENGTH);
return null;
}
return LocalTime.of(parts[3], parts[4], parts[5], parts[6]);
case TIME:
parts = parseTime(buf, length, column);
parts[1] = parts[1] % 24;
if (parts[0] == -1) {
// negative
long seconds = (24 * 60 * 60 - (parts[1] * 3600 + parts[2] * 60L + parts[3]));
return LocalTime.ofNanoOfDay(seconds * 1_000_000_000 - parts[4]);
}
return LocalTime.of(parts[1] % 24, parts[2], parts[3], parts[4]);
case BLOB:
case TINYBLOB:
case MEDIUMBLOB:
case LONGBLOB:
if (column.isBinary()) {
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalTime", column.getType()));
}
// expected fallthrough
// BLOB is considered as String if it has a collation (this is TEXT column)
case VARSTRING:
case VARCHAR:
case STRING:
String val = buf.readString(length.get());
try {
if (val.contains(" ")) {
ZoneId tz =
cal != null ? cal.getTimeZone().toZoneId() : TimeZone.getDefault().toZoneId();
return LocalDateTime.parse(val, LocalDateTimeCodec.MARIADB_LOCAL_DATE_TIME.withZone(tz))
.toLocalTime();
} else {
return LocalTime.parse(val);
}
} catch (DateTimeParseException e) {
throw new SQLDataException(
String.format(
"value '%s' (%s) cannot be decoded as LocalTime", val, column.getType()));
}
default:
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalTime", column.getType()));
}
}
@Override
@SuppressWarnings("fallthrough")
public LocalTime decodeBinary(
ReadableByteBuf buf, MutableInt length, ColumnDecoder column, Calendar cal)
throws SQLDataException {
int hour = 0;
int minutes = 0;
int seconds = 0;
long microseconds = 0;
switch (column.getType()) {
case TIMESTAMP:
case DATETIME:
if (length.get() == 0) {
length.set(NULL_LENGTH);
return null;
}
int year = buf.readUnsignedShort();
int month = buf.readByte();
int dayOfMonth = buf.readByte();
if (length.get() > 4) {
hour = buf.readByte();
minutes = buf.readByte();
seconds = buf.readByte();
if (length.get() > 7) {
microseconds = buf.readInt();
}
}
// xpand workaround https://jira.mariadb.org/browse/XPT-274
if (year == 0
&& month == 0
&& dayOfMonth == 0
&& hour == 0
&& minutes == 0
&& seconds == 0) {
length.set(NULL_LENGTH);
return null;
}
return LocalTime.of(hour, minutes, seconds).plusNanos(microseconds * 1000);
case TIME:
if (length.get() > 0) {
boolean negate = buf.readByte() == 1;
if (length.get() > 4) {
buf.skip(4); // skip days
if (length.get() > 7) {
hour = buf.readByte();
minutes = buf.readByte();
seconds = buf.readByte();
if (length.get() > 8) {
microseconds = buf.readInt();
}
}
}
if (negate) {
// negative
long nanos = (24 * 60 * 60 - (hour * 3600 + minutes * 60 + seconds));
return LocalTime.ofNanoOfDay(nanos * 1_000_000_000 - microseconds * 1000);
}
}
return LocalTime.of(hour % 24, minutes, seconds, (int) microseconds * 1000);
case BLOB:
case TINYBLOB:
case MEDIUMBLOB:
case LONGBLOB:
if (column.isBinary()) {
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalTime", column.getType()));
}
// expected fallthrough
// BLOB is considered as String if it has a collation (this is TEXT column)
case VARSTRING:
case VARCHAR:
case STRING:
String val = buf.readString(length.get());
try {
if (val.contains(" ")) {
ZoneId tz =
cal != null ? cal.getTimeZone().toZoneId() : TimeZone.getDefault().toZoneId();
return LocalDateTime.parse(val, LocalDateTimeCodec.MARIADB_LOCAL_DATE_TIME.withZone(tz))
.toLocalTime();
} else {
return LocalTime.parse(val);
}
} catch (DateTimeParseException e) {
throw new SQLDataException(
String.format(
"value '%s' (%s) cannot be decoded as LocalTime", val, column.getType()));
}
default:
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalTime", column.getType()));
}
}
@Override
public void encodeText(Writer encoder, Context context, Object value, Calendar cal, Long maxLen)
throws IOException {
LocalTime val = (LocalTime) value;
StringBuilder dateString = new StringBuilder(15);
dateString
.append(val.getHour() < 10 ? "0" : "")
.append(val.getHour())
.append(val.getMinute() < 10 ? ":0" : ":")
.append(val.getMinute())
.append(val.getSecond() < 10 ? ":0" : ":")
.append(val.getSecond());
int microseconds = val.getNano() / 1000;
if (microseconds > 0) {
dateString.append(".");
if (microseconds % 1000 == 0) {
dateString.append(Integer.toString(microseconds / 1000 + 1000).substring(1));
} else {
dateString.append(Integer.toString(microseconds + 1000000).substring(1));
}
}
encoder.writeByte('\'');
encoder.writeAscii(dateString.toString());
encoder.writeByte('\'');
}
@Override
public void encodeBinary(Writer encoder, Object value, Calendar cal, Long maxLength)
throws IOException {
LocalTime val = (LocalTime) value;
int nano = val.getNano();
if (nano > 0) {
encoder.writeByte((byte) 12);
encoder.writeByte((byte) 0);
encoder.writeInt(0);
encoder.writeByte((byte) val.getHour());
encoder.writeByte((byte) val.getMinute());
encoder.writeByte((byte) val.getSecond());
encoder.writeInt(nano / 1000);
} else {
encoder.writeByte((byte) 8);
encoder.writeByte((byte) 0);
encoder.writeInt(0);
encoder.writeByte((byte) val.getHour());
encoder.writeByte((byte) val.getMinute());
encoder.writeByte((byte) val.getSecond());
}
}
public int getBinaryEncodeType() {
return DataType.TIME.get();
}
}