| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 org.apache.drill.exec.vector.complex.impl; |
| |
| import org.apache.drill.common.exceptions.DrillRuntimeException; |
| import org.apache.drill.exec.expr.holders.DictHolder; |
| import org.apache.drill.exec.vector.complex.DictVector; |
| import org.apache.drill.exec.vector.complex.writer.BaseWriter; |
| import org.apache.drill.exec.vector.complex.writer.FieldWriter; |
| |
| public class SingleDictWriter extends AbstractRepeatedMapWriter<DictVector> implements BaseWriter.DictWriter { |
| |
| private boolean mapStarted; |
| |
| public SingleDictWriter(DictVector container, FieldWriter parent, boolean unionEnabled) { |
| super(container, parent, unionEnabled); |
| } |
| |
| public SingleDictWriter(DictVector container, FieldWriter parent) { |
| this(container, parent, false); |
| } |
| |
| @Override |
| public void start() { |
| assert !mapStarted : "Map should not be started"; |
| |
| // Make sure that the current vector can support the end position of this list. |
| if (container.getValueCapacity() <= idx()) { |
| container.getMutator().setValueCount(idx() + 1); |
| } |
| |
| DictHolder h = new DictHolder(); |
| container.getAccessor().get(idx(), h); |
| if (h.start >= h.end) { |
| container.getMutator().startNewValue(idx()); |
| } |
| |
| mapStarted = true; |
| } |
| |
| @Override |
| public void end() { |
| checkStarted(); |
| mapStarted = false; |
| } |
| |
| @Override |
| public void startKeyValuePair() { |
| checkStarted(); |
| currentChildIndex = container.getMutator().add(idx()); |
| for (FieldWriter w : fields.values()) { |
| w.setPosition(currentChildIndex); |
| } |
| } |
| |
| @Override |
| public void endKeyValuePair() { |
| checkStarted(); |
| // Check whether key was written |
| if (container.getKeys().getAccessor().getValueCount() == currentChildIndex) { |
| throw new DrillRuntimeException("Key in DICT cannot be null. Index: " + idx()); |
| } |
| // If value was not written, write it as null explicitly if supported |
| if (container.getValues() != null && container.isValueNullable() |
| && container.getValues().getAccessor().getValueCount() == currentChildIndex) { |
| ((AbstractFieldWriter) getValueWriter()).writeNull(); |
| } |
| } |
| |
| @Override |
| public FieldWriter getKeyWriter() { |
| return fields.get(DictVector.FIELD_KEY_NAME); |
| } |
| |
| @Override |
| public FieldWriter getValueWriter() { |
| return fields.get(DictVector.FIELD_VALUE_NAME); |
| } |
| |
| private void checkStarted() { |
| assert mapStarted : "Must start map (startRow()) before"; |
| } |
| } |