hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
6979025cf4eb19d05314a1748b96574fc2110d43
oceanbase/src/sql/monitor/flt/ob_flt_utils.cpp

601 lines
21 KiB
C++
Raw Blame History

// Copyright 2010-2016 Alibaba Inc. All Rights Reserved.
// Author:
// juehui.lgy@alibaba-inc.com
// Normalizer:
// juehui.lgy@alibaba-inc.com
// this file defines implementation of full link trace span manager
#define USING_LOG_PREFIX SERVER
#include "sql/monitor/flt/ob_flt_utils.h"
#include "share/ob_define.h"
#include "sql/session/ob_basic_session_info.h"
#include "lib/trace/ob_trace.h"
#include "lib/trace/ob_trace_def.h"
#include "sql/monitor/flt/ob_flt_extra_info.h"
#include "sql/monitor/flt/ob_flt_control_info_mgr.h"
#include "sql/session/ob_sql_session_info.h"
namespace oceanbase
{
namespace sql
{
int ObFLTUtils::init_flt_show_trace_env(sql::ObSQLSessionInfo &session) {
int ret = OB_SUCCESS;
// if use trace log, init trace framework
if (session.is_inner()) {
// inner sql will reuse trace framework, do nothing
} else {
if (session.is_use_trace_log()) {
// query level trace
if (session.is_query_trc_granuality()) {
FLT_BEGIN_TRACE();
FLT_SET_TRACE_LEVEL(1);
FLT_SET_AUTO_FLUSH(true);
// transaction level trace
} else {
if (session.is_server_status_in_transaction()) {
// do nothing
} else {
FLT_BEGIN_TRACE();
FLT_SET_TRACE_LEVEL(1);
FLT_SET_AUTO_FLUSH(true);
}
}
} else {
FLT_SET_TRACE_LEVEL(0);
FLT_SET_AUTO_FLUSH(false);
}
}
return ret;
}
int ObFLTUtils::record_flt_last_trace_id(sql::ObSQLSessionInfo &session) {
int ret = OB_SUCCESS;
trace::UUID trc_uuid = OBTRACE->get_trace_id();
trace::UUID span_id;
char last_trace_id[OB_MAX_UUID_STR_LENGTH + 1];
char last_span_id_buf[OB_MAX_UUID_STR_LENGTH + 1];
ObString last_span_id;
session.get_flt_span_id(last_span_id);
int64_t pos = 0;
span_id.deserialize(last_span_id.ptr(), last_span_id.length(), pos);
pos = 0;
trc_uuid.tostring(last_trace_id, OB_MAX_UUID_STR_LENGTH + 1, pos);
pos = 0;
span_id.tostring(last_span_id_buf, OB_MAX_UUID_STR_LENGTH + 1, pos);
if (last_trace_id[0] == '\0') {
// do nothing
} else if (OB_FAIL(session.set_last_flt_trace_id(ObString(OB_MAX_UUID_STR_LENGTH + 1, last_trace_id)))) {
LOG_WARN("failed to set last flt trace id", K(ret), K(ObString(OB_MAX_UUID_STR_LENGTH + 1, last_trace_id)));
} else if (OB_FAIL(session.set_last_flt_span_id(ObString(OB_MAX_UUID_STR_LENGTH + 1, last_span_id_buf)))) {
LOG_WARN("failed to set last flt span id", K(ret), K(ObString(OB_MAX_UUID_STR_LENGTH + 1, last_span_id_buf)));
} else {
// do nothing
}
return ret;
}
void ObFLTUtils::clean_flt_env(sql::ObSQLSessionInfo &session) {
if (session.is_query_trc_granuality()) {
FLT_END_TRACE();
} else {
if (session.get_in_transaction()) {
// do nothing
} else {
FLT_END_TRACE();
}
}
}
int ObFLTUtils::clean_flt_show_trace_env(sql::ObSQLSessionInfo &session) {
int ret = OB_SUCCESS;
if (session.is_query_trc_granuality()) {
// record current trace id
if (OB_FAIL(record_flt_last_trace_id(session))) {
LOG_WARN("failed to record last flt trace id", K(ret));
} else {
// do nothing
}
// transaction level show trace
} else {
if (session.get_in_transaction()) {
// do nothing
} else {
// record current trace id
if (OB_FAIL(record_flt_last_trace_id(session))) {
LOG_WARN("failed to record last flt trace id", K(ret));
} else {
// do nothing
}
}
}
return ret;
}
int ObFLTUtils::init_flt_info(Ob20ExtraInfo extra_info,
sql::ObSQLSessionInfo &session,
bool is_client_support_flt)
{
int ret = OB_SUCCESS;
if (extra_info.exist_full_link_trace()) {
OZ(process_flt_extra_info(extra_info.get_full_link_trace().ptr(),
extra_info.get_full_link_trace().length(),
session));
extra_info.get_full_link_trace().reset();
}
if (session.get_control_info().is_valid()){
OZ(init_flt_log_framework(session, is_client_support_flt));
} else {
FLT_SET_TRACE_LEVEL(0);
FLT_SET_AUTO_FLUSH(false);
session.set_auto_flush_trace(false);
session.set_trace_enable(false);
}
return ret;
}
int ObFLTUtils::append_flt_extra_info(common::ObIAllocator &allocator,
ObIArray<obmysql::ObObjKV> *extra_info,
ObIArray<obmysql::Obp20Encoder*> *extra_info_ecds,
sql::ObSQLSessionInfo &sess,
bool is_new_extra_info)
{
int ret = OB_SUCCESS;
char *buf = NULL;
int size = 0;
FLTQueryInfo query_info;
// reserver memory for control info
// if sys config in control info and sys parameter has modified, resend this control info.
if (sess.get_control_info().is_valid_sys_config()
&& !((sess.get_control_info().print_sample_pct_ == ((double)(sess.get_tenant_print_sample_ppm()))/1000000)
&& (sess.get_control_info().slow_query_thres_ == GCONF.trace_log_slow_query_watermark))) {
sess.set_send_control_info(false);
}
if (!sess.is_send_control_info()) {
size += sess.get_control_info().get_serialize_size();
}
// reserver memmory for query info
if (sess.is_trace_enable()) {
query_info.query_start_time_ = sess.get_query_start_time();
query_info.query_end_time_ = ::oceanbase::common::ObTimeUtility::current_time();
size += query_info.get_serialize_size();
}
if (size == 0){
// has not flt extra info, do nothing
} else if (OB_UNLIKELY(size < 0)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("Invalid buffer length", K(ret), K(size));
} else if (NULL == (buf = static_cast<char *>(allocator.alloc(size)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_ERROR("fail to alloc mem", K(size), K(ret));
} else {
int64_t pos = 0;
// assamble control info
if (!sess.is_send_control_info()) {
FLTControlInfo con = sess.get_control_info();
if (!con.is_valid()) {
con.reset();
}
con.print_sample_pct_ = ((double)(sess.get_tenant_print_sample_ppm()))/1000000;
con.slow_query_thres_ = GCONF.trace_log_slow_query_watermark;
sess.set_flt_control_info(con);
if (OB_FAIL(con.serialize(buf, size, pos))) {
LOG_WARN("failed to serialize control info", K(pos), K(size));
} else {
sess.set_send_control_info(true);
}
}
// assamble query info
if (OB_FAIL(ret)) {
// do nothing
} else if (sess.is_trace_enable()) {
if (OB_FAIL(query_info.serialize(buf, size, pos))) {
LOG_WARN("failed to serialize query info", K(pos), K(size));
} else {
sess.set_trace_enable(false);
}
}
}
// set session info to extra info
if (OB_FAIL(ret)) {
// do nothing
} else if (size == 0) {
// nothing to write, do nothing
} else if (is_new_extra_info) {
Obp20FullTrcEncoder* full_trc_ecd = NULL;
void* ecd_buf = NULL;
if (OB_ISNULL(ecd_buf = allocator.alloc(sizeof(Obp20FullTrcEncoder)))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
LOG_WARN("failed to allocate memory.", K(ret));
} else {
full_trc_ecd = new(ecd_buf)Obp20FullTrcEncoder();
full_trc_ecd->full_trc_.assign(buf, size);
if (OB_FAIL(extra_info_ecds->push_back(full_trc_ecd))) {
LOG_WARN("failed to add extra info kv", K(full_trc_ecd), K(ret));
}
}
} else {
ObObjKV kv;
common::ObObj key;
common::ObObj value;
ObString key_str = "full_trc";
key.set_varchar(key_str);
value.set_varchar(ObString(size, buf));
kv.key_ = key;
kv.value_ = value;
if (OB_FAIL(extra_info->push_back(kv))) {
LOG_WARN("failed to add extra info kv", K(kv), K(ret));
}
}
return ret;
}
int ObFLTUtils::process_flt_extra_info(const char *buf,
const int64_t len, sql::ObSQLSessionInfo &sess)
{
int ret = OB_SUCCESS;
int64_t pos = 0;
LOG_TRACE("recieve flt extra info", KP(buf), K(len), KPHEX(buf, len));
while (OB_SUCC(ret) && pos < len) {
FullLinkTraceExtraInfoType extra_type;
int32_t v_len = 0;
LOG_TRACE("process single flt extra info", KP(buf), K(pos), K(len), KPHEX(buf+pos, len-pos));
if (OB_FAIL(FLTExtraInfo::resolve_type_and_len(buf, len, pos, extra_type, v_len))) {
LOG_WARN("failed to resolve type and len", K(len), K(pos));
} else if (pos+v_len > len) {
ret = OB_SIZE_OVERFLOW;
LOG_WARN("buf size overflow", K(ret), K(pos), K(v_len), K(len));
} else {
switch (extra_type) {
// for drv types:
case FLT_TYPE_DRV_LOG: {
FLTDrvSpan drv_span;
if (OB_FAIL(drv_span.deserialize(buf, pos+v_len, pos))) {
LOG_WARN("failed to deserialize full link trace extra info",
KP(buf), K(ret), K(pos), K(v_len));
} else {
_OBTRACE_LOG(INFO, "%s", drv_span.span_info_.ptr());
}
break;
}
// for proxy types:
// for public types:
case FLT_TYPE_APP_INFO: {
// do nothing
FLTAppInfo app_info;
FLTControlInfo con;
ObFLTControlInfoManager mgr(sess.get_effective_tenant_id());
if (OB_FAIL(app_info.deserialize(buf, pos+v_len, pos))) {
LOG_WARN("failed to deserialize full link trace extra info",
KP(buf), K(ret), K(pos), K(v_len));
} else if (OB_FAIL(mgr.init())) {
LOG_WARN("failed to init full link trace info manager", K(ret));
} else {
if (app_info.trace_client_info_.empty()) {
// do nothing
} else if (OB_FAIL(sess.get_app_info_encoder()
.set_client_info(&sess, app_info.trace_client_info_))) {
LOG_WARN("failed to set client info name", K(ret));
}
if (OB_FAIL(ret)) {
// do nothing
} else if (OB_FAIL(init_app_info(sess, app_info))) {
LOG_WARN("failed to init app info from session", K(ret));
} else if (OB_FAIL(mgr.find_appropriate_con_info(sess))) {
LOG_WARN("failed to find appropriate control info", K(ret));
} else {
// do nothing
}
}
break;
}
case FLT_TYPE_QUERY_INFO: {
// this extra info is written by server, and driver/proxy cannot send this to server;
ret = OB_NOT_SUPPORTED;
LOG_WARN("invalid extra_type", K(extra_type), K(ret));
break;
}
case FLT_TYPE_CONTROL_INFO: {
// this extra info is written by server, and driver/proxy cannot send this to server;
ret = OB_NOT_SUPPORTED;
LOG_WARN("invalid extra_type", K(extra_type), K(ret));
break;
}
case FLT_TYPE_SPAN_INFO: {
FLTSpanInfo span_info;
if (OB_FAIL(span_info.deserialize(buf, pos+v_len, pos))) {
LOG_WARN("failed to deserialize full link trace extra info",
KP(buf), K(ret), K(pos), K(v_len));
} else if (span_info.trace_enable_) {
if (OB_FAIL(sess.set_flt_trace_id(span_info.trace_id_))) {
LOG_WARN("failed to set trace id", K(ret));
} else if (OB_FAIL(sess.set_flt_span_id(span_info.span_id_))) {
LOG_WARN("failed to set span id", K(ret));
} else {
sess.set_trace_enable(span_info.trace_enable_);
sess.set_auto_flush_trace(span_info.force_print_);
}
}
break;
}
case FLT_TYPE_SHOW_TRACE: {
FLTShowTrace trace;
if (OB_FAIL(trace.deserialize(buf, pos+v_len, pos))) {
LOG_WARN("failed to deserialize full link trace extra info", KP(buf), K(ret), K(pos), K(v_len));
} else {
// add to span
ObString empty_str;
empty_str.reset();
sess.set_last_flt_span_id(empty_str);
}
break;
}
default: {
ret = OB_NOT_SUPPORTED;
LOG_WARN("invalid extra_type", K(extra_type), K(ret));
break;
}
} // switch ends
}
} // while ends
return ret;
}
int ObFLTUtils::init_flt_log_framework(sql::ObSQLSessionInfo &sess, bool is_client_support_flt)
{
// initialize log framework
int ret = OB_SUCCESS;
FLT_RESET_SPAN();
if (sess.is_query_trc_granuality()) {
// reset trace_enable and flush trace in transcation granularity
// init flush trace and trace enable
if (!is_client_support_flt) {
// reset log config
sess.set_trace_enable(false);
sess.set_auto_flush_trace(false);
OZ(update_flush_policy_by_control_info(sess));
} else {
//set by client
}
} else {
if (sess.is_server_status_in_transaction()) {
// in the trans, do nothing
} else {
// reset trace_enable and flush trace in transcation granularity
// init flush trace and trace enable
if (!is_client_support_flt) {
// reset log config
sess.set_trace_enable(false);
sess.set_auto_flush_trace(false);
OZ(update_flush_policy_by_control_info(sess));
} else {
//set by client
}
}
}
// init log frame
if (OB_FAIL(ret)) {
// do nothing
} else if (!sess.get_control_info().is_valid() || !sess.is_trace_enable()) {
FLT_SET_TRACE_LEVEL(0);
FLT_SET_AUTO_FLUSH(false);
sess.set_auto_flush_trace(false);
sess.set_trace_enable(false);
} else if (is_client_support_flt) {
trace::UUID tid, sid;
int64_t pos = 0;
ObString span_id;
ObString trace_id;
sess.get_flt_span_id(span_id);
sess.get_flt_trace_id(trace_id);
if (span_id.empty() || trace_id.empty()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid span_id or trace_id", K(span_id), K(trace_id), K(ret));
} else {
tid.deserialize(trace_id.ptr(), trace_id.length(), pos);
pos = 0;
sid.deserialize(span_id.ptr(), span_id.length(), pos);
OBTRACE->init(tid, sid);
FLT_SET_TRACE_LEVEL(sess.get_control_info().level_);
if (sess.is_use_trace_log()) {
sess.set_auto_flush_trace(true);
}
FLT_SET_AUTO_FLUSH(sess.is_auto_flush_trace());
char last_trace_id[OB_MAX_UUID_STR_LENGTH + 1];
pos = 0;
sid.tostring(last_trace_id, OB_MAX_UUID_STR_LENGTH + 1, pos);
}
// update trace_id by server self
} else {
// update trace_id in query granularity
if (sess.is_query_trc_granuality()) {
FLT_BEGIN_TRACE();
} else {
// update trace_id in transaction granularity
if (!sess.get_in_transaction()) {
// begin new trace
FLT_BEGIN_TRACE();
} else {
// do nothing
}
}
FLT_SET_TRACE_LEVEL(sess.get_control_info().level_);
if (sess.is_use_trace_log()) {
sess.set_auto_flush_trace(true);
}
FLT_SET_AUTO_FLUSH(sess.is_auto_flush_trace());
}
LOG_TRACE("flt init log", K(sess.is_trace_enable()),
K(sess.is_auto_flush_trace()), K(sess.get_control_info()));
return ret;
}
int ObFLTUtils::update_flush_policy_by_control_info(sql::ObSQLSessionInfo &sess)
{
int ret = OB_SUCCESS;
FLTControlInfo con = sess.get_control_info();
if (con.is_valid()) {
// init trace enable
con.print_sample_pct_ = ((double)(sess.get_tenant_print_sample_ppm()))/1000000;
ObRandom r;
double rand_num = 1.0 * (r.rand(0, RAND_MAX)/RAND_MAX);
if (rand_num < con.sample_pct_) {
sess.set_trace_enable(true);
} else {
sess.set_trace_enable(false);
}
// init flush trace
if (con.rp_ == FLTControlInfo::RecordPolicy::RP_ALL) {
if (sess.is_trace_enable()) {
sess.set_auto_flush_trace(true);
} else {
sess.set_auto_flush_trace(false);
}
} else if (con.rp_ == FLTControlInfo::RecordPolicy::RP_ONLY_SLOW_QUERY) {
// do nothing, slow query will must flush
} else if (con.rp_ == FLTControlInfo::RecordPolicy::RP_SAMPLE_AND_SLOW_QUERY) {
double rand_num2 = 1.0 * (r.rand(0, RAND_MAX)/RAND_MAX);
if (rand_num2 < con.print_sample_pct_) {
sess.set_auto_flush_trace(true);
} else {
sess.set_auto_flush_trace(false);
}
}
} else {
// invalid control info, do nothing
}
return ret;
}
int ObFLTUtils::init_app_info(sql::ObSQLSessionInfo &sess, FLTAppInfo &app_info)
{
int ret = OB_SUCCESS;
if (!app_info.trace_module_.empty() &&
OB_FAIL(sess.get_app_info_encoder().
set_module_name(&sess, app_info.trace_module_))) {
LOG_WARN("failed to set mod name", K(ret));
} else if (!app_info.trace_action_.empty() &&
OB_FAIL(sess.get_app_info_encoder().
set_action_name(&sess, app_info.trace_action_))) {
LOG_WARN("failed to set action name", K(ret));
} else if (!app_info.trace_client_identifier_.empty() &&
OB_FAIL(sess.set_client_id(app_info.trace_client_identifier_))) {
LOG_WARN("failed to set client id", K(ret));
} else {
// do nothing
}
return ret;
}
int handle_span_record(ObFLTSpanMgr *flt_span_manager, char* tag_buf, int64_t tag_len, ::oceanbase::trace::ObSpanCtx* span)
{
int ret = OB_SUCCESS;
if (OB_ISNULL(flt_span_manager)) {
// failed to get flt span manager, maybe tenant has been dropped, NOT NEED TO record;
} else {
ObFLTSpanData data;
ObArenaAllocator allocator;
const int64_t len = MAX_TRACE_LOG_SIZE;
char *buf = NULL;
int64_t pos = 0;
int64_t org_pos = 0;
if (NULL == (buf = (char*)allocator.alloc(len))) {
ret = OB_ALLOCATE_MEMORY_FAILED;
if (REACH_TIME_INTERVAL(100 * 1000)) {
SERVER_LOG(WARN, "record concurrent fifoallocator alloc mem failed", K(len), K(ret));
}
} else {
data.tenant_id_ = flt_span_manager->get_tenant_id();
// trace id
org_pos = pos;
if (OB_FAIL(OBTRACE->get_trace_id().tostring(buf, len, pos))) {
LOG_WARN ("failed to deserialize uuid", K(ret), K(buf), K(pos));
} else {
data.trace_id_.assign(buf+org_pos, pos - org_pos);
}
// span id
org_pos = pos;
if (OB_FAIL(span->span_id_.tostring(buf, len, pos))) {
LOG_WARN ("failed to deserialize uuid", K(ret), K(buf), K(pos));
} else {
data.span_id_.assign(buf+org_pos, pos - org_pos);
}
// span name
data.span_name_.assign(const_cast<char *>(trace::__span_type_mapper[span->span_type_]),
static_cast<int32_t>(strlen(trace::__span_type_mapper[span->span_type_])));
//parent_span_id_
org_pos = pos;
if (OB_ISNULL(span->source_span_) &&
OB_FAIL(OBTRACE->get_root_span_id().tostring(buf, len, pos))) {
LOG_WARN ("failed to deserialize uuid", K(ret), K(buf), K(pos));
} else if (!OB_ISNULL(span->source_span_) &&
OB_FAIL(span->source_span_->span_id_.tostring(buf, len, pos))){
LOG_WARN ("failed to deserialize uuid", K(ret), K(buf), K(pos));
} else {
data.parent_span_id_.assign(buf+org_pos, pos - org_pos);
}
//data.parent_span_id_ = buf+org_pos;
//data.parent_span_id_ = buf+org_pos;
if (OB_NOT_NULL(tag_buf) && tag_len != 0) {
//skip "tags":[ and trim ]
data.tags_.assign(tag_buf+sizeof("\"tags\":["), static_cast<int32_t>(tag_len-sizeof("\"tags\":[")-2));
//data.tags_.assign(tag_buf, tag_len);
}
// start ts
data.start_ts_ = span->start_ts_;
// end_ts
data.end_ts_ = span->end_ts_;
// is_follow
data.ref_type_ = span->is_follow_;
}
// record span
if (OB_FAIL(ret)) {
// do nothing
} else {
while (true) {
if (OB_FAIL(flt_span_manager->record_span(data))) {
if (OB_SIZE_OVERFLOW == ret || OB_ALLOCATE_MEMORY_FAILED == ret) {
LOG_TRACE("cannot allocate mem for record", K(ret));
ret = OB_SUCCESS;
} else {
if (REACH_TIME_INTERVAL(100 * 1000)) { // in case logging is too frequent
LOG_WARN("failed to record request info in request manager", K(ret));
}
}
}
// If the slot is full, a certain amount of
// internal space will be released,
// and it needs to be added again
if (ret != OB_ENTRY_NOT_EXIST) break;
}
}
}
return ret;
}
} // end of namespace sql
} // end of namespace oceanbase
Reference in New Issue View Git Blame Copy Permalink