Examples with ArrayByteIterable jetbrains.exodus.ArrayByteIterable used on opensource projects

Example 1 with ArrayByteIterable

use of jetbrains.exodus.ArrayByteIterable in project xodus by JetBrains.

the class CompoundByteIterator method hasNextImpl.

private boolean hasNextImpl() {
    while (!current.hasNext()) {
        currentAddress += read;
        final int alignment = ((int) currentAddress) & (log.getCachePageSize() - 1);
        final long alignedAddress = currentAddress - alignment;
        final ArrayByteIterable page = log.cache.getPageIterable(log, alignedAddress);
        final int readBytes = page.getLength();
        if (readBytes <= alignment) {
            // alignment is >= 0 for sure
            read = 0;
            offset = 0;
            return false;
        }
        read = readBytes - alignment;
        current = page.iterator(alignment);
        offset = current.getOffset();
    }
    return true;
}

Example 2 with ArrayByteIterable

use of jetbrains.exodus.ArrayByteIterable in project xodus by JetBrains.

the class EnvironmentTestInMemory method deleteRandomKey.

private void deleteRandomKey(Store primary, Store secondary, Transaction txn, int keysCount, Persistent23TreeMap.MutableMap<Integer, Integer> testMap) {
    final int key = rnd.nextInt(keysCount);
    testMap.remove(key);
    final ArrayByteIterable keyEntry = IntegerBinding.intToCompressedEntry(key);
    final ByteIterable oldValue = primary.get(txn, keyEntry);
    primary.delete(txn, keyEntry);
    if (oldValue != null) {
        try (Cursor cursor = secondary.openCursor(txn)) {
            Assert.assertTrue(cursor.getSearchBoth(oldValue, keyEntry));
            Assert.assertTrue(cursor.deleteCurrent());
        }
    }
}

Example 3 with ArrayByteIterable

use of jetbrains.exodus.ArrayByteIterable in project xodus by JetBrains.

the class PatriciaTreeMutable method put.

@Override
public boolean put(@NotNull final ByteIterable key, @NotNull final ByteIterable value) {
    final ByteIterator it = key.iterator();
    MutableNode node = root;
    MutableNode prev = null;
    byte prevFirstByte = (byte) 0;
    while (true) {
        final long matchResult = node.matchesKeySequence(it);
        final int matchingLength = NodeBase.MatchResult.getMatchingLength(matchResult);
        if (matchingLength < 0) {
            final MutableNode prefix = node.splitKey(-matchingLength - 1, NodeBase.MatchResult.getKeyByte(matchResult));
            if (NodeBase.MatchResult.hasNext(matchResult)) {
                prefix.hang(NodeBase.MatchResult.getNextByte(matchResult), it).setValue(value);
            } else {
                prefix.setValue(value);
            }
            if (prev == null) {
                root = new MutableRoot(prefix, root.sourceAddress);
            } else {
                prev.setChild(prevFirstByte, prefix);
            }
            ++size;
            return true;
        }
        if (!it.hasNext()) {
            final ByteIterable oldValue = node.getValue();
            node.setValue(value);
            if (oldValue == null) {
                ++size;
                return true;
            }
            return !oldValue.equals(value);
        }
        final byte nextByte = it.next();
        final NodeBase child = node.getChild(this, nextByte);
        if (child == null) {
            if (node.hasChildren() || node.hasKey() || node.hasValue()) {
                node.hang(nextByte, it).setValue(value);
            } else {
                node.setKeySequence(new ArrayByteIterable(nextByte, it));
                node.setValue(value);
            }
            ++size;
            return true;
        }
        prev = node;
        prevFirstByte = nextByte;
        final MutableNode mutableChild = child.getMutableCopy(this);
        if (!child.isMutable()) {
            node.setChild(nextByte, mutableChild);
        }
        node = mutableChild;
    }
}

Example 4 with ArrayByteIterable

use of jetbrains.exodus.ArrayByteIterable in project xodus by JetBrains.

the class PatriciaTreeMutable method add.

@Override
public boolean add(@NotNull final ByteIterable key, @NotNull final ByteIterable value) {
    final ByteIterator it = key.iterator();
    NodeBase node = root;
    MutableNode mutableNode = null;
    final Deque<ChildReferenceTransient> stack = new ArrayDeque<>();
    while (true) {
        final long matchResult = node.matchesKeySequence(it);
        final int matchingLength = NodeBase.MatchResult.getMatchingLength(matchResult);
        if (matchingLength < 0) {
            final MutableNode prefix = node.getMutableCopy(this).splitKey(-matchingLength - 1, NodeBase.MatchResult.getKeyByte(matchResult));
            if (NodeBase.MatchResult.hasNext(matchResult)) {
                prefix.hang(NodeBase.MatchResult.getNextByte(matchResult), it).setValue(value);
            } else {
                prefix.setValue(value);
            }
            if (stack.isEmpty()) {
                root = new MutableRoot(prefix, root.sourceAddress);
            } else {
                final ChildReferenceTransient parent = stack.pop();
                mutableNode = parent.mutate(this);
                mutableNode.setChild(parent.firstByte, prefix);
            }
            break;
        }
        if (!it.hasNext()) {
            if (node.hasValue()) {
                return false;
            }
            mutableNode = node.getMutableCopy(this);
            mutableNode.setValue(value);
            break;
        }
        final byte nextByte = it.next();
        final NodeBase child = node.getChild(this, nextByte);
        if (child == null) {
            mutableNode = node.getMutableCopy(this);
            if (mutableNode.hasChildren() || mutableNode.hasKey() || mutableNode.hasValue()) {
                mutableNode.hang(nextByte, it).setValue(value);
            } else {
                mutableNode.setKeySequence(new ArrayByteIterable(nextByte, it));
                mutableNode.setValue(value);
            }
            break;
        }
        stack.push(new ChildReferenceTransient(nextByte, node));
        node = child;
    }
    ++size;
    mutateUp(stack, mutableNode);
    return true;
}

Example 5 with ArrayByteIterable

use of jetbrains.exodus.ArrayByteIterable in project xodus by JetBrains.

the class PatriciaTreeWithDuplicates method get.

@Nullable
@Override
public ByteIterable get(@NotNull final ByteIterable key) {
    try (ITreeCursor cursor = treeNoDuplicates.openCursor()) {
        final ByteIterable value = cursor.getSearchKeyRange(getEscapedKeyWithSeparator(key));
        if (value != null && value != ByteIterable.EMPTY) {
            int keyLength = CompressedUnsignedLongByteIterable.getInt(value);
            if (key.getLength() == keyLength) {
                final ByteIterable noDupKey = new UnEscapingByteIterable(cursor.getKey());
                final byte[] noDupKeyBytes = noDupKey.getBytesUnsafe();
                if (ByteIterableUtil.compare(key.getBytesUnsafe(), keyLength, noDupKeyBytes, keyLength) == 0) {
                    return new ArrayByteIterable(Arrays.copyOfRange(noDupKeyBytes, // skip separator
                    keyLength + 1, noDupKey.getLength()));
                }
            }
        }
        return null;
    }
}