package org.apache.lucene.codecs.mocksep; /* * 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. */ import org.apache.lucene.store.IOContext; import org.apache.lucene.store.DataOutput; import org.apache.lucene.store.IndexOutput; import org.apache.lucene.store.Directory; import org.apache.lucene.util.IOUtils; import org.apache.lucene.codecs.CodecUtil; import org.apache.lucene.codecs.sep.IntIndexOutput; import java.io.IOException; /** Writes ints directly to the file (not in blocks) as * vInt. * * @lucene.experimental */ public class MockSingleIntIndexOutput extends IntIndexOutput { private final IndexOutput out; final static String CODEC = "SINGLE_INTS"; final static int VERSION_START = 0; final static int VERSION_CURRENT = VERSION_START; public MockSingleIntIndexOutput(Directory dir, String fileName, IOContext context) throws IOException { out = dir.createOutput(fileName, context); boolean success = false; try { CodecUtil.writeHeader(out, CODEC, VERSION_CURRENT); success = true; } finally { if (!success) { IOUtils.closeWhileHandlingException(out); } } } /** Write an int to the primary file */ @Override public void write(int v) throws IOException { out.writeVInt(v); } @Override public Index index() { return new MockSingleIntIndexOutputIndex(); } @Override public void close() throws IOException { out.close(); } @Override public String toString() { return "MockSingleIntIndexOutput fp=" + out.getFilePointer(); } private class MockSingleIntIndexOutputIndex extends IntIndexOutput.Index { long fp; long lastFP; @Override public void mark() { fp = out.getFilePointer(); } @Override public void copyFrom(IntIndexOutput.Index other, boolean copyLast) { fp = ((MockSingleIntIndexOutputIndex) other).fp; if (copyLast) { lastFP = ((MockSingleIntIndexOutputIndex) other).fp; } } @Override public void write(DataOutput indexOut, boolean absolute) throws IOException { if (absolute) { indexOut.writeVLong(fp); } else { indexOut.writeVLong(fp - lastFP); } lastFP = fp; } @Override public String toString() { return Long.toString(fp); } } }