ru.yandex.clickhouse.util.ClickHouseValueFormatter Maven / Gradle / Ivy
The newest version!
package ru.yandex.clickhouse.util;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.sql.Array;
import java.sql.Date;
import java.sql.Time;
import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.OffsetDateTime;
import java.time.OffsetTime;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Collection;
import java.util.Map;
import java.util.TimeZone;
import java.util.UUID;
import ru.yandex.clickhouse.ClickHouseUtil;
import ru.yandex.clickhouse.domain.ClickHouseDataType;
public final class ClickHouseValueFormatter {
public static final String NULL_MARKER = "\\N";
private static final DateTimeFormatter DATE_FORMATTER =
DateTimeFormatter.ofPattern("yyyy-MM-dd");
private static final DateTimeFormatter DATE_TIME_FORMATTER =
DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
private static final DateTimeFormatter TIME_FORMATTER =
DateTimeFormatter.ofPattern("HH:mm:ss");
private static ThreadLocal dateFormat = new ThreadLocal() {
@Override
protected SimpleDateFormat initialValue() {
return new SimpleDateFormat("yyyy-MM-dd");
}
};
private static ThreadLocal dateTimeFormat = new ThreadLocal() {
@Override
protected SimpleDateFormat initialValue() {
return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
}
};
public static String formatBytes(byte[] bytes) {
if (bytes == null) {
return null;
}
char[] hexArray =
{'0','1','2','3','4','5','6','7','8','9','A','B','C','D','E','F'};
char[] hexChars = new char[bytes.length * 4];
int v;
for ( int j = 0; j < bytes.length; j++ ) {
v = bytes[j] & 0xFF;
hexChars[j * 4] = '\\';
hexChars[j * 4 + 1] = 'x';
hexChars[j * 4 + 2] = hexArray[v/16];
hexChars[j * 4 + 3] = hexArray[v%16];
}
return new String(hexChars);
}
public static String formatInt(int myInt) {
return Integer.toString(myInt);
}
public static String formatDouble(double myDouble) {
return Double.toString(myDouble);
}
public static String formatChar(char myChar) {
return Character.toString(myChar);
}
public static String formatLong(long myLong) {
return Long.toString(myLong);
}
public static String formatFloat(float myFloat) {
return Float.toString(myFloat);
}
public static String formatBigDecimal(BigDecimal myBigDecimal) {
return myBigDecimal != null ? myBigDecimal.toPlainString() : NULL_MARKER;
}
public static String formatShort(short myShort) {
return Short.toString(myShort);
}
public static String formatString(String myString) {
return ClickHouseUtil.escape(myString);
}
public static String formatNull() {
return NULL_MARKER;
}
public static String formatByte(byte myByte) {
return Byte.toString(myByte);
}
public static String formatBoolean(boolean myBoolean) {
return myBoolean ? "1" : "0";
}
public static String formatDate(Date date, TimeZone timeZone) {
SimpleDateFormat formatter = getDateFormat();
formatter.setTimeZone(timeZone);
return formatter.format(date);
}
public static String formatTime(Time time, TimeZone timeZone) {
return TIME_FORMATTER.format(
Instant.ofEpochMilli(time.getTime())
.atZone(timeZone.toZoneId())
.toLocalTime());
// getDateTimeFormat().setTimeZone(timeZone);
// return getDateTimeFormat().format(time);
}
public static String formatTimestamp(Timestamp time, TimeZone timeZone) {
SimpleDateFormat formatter = getDateTimeFormat();
formatter.setTimeZone(timeZone);
StringBuilder formatted = new StringBuilder(formatter.format(time));
// TODO implement a true prepared statement to format according to parameter type
if (time != null && time.getNanos() % 1000000 > 0) {
formatted.append('.').append(time.getNanos());
}
return formatted.toString();
}
public static String formatUUID(UUID x) {
return x.toString();
}
public static String formatBigInteger(BigInteger x) {
return x.toString();
}
public static String formatLocalDate(LocalDate x) {
return DATE_FORMATTER.format(x);
}
public static String formatLocalDateTime(LocalDateTime x) {
return DATE_TIME_FORMATTER.format(x);
}
/**
* Formats a {@link LocalTime} as "HH:mm:ss". There isn't any
* dedicated ClickHouse data type for times, so this is the most
* straightforward thing to do. It would be wrong for the JDBC driver to
* construct an artificial {@link ClickHouseDataType#DateTime DateTime}
* representation using a dummy date, e.g. 1970-01-01.
*
* @param x
* a {@link LocalTime} parameter
* @return {@code x} formatted as "HH:mm:ss"
*/
public static String formatLocalTime(LocalTime x) {
return TIME_FORMATTER.format(x);
}
public static String formatOffsetTime(OffsetTime x) {
return DateTimeFormatter.ISO_OFFSET_TIME.format(x);
}
public static String formatOffsetDateTime(OffsetDateTime x, TimeZone timeZone) {
return DATE_TIME_FORMATTER
.withZone(timeZone.toZoneId())
.format(x);
}
public static String formatZonedDateTime(ZonedDateTime x, TimeZone timeZone) {
return DATE_TIME_FORMATTER
.withZone(timeZone.toZoneId())
.format(x);
}
public static String formatInstant(Instant x, TimeZone timeZone) {
return DATE_TIME_FORMATTER
.withZone(timeZone.toZoneId())
.format(x);
}
public static String formatBitmap(ClickHouseBitmap bitmap) {
return bitmap.toBitmapBuildExpression();
}
public static String formatMap(Map map, TimeZone dateTimeZone, TimeZone dateTimeTimeZone) {
StringBuilder sb = new StringBuilder();
for (Map.Entry e : map.entrySet()) {
Object key = e.getKey();
Object value = e.getValue();
sb.append(',');
if (key instanceof String) {
sb.append('\'').append(formatString((String) key)).append('\'');
} else {
sb.append(formatObject(key, dateTimeZone, dateTimeTimeZone));
}
sb.append(':');
if (value instanceof String) {
sb.append('\'').append(formatString((String) value)).append('\'');
} else {
sb.append(formatObject(value, dateTimeZone, dateTimeTimeZone));
}
}
if (sb.length() > 0) {
sb.deleteCharAt(0);
}
return sb.insert(0, '{').append('}').toString();
}
public static String formatObject(Object x, TimeZone dateTimeZone,
TimeZone dateTimeTimeZone)
{
if (x == null) {
return null;
}
if (x instanceof Byte) {
return formatInt(((Byte) x).intValue());
} else if (x instanceof String) {
return formatString((String) x);
} else if (x instanceof BigDecimal) {
return formatBigDecimal((BigDecimal) x);
} else if (x instanceof Short) {
return formatShort(((Short) x).shortValue());
} else if (x instanceof Integer) {
return formatInt(((Integer) x).intValue());
} else if (x instanceof Long) {
return formatLong(((Long) x).longValue());
} else if (x instanceof Float) {
return formatFloat(((Float) x).floatValue());
} else if (x instanceof Double) {
return formatDouble(((Double) x).doubleValue());
} else if (x instanceof byte[]) {
return formatBytes((byte[]) x);
} else if (x instanceof Date) {
return formatDate((Date) x, dateTimeZone);
} else if (x instanceof LocalDate) {
return formatLocalDate((LocalDate) x);
} else if (x instanceof Time) {
return formatTime((Time) x, dateTimeTimeZone);
} else if (x instanceof LocalTime) {
return formatLocalTime((LocalTime) x);
} else if (x instanceof OffsetTime) {
return formatOffsetTime((OffsetTime) x);
} else if (x instanceof Timestamp) {
return formatTimestamp((Timestamp) x, dateTimeTimeZone);
} else if (x instanceof LocalDateTime) {
return formatLocalDateTime((LocalDateTime) x);
} else if (x instanceof OffsetDateTime) {
return formatOffsetDateTime((OffsetDateTime) x, dateTimeTimeZone);
} else if (x instanceof ZonedDateTime) {
return formatZonedDateTime((ZonedDateTime) x, dateTimeTimeZone);
} else if (x instanceof Boolean) {
return formatBoolean(((Boolean) x).booleanValue());
} else if (x instanceof UUID) {
return formatUUID((UUID) x);
} else if (x instanceof BigInteger) {
return formatBigInteger((BigInteger) x);
} else if (x instanceof Collection) {
return ClickHouseArrayUtil.toString((Collection) x, dateTimeZone, dateTimeTimeZone);
} else if (x instanceof ClickHouseBitmap) {
return formatBitmap((ClickHouseBitmap) x);
} else if (x instanceof Map) {
return formatMap((Map) x, dateTimeZone, dateTimeTimeZone);
} else if (x.getClass().isArray()) {
return ClickHouseArrayUtil.arrayToString(x, dateTimeZone, dateTimeTimeZone);
} else {
return String.valueOf(x);
}
}
public static boolean needsQuoting(Object o) {
if (o == null
|| o instanceof Array
|| o instanceof Boolean
|| o instanceof Collection
// || o instanceof Iterable
|| o instanceof Map
|| o instanceof Number
|| o.getClass().isArray()
|| o instanceof ClickHouseBitmap) {
return false;
}
return true;
}
private static SimpleDateFormat getDateFormat() {
return dateFormat.get();
}
private static SimpleDateFormat getDateTimeFormat() {
return dateTimeFormat.get();
}
private ClickHouseValueFormatter() { /* NOP */ }
}