org.neo4j.kernel.api.database.enrichment.ValuesReader Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-kernel Show documentation
Show all versions of neo4j-kernel Show documentation
Neo4j kernel is a lightweight, embedded Java database designed to
store data structured as graphs rather than tables. For more
information, see http://neo4j.org.
/*
* Copyright (c) "Neo4j"
* Neo4j Sweden AB [https://neo4j.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see BY_ID =
ByteObjectMaps.immutable.from(List.of(ValuesReader.values()), ValuesReader::id, v -> v);
private final byte id;
private final Class valueClass;
private final ValueReader reader;
ValuesReader(byte id, Class valueClass, ValueReader reader) {
this.id = id;
this.valueClass = valueClass;
this.reader = reader;
}
public AnyValue read(ByteBuffer buffer) {
return reader.read(buffer);
}
public static ValuesReader forValueClass(Class type) {
for (ValuesReader valuesReader : BY_ID) {
if (valuesReader.valueClass.isAssignableFrom(type)) {
return valuesReader;
}
}
throw new IllegalArgumentException("Unsupported value type: " + type);
}
public static AnyValue from(ByteBuffer buffer) {
final var type = buffer.get();
final var reader = BY_ID.get(type);
if (reader == null) {
throw new IllegalArgumentException("Unsupported value type: " + type);
}
return reader.read(buffer);
}
public static String readJavaString(ByteBuffer buffer) {
final var bytes = new byte[buffer.getInt()];
buffer.get(bytes, 0, bytes.length);
return new String(bytes, StandardCharsets.UTF_8);
}
private static ByteValue readByte(ByteBuffer buffer) {
return Values.byteValue(buffer.get());
}
private static ByteArray readByteArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new byte[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.get();
}
return Values.byteArray(values);
}
private static BooleanValue readBoolean(ByteBuffer buffer) {
return Values.booleanValue(readJavaBoolean(buffer));
}
private static BooleanArray readBooleanArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new boolean[length];
for (var i = 0; i < length; i++) {
values[i] = readJavaBoolean(buffer);
}
return Values.booleanArray(values);
}
private static CharValue readChar(ByteBuffer buffer) {
return Values.charValue(buffer.getChar());
}
private static CharArray readCharArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new char[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getChar();
}
return Values.charArray(values);
}
private static ShortValue readShort(ByteBuffer buffer) {
return Values.shortValue(buffer.getShort());
}
private static ShortArray readShortArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new short[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getShort();
}
return Values.shortArray(values);
}
private static IntValue readInt(ByteBuffer buffer) {
return Values.intValue(buffer.getInt());
}
private static IntArray readIntArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new int[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getInt();
}
return Values.intArray(values);
}
private static LongValue readLong(ByteBuffer buffer) {
return Values.longValue(buffer.getLong());
}
private static LongArray readLongArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new long[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getLong();
}
return Values.longArray(values);
}
private static FloatValue readFloat(ByteBuffer buffer) {
return Values.floatValue(buffer.getFloat());
}
private static FloatArray readFloatArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new float[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getFloat();
}
return Values.floatArray(values);
}
private static DoubleValue readDouble(ByteBuffer buffer) {
return Values.doubleValue(buffer.getDouble());
}
private static DoubleArray readDoubleArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new double[length];
for (var i = 0; i < length; i++) {
values[i] = buffer.getDouble();
}
return Values.doubleArray(values);
}
private static TextValue readString(ByteBuffer buffer) {
return Values.stringValue(readJavaString(buffer));
}
private static TextArray readStringArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new String[length];
for (var i = 0; i < length; i++) {
values[i] = readJavaString(buffer);
}
return Values.stringArray(values);
}
private static PointValue readPoint(ByteBuffer buffer) {
final var crs = CoordinateReferenceSystem.get(buffer.getInt());
final var coordinates = new double[crs.getDimension()];
for (var i = 0; i < coordinates.length; i++) {
coordinates[i] = buffer.getDouble();
}
return Values.pointValue(crs, coordinates);
}
private static PointArray readPointArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new PointValue[length];
for (var i = 0; i < length; i++) {
values[i] = readPoint(buffer);
}
return Values.pointArray(values);
}
private static DurationValue readDuration(ByteBuffer buffer) {
final var months = buffer.getLong();
final var days = buffer.getLong();
final var seconds = buffer.getLong();
final var nanos = buffer.getInt();
return DurationValue.duration(months, days, seconds, nanos);
}
private static DurationArray readDurationArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new DurationValue[length];
for (var i = 0; i < length; i++) {
values[i] = readDuration(buffer);
}
return Values.durationArray(values);
}
private static DateValue readDate(ByteBuffer buffer) {
return DateValue.epochDate(buffer.getLong());
}
private static DateArray readDateArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new LocalDate[length];
for (var i = 0; i < length; i++) {
values[i] = LocalDate.ofEpochDay(buffer.getLong());
}
return Values.dateArray(values);
}
private static TimeValue readTime(ByteBuffer buffer) {
return TimeValue.time(readRawTime(buffer));
}
private static TimeArray readTimeArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var values = new OffsetTime[length];
for (var i = 0; i < length; i++) {
values[i] = readRawTime(buffer);
}
return Values.timeArray(values);
}
private static DateTimeValue readDateTime(ByteBuffer buffer) {
final long epocSeconds = buffer.getLong();
final int nanos = buffer.getInt();
return DateTimeValue.datetime(epocSeconds, nanos, toZoneId(buffer.getInt()));
}
private static DateTimeArray readDateTimeArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var array = new ZonedDateTime[length];
for (var i = 0; i < length; i++) {
final var epocSeconds = buffer.getLong();
final var nanos = buffer.getInt();
final var z = buffer.getInt();
array[i] = ZonedDateTime.ofInstant(Instant.ofEpochSecond(epocSeconds, nanos), toZoneId(z));
}
return dateTimeArray(array);
}
private static LocalTimeValue readLocalTime(ByteBuffer buffer) {
return LocalTimeValue.localTime(buffer.getLong());
}
private static LocalTimeArray readLocalTimeArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var array = new LocalTime[length];
for (var i = 0; i < length; i++) {
array[i] = LocalTime.ofNanoOfDay(buffer.getLong());
}
return localTimeArray(array);
}
private static LocalDateTimeValue readLocalDateTime(ByteBuffer buffer) {
final var epochSecond = buffer.getLong();
final var nanos = buffer.getInt();
return LocalDateTimeValue.localDateTime(epochSecond, nanos);
}
private static LocalDateTimeArray readLocalDateTimeArray(ByteBuffer buffer) {
final var length = buffer.getInt();
final var array = new LocalDateTime[length];
for (var i = 0; i < length; i++) {
final var epochSecond = buffer.getLong();
final var nanos = buffer.getInt();
array[i] = LocalDateTime.ofEpochSecond(epochSecond, nanos, UTC);
}
return localDateTimeArray(array);
}
private static VirtualPathValue readPath(ByteBuffer buffer) {
final var isDirect = readJavaBoolean(buffer);
final var nodeCount = buffer.getInt();
if (isDirect) {
final var beforePos = buffer.position();
final var nodes = new NodeValue[nodeCount];
for (int i = 0; i < nodes.length; i++) {
nodes[i] = (NodeValue) readNode(buffer);
}
final var relationships = new RelationshipValue[nodeCount - 1];
for (int i = 0; i < relationships.length; i++) {
relationships[i] = (RelationshipValue) readRelationship(buffer);
}
final var payloadSize = buffer.position() - beforePos;
return VirtualValues.path(nodes, relationships, payloadSize);
} else {
final var nodes = new long[nodeCount];
for (int i = 0; i < nodes.length; i++) {
nodes[i] = buffer.getLong();
}
final var relationships = new long[nodeCount - 1];
for (int i = 0; i < relationships.length; i++) {
relationships[i] = buffer.getLong();
}
return VirtualValues.pathReference(nodes, relationships);
}
}
private static VirtualNodeValue readNode(ByteBuffer buffer) {
final var isDirect = readJavaBoolean(buffer);
final var nodeId = buffer.getLong();
if (isDirect) {
final var elementId = readJavaString(buffer);
final var isDeleted = readJavaBoolean(buffer);
final var labels = readStringArray(buffer);
final var properties = readMap(buffer);
return VirtualValues.nodeValue(nodeId, elementId, labels, properties, isDeleted);
} else {
return VirtualValues.node(nodeId);
}
}
private static VirtualRelationshipValue readRelationship(ByteBuffer buffer) {
final var isDirect = readJavaBoolean(buffer);
final var relId = buffer.getLong();
if (isDirect) {
final var elementId = readJavaString(buffer);
final var type = readString(buffer);
final var isDeleted = readJavaBoolean(buffer);
final var startId = buffer.getLong();
final var startElementId = readJavaString(buffer);
final var endId = buffer.getLong();
final var endElementId = readJavaString(buffer);
final var properties = readMap(buffer);
return VirtualValues.relationshipValue(
relId,
elementId,
VirtualValues.node(startId, startElementId),
VirtualValues.node(endId, endElementId),
type,
properties,
isDeleted);
} else {
return VirtualValues.relationship(relId);
}
}
private static ListValue readList(ByteBuffer buffer) {
final var values = new AnyValue[buffer.getInt()];
for (int i = 0; i < values.length; i++) {
values[i] = from(buffer);
}
return VirtualValues.list(values);
}
private static MapValue readMap(ByteBuffer buffer) {
final var entryCount = buffer.getInt();
final var keys = new String[entryCount];
final var values = new AnyValue[entryCount];
for (int i = 0; i < entryCount; i++) {
keys[i] = readJavaString(buffer);
values[i] = from(buffer);
}
return VirtualValues.map(keys, values);
}
private static boolean readJavaBoolean(ByteBuffer buffer) {
return buffer.get() == 1;
}
private static OffsetTime readRawTime(ByteBuffer buffer) {
final var nanosOfDayUTC = buffer.getLong();
final var offsetSeconds = buffer.getInt();
return OffsetTime.ofInstant(Instant.ofEpochSecond(0, nanosOfDayUTC), ZoneOffset.ofTotalSeconds(offsetSeconds));
}
private static ZoneId toZoneId(int z) {
// if lowest bit is set to 1 then it's a shifted zone id
if ((z & 1) != 0) {
final String zoneId = TimeZones.map((short) (z >> 1));
return ZoneId.of(zoneId);
}
// otherwise it's a shifted offset seconds value
// preserve sign-bit for negative offsets
return ZoneOffset.ofTotalSeconds(z >> 1);
}
public byte id() {
return id;
}
@FunctionalInterface
public interface ValueReader {
T read(ByteBuffer buffer);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy