feat(shannon): selective-offload-est-query-cost

Shannon-Data · Dec 29, 2024 · ad8d673 · ad8d673
1 parent 97e760b
commit ad8d673
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/storage/rapid_engine/handler/ha_shannon_rapid.cc b/storage/rapid_engine/handler/ha_shannon_rapid.cc
@@ -571,9 +571,7 @@ static bool RapidPrepareEstimateQueryCosts(THD *thd, LEX *lex) {
   if (thd->variables.use_secondary_engine == SECONDARY_ENGINE_OFF) {
     SetSecondaryEngineOffloadFailedReason(thd, "use_secondary_engine set to off.");
     return true;
-  }
-
-  else if (thd->variables.use_secondary_engine == SECONDARY_ENGINE_FORCED)
+  } else if (thd->variables.use_secondary_engine == SECONDARY_ENGINE_FORCED)
     return false;
 
   // gets the shannon statement context from thd, which stores in SecondaryEnginePrePrepareHook.
@@ -599,6 +597,7 @@ static bool RapidPrepareEstimateQueryCosts(THD *thd, LEX *lex) {
   }
 
   // 3: checks dict encoding projection, and varlen project size, etc.
+  if (ShannonBase::Utils::Util::check_dict_encoding_projection(thd)) return true;
 
   return false;
 }

diff --git a/storage/rapid_engine/utils/utils.cpp b/storage/rapid_engine/utils/utils.cpp
@@ -311,5 +311,9 @@ bool Util::dynamic_feature_normalization(THD *thd) {
   return false;
 }
 
+// check whether the dictionary encoding projection is supported or not.
+// returns true if supported to innodb, otherwise, false to secondary engine.
+bool Util::check_dict_encoding_projection(THD *thd) { return false; }
+
 }  // namespace Utils
 }  // namespace ShannonBase
diff --git a/storage/rapid_engine/utils/utils.h b/storage/rapid_engine/utils/utils.h
@@ -117,6 +117,8 @@ class Util {
   static bool decision_tree_classifier(THD *thd);
 
   static bool dynamic_feature_normalization(THD *thd);
+
+  static bool check_dict_encoding_projection(THD *thd);
 };
 
 }  // namespace Utils