From e49b2ce0c93c38f465e6167d6baf184237cbcced Mon Sep 17 00:00:00 2001 From: chaokunyang Date: Thu, 26 Dec 2024 21:23:33 +0800 Subject: [PATCH] fix arrow register --- .../apache/fury/format/vectorized/ArrowSerializers.java | 7 +++++-- python/pyfury/_util.pyx | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/java/fury-format/src/main/java/org/apache/fury/format/vectorized/ArrowSerializers.java b/java/fury-format/src/main/java/org/apache/fury/format/vectorized/ArrowSerializers.java index c6f3c83f9d..a03c29a06f 100644 --- a/java/fury-format/src/main/java/org/apache/fury/format/vectorized/ArrowSerializers.java +++ b/java/fury-format/src/main/java/org/apache/fury/format/vectorized/ArrowSerializers.java @@ -31,6 +31,7 @@ import org.apache.arrow.vector.ipc.message.IpcOption; import org.apache.arrow.vector.ipc.message.MessageSerializer; import org.apache.fury.Fury; +import org.apache.fury.config.Language; import org.apache.fury.io.MemoryBufferReadableChannel; import org.apache.fury.io.MemoryBufferWritableChannel; import org.apache.fury.io.MockWritableChannel; @@ -166,8 +167,10 @@ public MemoryBuffer toBuffer() { } public static void registerSerializers(Fury fury) { - fury.register(ArrowTable.class, Types.ARROW_TABLE); - fury.register(VectorSchemaRoot.class, Types.ARROW_RECORD_BATCH); + if (fury.getLanguage() != Language.JAVA) { + fury.register(ArrowTable.class, Types.ARROW_TABLE); + fury.register(VectorSchemaRoot.class, Types.ARROW_RECORD_BATCH); + } fury.registerSerializer(ArrowTable.class, new ArrowTableSerializer(fury)); fury.registerSerializer(VectorSchemaRoot.class, new VectorSchemaRootSerializer(fury)); } diff --git a/python/pyfury/_util.pyx b/python/pyfury/_util.pyx index 6aff3694f0..9baaca5c04 100644 --- a/python/pyfury/_util.pyx +++ b/python/pyfury/_util.pyx @@ -364,7 +364,7 @@ cdef class Buffer: cpdef inline int32_t read_varint32(self): cdef uint32_t v = self.read_varuint32() - return (v >> 1) ^ -(v & 1); + return (v >> 1) ^ -(v & 1) cpdef inline int32_t read_varuint32(self): cdef: