org.neo4j.kernel.api.impl.index.IndexReaderStub.java Source code

Introduction

Here is the source code for org.neo4j.kernel.api.impl.index.IndexReaderStub.java

Source

/*
 * Copyright (c) 2002-2016 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * Neo4j is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.neo4j.kernel.api.impl.index;

import org.apache.commons.lang3.math.NumberUtils;
import org.apache.lucene.index.BinaryDocValues;
import org.apache.lucene.index.DocValues;
import org.apache.lucene.index.DocValuesType;
import org.apache.lucene.index.FieldInfo;
import org.apache.lucene.index.FieldInfos;
import org.apache.lucene.index.Fields;
import org.apache.lucene.index.IndexOptions;
import org.apache.lucene.index.LeafReader;
import org.apache.lucene.index.NumericDocValues;
import org.apache.lucene.index.SortedDocValues;
import org.apache.lucene.index.SortedNumericDocValues;
import org.apache.lucene.index.SortedSetDocValues;
import org.apache.lucene.index.StoredFieldVisitor;
import org.apache.lucene.util.Bits;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.Collections;
import java.util.Map;
import java.util.function.Function;

public class IndexReaderStub extends LeafReader {

    private Fields fields;
    private boolean allDeleted;
    private String[] elements = new String[0];
    private Function<String, NumericDocValues> ndvs = s -> DocValues.emptyNumeric();

    private IOException throwOnFields;
    private static FieldInfo DummyFieldInfo = new FieldInfo("id", 0, false, true, false, IndexOptions.DOCS,
            DocValuesType.NONE, -1, Collections.<String, String>emptyMap());

    public IndexReaderStub(boolean allDeleted, final String... elements) {
        this.allDeleted = allDeleted;
        this.elements = elements;
    }

    public IndexReaderStub(Fields fields) {
        this.fields = fields;
    }

    public IndexReaderStub(final NumericDocValues ndv) {
        this.ndvs = s -> ndv;
    }

    public IndexReaderStub(final Map<String, NumericDocValues> ndvs) {
        this.ndvs = s -> {
            NumericDocValues dv = ndvs.get(s);
            if (dv == null) {
                return DocValues.emptyNumeric();
            }
            return dv;
        };
    }

    public void setElements(String[] elements) {
        this.elements = elements;
    }

    @Override
    public void addCoreClosedListener(CoreClosedListener listener) {

    }

    @Override
    public void removeCoreClosedListener(CoreClosedListener listener) {

    }

    @Override
    public Fields fields() throws IOException {
        if (throwOnFields != null) {
            IOException exception = this.throwOnFields;
            throwOnFields = null;
            throw exception;
        }
        return fields;
    }

    @Override
    public NumericDocValues getNumericDocValues(String field) throws IOException {
        return ndvs.apply(field);
    }

    @Override
    public BinaryDocValues getBinaryDocValues(String field) throws IOException {
        return DocValues.emptyBinary();
    }

    @Override
    public SortedDocValues getSortedDocValues(String field) throws IOException {
        return DocValues.emptySorted();
    }

    @Override
    public SortedNumericDocValues getSortedNumericDocValues(String field) throws IOException {
        return DocValues.emptySortedNumeric(elements.length);
    }

    @Override
    public SortedSetDocValues getSortedSetDocValues(String field) throws IOException {
        return DocValues.emptySortedSet();
    }

    @Override
    public Bits getDocsWithField(String field) throws IOException {
        throw new RuntimeException("Not yet implemented.");
    }

    @Override
    public NumericDocValues getNormValues(String field) throws IOException {
        return DocValues.emptyNumeric();
    }

    @Override
    public FieldInfos getFieldInfos() {
        throw new RuntimeException("Not yet implemented.");
    }

    @Override
    public Bits getLiveDocs() {
        return new Bits() {
            @Override
            public boolean get(int index) {
                if (index >= elements.length) {
                    throw new IllegalArgumentException("Doc id out of range");
                }
                return !allDeleted;
            }

            @Override
            public int length() {
                return elements.length;
            }
        };
    }

    @Override
    public void checkIntegrity() throws IOException {
    }

    @Override
    public Fields getTermVectors(int docID) throws IOException {
        throw new RuntimeException("Not yet implemented.");
    }

    @Override
    public int numDocs() {
        return allDeleted ? 0 : elements.length;
    }

    @Override
    public int maxDoc() {
        return Math.max(maxValue(), elements.length) + 1;
    }

    @Override
    public void document(int docID, StoredFieldVisitor visitor) throws IOException {
        visitor.stringField(DummyFieldInfo, String.valueOf(docID).getBytes(StandardCharsets.UTF_8));
    }

    @Override
    protected void doClose() throws IOException {
    }

    private int maxValue() {
        return Arrays.stream(elements).mapToInt(value -> NumberUtils.toInt(value, 0)).max().getAsInt();
    }
}