Fix CompletionException wrapped exception (#817)

metadataStore will wrap the actual exception in CompletionException.
streamnative · Oct 19, 2021 · 61a0fdd · 61a0fdd
1 parent be27160
commit 61a0fdd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...n/java/io/streamnative/pulsar/handlers/kop/coordinator/transaction/ProducerIdManager.java b/...n/java/io/streamnative/pulsar/handlers/kop/coordinator/transaction/ProducerIdManager.java
@@ -178,21 +178,21 @@ private CompletableFuture<Long> conditionalUpdateData(byte[] data, long expectVe
  }
  updateFuture.complete(stat.getVersion());
  }).exceptionally(ex -> {
- if (ex instanceof MetadataStoreException.BadVersionException) {
+ if (ex.getCause() instanceof MetadataStoreException.BadVersionException) {
  checkProducerIdBlockMetadata(data)
  .thenAccept(updateFuture::complete).exceptionally(e -> {
  updateFuture.completeExceptionally(e);
  return null;
  });
- } else if (ex instanceof MetadataStoreException.NotFoundException) {
+ } else if (ex.getCause() instanceof MetadataStoreException.NotFoundException) {
  log.error("Update of path {} with data {} and expected version {} failed due to {}",
  KOP_PID_BLOCK_ZNODE, getProducerIdBlockStr(data), expectVersion,
  "NoNode for path " + KOP_PID_BLOCK_ZNODE);
  updateFuture.completeExceptionally(new Exception("NoNode for path " + KOP_PID_BLOCK_ZNODE));
  } else {
  log.error("Update of path {} with data {} and expected version {} exception: {}",
- KOP_PID_BLOCK_ZNODE, getProducerIdBlockStr(data), expectVersion, ex);
- updateFuture.completeExceptionally(new Exception("Error to update data.", ex));
+ KOP_PID_BLOCK_ZNODE, getProducerIdBlockStr(data), expectVersion, ex.getCause());
+ updateFuture.completeExceptionally(new Exception("Error to update data.", ex.getCause()));
  }
  return null;
  });