fix keeper build

bigo-sg · Sep 10, 2023 · d326554 · d326554
1 parent 1b3da7f
commit d326554
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 3 deletions.
diff --git a/src/Coordination/Standalone/Context.cpp b/src/Coordination/Standalone/Context.cpp
@@ -268,6 +268,16 @@ ReadSettings Context::getReadSettings() const
     return ReadSettings{};
 }
 
+ResourceManagerPtr Context::getResourceManager() const
+{
+    return nullptr;
+}
+
+ClassifierPtr Context::getWorkloadClassifier() const
+{
+    return nullptr;
+}
+
 void Context::initializeKeeperDispatcher([[maybe_unused]] bool start_async) const
 {
     const auto & config_ref = getConfigRef();

diff --git a/src/Coordination/Standalone/Context.h b/src/Coordination/Standalone/Context.h
@@ -118,6 +118,10 @@ class Context : public std::enable_shared_from_this<Context>
 
     ReadSettings getReadSettings() const;
 
+    /// Resource management related
+    ResourceManagerPtr getResourceManager() const;
+    ClassifierPtr getWorkloadClassifier() const;
+
     std::shared_ptr<KeeperDispatcher> getKeeperDispatcher() const;
     std::shared_ptr<KeeperDispatcher> tryGetKeeperDispatcher() const;
     void initializeKeeperDispatcher(bool start_async) const;

diff --git a/src/Disks/ObjectStorages/DiskObjectStorage.cpp b/src/Disks/ObjectStorages/DiskObjectStorage.cpp
@@ -491,7 +491,7 @@ static inline Settings updateResourceLink(const Settings & settings, const Strin
     if (auto query_context = CurrentThread::getQueryContext())
     {
         Settings result(settings);
-        result.resource_link = query_context->getClassifier()->get(resource_name);
+        result.resource_link = query_context->getWorkloadClassifier()->get(resource_name);
         return result;
     }
     return settings;

diff --git a/src/Interpreters/Context.cpp b/src/Interpreters/Context.cpp
@@ -1294,7 +1294,7 @@ ResourceManagerPtr Context::getResourceManager() const
     return shared->resource_manager;
 }
 
-ClassifierPtr Context::getClassifier() const
+ClassifierPtr Context::getWorkloadClassifier() const
 {
     auto lock = getLock();
     if (!classifier)

diff --git a/src/Interpreters/Context.h b/src/Interpreters/Context.h
@@ -581,7 +581,7 @@ class Context: public std::enable_shared_from_this<Context>
 
     /// Resource management related
     ResourceManagerPtr getResourceManager() const;
-    ClassifierPtr getClassifier() const;
+    ClassifierPtr getWorkloadClassifier() const;
 
     /// We have to copy external tables inside executeQuery() to track limits. Therefore, set callback for it. Must set once.
     void setExternalTablesInitializer(ExternalTablesInitializer && initializer);