82 lines
3.0 KiB
C++
82 lines
3.0 KiB
C++
/* Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License, version 2.0, for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
#include <sql/current_thd.h>
|
|
#include <sql/mysqld_thd_manager.h>
|
|
#include <sql/sql_lex.h>
|
|
#include "components/mysql_server/mysql_ongoing_transaction_query.h"
|
|
#include "sql/sql_class.h" // THD
|
|
|
|
void mysql_server_ongoing_transactions_query_init() { return; }
|
|
|
|
class Get_running_transactions : public Do_THD_Impl {
|
|
public:
|
|
Get_running_transactions() {}
|
|
|
|
/*
|
|
This method relies on the assumption that a thread running query will either
|
|
have an active query plan, or is in the middle of a multi statement
|
|
transaction.
|
|
*/
|
|
virtual void operator()(THD *thd) {
|
|
if (thd->is_killed() || thd->is_error()) return;
|
|
|
|
thd->lock_query_plan();
|
|
|
|
LEX *lex = thd->query_plan.get_lex();
|
|
if (lex && lex->m_sql_cmd && lex->m_sql_cmd->is_dml())
|
|
thread_ids.push_back(thd->thread_id());
|
|
thd->unlock_query_plan();
|
|
|
|
if (thd->in_active_multi_stmt_transaction())
|
|
thread_ids.push_back(thd->thread_id());
|
|
}
|
|
|
|
ulong get_transaction_number() { return thread_ids.size(); }
|
|
|
|
void fill_transaction_ids(unsigned long **ids) {
|
|
size_t number_thd = thread_ids.size();
|
|
*ids = (unsigned long *)my_malloc(
|
|
PSI_NOT_INSTRUMENTED, number_thd * sizeof(unsigned long), MYF(MY_WME));
|
|
int index = 0;
|
|
for (std::vector<my_thread_id>::iterator it = thread_ids.begin();
|
|
it != thread_ids.end(); ++it) {
|
|
(*ids)[index] = *it;
|
|
index++;
|
|
}
|
|
}
|
|
|
|
private:
|
|
/* Status of all threads are summed into this. */
|
|
std::vector<my_thread_id> thread_ids;
|
|
};
|
|
|
|
DEFINE_BOOL_METHOD(
|
|
mysql_ongoing_transactions_query_imp::get_ongoing_server_transactions,
|
|
(unsigned long **thread_ids, unsigned long *lenght)) {
|
|
Get_running_transactions trx_counter;
|
|
Global_THD_manager::get_instance()->do_for_all_thd(&trx_counter);
|
|
trx_counter.fill_transaction_ids(thread_ids);
|
|
*lenght = trx_counter.get_transaction_number();
|
|
return false;
|
|
}
|