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

[FEAT MERGE]Query Interface Enhancement

Browse Source 
Co-authored-by: chinaxing <chen.yack@gmail.com>
This commit is contained in:
obdev
2023-04-28 02:41:45 +00:00
committed by ob-robot
parent d48cc591bd
commit 23b365a0a7
74 changed files with 2611 additions and 330 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -217,6 +217,16 @@ public:
virtual int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) = 0;
virtual int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos) = 0;
virtual int64_t get_serialize_size(ObSQLSessionInfo& sess) const = 0;
// When implementing new information synchronization,
// it is necessary to add a self-verification interface
// include fetch, compare, and error display.
virtual int fetch_sess_info(ObSQLSessionInfo &sess, char *buf,
const int64_t length, int64_t &pos) = 0;
virtual int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess)= 0;
virtual int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length) = 0;
virtual int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length) = 0;
bool is_changed_;
};
@ -227,6 +237,12 @@ public:
int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos);
int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos);
int64_t get_serialize_size(ObSQLSessionInfo& sess) const;
int fetch_sess_info(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos);
int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess);
int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length);
int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length);
};
class ObAppInfoEncoder : public ObSessInfoEncoder {
@ -236,6 +252,12 @@ public:
int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos);
int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos);
int64_t get_serialize_size(ObSQLSessionInfo& sess) const;
int fetch_sess_info(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos);
int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess);
int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length);
int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length);
int set_client_info(ObSQLSessionInfo* sess, const ObString &client_info);
int set_module_name(ObSQLSessionInfo* sess, const ObString &mod);
int set_action_name(ObSQLSessionInfo* sess, const ObString &act);
@ -258,6 +280,13 @@ public:
virtual int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) override;
virtual int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos) override;
virtual int64_t get_serialize_size(ObSQLSessionInfo& sess) const override;
virtual int fetch_sess_info(ObSQLSessionInfo &sess, char *buf,
const int64_t length, int64_t &pos);
virtual int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess);
virtual int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length);
virtual int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length);
};
class ObClientIdInfoEncoder : public ObSessInfoEncoder {
public:
@ -266,6 +295,13 @@ public:
virtual int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) override;
virtual int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos) override;
virtual int64_t get_serialize_size(ObSQLSessionInfo &sess) const override;
virtual int fetch_sess_info(ObSQLSessionInfo &sess, char *buf,
const int64_t length, int64_t &pos);
virtual int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess);
virtual int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length);
virtual int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length);
};
class ObControlInfoEncoder : public ObSessInfoEncoder {
@ -275,6 +311,13 @@ public:
virtual int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) override;
virtual int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos) override;
virtual int64_t get_serialize_size(ObSQLSessionInfo &sess) const override;
virtual int fetch_sess_info(ObSQLSessionInfo &sess, char *buf,
const int64_t length, int64_t &pos);
virtual int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess);
virtual int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length,
const char* last_sess_buf, int64_t last_sess_length);
virtual int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf,
int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length);
static const int16_t CONINFO_BY_SESS = 0xC078;
};
@ -284,6 +327,10 @@ public: \
int serialize(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) override; \
int deserialize(ObSQLSessionInfo &sess, const char *buf, const int64_t length, int64_t &pos) override; \
int64_t get_serialize_size(ObSQLSessionInfo &sess) const override; \
int fetch_sess_info(ObSQLSessionInfo &sess, char *buf, const int64_t length, int64_t &pos) override; \
int64_t get_fetch_sess_info_size(ObSQLSessionInfo& sess) override; \
int compare_sess_info(const char* current_sess_buf, int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length) override; \
int display_sess_info(ObSQLSessionInfo &sess, const char* current_sess_buf, int64_t current_sess_length, const char* last_sess_buf, int64_t last_sess_length) override; \
};
DEF_SESSION_TXN_ENCODER(ObTxnStaticInfoEncoder);
DEF_SESSION_TXN_ENCODER(ObTxnDynamicInfoEncoder);
@ -345,6 +392,22 @@ struct ObInnerContextMap {
}
return ret;
}
inline bool operator==(const ObInnerContextMap &other) const {
bool equal1 = context_name_ == other.context_name_ &&
context_map_->size() == other.context_map_->size();
bool equal2 = true;
if (equal1) {
auto it2 = other.context_map_->begin();
for (auto it1 = context_map_->begin(); it1 != context_map_->end() &&
it2 != other.context_map_->end(); ++it1, ++it2) {
if (it1->second != it2->second) {
equal2 = false;
}
}
}
return equal1 && equal2;
}
ObString context_name_;
ObInnerContextHashMap *context_map_;
common::ObIAllocator &alloc_;
@ -543,6 +606,7 @@ public:
TO_STRING_KV(K_(module_name), K_(action_name), K_(client_info));
};
public:
ObSQLSessionInfo();
virtual ~ObSQLSessionInfo();
@ -563,6 +627,13 @@ public:
const common::ObWarningBuffer &get_show_warnings_buffer() const { return show_warnings_buf_; }
const common::ObWarningBuffer &get_warnings_buffer() const { return warnings_buf_; }
common::ObWarningBuffer &get_warnings_buffer() { return warnings_buf_; }
// self-verification add.
bool is_has_query_executed() {return has_query_executed_; }
void set_has_query_executed(bool has_query_executed) {
has_query_executed_ = has_query_executed; }
void reset_warnings_buf()
{
warnings_buf_.reset();
@ -872,6 +943,9 @@ public:
int set_module_name(const common::ObString &mod);
int set_action_name(const common::ObString &act);
int set_client_info(const common::ObString &client_info);
int set_verify_info_sess_id(const uint32_t sess_id);
int set_verify_info_proxy_sess_id(const uint64_t proxy_sess_id);
int set_verify_info_addr(const ObAddr addr);
ApplicationInfo& get_client_app_info() { return client_app_info_; }
int get_sess_encoder(const SessionSyncInfoType sess_sync_info_type, ObSessInfoEncoder* &encoder);
const common::ObString& get_module_name() const { return client_app_info_.module_name_; }
@ -880,6 +954,7 @@ public:
const FLTControlInfo& get_control_info() const { return flt_control_info_; }
FLTControlInfo& get_control_info() { return flt_control_info_; }
void set_flt_control_info(const FLTControlInfo &con_info);
void set_flt_control_info_no_sync(const FLTControlInfo &con_info);
bool is_send_control_info() { return is_send_control_info_; }
void set_send_control_info(bool is_send) { is_send_control_info_ = is_send; }
bool is_coninfo_set_by_sess() { return coninfo_set_by_sess_; }
@ -1164,6 +1239,9 @@ private:
// To customize the memory length, you need to use malloc_alloctor of mem_context
lib::MemoryContext mem_context_;
ApplicationInfo client_app_info_;
// There is a scenario, when the connection is established for the first time,
// the route is immediately switched and no request is initiated, and no verification is required at this time
bool has_query_executed_; //add for routing without synchronizing session information
char module_buf_[common::OB_MAX_MOD_NAME_LENGTH];
char action_buf_[common::OB_MAX_ACT_NAME_LENGTH];
char client_info_buf_[common::OB_MAX_CLIENT_INFO_LENGTH];