org.apache.activemq.store.kahadb.disk.util.SequenceSet Maven / Gradle / Ivy
/**
* 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.activemq.store.kahadb.disk.util;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;
/**
* Keeps track of a added long values. Collapses ranges of numbers using a
* Sequence representation. Use to keep track of received message ids to find
* out if a message is duplicate or if there are any missing messages.
*
* @author chirino
*/
public class SequenceSet extends LinkedNodeList implements Iterable {
public static class Marshaller implements org.apache.activemq.store.kahadb.disk.util.Marshaller {
public static final Marshaller INSTANCE = new Marshaller();
public SequenceSet readPayload(DataInput in) throws IOException {
SequenceSet value = new SequenceSet();
int count = in.readInt();
for (int i = 0; i < count; i++) {
if( in.readBoolean() ) {
Sequence sequence = new Sequence(in.readLong(), in.readLong());
value.addLast(sequence);
} else {
Sequence sequence = new Sequence(in.readLong());
value.addLast(sequence);
}
}
return value;
}
public void writePayload(SequenceSet value, DataOutput out) throws IOException {
out.writeInt(value.size());
Sequence sequence = value.getHead();
while (sequence != null ) {
if( sequence.range() > 1 ) {
out.writeBoolean(true);
out.writeLong(sequence.first);
out.writeLong(sequence.last);
} else {
out.writeBoolean(false);
out.writeLong(sequence.first);
}
sequence = sequence.getNext();
}
}
public int getFixedSize() {
return -1;
}
public SequenceSet deepCopy(SequenceSet value) {
SequenceSet rc = new SequenceSet();
Sequence sequence = value.getHead();
while (sequence != null ) {
rc.add(new Sequence(sequence.first, sequence.last));
sequence = sequence.getNext();
}
return rc;
}
public boolean isDeepCopySupported() {
return true;
}
}
public void add(Sequence value) {
// TODO we can probably optimize this a bit
for(long i=value.first; i count ) {
Sequence rc = new Sequence(sequence.first, sequence.first+count-1);
sequence.first+=count;
return rc;
}
sequence = sequence.getNext();
}
return null;
}
/**
* @return all the id Sequences that are missing from this set that are not
* in between the range provided.
*/
public List getMissing(long first, long last) {
ArrayList rc = new ArrayList();
if (first > last) {
throw new IllegalArgumentException("First cannot be more than last");
}
if (isEmpty()) {
// We are missing all the messages.
rc.add(new Sequence(first, last));
return rc;
}
Sequence sequence = getHead();
while (sequence != null && first <= last) {
if (sequence.contains(first)) {
first = sequence.last + 1;
} else {
if (first < sequence.first) {
if (last < sequence.first) {
rc.add(new Sequence(first, last));
return rc;
} else {
rc.add(new Sequence(first, sequence.first - 1));
first = sequence.last + 1;
}
}
}
sequence = sequence.getNext();
}
if (first <= last) {
rc.add(new Sequence(first, last));
}
return rc;
}
/**
* @return all the Sequence that are in this list
*/
public List getReceived() {
ArrayList rc = new ArrayList(size());
Sequence sequence = getHead();
while (sequence != null) {
rc.add(new Sequence(sequence.first, sequence.last));
sequence = sequence.getNext();
}
return rc;
}
/**
* Returns true if the value given is contained within one of the
* sequences held in this set.
*
* @param value
* The value to search for in the set.
*
* @return true if the value is contained in the set.
*/
public boolean contains(long value) {
if (isEmpty()) {
return false;
}
Sequence sequence = getHead();
while (sequence != null) {
if (sequence.contains(value)) {
return true;
}
sequence = sequence.getNext();
}
return false;
}
public boolean contains(int first, int last) {
if (isEmpty()) {
return false;
}
Sequence sequence = getHead();
while (sequence != null) {
if (sequence.first <= first && first <= sequence.last ) {
return last <= sequence.last;
}
sequence = sequence.getNext();
}
return false;
}
/**
* Computes the size of this Sequence by summing the values of all
* the contained sequences.
*
* @return the total number of values contained in this set if it
* were to be iterated over like an array.
*/
public long rangeSize() {
long result = 0;
Sequence sequence = getHead();
while (sequence != null) {
result += sequence.range();
sequence = sequence.getNext();
}
return result;
}
public Iterator iterator() {
return new SequenceIterator();
}
private class SequenceIterator implements Iterator {
private Sequence currentEntry;
private long lastReturned = -1;
public SequenceIterator() {
currentEntry = getHead();
if (currentEntry != null) {
lastReturned = currentEntry.first - 1;
}
}
public boolean hasNext() {
return currentEntry != null;
}
public Long next() {
if (currentEntry == null) {
throw new NoSuchElementException();
}
if(lastReturned < currentEntry.first) {
lastReturned = currentEntry.first;
if (currentEntry.range() == 1) {
currentEntry = currentEntry.getNext();
}
} else {
lastReturned++;
if (lastReturned == currentEntry.last) {
currentEntry = currentEntry.getNext();
}
}
return lastReturned;
}
public void remove() {
throw new UnsupportedOperationException();
}
}
}