All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
ru.yandex.clickhouse.response.ByteFragmentUtils Maven / Gradle / Ivy
package ru.yandex.clickhouse.response;
import com.google.common.primitives.Primitives;
import java.math.BigInteger;
import java.sql.Date;
import java.sql.Timestamp;
import java.text.ParseException;
import java.text.SimpleDateFormat;
final class ByteFragmentUtils {
private static final char ARRAY_ELEMENTS_SEPARATOR = ',';
private static final char STRING_QUOTATION = '\'';
private ByteFragmentUtils() {
}
static int parseInt(ByteFragment s) throws NumberFormatException {
if (s == null) {
throw new NumberFormatException("null");
}
if (s.isNull()) {
return 0; //jdbc spec
}
int result = 0;
boolean negative = false;
int i = 0, max = s.length();
int limit;
int multmin;
int digit;
if (max > 0) {
if (s.charAt(0) == '-') {
negative = true;
limit = Integer.MIN_VALUE;
i++;
} else {
limit = -Integer.MAX_VALUE;
}
multmin = limit / 10;
if (i < max) {
digit = s.charAt(i++) - 0x30; //Character.digit(s.charAt(i++), 10);
if (digit < 0) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
} else {
result = -digit;
}
}
while (i < max) {
// Accumulating negatively avoids surprises near MAX_VALUE
digit = s.charAt(i++) - 0x30; // Character.digit(s.charAt(i++), 10);
if (digit < 0 || digit > 9) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
if (result < multmin) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
result *= 10;
if (result < limit + digit) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
result -= digit;
}
} else {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
if (negative) {
if (i > 1) {
return result;
} else { /* Only got "-" */
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
} else {
return -result;
}
}
static long parseLong(ByteFragment s) throws NumberFormatException {
if (s == null) {
throw new NumberFormatException("null");
}
if (s.isNull()) {
return 0; //jdbc spec
}
long result = 0;
boolean negative = false;
int i = 0, max = s.length();
long limit;
long multmin;
int digit;
if (max > 0) {
if (s.charAt(0) == '-') {
negative = true;
limit = Long.MIN_VALUE;
i++;
} else {
limit = -Long.MAX_VALUE;
}
multmin = limit / 10;
if (i < max) {
digit = s.charAt(i++) - 0x30; // Character.digit(s.charAt(i++), 10);
if (digit < 0 || digit > 9) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
} else {
result = -digit;
}
}
while (i < max) {
// Accumulating negatively avoids surprises near MAX_VALUE
digit = s.charAt(i++) - 0x30; // Character.digit(s.charAt(i++), 10);
if (digit < 0 || digit > 9) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
if (result < multmin) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
result *= 10;
if (result < limit + digit) {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
result -= digit;
}
} else {
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
if (negative) {
if (i > 1) {
return result;
} else { /* Only got "-" */
throw new NumberFormatException("For input string: \"" + s.asString() + '"');
}
} else {
return -result;
}
}
static Object parseArray(ByteFragment value, Class elementClass) {
return parseArray(value, elementClass, false, null);
}
static Object parseArray(ByteFragment value, Class elementClass, SimpleDateFormat dateFormat) {
return parseArray(value, elementClass, false, dateFormat);
}
static Object parseArray(ByteFragment value, Class elementClass, boolean useObjects) {
return parseArray(value, elementClass, useObjects, null);
}
static Object parseArray(ByteFragment value, Class elementClass, boolean useObjects, SimpleDateFormat dateFormat) {
if (value.isNull()) {
return null;
}
if (value.charAt(0) != '[' || value.charAt(value.length() - 1) != ']') {
throw new IllegalArgumentException("not an array: " + value);
}
if ((elementClass == Date.class || elementClass == Timestamp.class) && dateFormat == null) {
throw new IllegalArgumentException("DateFormat must be provided for date/dateTime array");
}
ByteFragment trim = value.subseq(1, value.length() - 2);
int index = 0;
Object array = java.lang.reflect.Array.newInstance(
useObjects ? elementClass : Primitives.unwrap(elementClass),
getArrayLength(trim)
);
int fieldStart = 0;
boolean inQuotation = false;
for (int chIdx = 0; chIdx < trim.length(); chIdx++) {
int ch = trim.charAt(chIdx);
if (ch == '\\') {
chIdx++;
}
inQuotation = ch == STRING_QUOTATION ^ inQuotation;
if (!inQuotation && ch == ARRAY_ELEMENTS_SEPARATOR || chIdx == trim.length() - 1) {
int fieldEnd = chIdx == trim.length() - 1 ? chIdx + 1 : chIdx;
if (trim.charAt(fieldStart) == '\'') {
fieldStart++;
fieldEnd--;
}
ByteFragment fragment = trim.subseq(fieldStart, fieldEnd - fieldStart);
if (elementClass == String.class) {
String stringValue = fragment.asString(true);
java.lang.reflect.Array.set(array, index++, stringValue);
} else if (elementClass == Long.class) {
long longValue = parseLong(fragment);
java.lang.reflect.Array.set(array, index++, longValue);
} else if (elementClass == Integer.class) {
int longValue = parseInt(fragment);
java.lang.reflect.Array.set(array, index++, longValue);
} else if (elementClass == BigInteger.class){
BigInteger bigIntegerValue = new BigInteger(fragment.asString(true));
java.lang.reflect.Array.set(array, index++, bigIntegerValue);
} else if (elementClass == Float.class) {
Float floatValue = Float.parseFloat(fragment.asString());
java.lang.reflect.Array.set(array, index++, floatValue);
} else if (elementClass == Double.class) {
Double doubleValue = Double.parseDouble(fragment.asString());
java.lang.reflect.Array.set(array, index++, doubleValue);
} else if (elementClass == Date.class) {
Date dateValue;
try {
dateValue = new Date(dateFormat.parse(fragment.asString()).getTime());
} catch (ParseException e) {
throw new IllegalArgumentException(e);
}
java.lang.reflect.Array.set(array, index++, dateValue);
} else if (elementClass == Timestamp.class) {
Timestamp dateTimeValue;
try {
dateTimeValue = new Timestamp(dateFormat.parse(fragment.asString()).getTime());
} catch (ParseException e) {
throw new IllegalArgumentException(e);
}
java.lang.reflect.Array.set(array, index++, dateTimeValue);
} else {
throw new IllegalStateException();
}
fieldStart = chIdx + 1;
}
}
return array;
}
private static int getArrayLength(ByteFragment value) {
if (value.length() == 0) {
return 0;
}
int length = 1;
boolean inQuotation = false;
for (int i = 0; i < value.length(); i++) {
int ch = value.charAt(i);
if (ch == '\\') {
i++;
}
inQuotation = ch == STRING_QUOTATION ^ inQuotation;
if (!inQuotation && ch == ARRAY_ELEMENTS_SEPARATOR) {
length++;
}
}
return length;
}
}
