org.teamapps.universaldb.index.numeric.ShortIndex Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of universal-db Show documentation
Show all versions of universal-db Show documentation
Ultra fast TeamApps database
/*-
* ========================LICENSE_START=================================
* UniversalDB
* ---
* Copyright (C) 2014 - 2023 TeamApps.org
* ---
* 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.
* =========================LICENSE_END==================================
*/
package org.teamapps.universaldb.index.numeric;
import org.teamapps.universaldb.context.UserContext;
import org.teamapps.universaldb.index.*;
import org.teamapps.universaldb.index.buffer.common.PrimitiveEntryAtomicStore;
import org.teamapps.universaldb.model.FieldModel;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.EOFException;
import java.io.IOException;
import java.util.*;
public class ShortIndex extends AbstractIndex implements NumericIndex {
private PrimitiveEntryAtomicStore atomicStore;
public ShortIndex(FieldModel fieldModel, TableIndex tableIndex) {
super(fieldModel, tableIndex);
atomicStore = new PrimitiveEntryAtomicStore(tableIndex.getDataPath(), fieldModel.getName());
}
@Override
public IndexType getType() {
return IndexType.SHORT;
}
@Override
public Short getGenericValue(int id) {
return getValue(id);
}
@Override
public boolean isEmpty(int id) {
return getValue(id) == 0;
}
@Override
public void setGenericValue(int id, Short value) {
setValue(id, value);
}
@Override
public void removeValue(int id) {
setValue(id, (short) 0);
}
public short getValue(int id) {
return atomicStore.getShort(id);
}
public void setValue(int id, short value) {
atomicStore.setShort(id, value);
}
public List sortRecords(List sortEntries, boolean ascending, UserContext userContext) {
int order = ascending ? 1 : -1;
sortEntries.sort((o1, o2) -> {
short value1 = getValue(o1.getLeafId());
short value2 = getValue(o2.getLeafId());
return Short.compare(value1, value2) * order;
});
return sortEntries;
}
@Override
public void dumpIndex(DataOutputStream dataOutputStream, BitSet records) throws IOException {
for (int id = records.nextSetBit(0); id >= 0; id = records.nextSetBit(id + 1)) {
short value = getValue(id);
dataOutputStream.writeInt(id);
dataOutputStream.writeShort(value);
}
}
@Override
public void restoreIndex(DataInputStream dataInputStream) throws IOException {
try {
int id = dataInputStream.readInt();
short value = dataInputStream.readShort();
setValue(id, value);
} catch (EOFException ignore) {}
}
@Override
public void close() {
atomicStore.close();
}
@Override
public void drop() {
atomicStore.drop();
}
@Override
public BitSet filter(BitSet records, NumericFilter numericFilter) {
Set set = new HashSet<>();
if (numericFilter.getValues() != null) {
for (Number value : numericFilter.getValues()) {
set.add(value.shortValue());
}
}
switch (numericFilter.getFilterType()) {
case EQUALS:
return filterEquals(records, numericFilter.getValue1().shortValue());
case NOT_EQUALS:
return filterNotEquals(records, numericFilter.getValue1().shortValue());
case GREATER:
return filterGreater(records, numericFilter.getValue1().shortValue());
case GREATER_EQUALS:
return filterGreaterOrEquals(records, numericFilter.getValue1().shortValue());
case SMALLER:
return filterSmaller(records, numericFilter.getValue1().shortValue());
case SMALLER_EQUALS:
return filterSmallerOrEquals(records, numericFilter.getValue1().shortValue());
case BETWEEN:
return filterBetween(records, numericFilter.getValue1().shortValue(), numericFilter.getValue2().shortValue());
case BETWEEN_EXCLUSIVE:
return filterBetweenExclusive(records, numericFilter.getValue1().shortValue(), numericFilter.getValue2().shortValue());
case CONTAINS:
return filterContains(records, set);
case CONTAINS_NOT:
return filterContainsNot(records, set);
}
return null;
}
public BitSet filterEquals(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value == compare) {
result.set(id);
}
}
return result;
}
public BitSet filterNotEquals(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value != compare) {
result.set(id);
}
}
return result;
}
public BitSet filterGreater(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value > compare) {
result.set(id);
}
}
return result;
}
public BitSet filterGreaterOrEquals(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value >= compare) {
result.set(id);
}
}
return result;
}
public BitSet filterSmaller(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value < compare) {
result.set(id);
}
}
return result;
}
public BitSet filterSmallerOrEquals(BitSet bitSet, short compare) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value <= compare) {
result.set(id);
}
}
return result;
}
public BitSet filterBetween(BitSet bitSet, short start, short end) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value >= start && value <= end) {
result.set(id);
}
}
return result;
}
public BitSet filterBetweenExclusive(BitSet bitSet, short start, short end) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (value > start && value < end) {
result.set(id);
}
}
return result;
}
public BitSet filterContains(BitSet bitSet, Set set) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (set.contains(value)) {
result.set(id);
}
}
return result;
}
public BitSet filterContainsNot(BitSet bitSet, Set set) {
BitSet result = new BitSet();
for (int id = bitSet.nextSetBit(0); id >= 0; id = bitSet.nextSetBit(id + 1)) {
short value = getValue(id);
if (!set.contains(value)) {
result.set(id);
}
}
return result;
}
}