Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,9 @@
import org.apache.arrow.vector.UInt8Vector;
import org.apache.arrow.vector.ValueVector;
import org.apache.arrow.vector.VarBinaryVector;
import org.apache.arrow.vector.ViewVarBinaryVector;
import org.apache.arrow.vector.VarCharVector;
import org.apache.arrow.vector.ViewVarCharVector;
import org.apache.arrow.vector.complex.DenseUnionVector;
import org.apache.arrow.vector.complex.FixedSizeListVector;
import org.apache.arrow.vector.complex.LargeListVector;
Expand Down Expand Up @@ -130,6 +132,9 @@ public static ArrowFlightJdbcAccessor createAccessor(
} else if (vector instanceof VarBinaryVector) {
return new ArrowFlightJdbcBinaryVectorAccessor(
(VarBinaryVector) vector, getCurrentRow, setCursorWasNull);
} else if (vector instanceof ViewVarBinaryVector) {
return new ArrowFlightJdbcBinaryVectorAccessor(
(ViewVarBinaryVector) vector, getCurrentRow, setCursorWasNull);
} else if (vector instanceof LargeVarBinaryVector) {
return new ArrowFlightJdbcBinaryVectorAccessor(
(LargeVarBinaryVector) vector, getCurrentRow, setCursorWasNull);
Expand Down Expand Up @@ -163,6 +168,9 @@ public static ArrowFlightJdbcAccessor createAccessor(
} else if (vector instanceof LargeVarCharVector) {
return new ArrowFlightJdbcVarCharVectorAccessor(
(LargeVarCharVector) vector, getCurrentRow, setCursorWasNull);
} else if (vector instanceof ViewVarCharVector) {
return new ArrowFlightJdbcVarCharVectorAccessor(
(ViewVarCharVector) vector, getCurrentRow, setCursorWasNull);
} else if (vector instanceof DurationVector) {
return new ArrowFlightJdbcDurationVectorAccessor(
(DurationVector) vector, getCurrentRow, setCursorWasNull);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
import org.apache.arrow.vector.FixedSizeBinaryVector;
import org.apache.arrow.vector.LargeVarBinaryVector;
import org.apache.arrow.vector.VarBinaryVector;
import org.apache.arrow.vector.ViewVarBinaryVector;

/**
* Accessor for the Arrow types: {@link FixedSizeBinaryVector}, {@link VarBinaryVector} and {@link
Expand Down Expand Up @@ -61,6 +62,13 @@ public ArrowFlightJdbcBinaryVectorAccessor(
this(vector::get, currentRowSupplier, setCursorWasNull);
}

public ArrowFlightJdbcBinaryVectorAccessor(
ViewVarBinaryVector vector,
IntSupplier currentRowSupplier,
ArrowFlightJdbcAccessorFactory.WasNullConsumer setCursorWasNull) {
this(vector::get, currentRowSupplier, setCursorWasNull);
}

private ArrowFlightJdbcBinaryVectorAccessor(
ByteArrayGetter getter,
IntSupplier currentRowSupplier,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@
import org.apache.arrow.driver.jdbc.utils.DateTimeUtils;
import org.apache.arrow.vector.LargeVarCharVector;
import org.apache.arrow.vector.VarCharVector;
import org.apache.arrow.vector.ViewVarCharVector;
import org.apache.arrow.vector.util.Text;

/** Accessor for the Arrow types: {@link VarCharVector} and {@link LargeVarCharVector}. */
Expand Down Expand Up @@ -62,6 +63,13 @@ public ArrowFlightJdbcVarCharVectorAccessor(
this(vector::get, currentRowSupplier, setCursorWasNull);
}

public ArrowFlightJdbcVarCharVectorAccessor(
ViewVarCharVector vector,
IntSupplier currentRowSupplier,
ArrowFlightJdbcAccessorFactory.WasNullConsumer setCursorWasNull) {
this(vector::get, currentRowSupplier, setCursorWasNull);
}

ArrowFlightJdbcVarCharVectorAccessor(
Getter getter,
IntSupplier currentRowSupplier,
Expand Down
Loading