From f084447f34defc5b386df43f6771dc2c33936bb1 Mon Sep 17 00:00:00 2001
From: Philipp Bogensberger <philipp@crate.io>
Date: Wed, 6 May 2015 15:11:54 +0200
Subject: [PATCH] fixup! added UpsertByIdContext in order to register
 UpsertByIdTask in JobContextService

---
 sql/src/main/java/io/crate/jobs/UpsertByIdContext.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sql/src/main/java/io/crate/jobs/UpsertByIdContext.java b/sql/src/main/java/io/crate/jobs/UpsertByIdContext.java
index b0e952bb1f2d..99554f6daf85 100644
--- a/sql/src/main/java/io/crate/jobs/UpsertByIdContext.java
+++ b/sql/src/main/java/io/crate/jobs/UpsertByIdContext.java
@@ -77,6 +77,7 @@ public void onResponse(ShardUpsertResponse updateResponse) {
                     }
                     futureResult.set(TaskResult.ZERO);
                 }
+                close();
             }
 
             @Override
@@ -90,6 +91,7 @@ public void onFailure(Throwable e) {
                 } else {
                     futureResult.setException(e);
                 }
+                close();
             }
         });
     }