From 3e788f539121ef54416a7a9bd7b9bfeed23050d5 Mon Sep 17 00:00:00 2001
From: obdev <obdev@oceanbase.com>
Date: Wed, 23 Nov 2022 12:11:59 +0000
Subject: [PATCH] Add ctx id from connect by allocator.

---
 src/sql/engine/connect_by/ob_cnnt_by_pump.cpp     | 1 +
 src/sql/engine/connect_by/ob_cnnt_by_pump.h       | 4 ++++
 src/sql/engine/connect_by/ob_cnnt_by_pump_bfs.cpp | 1 +
 3 files changed, 6 insertions(+)

diff --git a/src/sql/engine/connect_by/ob_cnnt_by_pump.cpp b/src/sql/engine/connect_by/ob_cnnt_by_pump.cpp
index 07ba89a3d7..dfca111769 100644
--- a/src/sql/engine/connect_by/ob_cnnt_by_pump.cpp
+++ b/src/sql/engine/connect_by/ob_cnnt_by_pump.cpp
@@ -248,6 +248,7 @@ int ObConnectByOpPump::init(const ObNLConnectBySpec &connect_by, ObNLConnectByOp
   } else {
     uint64_t tenant_id = eval_ctx.exec_ctx_.get_my_session()->get_effective_tenant_id();
     allocator_.set_tenant_id(tenant_id);
+    allocator_.set_ctx_id(ObCtxIds::WORK_AREA);
     connect_by_prior_exprs_ = &connect_by.connect_by_prior_exprs_;
     eval_ctx_ = &eval_ctx;
     connect_by_ = &connect_by_op;
diff --git a/src/sql/engine/connect_by/ob_cnnt_by_pump.h b/src/sql/engine/connect_by/ob_cnnt_by_pump.h
index 267973d310..efa6cafc7e 100644
--- a/src/sql/engine/connect_by/ob_cnnt_by_pump.h
+++ b/src/sql/engine/connect_by/ob_cnnt_by_pump.h
@@ -56,6 +56,10 @@ private:
     {
       allocator_.set_tenant_id(tenant_id);
     }
+    void set_ctx_id(int64_t ctx_id)
+    {
+      allocator_.set_ctx_id(ctx_id);
+    }
   private:
     common::ObArenaAllocator allocator_;
     int64_t alloc_cnt_;
diff --git a/src/sql/engine/connect_by/ob_cnnt_by_pump_bfs.cpp b/src/sql/engine/connect_by/ob_cnnt_by_pump_bfs.cpp
index ef79c8cefb..f70df541cf 100644
--- a/src/sql/engine/connect_by/ob_cnnt_by_pump_bfs.cpp
+++ b/src/sql/engine/connect_by/ob_cnnt_by_pump_bfs.cpp
@@ -279,6 +279,7 @@ int ObConnectByOpBFSPump::init(ObNLConnectByWithIndexSpec &connect_by,
   } else {
     uint64_t tenant_id = eval_ctx.exec_ctx_.get_my_session()->get_effective_tenant_id();
     allocator_.set_tenant_id(tenant_id);
+    allocator_.set_ctx_id(ObCtxIds::WORK_AREA);
     connect_by_ = &connect_by_op;
     connect_by_prior_exprs_ = &connect_by.connect_by_prior_exprs_;
     left_prior_exprs_ = &connect_by.left_prior_exprs_;