summaryrefslogtreecommitdiffstats
path: root/infra/translate-api/src
diff options
context:
space:
mode:
Diffstat (limited to 'infra/translate-api/src')
-rw-r--r--infra/translate-api/src/main/java/io/fd/honeycomb/translate/write/registry/WriterRegistry.java35
-rw-r--r--infra/translate-api/src/test/java/io/fd/honeycomb/translate/write/registry/BulkUpdateExceptionTest.java9
2 files changed, 39 insertions, 5 deletions
diff --git a/infra/translate-api/src/main/java/io/fd/honeycomb/translate/write/registry/WriterRegistry.java b/infra/translate-api/src/main/java/io/fd/honeycomb/translate/write/registry/WriterRegistry.java
index d281ee558..51e6415a2 100644
--- a/infra/translate-api/src/main/java/io/fd/honeycomb/translate/write/registry/WriterRegistry.java
+++ b/infra/translate-api/src/main/java/io/fd/honeycomb/translate/write/registry/WriterRegistry.java
@@ -152,10 +152,12 @@ public interface WriterRegistry {
/**
* Reverts changes that were successfully applied during bulk update before failure occurred.
*
+ * @param writeContext Non-closed {@code WriteContext} to be used by reverting logic.<br> <b>Do not use same
+ * write context as was used in previous write</b>
* @throws Reverter.RevertFailedException if revert fails
*/
- public void revertChanges() throws Reverter.RevertFailedException {
- reverter.revert();
+ public void revertChanges(@Nonnull final WriteContext writeContext) throws Reverter.RevertFailedException {
+ reverter.revert(writeContext);
}
public Set<InstanceIdentifier<?>> getFailedIds() {
@@ -172,10 +174,13 @@ public interface WriterRegistry {
/**
* Reverts changes that were successfully applied during bulk update before failure occurred. Changes are
* reverted in reverse order they were applied.
+ * Used {@code WriteContext} needs to be in non-closed state, creating fresh one for revert
+ * is recommended, same way as for write, to allow {@code Reverter} use same logic as write.
*
+ * @param writeContext Non-closed {@code WriteContext} to be used by reverting logic
* @throws RevertFailedException if not all of applied changes were successfully reverted
*/
- void revert() throws RevertFailedException;
+ void revert(@Nonnull final WriteContext writeContext) throws RevertFailedException;
/**
* Thrown when some of the changes applied during bulk update were not reverted.
@@ -209,5 +214,29 @@ public interface WriterRegistry {
return notRevertedChanges;
}
}
+
+ /**
+ * Thrown after bulk operation was successfully reverted,
+ * to maintain marking of transaction as failed,without double logging of
+ * cause of update fail(its logged before reverting in ModifiableDataTreeDelegator
+ */
+ @Beta
+ class RevertSuccessException extends TranslationException {
+ private final Set<InstanceIdentifier<?>> failedIds;
+
+ /**
+ * Constructs an RevertSuccessException.
+ *
+ * @param failedIds instance identifiers of the data objects that were not processed during bulk update.
+ */
+ public RevertSuccessException(@Nonnull final Set<InstanceIdentifier<?>> failedIds) {
+ super("Bulk update failed for: " + failedIds);
+ this.failedIds = failedIds;
+ }
+
+ public Set<InstanceIdentifier<?>> getFailedIds() {
+ return failedIds;
+ }
+ }
}
} \ No newline at end of file
diff --git a/infra/translate-api/src/test/java/io/fd/honeycomb/translate/write/registry/BulkUpdateExceptionTest.java b/infra/translate-api/src/test/java/io/fd/honeycomb/translate/write/registry/BulkUpdateExceptionTest.java
index c1880ca26..ae0c36d5b 100644
--- a/infra/translate-api/src/test/java/io/fd/honeycomb/translate/write/registry/BulkUpdateExceptionTest.java
+++ b/infra/translate-api/src/test/java/io/fd/honeycomb/translate/write/registry/BulkUpdateExceptionTest.java
@@ -20,6 +20,7 @@ import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.verify;
import com.google.common.collect.Sets;
+import io.fd.honeycomb.translate.write.WriteContext;
import java.util.HashSet;
import org.junit.Before;
import org.junit.Test;
@@ -31,6 +32,10 @@ import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
public class BulkUpdateExceptionTest {
private InstanceIdentifier<?> id = InstanceIdentifier.create(DataObject.class);
+
+ @Mock
+ private WriteContext writeContext;
+
@Mock
private WriterRegistry.Reverter reverter;
@@ -47,7 +52,7 @@ public class BulkUpdateExceptionTest {
assertEquals(failedIds, bulkUpdateException.getFailedIds());
- bulkUpdateException.revertChanges();
- verify(reverter).revert();
+ bulkUpdateException.revertChanges(writeContext);
+ verify(reverter).revert(writeContext);
}
} \ No newline at end of file