init push

src/sql/monitor/ob_exec_stat_collector.h

@@ -0,0 +1,117 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#ifndef OCEANBASE_SQL_OB_EXEC_STAT_COLLECTOR_H_
+#define OCEANBASE_SQL_OB_EXEC_STAT_COLLECTOR_H_
+
+#include "lib/utility/ob_macro_utils.h"
+#include "lib/string/ob_string.h"
+#include "lib/trace/ob_trace_event.h"
+#include "sql/monitor/ob_phy_operator_monitor_info.h"
+#include "sql/monitor/ob_exec_stat.h"
+#include "lib/net/ob_addr.h"
+namespace oceanbase {
+namespace observer {
+class ObInnerSQLResult;
+}
+namespace sql {
+class ObExecContext;
+class ObPhyPlanMonitorInfo;
+class ObPhyOperatorMonitorInfo;
+class ObSql;
+class ObPhysicalPlan;
+class ObSQLSessionInfo;
+class ObExecStatCollector {
+  public:
+  ObExecStatCollector() : length_(0)
+  {}
+  ~ObExecStatCollector()
+  {}
+  int collect_monitor_info(uint64_t job_id, uint64_t task_id, ObPhyOperatorMonitorInfo& op_info);
+
+  int collect_plan_monitor_info(uint64_t job_id, uint64_t task_id, ObPhyPlanMonitorInfo* monitor_info);
+  int add_raw_stat(const common::ObString& str);
+  int get_extend_info(common::ObIAllocator& allocator, common::ObString& str);
+  void reset();
+
+  private:
+  template <class T>
+  int add_stat(const T* value);
+
+  /* functions */
+  DISALLOW_COPY_AND_ASSIGN(ObExecStatCollector);
+  static const int64_t MAX_STAT_BUF_COUNT = 10240;
+  char extend_buf_[MAX_STAT_BUF_COUNT];
+  int64_t length_;
+};
+
+class ObExecStatDispatch {
+  public:
+  ObExecStatDispatch() : stat_str_(), pos_(0){};
+  ~ObExecStatDispatch(){};
+  int set_extend_info(const common::ObString& str);
+  int dispatch(bool need_add_monitor, ObPhyPlanMonitorInfo* monitor_info, bool need_update_plan, ObPhysicalPlan* plan);
+
+  private:
+  int get_next_type(StatType& type);
+  template <class T>
+  int get_value(T* value);
+
+  private:
+  DISALLOW_COPY_AND_ASSIGN(ObExecStatDispatch);
+  common::ObString stat_str_;
+  int64_t pos_;
+};
+
+class ObExecStatUtils {
+  public:
+  template <class T>
+  OB_INLINE static void record_exec_timestamp(const T& process,
+      bool is_first,  // fir execution, not retry
+      ObExecTimestamp& exec_timestamp)
+  {
+    exec_timestamp.rpc_send_ts_ = process.get_send_timestamp();
+    exec_timestamp.receive_ts_ = process.get_receive_timestamp();
+    exec_timestamp.enter_queue_ts_ = process.get_enqueue_timestamp();
+    exec_timestamp.run_ts_ = process.get_run_timestamp();
+    exec_timestamp.before_process_ts_ = process.get_process_timestamp();
+    exec_timestamp.single_process_ts_ = process.get_single_process_timestamp();
+    exec_timestamp.process_executor_ts_ = process.get_exec_start_timestamp();
+    exec_timestamp.executor_end_ts_ = process.get_exec_end_timestamp();
+
+    if (is_first) {
+      /* packet event order:
+       * send -> receive -> enter_queue -> run -> before_process -> single process -> executor
+       */
+      exec_timestamp.net_t_ = exec_timestamp.receive_ts_ - exec_timestamp.rpc_send_ts_;
+      exec_timestamp.net_wait_t_ = exec_timestamp.enter_queue_ts_ - exec_timestamp.receive_ts_;
+    }
+    // SQL_MONITOR_LOG(DEBUG, "set exec timestamp", K(exec_timestamp.exec_type_),
+    //                K(exec_timestamp.before_process_ts_),
+    //                K(exec_timestamp.process_executor_ts_),
+    //                K(exec_timestamp.executor_end_ts_),
+    //                K(exec_timestamp.receive_ts_),
+    //                K(exec_timestamp.enter_queue_ts_),
+    //                K(exec_timestamp.run_ts_),
+    //                K(exec_timestamp.single_process_ts_));
+  }
+
+  private:
+  DISALLOW_COPY_AND_ASSIGN(ObExecStatUtils);
+  ObExecStatUtils();
+  ~ObExecStatUtils();
+};
+
+}  // namespace sql
+}  // namespace oceanbase
+#endif /* OCEANBASE_COMMON_STAT_OB_EXEC_STAT_COLLECTOR_H_ */
+//// end of header file