From 1dbd2159092552bfe882dfb82f5c635a3744939f Mon Sep 17 00:00:00 2001
From: Karol Sobczak <karol.sobczak@karolsobczak.com>
Date: Wed, 13 Mar 2019 12:03:15 +0100
Subject: [PATCH] Add missing WindowOperator#isBlocked method

WindowOperator can block when spilling,
therefore it needs to explicitly support isBlocked
method.
---
 .../java/io/prestosql/operator/WindowOperator.java    | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/presto-main/src/main/java/io/prestosql/operator/WindowOperator.java b/presto-main/src/main/java/io/prestosql/operator/WindowOperator.java
index 4343974094c56..1796e1c5b0578 100644
--- a/presto-main/src/main/java/io/prestosql/operator/WindowOperator.java
+++ b/presto-main/src/main/java/io/prestosql/operator/WindowOperator.java
@@ -334,6 +334,17 @@ public boolean isFinished()
         return outputPages.isFinished();
     }
 
+    @Override
+    public ListenableFuture<?> isBlocked()
+    {
+        // We can block e.g. because of self-triggered spill
+        if (outputPages.isBlocked()) {
+            return outputPages.getBlockedFuture();
+        }
+
+        return NOT_BLOCKED;
+    }
+
     @Override
     public boolean needsInput()
     {