Specify tenant id for RemoteExecuteStreamHandle

This commit is contained in:
obdev
2023-09-14 23:40:34 +00:00
committed by ob-robot
parent 7f11ed52a4
commit 4c4c68977b
2 changed files with 6 additions and 6 deletions

View File

@ -43,8 +43,8 @@ class MyStreamHandle
public: public:
typedef typename obrpc::ObExecutorRpcProxy::SSHandle<pcode> MyHandle; typedef typename obrpc::ObExecutorRpcProxy::SSHandle<pcode> MyHandle;
typedef common::ObScanner MyResult; typedef common::ObScanner MyResult;
explicit MyStreamHandle(const char *label) explicit MyStreamHandle(const char *label, uint64_t tenant_id)
: result_(label), : result_(label, NULL, common::ObScanner::DEFAULT_MAX_SERIALIZE_SIZE, tenant_id),
rc_(common::OB_SUCCESS) rc_(common::OB_SUCCESS)
{ {
} }
@ -99,10 +99,10 @@ typedef MyStreamHandle<obrpc::OB_TASK_FETCH_RESULT> FetchResultStreamHandle;
class RemoteExecuteStreamHandle class RemoteExecuteStreamHandle
{ {
public: public:
RemoteExecuteStreamHandle(const char *label) : RemoteExecuteStreamHandle(const char *label, uint64_t tenant_id) :
use_remote_protocol_v2_(false), use_remote_protocol_v2_(false),
sync_stream_handle_(label), sync_stream_handle_(label, tenant_id),
sync_stream_handle_v2_(label) sync_stream_handle_v2_(label, tenant_id)
{ {
} }
~RemoteExecuteStreamHandle() = default; ~RemoteExecuteStreamHandle() = default;

View File

@ -240,7 +240,7 @@ int ObTaskExecutorCtx::reset_and_init_stream_handler()
ret = OB_ALLOCATE_MEMORY_FAILED; ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("fail to alloc memory for RemoteExecuteStreamHandle", K(ret)); LOG_WARN("fail to alloc memory for RemoteExecuteStreamHandle", K(ret));
} else { } else {
task_resp_handler_ = new (buffer) RemoteExecuteStreamHandle("RemoteExecStream"); task_resp_handler_ = new (buffer) RemoteExecuteStreamHandle("RemoteExecStream", MTL_ID());
} }
} }
return ret; return ret;