/**
* Copyright (C) 2014-2016 LinkedIn Corp. (pinot-core@linkedin.com)
*
* 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.linkedin.pinot.core.operator.blocks;
import com.linkedin.pinot.common.data.FieldSpec;
import com.linkedin.pinot.common.data.FieldSpec.DataType;
import com.linkedin.pinot.core.common.Block;
import com.linkedin.pinot.core.common.BlockDocIdSet;
import com.linkedin.pinot.core.common.BlockDocIdValueSet;
import com.linkedin.pinot.core.common.BlockId;
import com.linkedin.pinot.core.common.BlockMetadata;
import com.linkedin.pinot.core.common.BlockValSet;
import com.linkedin.pinot.core.common.Predicate;
import com.linkedin.pinot.core.io.readerwriter.impl.FixedByteSingleColumnSingleValueReaderWriter;
import com.linkedin.pinot.core.operator.docvalsets.RealtimeSingleValueSet;
import com.linkedin.pinot.core.realtime.impl.dictionary.BaseOnHeapMutableDictionary;
import com.linkedin.pinot.core.segment.index.readers.Dictionary;
import org.roaringbitmap.buffer.MutableRoaringBitmap;
public class RealtimeSingleValueBlock implements Block {
private final MutableRoaringBitmap filteredBitmap;
final FieldSpec spec;
private final BaseOnHeapMutableDictionary dictionary;
final int docIdSearchableOffset;
final FixedByteSingleColumnSingleValueReaderWriter reader;
private Predicate p;
public RealtimeSingleValueBlock(MutableRoaringBitmap filteredBitmap, FieldSpec spec, BaseOnHeapMutableDictionary dictionary,
int offset, FixedByteSingleColumnSingleValueReaderWriter indexReader) {
this.spec = spec;
this.dictionary = dictionary;
this.filteredBitmap = filteredBitmap;
this.docIdSearchableOffset = offset;
this.reader = indexReader;
}
@Override
public BlockId getId() {
return null;
}
@Override
public boolean applyPredicate(Predicate predicate) {
this.p = predicate;
return true;
}
@Override
public BlockDocIdSet getBlockDocIdSet() {
if (this.p != null) {
return BlockFactory.getBlockDocIdSetBackedByBitmap(filteredBitmap);
}
return BlockFactory.getDummyBlockDocIdSet(docIdSearchableOffset);
}
@Override
public BlockValSet getBlockValueSet() {
return new RealtimeSingleValueSet(reader, docIdSearchableOffset + 1, spec.getDataType());
}
public FixedByteSingleColumnSingleValueReaderWriter getReader() {
return reader;
}
@Override
public BlockDocIdValueSet getBlockDocIdValueSet() {
return null;
}
@Override
public BlockMetadata getMetadata() {
return getDimensionOrTimeBlockMetadata();
}
private BlockMetadata getDimensionOrTimeBlockMetadata() {
return new BlockMetadata() {
@Override
public int getMaxNumberOfMultiValues() {
return 0;
}
@Override
public boolean isSparse() {
return false;
}
@Override
public boolean isSorted() {
return false;
}
@Override
public boolean isSingleValue() {
return true;
}
@Override
public boolean hasInvertedIndex() {
return true;
}
@Override
public boolean hasDictionary() {
return true;
}
@Override
public int getStartDocId() {
return 0;
}
@Override
public int getSize() {
return docIdSearchableOffset + 1;
}
@Override
public int getLength() {
return docIdSearchableOffset + 1;
}
@Override
public int getEndDocId() {
return docIdSearchableOffset;
}
@Override
public Dictionary getDictionary() {
return dictionary;
}
@Override
public DataType getDataType() {
return spec.getDataType();
}
};
}
}