From 39a4cbf0d6fd5ca96e327e36185299176abd3b72 Mon Sep 17 00:00:00 2001
From: Andrey Abramov <gnusmas@kvolitek.ru>
Date: Mon, 7 May 2018 21:17:48 +0300
Subject: [PATCH] fix valgrind issues (#5283)

---
 arangod/Aql/OptimizerRules.cpp | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/arangod/Aql/OptimizerRules.cpp b/arangod/Aql/OptimizerRules.cpp
index b9ba494c76..3b11f55805 100644
--- a/arangod/Aql/OptimizerRules.cpp
+++ b/arangod/Aql/OptimizerRules.cpp
@@ -3876,14 +3876,20 @@ class RestrictToSingleShardChecker final : public WalkerWorker<ExecutionNode> {
         break;
       }
       
-      case EN::ENUMERATE_COLLECTION: 
-      case EN::UPSERT: {
+      case EN::ENUMERATE_COLLECTION: {
         // track usage of the collection
         auto collection = static_cast<EnumerateCollectionNode const*>(en)->collection();
         _shardsUsed[collection].emplace("all");
         break;
       }
-      
+
+      case EN::UPSERT: {
+        // track usage of the collection
+        auto collection = static_cast<ModificationNode const*>(en)->collection();
+        _shardsUsed[collection].emplace("all");
+        break;
+      }
+
       case EN::INSERT:
       case EN::REPLACE:
       case EN::UPDATE: