From 0a47bace72753a4d6dbee1c894030c47cc88c128 Mon Sep 17 00:00:00 2001
From: florijan <florijan@memgraph.io>
Date: Mon, 3 Apr 2017 14:56:16 +0200
Subject: [PATCH] Query::Interpreter-added support for missing ops

Reviewers: teon.banek, buda

Reviewed By: teon.banek

Subscribers: pullbot

Differential Revision: https://phabricator.memgraph.io/D215
---
 src/query/interpreter.hpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/query/interpreter.hpp b/src/query/interpreter.hpp
index 86fd25463..323e002f3 100644
--- a/src/query/interpreter.hpp
+++ b/src/query/interpreter.hpp
@@ -65,11 +65,17 @@ void Interpret(const std::string &query, GraphDbAccessor &db_accessor,
     }
   } else if (dynamic_cast<plan::CreateNode *>(logical_plan.get()) ||
              dynamic_cast<plan::CreateExpand *>(logical_plan.get()) ||
+             dynamic_cast<plan::SetProperty *>(logical_plan.get()) ||
+             dynamic_cast<plan::SetProperties *>(logical_plan.get()) ||
+             dynamic_cast<plan::SetLabels *>(logical_plan.get()) ||
+             dynamic_cast<plan::RemoveProperty *>(logical_plan.get()) ||
+             dynamic_cast<plan::RemoveLabels *>(logical_plan.get()) ||
              dynamic_cast<Delete *>(logical_plan.get())) {
     stream.Header(header);
     auto cursor = logical_plan.get()->MakeCursor(db_accessor);
-    while (cursor->Pull(frame, symbol_table))
-      continue;
+    while (cursor->Pull(frame, symbol_table)) continue;
+  } else {
+    throw QueryRuntimeException("Unknown top level LogicalOp");
   }
 
   clock_t end_time = clock();