fix build

Signed-off-by: guo-shaoge <shaoge1994@163.com>
pingcap · Mar 29, 2023 · df04360 · df04360
1 parent c93b49b
commit df04360
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dbms/src/DataStreams/GeneratedColumnPlaceholderBlockInputStream.h b/dbms/src/DataStreams/GeneratedColumnPlaceholderBlockInputStream.h
@@ -54,11 +54,11 @@ class GeneratedColumnPlaceholderBlockInputStream : public IProfilingBlockInputSt
 protected:
     void readPrefix() override
     {
-        RUNTIME_CHECK(!generated_column_infos.empty());
+        RUNTIME_CHECK(!generated_column_infos.empty(), Exception, "generated_column_infos cannot be empty");
         // Validation check.
         for (size_t i = 1; i < generated_column_infos.size(); ++i)
         {
-            RUNTIME_CHECK(std::get<0>(generated_column_infos[i]) > std::get<0>(generated_column_infos[i - 1]));
+            RUNTIME_CHECK(std::get<0>(generated_column_infos[i]) > std::get<0>(generated_column_infos[i - 1]), Exception, "generated column index should be ordered");
         }
     }
 

diff --git a/dbms/src/Flash/Coprocessor/DAGStorageInterpreter.cpp b/dbms/src/Flash/Coprocessor/DAGStorageInterpreter.cpp
@@ -965,7 +965,7 @@ std::tuple<Names, NamesAndTypes, std::vector<ExtraCastAfterTSMode>> DAGStorageIn
         auto const & ci = table_scan.getColumns()[i];
         ColumnID cid = ci.column_id();
 
-        if (ci.hasGeneratedColumnFlag())
+        if (ci.has_flag() && ((ci.flag() & TiDB::ColumnFlagGeneratedColumn) != 0))
         {
             LOG_DEBUG(log, "got column({}) with generated column flag", i);
             const auto & data_type = getDataTypeByColumnInfoForComputingLayer(ci);