Skip to content

Commit 2067e01

Browse files
committed
update format
1 parent aacf61e commit 2067e01

31 files changed

+417
-195
lines changed

java/vector/src/main/codegen/data/ArrowTypes.tdd

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,15 @@
3030
},
3131
{
3232
name: "Union",
33-
fields: [{name: "mode", type: int}]
33+
fields: [{name: "mode", type: short}]
3434
},
3535
{
3636
name: "Int",
3737
fields: [{name: "bitWidth", type: int}, {name: "isSigned", type: boolean}]
3838
},
3939
{
4040
name: "FloatingPoint",
41-
fields: [{name: precision, type: int}]
41+
fields: [{name: precision, type: short}]
4242
},
4343
{
4444
name: "Utf8",

java/vector/src/main/codegen/templates/ArrowType.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,14 @@ public int getType(FlatBufferBuilder builder) {
8787
return ${field.name};
8888
}
8989
</#list>
90+
91+
public String toString() {
92+
return "${name}{"
93+
<#list fields as field>
94+
+ ", " + ${field.name}
95+
</#list>
96+
+ "}";
97+
}
9098

9199
@Override
92100
public int hashCode() {

java/vector/src/main/codegen/templates/NullableValueVectors.java

Lines changed: 29 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@
2929

3030
package org.apache.arrow.vector;
3131

32+
import org.apache.arrow.vector.schema.ArrowFieldNode;
33+
import java.util.Collections;
34+
3235
<#include "/@includes/vv_imports.ftl" />
3336

3437
/**
@@ -39,7 +42,7 @@
3942
* NB: this class is automatically generated from ${.template_name} and ValueVectorTypes.tdd using FreeMarker.
4043
*/
4144
@SuppressWarnings("unused")
42-
public final class ${className} extends BaseDataValueVector implements <#if type.major == "VarLen">VariableWidth<#else>FixedWidth</#if>Vector, NullableVector{
45+
public final class ${className} extends BaseDataValueVector implements <#if type.major == "VarLen">VariableWidth<#else>FixedWidth</#if>Vector, NullableVector, FieldVector{
4346
private static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(${className}.class);
4447

4548
private final FieldReader reader = new ${minor.class}ReaderImpl(Nullable${minor.class}Vector.this);
@@ -88,9 +91,9 @@ public final class ${className} extends BaseDataValueVector implements <#if type
8891
<#elseif minor.class == "Time">
8992
field = new Field(name, true, new org.apache.arrow.vector.types.pojo.ArrowType.Time(), null);
9093
<#elseif minor.class == "Float4">
91-
field = new Field(name, true, new FloatingPoint(0), null);
94+
field = new Field(name, true, new FloatingPoint(org.apache.arrow.flatbuf.Precision.SINGLE), null);
9295
<#elseif minor.class == "Float8">
93-
field = new Field(name, true, new FloatingPoint(1), null);
96+
field = new Field(name, true, new FloatingPoint(org.apache.arrow.flatbuf.Precision.DOUBLE), null);
9497
<#elseif minor.class == "TimeStamp">
9598
field = new Field(name, true, new org.apache.arrow.vector.types.pojo.ArrowType.Timestamp(""), null);
9699
<#elseif minor.class == "IntervalDay">
@@ -107,6 +110,29 @@ public final class ${className} extends BaseDataValueVector implements <#if type
107110
}
108111
</#if>
109112

113+
/**
114+
* Initializes the child vectors
115+
* to be later loaded with loadBuffers
116+
* @param children
117+
*/
118+
public void initializeChildrenFromFields(List<Field> children) {
119+
throw new UnsupportedOperationException();
120+
}
121+
122+
public List<FieldVector> getChildrenFromFields() {
123+
return Collections.emptyList();
124+
}
125+
126+
public void loadFieldBuffers(ArrowFieldNode fieldNode, List<ArrowBuf> ownBuffers) {
127+
throw new UnsupportedOperationException();
128+
}
129+
130+
public List<ArrowBuf> getFieldBuffers() {
131+
bits.getBuffer().readerIndex(0);
132+
values.getBuffer().readerIndex(0);
133+
return Arrays.asList(bits.getBuffer(), values.getBuffer());
134+
}
135+
110136
@Override
111137
public Field getField() {
112138
return field;

java/vector/src/main/java/org/apache/arrow/schema/ArrowVectorType.java

Lines changed: 0 additions & 11 deletions
This file was deleted.
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
package org.apache.arrow.vector;
2+
3+
import java.util.List;
4+
5+
import org.apache.arrow.vector.schema.ArrowFieldNode;
6+
import org.apache.arrow.vector.types.pojo.Field;
7+
8+
import io.netty.buffer.ArrowBuf;
9+
10+
public interface FieldVector extends ValueVector {
11+
12+
/**
13+
* Initializes the child vectors
14+
* to be later loaded with loadBuffers
15+
* @param children
16+
*/
17+
void initializeChildrenFromFields(List<Field> children);
18+
19+
List<FieldVector> getChildrenFromFields();
20+
21+
void loadFieldBuffers(ArrowFieldNode fieldNode, List<ArrowBuf> ownBuffers);
22+
23+
/**
24+
* Returns the own buffers for this vector
25+
* @return the
26+
*/
27+
List<ArrowBuf> getFieldBuffers();
28+
}

java/vector/src/main/java/org/apache/arrow/vector/ValueVector.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@
1919

2020
import java.io.Closeable;
2121

22-
import io.netty.buffer.ArrowBuf;
23-
2422
import org.apache.arrow.memory.BufferAllocator;
2523
import org.apache.arrow.memory.OutOfMemoryException;
2624
import org.apache.arrow.vector.complex.reader.FieldReader;
2725
import org.apache.arrow.vector.types.Types.MinorType;
28-
import org.apache.arrow.vector.util.TransferPair;
2926
import org.apache.arrow.vector.types.pojo.Field;
27+
import org.apache.arrow.vector.util.TransferPair;
28+
29+
import io.netty.buffer.ArrowBuf;
3030

3131
/**
3232
* An abstraction that is used to store a sequence of values in an individual column.
@@ -137,6 +137,8 @@ public interface ValueVector extends Closeable, Iterable<ValueVector> {
137137
*/
138138
// SerializedField getMetadata();
139139

140+
// TypeLayout getTypeLayout();
141+
140142
/**
141143
* Returns the number of bytes that is used by this vector instance.
142144
*/
@@ -221,4 +223,5 @@ interface Mutator {
221223
@Deprecated
222224
void generateTestData(int values);
223225
}
226+
224227
}

java/vector/src/main/java/org/apache/arrow/vector/VectorLoader.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,11 @@
2121
import java.util.List;
2222

2323
import org.apache.arrow.flatbuf.Precision;
24-
import org.apache.arrow.schema.ArrowFieldNode;
2524
import org.apache.arrow.vector.complex.ComplexVectorLoader;
2625
import org.apache.arrow.vector.complex.ListVector;
2726
import org.apache.arrow.vector.complex.MapVector;
2827
import org.apache.arrow.vector.complex.NestedVector;
28+
import org.apache.arrow.vector.schema.ArrowFieldNode;
2929
import org.apache.arrow.vector.types.Types;
3030
import org.apache.arrow.vector.types.Types.MinorType;
3131
import org.apache.arrow.vector.types.pojo.ArrowType;

java/vector/src/main/java/org/apache/arrow/vector/complex/ComplexVectorLoader.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
1919

2020
import java.util.Iterator;
2121

22-
import org.apache.arrow.schema.ArrowFieldNode;
2322
import org.apache.arrow.vector.VectorLoader;
23+
import org.apache.arrow.vector.schema.ArrowFieldNode;
2424

2525
import io.netty.buffer.ArrowBuf;
2626

java/vector/src/main/java/org/apache/arrow/vector/complex/MapVector.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,12 @@
2727

2828
import org.apache.arrow.memory.BufferAllocator;
2929
import org.apache.arrow.vector.BaseValueVector;
30+
import org.apache.arrow.vector.FieldVector;
3031
import org.apache.arrow.vector.ValueVector;
3132
import org.apache.arrow.vector.complex.impl.SingleMapReaderImpl;
3233
import org.apache.arrow.vector.complex.reader.FieldReader;
3334
import org.apache.arrow.vector.holders.ComplexHolder;
35+
import org.apache.arrow.vector.types.Types;
3436
import org.apache.arrow.vector.types.Types.MinorType;
3537
import org.apache.arrow.vector.types.pojo.ArrowType.Tuple;
3638
import org.apache.arrow.vector.types.pojo.Field;
@@ -323,4 +325,25 @@ public void close() {
323325

324326
super.close();
325327
}
328+
329+
private List<FieldVector> fieldChildren;
330+
331+
public void initializeChildren(List<Field> children) {
332+
if (fieldChildren != null) {
333+
throw new IllegalArgumentException(children.toString()); //TODO
334+
}
335+
for (Field field : children) {
336+
MinorType minorType = Types.getMinorTypeForArrowType(field.getType());
337+
FieldVector vector = (FieldVector)this.add(field.getName(), minorType);
338+
fieldChildren.add(vector);
339+
// TODO: clean this up
340+
vector.initializeChildrenFromFields(field.getChildren());
341+
}
342+
}
343+
344+
public List<FieldVector> getFieldVectors() {
345+
// TODO: clean this up
346+
return (List<FieldVector>)(List<?>)getChildren();
347+
}
348+
326349
}

java/vector/src/main/java/org/apache/arrow/file/ArrowBlock.java renamed to java/vector/src/main/java/org/apache/arrow/vector/file/ArrowBlock.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515
* See the License for the specific language governing permissions and
1616
* limitations under the License.
1717
*/
18-
package org.apache.arrow.file;
18+
package org.apache.arrow.vector.file;
1919

2020
import org.apache.arrow.flatbuf.Block;
21-
import org.apache.arrow.schema.FBSerializable;
21+
import org.apache.arrow.vector.schema.FBSerializable;
2222

2323
import com.google.flatbuffers.FlatBufferBuilder;
2424

0 commit comments

Comments
 (0)