Technoboy- commented on code in PR #14149:
URL: https://github.com/apache/pulsar/pull/14149#discussion_r848435679
##########
pulsar-broker/src/main/java/org/apache/pulsar/broker/admin/impl/PersistentTopicsBase.java:
##########
@@ -327,49 +327,61 @@ protected void internalDeleteTopicForcefully(boolean
authoritative, boolean dele
}
}
- private void revokePermissions(String topicUri, String role) {
+ private CompletableFuture<Void> revokePermissions(String topicUri, String
role) {
Policies policies;
try {
policies = namespaceResources().getPolicies(namespaceName)
.orElseThrow(() -> new RestException(Status.NOT_FOUND,
"Namespace does not exist"));
} catch (Exception e) {
log.error("[{}] Failed to revoke permissions for topic {}",
clientAppId(), topicUri, e);
- throw new RestException(e);
+ return FutureUtil.failedFuture(new RestException(e));
}
if
(!policies.auth_policies.getTopicAuthentication().containsKey(topicUri)
||
!policies.auth_policies.getTopicAuthentication().get(topicUri).containsKey(role))
{
log.warn("[{}] Failed to revoke permission from role {} on topic:
Not set at topic level {}", clientAppId(),
role, topicUri);
- throw new RestException(Status.PRECONDITION_FAILED, "Permissions
are not set at the topic level");
- }
- try {
- // Write the new policies to metadata store
- namespaceResources().setPolicies(namespaceName, p -> {
-
p.auth_policies.getTopicAuthentication().get(topicUri).remove(role);
- return p;
- });
- log.info("[{}] Successfully revoke access for role {} - topic {}",
clientAppId(), role, topicUri);
- } catch (Exception e) {
- log.error("[{}] Failed to revoke permissions for topic {}",
clientAppId(), topicUri, e);
- throw new RestException(e);
+ return FutureUtil.failedFuture(new
RestException(Status.PRECONDITION_FAILED,
+ "Permissions are not set at the topic level"));
}
+ // Write the new policies to metadata store
+ return namespaceResources().setPoliciesAsync(namespaceName, p -> {
+
p.auth_policies.getTopicAuthentication().get(topicUri).remove(role);
+ return p;
+ }).thenAccept(__ -> log.info("[{}] Successfully revoke access for role
{} - topic {}", clientAppId(), role,
+ topicUri)
+
+ ).exceptionally(ex -> {
+ Throwable realCause = FutureUtil.unwrapCompletionException(ex);
+ log.error("[{}] Failed revoke access for role {} - topic {}",
clientAppId(), role, topicUri, realCause);
+ throw new RestException(realCause);
+
+ });
}
- protected void internalRevokePermissionsOnTopic(String role) {
+ protected void internalRevokePermissionsOnTopic(AsyncResponse
asyncResponse, String role) {
// This operation should be reading from zookeeper and it should be
allowed without having admin privileges
validateAdminAccessForTenant(namespaceName.getTenant());
- validatePoliciesReadOnlyAccess();
-
- PartitionedTopicMetadata meta = getPartitionedTopicMetadata(topicName,
true, false);
- int numPartitions = meta.partitions;
- if (numPartitions > 0) {
- for (int i = 0; i < numPartitions; i++) {
- TopicName topicNamePartition = topicName.getPartition(i);
- revokePermissions(topicNamePartition.toString(), role);
- }
- }
- revokePermissions(topicName.toString(), role);
+ validatePoliciesReadOnlyAccessAsync().thenCompose(__ ->
+ getPartitionedTopicMetadataAsync(topicName, true, false)
+ .thenCompose(metadata -> {
+ int numPartitions = metadata.partitions;
+ CompletableFuture<Void> future =
CompletableFuture.completedFuture(null);
+ if (numPartitions > 0) {
+ for (int i = 0; i < numPartitions; i++) {
+ TopicName topicNamePartition =
topicName.getPartition(i);
+ future = future.thenComposeAsync(unused ->
revokePermissions(topicNamePartition.toString(),
+ role));
+ }
+ }
+ return future.thenComposeAsync(unused ->
revokePermissions(topicName.toString(), role))
Review Comment:
Why will this cause deadlock ?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
