/
RecordPageSink.java
89 lines (79 loc) · 2.75 KB
/
RecordPageSink.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.facebook.presto.spi;
import com.facebook.presto.spi.block.Block;
import com.facebook.presto.spi.type.Type;
import io.airlift.slice.Slice;
import java.util.Collection;
import java.util.List;
import static com.facebook.presto.spi.type.BigintType.BIGINT;
import static java.util.Objects.requireNonNull;
public class RecordPageSink
implements ConnectorPageSink
{
private final RecordSink recordSink;
public RecordPageSink(RecordSink recordSink)
{
this.recordSink = requireNonNull(recordSink, "recordSink is null");
}
@Override
public Collection<Slice> finish()
{
return recordSink.commit();
}
@Override
public void abort()
{
recordSink.rollback();
}
@Override
public void appendPage(Page page, Block sampleWeightBlock)
{
Block[] blocks = page.getBlocks();
List<Type> columnTypes = recordSink.getColumnTypes();
for (int position = 0; position < page.getPositionCount(); position++) {
long sampleWeight = 1;
if (sampleWeightBlock != null) {
sampleWeight = BIGINT.getLong(sampleWeightBlock, position);
}
recordSink.beginRecord(sampleWeight);
for (int i = 0; i < blocks.length; i++) {
writeField(position, blocks[i], columnTypes.get(i));
}
recordSink.finishRecord();
}
}
private void writeField(int position, Block block, Type type)
{
if (block.isNull(position)) {
recordSink.appendNull();
return;
}
if (type.getJavaType() == boolean.class) {
recordSink.appendBoolean(type.getBoolean(block, position));
}
else if (type.getJavaType() == long.class) {
recordSink.appendLong(type.getLong(block, position));
}
else if (type.getJavaType() == double.class) {
recordSink.appendDouble(type.getDouble(block, position));
}
else if (type.getJavaType() == Slice.class) {
recordSink.appendString(type.getSlice(block, position).getBytes());
}
else {
recordSink.appendObject(type.getObject(block, position));
}
}
}