Examples with AbstractBatchedModificationsCursor org.opendaylight.controller.cluster.datastore.utils.AbstractBatchedModificationsCursor used on opensource projects

Example 1 with AbstractBatchedModificationsCursor

use of org.opendaylight.controller.cluster.datastore.utils.AbstractBatchedModificationsCursor in project controller by opendaylight.

the class ShardCommitCoordinator method createForwardedBatchedModifications.

Collection<BatchedModifications> createForwardedBatchedModifications(final BatchedModifications from, final int maxModificationsPerBatch) {
    CohortEntry cohortEntry = cohortCache.remove(from.getTransactionId());
    if (cohortEntry == null || cohortEntry.getTransaction() == null) {
        return Collections.singletonList(from);
    }
    cohortEntry.applyModifications(from.getModifications());
    final LinkedList<BatchedModifications> newModifications = new LinkedList<>();
    cohortEntry.getTransaction().getSnapshot().applyToCursor(new AbstractBatchedModificationsCursor() {

        @Override
        protected BatchedModifications getModifications() {
            if (newModifications.isEmpty() || newModifications.getLast().getModifications().size() >= maxModificationsPerBatch) {
                newModifications.add(new BatchedModifications(from.getTransactionId(), from.getVersion()));
            }
            return newModifications.getLast();
        }
    });
    BatchedModifications last = newModifications.getLast();
    last.setDoCommitOnReady(from.isDoCommitOnReady());
    last.setReady(from.isReady());
    last.setTotalMessagesSent(newModifications.size());
    return newModifications;
}

Example 2 with AbstractBatchedModificationsCursor

use of org.opendaylight.controller.cluster.datastore.utils.AbstractBatchedModificationsCursor in project controller by opendaylight.

the class ShardCommitCoordinator method convertPendingTransactionsToMessages.

Collection<?> convertPendingTransactionsToMessages(final int maxModificationsPerBatch) {
    final Collection<VersionedExternalizableMessage> messages = new ArrayList<>();
    for (ShardDataTreeCohort cohort : dataTree.getAndClearPendingTransactions()) {
        CohortEntry cohortEntry = cohortCache.remove(cohort.getIdentifier());
        if (cohortEntry == null) {
            continue;
        }
        final Deque<BatchedModifications> newMessages = new ArrayDeque<>();
        cohortEntry.getDataTreeModification().applyToCursor(new AbstractBatchedModificationsCursor() {

            @Override
            protected BatchedModifications getModifications() {
                final BatchedModifications lastBatch = newMessages.peekLast();
                if (lastBatch != null && lastBatch.getModifications().size() >= maxModificationsPerBatch) {
                    return lastBatch;
                }
                // Allocate a new message
                final BatchedModifications ret = new BatchedModifications(cohortEntry.getTransactionId(), cohortEntry.getClientVersion());
                newMessages.add(ret);
                return ret;
            }
        });
        final BatchedModifications last = newMessages.peekLast();
        if (last != null) {
            final boolean immediate = cohortEntry.isDoImmediateCommit();
            last.setDoCommitOnReady(immediate);
            last.setReady(true);
            last.setTotalMessagesSent(newMessages.size());
            messages.addAll(newMessages);
            if (!immediate) {
                switch(cohort.getState()) {
                    case CAN_COMMIT_COMPLETE:
                    case CAN_COMMIT_PENDING:
                        messages.add(new CanCommitTransaction(cohortEntry.getTransactionId(), cohortEntry.getClientVersion()));
                        break;
                    case PRE_COMMIT_COMPLETE:
                    case PRE_COMMIT_PENDING:
                        messages.add(new CommitTransaction(cohortEntry.getTransactionId(), cohortEntry.getClientVersion()));
                        break;
                    default:
                        break;
                }
            }
        }
    }
    return messages;
}