使 geoQuery.addGeoQueryEventListener() 成为阻塞操作

Question

我已经为 Firestore 中的每个文档保存了包含一些虚拟数据和纬度/经度的商店列表。我想在 poi 附近获取文件。为此，我使用了 GeoFire 库。 现在，我可以对 geoQuery.addGeoQueryEventListener() 方法内部的数据进行排序，但此方法不是阻塞操作，因此我的代码在 geoQuery.addGeoQueryEventListener() 可以返回排序列表之前退出。下面是代码sn-p：

    List<QueryDocumentSnapshot> doc = new ArrayList<>();
    List<String> businessIds = new ArrayList<>();
    Set<Business> businessSet = new HashSet<>();

    GeoQuery geoQuery = geoFire.queryAtLocation(new GeoLocation(latitude, longitude), 2000);
    geoQuery.addGeoQueryEventListener(new GeoQueryEventListener() {

        @Override
        public void onKeyEntered(String key, GeoLocation location) {
            // TODO Auto-generated method stub
            businessIds.add(key);
        }

        @Override
        public void onKeyExited(String key) {
            // TODO Auto-generated method stub
        }

        @Override
        public void onKeyMoved(String key, GeoLocation location) {
            // TODO Auto-generated method stub
        }

        @Override
        public void onGeoQueryReady() {
            // TODO Auto-generated method stub
            System.out.println("All initial data has been loaded and events have been fired!");
        }

        @Override
        public void onGeoQueryError(DatabaseError error) {
            // TODO Auto-generated method stub
            System.err.println("There was an error with this query: " + error);
        }

    });
    geoQuery.removeAllListeners();

    ApiFuture<QuerySnapshot> query = firestore.collection("business").whereIn("businessId", businessIds).get();
    businessSet = query.get().getDocuments().stream()
            .map(b -> b.toObject(Business.class)).collect(Collectors.toSet());
    return businessSet;

在上面的代码中，我的程序在 onKeyEntered() 中可以添加到 businessIds 之前退出。

Answer 1

请注意，我不知道您正在使用哪些 API，但理论上，您应该能够使用 CompletableFuture，例如使用类似这样的东西（未测试）：

private CompletableFuture<List<String>> runGeoQuery(double latitude, double longitude) {
  CompletableFuture<List<String>> future = new CompletableFuture<>();
  GeoQuery geoQuery = geoFire.queryAtLocation(new GeoLocation(latitude, longitude), 2000);
  geoQuery.addGeoQueryEventListener(new GeoQueryEventListener() {

    private final List<String> businessIds = new ArrayList<>();

    @Override
    public void onKeyEntered(String key, GeoLocation location) {
      businessIds.add(key);
    }

    @Override
    public void onKeyExited(String key) {
    }

    @Override
    public void onKeyMoved(String key, GeoLocation location) {
    }

    @Override
    public void onGeoQueryReady() {
      future.complete(businessIds);
    }

    @Override
    public void onGeoQueryError(DatabaseError error) {
      future.completeExceptionally(error);
    }

  });
  return future;
}

然后，您可以使用#thenApply 或#thenCompose 对结果进行撰写，或者您可以简单地使用#get 阻止。

旁注，请确保在您的情况下可以进行屏蔽。 不应该没问题的一个例子是，当运行此代码的线程来自 HTTP 请求池（例如 Netty 的）时，因为这可能会产生争用并停止您的 Web 服务器处理新请求。