diff --git a/src/main/java/com/github/kerbity/kerb/result/CompletableResultSet.java b/src/main/java/com/github/kerbity/kerb/result/CompletableResultSet.java index 058d2ab..dfb7a7b 100644 --- a/src/main/java/com/github/kerbity/kerb/result/CompletableResultSet.java +++ b/src/main/java/com/github/kerbity/kerb/result/CompletableResultSet.java @@ -61,6 +61,23 @@ public CompletableResultSet(int size) { this.containsCancelled = false; } + /** + * Wait for the results to be completed. + * + * @return This instance with completed result set. + */ + public @NotNull CompletableResultSet waitForComplete() { + while (!this.isComplete()) { + try { + Thread.sleep(LOCK_TIME_MILLS); + } catch (InterruptedException e) { + throw new RuntimeException(e); + } + } + + return this; + } + /** * Used to wait for the final result. * This will park the current thread until the