- private void executeModification(final AbstractModification modification) {
- final TransactionContextWrapper contextWrapper = getContextWrapper(modification.getPath());
- contextWrapper.maybeExecuteTransactionOperation(new TransactionOperation() {
- @Override
- protected void invoke(final TransactionContext transactionContext, final Boolean havePermit) {
- transactionContext.executeModification(modification, havePermit);
- }
- });
+ private void writeAllData(final ContainerNode rootData) {
+ // Open builders for all shards
+ final Map<String, DataContainerNodeBuilder<NodeIdentifier, ContainerNode>> rootBuilders = new HashMap<>();
+ for (String shardName : getActorUtils().getConfiguration().getAllShardNames()) {
+ rootBuilders.put(shardName, Builders.containerBuilder().withNodeIdentifier(rootData.getIdentifier()));
+ }
+
+ // Now distribute children as needed
+ for (DataContainerChild child : rootData.body()) {
+ final String shardName = shardNameFromRootChild(child);
+ verifyNotNull(rootBuilders.get(shardName), "Failed to find builder for %s", shardName).addChild(child);
+ }
+
+ // Now dispatch all writes
+ for (Entry<String, DataContainerNodeBuilder<NodeIdentifier, ContainerNode>> entry : rootBuilders.entrySet()) {
+ getContextWrapper(entry.getKey()).maybeExecuteTransactionOperation(new WriteOperation(
+ YangInstanceIdentifier.empty(), entry.getValue().build()));
+ }
+ }
+
+ private void executeModification(final TransactionModificationOperation operation) {
+ getContextWrapper(operation.path()).maybeExecuteTransactionOperation(operation);
+ }
+
+ private static ContainerNode checkRootData(final NormalizedNode data) {
+ // Root has to be a container
+ checkArgument(data instanceof ContainerNode, "Invalid root data %s", data);
+ return (ContainerNode) data;