From f41ee5fb174931ee73249311e76cf0a135b19dce Mon Sep 17 00:00:00 2001
From: tata24 <32012883+tata24@users.noreply.github.com>
Date: Fri, 10 Dec 2021 14:12:22 +0800
Subject: [PATCH] fixed bug of issue #606 (#609)

---
 src/sql/code_generator/ob_column_index_provider.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/sql/code_generator/ob_column_index_provider.cpp b/src/sql/code_generator/ob_column_index_provider.cpp
index 6e465c8d23..ce2d6f64ff 100644
--- a/src/sql/code_generator/ob_column_index_provider.cpp
+++ b/src/sql/code_generator/ob_column_index_provider.cpp
@@ -77,7 +77,9 @@ int RowDesc::add_column(ObRawExpr* raw_expr)
     }
   }
   if (OB_SUCC(ret) && !raw_expr->has_flag(IS_COLUMNLIZED)) {
-    raw_expr->add_flag(IS_COLUMNLIZED);
+    if(OB_FAIL(raw_expr->add_flag(IS_COLUMNLIZED))) {
+      SQL_CG_LOG(WARN, "failed to add flag IS_COLUMNLIZED", K(ret));
+    }
   }
   return ret;
 }
@@ -104,7 +106,9 @@ int RowDesc::replace_column(ObRawExpr* old_expr, ObRawExpr* new_expr)
     exprs_.at(idx) = new_expr;
   }
   if (OB_SUCC(ret) && !new_expr->has_flag(IS_COLUMNLIZED)) {
-    new_expr->add_flag(IS_COLUMNLIZED);
+    if(OB_FAIL(new_expr->add_flag(IS_COLUMNLIZED))) {
+      SQL_CG_LOG(WARN, "failed to add flag IS_COLUMNLIZED", K(ret));
+    }
   }
   return ret;
 }