org.mariadb.jdbc.plugin.codec.LocalDateTimeCodec 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.DateTimeException;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.time.format.DateTimeFormatterBuilder;
import java.util.Calendar;
import java.util.EnumSet;
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;
/** LocalDateTime codec */
public class LocalDateTimeCodec implements Codec {
/** default instance */
public static final LocalDateTimeCodec INSTANCE = new LocalDateTimeCodec();
/** timestamp with fractional part formatter */
public static final DateTimeFormatter TIMESTAMP_FORMAT =
DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss.SSSSSS");
/** timestamp without fractional part formatter */
public static final DateTimeFormatter TIMESTAMP_FORMAT_NO_FRACTIONAL =
DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
/** formatter */
public static final DateTimeFormatter MARIADB_LOCAL_DATE_TIME;
private static final EnumSet COMPATIBLE_TYPES =
EnumSet.of(
DataType.DATETIME,
DataType.TIMESTAMP,
DataType.VARSTRING,
DataType.VARCHAR,
DataType.STRING,
DataType.TIME,
DataType.YEAR,
DataType.DATE,
DataType.BLOB,
DataType.TINYBLOB,
DataType.MEDIUMBLOB,
DataType.LONGBLOB);
static {
MARIADB_LOCAL_DATE_TIME =
new DateTimeFormatterBuilder()
.parseCaseInsensitive()
.append(DateTimeFormatter.ISO_LOCAL_DATE)
.appendLiteral(' ')
.append(DateTimeFormatter.ISO_LOCAL_TIME)
.toFormatter();
}
/**
* Parse timestamp to date/month/year int array
*
* @param raw string data
* @return date/month/year int array
* @throws DateTimeException if wrong format
*/
public static int[] parseTimestamp(String raw) throws DateTimeException {
int nanoLen = -1;
int[] timestampsPart = new int[] {0, 0, 0, 0, 0, 0, 0};
int partIdx = 0;
for (int idx = 0; idx < raw.length(); idx++) {
char b = raw.charAt(idx);
if (b == '-' || b == ' ' || b == ':') {
partIdx++;
continue;
}
if (b == '.') {
partIdx++;
nanoLen = 0;
continue;
}
if (nanoLen >= 0) nanoLen++;
timestampsPart[partIdx] = timestampsPart[partIdx] * 10 + b - 48;
}
if (partIdx < 2) throw new DateTimeException("Wrong timestamp format");
if (timestampsPart[0] == 0 && timestampsPart[1] == 0 && timestampsPart[2] == 0) {
if (timestampsPart[3] == 0
&& timestampsPart[4] == 0
&& timestampsPart[5] == 0
&& timestampsPart[6] == 0) return null;
timestampsPart[1] = 1;
timestampsPart[2] = 1;
}
// fix non-leading tray for nanoseconds
if (nanoLen >= 0) {
for (int begin = 0; begin < 6 - nanoLen; begin++) {
timestampsPart[6] = timestampsPart[6] * 10;
}
timestampsPart[6] = timestampsPart[6] * 1000;
}
return timestampsPart;
}
public String className() {
return LocalDateTime.class.getName();
}
public boolean canDecode(ColumnDecoder column, Class type) {
return COMPATIBLE_TYPES.contains(column.getType())
&& type.isAssignableFrom(LocalDateTime.class);
}
public boolean canEncode(Object value) {
return value instanceof LocalDateTime;
}
@Override
@SuppressWarnings("fallthrough")
public LocalDateTime decodeText(
ReadableByteBuf buf, MutableInt length, ColumnDecoder column, Calendar cal)
throws SQLDataException {
int[] parts;
switch (column.getType()) {
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 LocalDateTime", column.getType()));
}
// expected fallthrough
// BLOB is considered as String if it has a collation (this is TEXT column)
case STRING:
case VARCHAR:
case VARSTRING:
String val = buf.readString(length.get());
try {
parts = parseTimestamp(val);
if (parts == null) {
length.set(NULL_LENGTH);
return null;
}
return LocalDateTime.of(parts[0], parts[1], parts[2], parts[3], parts[4], parts[5])
.plusNanos(parts[6]);
} catch (DateTimeException dte) {
throw new SQLDataException(
String.format(
"value '%s' (%s) cannot be decoded as LocalDateTime", val, column.getType()));
}
case DATE:
parts = LocalDateCodec.parseDate(buf, length);
if (parts == null) {
length.set(NULL_LENGTH);
return null;
}
return LocalDateTime.of(parts[0], parts[1], parts[2], 0, 0, 0);
case DATETIME:
case TIMESTAMP:
parts = parseTimestamp(buf.readAscii(length.get()));
if (parts == null) {
length.set(NULL_LENGTH);
return null;
}
return LocalDateTime.of(parts[0], parts[1], parts[2], parts[3], parts[4], parts[5])
.plusNanos(parts[6]);
case TIME:
parts = LocalTimeCodec.parseTime(buf, length, column);
if (parts[0] == -1) {
return LocalDateTime.of(1970, 1, 1, 0, 0)
.minusHours(parts[1] % 24)
.minusMinutes(parts[2])
.minusSeconds(parts[3])
.minusNanos(parts[4]);
}
return LocalDateTime.of(1970, 1, 1, parts[1] % 24, parts[2], parts[3]).plusNanos(parts[4]);
case YEAR:
int year = Integer.parseInt(buf.readAscii(length.get()));
if (column.getColumnLength() <= 2) year += year >= 70 ? 1900 : 2000;
return LocalDateTime.of(year, 1, 1, 0, 0);
default:
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalDateTime", column.getType()));
}
}
@Override
@SuppressWarnings("fallthrough")
public LocalDateTime decodeBinary(
ReadableByteBuf buf, MutableInt length, ColumnDecoder column, Calendar cal)
throws SQLDataException {
int year = 1970;
int month = 1;
long dayOfMonth = 1;
int hour = 0;
int minutes = 0;
int seconds = 0;
long microseconds = 0;
switch (column.getType()) {
case TIME:
if (length.get() > 0) {
// specific case for TIME, to handle value not in 00:00:00-23:59:59
boolean negate = buf.readByte() == 1;
int day = buf.readInt();
hour = buf.readByte();
minutes = buf.readByte();
seconds = buf.readByte();
if (length.get() > 8) {
microseconds = buf.readUnsignedInt();
}
if (negate) {
return LocalDateTime.of(1970, 1, 1, 0, 0)
.minusDays(day)
.minusHours(hour)
.minusMinutes(minutes)
.minusSeconds(seconds)
.minusNanos(microseconds * 1000);
}
}
break;
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 LocalDateTime", column.getType()));
}
// expected fallthrough
// BLOB is considered as String if it has a collation (this is TEXT column)
case STRING:
case VARCHAR:
case VARSTRING:
String val = buf.readString(length.get());
try {
int[] parts = parseTimestamp(val);
if (parts == null) {
length.set(NULL_LENGTH);
return null;
}
return LocalDateTime.of(parts[0], parts[1], parts[2], parts[3], parts[4], parts[5])
.plusNanos(parts[6]);
} catch (DateTimeException dte) {
throw new SQLDataException(
String.format(
"value '%s' (%s) cannot be decoded as LocalDateTime", val, column.getType()));
}
case DATE:
case TIMESTAMP:
case DATETIME:
if (length.get() == 0) {
length.set(NULL_LENGTH);
return null;
}
year = buf.readUnsignedShort();
month = buf.readByte();
dayOfMonth = buf.readByte();
if (length.get() > 4) {
hour = buf.readByte();
minutes = buf.readByte();
seconds = buf.readByte();
if (length.get() > 7) {
microseconds = buf.readUnsignedInt();
}
}
// 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;
}
break;
case YEAR:
year = buf.readUnsignedShort();
if (column.getColumnLength() <= 2) year += year >= 70 ? 1900 : 2000;
break;
default:
buf.skip(length.get());
throw new SQLDataException(
String.format("Data type %s cannot be decoded as LocalDateTime", column.getType()));
}
return LocalDateTime.of(year, month, (int) dayOfMonth, hour, minutes, seconds)
.plusNanos(microseconds * 1000);
}
@Override
public void encodeText(Writer encoder, Context context, Object value, Calendar cal, Long maxLen)
throws IOException {
LocalDateTime val = (LocalDateTime) value;
encoder.writeByte('\'');
encoder.writeAscii(
val.format(val.getNano() != 0 ? TIMESTAMP_FORMAT : TIMESTAMP_FORMAT_NO_FRACTIONAL));
encoder.writeByte('\'');
}
@Override
public void encodeBinary(Writer encoder, Object value, Calendar cal, Long maxLength)
throws IOException {
LocalDateTime val = (LocalDateTime) value;
int nano = val.getNano();
if (nano > 0) {
encoder.writeByte((byte) 11);
encoder.writeShort((short) val.getYear());
encoder.writeByte(val.getMonthValue());
encoder.writeByte(val.getDayOfMonth());
encoder.writeByte(val.getHour());
encoder.writeByte(val.getMinute());
encoder.writeByte(val.getSecond());
encoder.writeInt(nano / 1000);
} else {
encoder.writeByte((byte) 7);
encoder.writeShort((short) val.getYear());
encoder.writeByte(val.getMonthValue());
encoder.writeByte(val.getDayOfMonth());
encoder.writeByte(val.getHour());
encoder.writeByte(val.getMinute());
encoder.writeByte(val.getSecond());
}
}
public int getBinaryEncodeType() {
return DataType.DATETIME.get();
}
}