GCP发布/订阅客户端,GrpcSubscriberStub,超时

时间:2020-01-23 15:06:32

标签: java google-cloud-pubsub

我正在尝试将GrpcSubscriberStub用于“同步拉” 在示例https://cloud.google.com/pubsub/docs/pull

SubscriberStubSettings subscriberStubSettings =
    SubscriberStubSettings.newBuilder()
        .setTransportChannelProvider(
            SubscriberStubSettings.defaultGrpcTransportProviderBuilder()
                .setMaxInboundMessageSize(20 << 20) // 20MB
                .build())
        .build();

try (SubscriberStub subscriber = GrpcSubscriberStub.create(subscriberStubSettings)) {
  // String projectId = "my-project-id";
  // String subscriptionId = "my-subscription-id";
  // int numOfMessages = 10;   // max number of messages to be pulled
  String subscriptionName = ProjectSubscriptionName.format(projectId, subscriptionId);
  PullRequest pullRequest =
      PullRequest.newBuilder()
          .setMaxMessages(numOfMessages)
          .setReturnImmediately(false) // return immediately if messages are not available
          .setSubscription(subscriptionName)
          .build();

  // use pullCallable().futureCall to asynchronously perform this operation
  PullResponse pullResponse = subscriber.pullCallable().call(pullRequest);
  List<String> ackIds = new ArrayList<>();
  for (ReceivedMessage message : pullResponse.getReceivedMessagesList()) {
    // handle received message
    // ...
    ackIds.add(message.getAckId());
  }
  // acknowledge received messages
  AcknowledgeRequest acknowledgeRequest =
      AcknowledgeRequest.newBuilder()
          .setSubscription(subscriptionName)
          .addAllAckIds(ackIds)
          .build();
  // use acknowledgeCallable().futureCall to asynchronously perform this operation
  subscriber.acknowledgeCallable().call(acknowledgeRequest);
  return pullResponse.getReceivedMessagesList();
}

但是我不知道如何使用选项“ setReturnImmediately(false)”设置拉取超时。

你能帮我吗?

1 个答案:

答案 0 :(得分:1)

要更改提取超时,您需要修改SubscriberStubSettings

例如,将拉动的总超时设置为30秒:

SubscriberStubSettings.Builder subscriberStubSettingsBuilder =
    SubscriberStubSettings.newBuilder();
subscriberStubSettingsBuilder
    .pullSettings()
    .setRetrySettings(
        subscriberStubSettingsBuilder.pullSettings().getRetrySettings().toBuilder()
            .setTotalTimeout(Duration.ofSeconds(30))
            .build());
SubscriberStubSettings subscriberStubSettings = subscriberStubSettingsBuilder.build();

这基于SubscriberStubSettings文档here