Scheduler: restore a couple of missing pieces from the last merge

auto-ndp · Dec 9, 2021 · 98dd0ec · 98dd0ec
1 parent dad6942
commit 98dd0ec
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 11 deletions.
diff --git a/.clang-tidy b/.clang-tidy
@@ -1,13 +1,17 @@
 ---
 
 Checks: >
-    clang-analyzer-*,
-    -clang-diagnostic-unknown-attributes,
-    -clang-diagnostic-return-type
-    modernize-*,
-    -modernize-use-trailing-return-type,
-    readability-*,
-    -readability-magic-numbers,
-    -readability-redundant-string-cstr,
-    -readability-implicit-bool-conversion
+  clang-analyzer-*,
+  -clang-diagnostic-unknown-attributes,
+  -clang-diagnostic-return-type
+  modernize-*,
+  -modernize-use-trailing-return-type,
+  readability-*,
+  -readability-magic-numbers,
+  -readability-redundant-string-cstr,
+  -readability-implicit-bool-conversion,
+  -readability-function-cognitive-complexity,
+  -readability-function-size,
+  -readability-identifier-length,
+  -readability-else-after-return
 
diff --git a/src/scheduler/Scheduler.cpp b/src/scheduler/Scheduler.cpp
@@ -442,11 +442,11 @@ faabric::util::SchedulingDecision Scheduler::makeSchedulingDecision(
         if (remainder > 0) {
             std::vector<std::string> unregisteredHosts;
             if (hostKindDifferent) {
-                unregisteredHosts = getUnregisteredHosts(funcStr);
-            } else {
                 for (auto&& h : getAvailableHostsForFunction(firstMsg)) {
                     unregisteredHosts.push_back(std::move(h));
                 }
+            } else {
+                unregisteredHosts = getUnregisteredHosts(funcStr);
             }
 
             for (const auto& h : unregisteredHosts) {
@@ -737,6 +737,11 @@ faabric::util::SchedulingDecision Scheduler::doCallFunctions(
                 auto* newMsg = hostRequest->add_messages();
                 *newMsg = req->messages().at(msgIdx);
                 newMsg->set_executeslocally(false);
+                if (!newMsg->directresulthost().empty()) {
+                    faabric::util::UniqueLock resultsLock(localResultsMutex);
+                    localResults.insert(
+                      { newMsg->id(), std::make_shared<MessageLocalResult>() });
+                }
             }
 
             // Dispatch the calls