hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[FEAT MERGE] [CP] query_interface 422

Browse Source
This commit is contained in:
obdev
2024-04-15 16:23:22 +00:00
committed by ob-robot
parent 6423e587c1
commit dd7737c7ab
28 changed files with 687 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/sql/session/ob_sess_info_verify.h
View File

@ -14,6 +14,7 @@
#define OCEANBASE_SQL_OB_SESS_INFO_VERI_H_
#include "share/ob_define.h"
#include "share/system_variable/ob_sys_var_class_type.h"
#include "lib/string/ob_string.h"
#include "lib/atomic/ob_atomic.h"
#include "rpc/obrpc/ob_rpc_proxy.h"
@ -32,6 +33,10 @@ namespace obrpc
{
class ObSrvRpcProxy;
}
namespace share {
class ObBasicSysVar;
enum ObSysVarClassType;
}
namespace sql
{

19
src/sql/session/ob_sql_session_info.cpp
View File

@ -2919,7 +2919,7 @@ int ObSQLSessionInfo::ps_use_stream_result_set(bool &use_stream) {
return ret;
}
ObPieceCache* ObSQLSessionInfo::get_piece_cache(bool need_init) {
::oceanbase::observer::ObPieceCache* ObSQLSessionInfo::get_piece_cache(bool need_init) {
if (NULL == piece_cache_ && need_init) {
void *buf = get_session_allocator().alloc(sizeof(ObPieceCache));
if (NULL != buf) {
@ -2936,6 +2936,23 @@ ObPieceCache* ObSQLSessionInfo::get_piece_cache(bool need_init) {
return piece_cache_;
}
int ObSQLSessionInfo::set_login_info(const share::schema::ObUserLoginInfo &login_info)
{
int ret = OB_SUCCESS;
OZ (ob_write_string(get_session_allocator(), login_info.tenant_name_, login_info_.tenant_name_));
OZ (ob_write_string(get_session_allocator(), login_info.user_name_, login_info_.user_name_));
OZ (ob_write_string(get_session_allocator(), login_info.client_ip_, login_info_.client_ip_));
OZ (ob_write_string(get_session_allocator(), login_info.passwd_, login_info_.passwd_));
OZ (ob_write_string(get_session_allocator(), login_info.db_, login_info_.db_));
OZ (ob_write_string(get_session_allocator(), login_info.scramble_str_, login_info_.scramble_str_));
return ret;
}
int ObSQLSessionInfo::set_login_auth_data(const ObString &auth_data) {
int ret = OB_SUCCESS;
OZ (ob_write_string(get_session_allocator(), auth_data, login_info_.passwd_));
return ret;
}

4
src/sql/session/ob_sql_session_info.h
View File

@ -1333,6 +1333,9 @@ public:
// piece
observer::ObPieceCache *get_piece_cache(bool need_init = false);
share::schema::ObUserLoginInfo get_login_info () { return login_info_; }
int set_login_info(const share::schema::ObUserLoginInfo &login_info);
int set_login_auth_data(const ObString &auth_data);
void set_load_data_exec_session(bool v) { is_load_data_exec_session_ = v; }
bool is_load_data_exec_session() const { return is_load_data_exec_session_; }
inline ObSqlString &get_pl_exact_err_msg() { return pl_exact_err_msg_; }
@ -1596,6 +1599,7 @@ private:
common::ObSEArray<ObSequenceSchema*, 2> dblink_sequence_schemas_;
bool client_non_standard_;
bool is_session_sync_support_; // session_sync_support flag.
share::schema::ObUserLoginInfo login_info_;
};
inline bool ObSQLSessionInfo::is_terminate(int &ret) const