Add TrxBoundaryMatcher
A class capble of detecting statements that change the transaction state and autocommit mode. The detection is done using regexes. There is still some expanding and optimization to be done.
This commit is contained in:
Johan Wikman
@ -1,4 +1,4 @@
|
||||
add_library(maxscale-common SHARED adminusers.c alloc.c authenticator.c atomic.c buffer.c config.c config_runtime.c dcb.c filter.c filter.cc externcmd.c paths.c hashtable.c hint.c housekeeper.c load_utils.c log_manager.cc maxscale_pcre2.c misc.c mlist.c modutil.c monitor.c queuemanager.c query_classifier.cc poll.c random_jkiss.c resultset.c secrets.c server.c service.c session.c spinlock.c thread.c users.c utils.c skygw_utils.cc statistics.c listener.c ssl.c mysql_utils.c mysql_binlog.c modulecmd.c )
|
||||
add_library(maxscale-common SHARED adminusers.c alloc.c authenticator.c atomic.c buffer.c config.c config_runtime.c dcb.c filter.c filter.cc externcmd.c paths.c hashtable.c hint.c housekeeper.c load_utils.c log_manager.cc maxscale_pcre2.c misc.c mlist.c modutil.c monitor.c queuemanager.c query_classifier.cc poll.c random_jkiss.c resultset.c secrets.c server.c service.c session.c spinlock.c thread.c users.c utils.c skygw_utils.cc statistics.c listener.c ssl.c mysql_utils.c mysql_binlog.c modulecmd.c trxboundarymatcher.cc)
|
||||
|
||||
if(WITH_JEMALLOC)
|
||||
target_link_libraries(maxscale-common ${JEMALLOC_LIBRARIES})
|
||||
|
||||
87
server/core/maxscale/trxboundarymatcher.hh
Normal file
87
server/core/maxscale/trxboundarymatcher.hh
Normal file
@ -0,0 +1,87 @@
|
||||
#pragma once
|
||||
/*
|
||||
* Copyright (c) 2016 MariaDB Corporation Ab
|
||||
*
|
||||
* Use of this software is governed by the Business Source License included
|
||||
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
||||
*
|
||||
* Change Date: 2019-07-01
|
||||
*
|
||||
* On the date above, in accordance with the Business Source License, use
|
||||
* of this software will be governed by version 2 or later of the General
|
||||
* Public License.
|
||||
*/
|
||||
|
||||
#include <maxscale/cppdefs.hh>
|
||||
#include <maxscale/pcre2.h>
|
||||
#include <maxscale/query_classifier.h>
|
||||
|
||||
namespace maxscale
|
||||
{
|
||||
|
||||
/**
|
||||
* @class TrxBoundaryMatcher
|
||||
*
|
||||
* @ TrxBoundaryMatcher is a class capable of, using regexes, to recognize
|
||||
* and return the correct type mask of statements affecting the transaction
|
||||
* state and* autocommit mode.
|
||||
*/
|
||||
class TrxBoundaryMatcher
|
||||
{
|
||||
public:
|
||||
/**
|
||||
* To be called once at process startup. After a successful call,
|
||||
* @type_mask_of() can be used in the calling thread.
|
||||
*
|
||||
* @return True, if the initialization succeeded, false otherwise.
|
||||
*/
|
||||
static bool process_init();
|
||||
|
||||
/**
|
||||
* To be called once at process shut down.
|
||||
*/
|
||||
static void process_end();
|
||||
|
||||
/**
|
||||
* To be called once in each thread, other that the one where @process_init()
|
||||
* was called. After a successful call, @type_mask_of() can be used in the
|
||||
* calling thread.
|
||||
*
|
||||
* @return True, if the initialization succeeded, false otherwise.
|
||||
*/
|
||||
static bool thread_init();
|
||||
|
||||
/**
|
||||
* To be called once at thread shut down.
|
||||
*/
|
||||
static void thread_end();
|
||||
|
||||
/**
|
||||
* Return the type mask of a statement, provided the statement affects
|
||||
* transaction state or autocommit mode.
|
||||
*
|
||||
* @param pSql Sql statement.
|
||||
* @param len Length of @c pSql.
|
||||
*
|
||||
* @return The corresponding type mask or 0, if the statement does not
|
||||
* affect transaction state or autocommit mode.
|
||||
*/
|
||||
static uint32_t type_mask_of(const char* pSql, size_t len);
|
||||
|
||||
/**
|
||||
* Return the type mask of a statement, provided the statement affects
|
||||
* transaction state or autocommit mode.
|
||||
*
|
||||
* @param pBuf A COM_QUERY
|
||||
*
|
||||
* @return The corresponding type mask or 0, if the statement does not
|
||||
* affect transaction state or autocommit mode.
|
||||
*/
|
||||
static uint32_t type_mask_of(GWBUF* pBuf);
|
||||
|
||||
private:
|
||||
TrxBoundaryMatcher(const TrxBoundaryMatcher&);
|
||||
TrxBoundaryMatcher& operator = (const TrxBoundaryMatcher&);
|
||||
};
|
||||
|
||||
}
|
||||
268
server/core/trxboundarymatcher.cc
Normal file
268
server/core/trxboundarymatcher.cc
Normal file
@ -0,0 +1,268 @@
|
||||
/*
|
||||
* Copyright (c) 2016 MariaDB Corporation Ab
|
||||
*
|
||||
* Use of this software is governed by the Business Source License included
|
||||
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
||||
*
|
||||
* Change Date: 2019-07-01
|
||||
*
|
||||
* On the date above, in accordance with the Business Source License, use
|
||||
* of this software will be governed by version 2 or later of the General
|
||||
* Public License.
|
||||
*/
|
||||
|
||||
#include "maxscale/trxboundarymatcher.hh"
|
||||
#include <maxscale/platform.h>
|
||||
#include <maxscale/modutil.h>
|
||||
|
||||
namespace
|
||||
{
|
||||
|
||||
struct REGEX_DATA
|
||||
{
|
||||
const char* zMatch;
|
||||
uint32_t type_mask;
|
||||
pcre2_code* pCode;
|
||||
};
|
||||
|
||||
REGEX_DATA this_unit_regexes[] =
|
||||
{
|
||||
{
|
||||
"^\\s*BEGIN(\\s+WORK)?\\s*;?\\s*$",
|
||||
QUERY_TYPE_BEGIN_TRX
|
||||
},
|
||||
{
|
||||
"^\\s*COMMIT(\\s+WORK)?\\s*;?\\s*$",
|
||||
QUERY_TYPE_COMMIT,
|
||||
},
|
||||
{
|
||||
"^\\s*ROLLBACK(\\s+WORK)?\\s*;?\\s*$",
|
||||
QUERY_TYPE_ROLLBACK
|
||||
},
|
||||
{
|
||||
"^\\s*START\\s+TRANSACTION\\s+READ\\s+ONLY\\s*;?\\s*$",
|
||||
QUERY_TYPE_BEGIN_TRX | QUERY_TYPE_READ
|
||||
},
|
||||
{
|
||||
"^\\s*START\\s+TRANSACTION\\s+READ\\s+WRITE\\s*;?\\s*$",
|
||||
QUERY_TYPE_BEGIN_TRX | QUERY_TYPE_WRITE
|
||||
},
|
||||
{
|
||||
"^\\s*START\\s+TRANSACTION(\\s*;?\\s*|(\\s+.*))$",
|
||||
QUERY_TYPE_BEGIN_TRX
|
||||
},
|
||||
{
|
||||
"^\\s*SET\\s+AUTOCOMMIT\\s*\\=\\s*(1|true)\\s*;?\\s*$",
|
||||
QUERY_TYPE_COMMIT|QUERY_TYPE_ENABLE_AUTOCOMMIT
|
||||
},
|
||||
{
|
||||
"^\\s*SET\\s+AUTOCOMMIT\\s*\\=\\s*(0|false)\\s*;?\\s*$",
|
||||
QUERY_TYPE_BEGIN_TRX|QUERY_TYPE_DISABLE_AUTOCOMMIT
|
||||
}
|
||||
};
|
||||
|
||||
const size_t N_REGEXES = sizeof(this_unit_regexes)/sizeof(this_unit_regexes[0]);
|
||||
|
||||
struct this_unit
|
||||
{
|
||||
REGEX_DATA* pRegexes;
|
||||
} this_unit =
|
||||
{
|
||||
.pRegexes = this_unit_regexes
|
||||
};
|
||||
|
||||
thread_local struct this_thread
|
||||
{
|
||||
pcre2_match_data* match_datas[N_REGEXES];
|
||||
} this_thread;
|
||||
|
||||
|
||||
bool compile_regexes();
|
||||
void free_regexes();
|
||||
|
||||
bool create_thread_data();
|
||||
void free_thread_data();
|
||||
|
||||
bool compile_regexes()
|
||||
{
|
||||
REGEX_DATA* i = this_unit.pRegexes;
|
||||
REGEX_DATA* end = i + N_REGEXES;
|
||||
|
||||
bool success = true;
|
||||
|
||||
while (success && (i < end))
|
||||
{
|
||||
int errcode;
|
||||
PCRE2_SIZE erroffset;
|
||||
i->pCode = pcre2_compile((PCRE2_SPTR)i->zMatch, PCRE2_ZERO_TERMINATED, PCRE2_CASELESS,
|
||||
&errcode, &erroffset, NULL);
|
||||
|
||||
if (!i->pCode)
|
||||
{
|
||||
success = false;
|
||||
PCRE2_UCHAR errbuf[512];
|
||||
pcre2_get_error_message(errcode, errbuf, sizeof(errbuf));
|
||||
|
||||
MXS_ERROR("Regex compilation failed at %lu for regex '%s': %s.",
|
||||
erroffset, i->zMatch, errbuf);
|
||||
}
|
||||
|
||||
++i;
|
||||
}
|
||||
|
||||
if (!success)
|
||||
{
|
||||
free_regexes();
|
||||
}
|
||||
|
||||
return success;
|
||||
}
|
||||
|
||||
void free_regexes()
|
||||
{
|
||||
REGEX_DATA* begin = this_unit.pRegexes;
|
||||
REGEX_DATA* i = begin + N_REGEXES;
|
||||
|
||||
while (i > begin)
|
||||
{
|
||||
--i;
|
||||
|
||||
if (i->pCode)
|
||||
{
|
||||
pcre2_code_free(i->pCode);
|
||||
i->pCode = NULL;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool create_thread_data()
|
||||
{
|
||||
bool success = true;
|
||||
|
||||
REGEX_DATA* i = this_unit.pRegexes;
|
||||
REGEX_DATA* end = i + N_REGEXES;
|
||||
|
||||
pcre2_match_data** ppData = this_thread.match_datas;
|
||||
|
||||
while (success && (i < end))
|
||||
{
|
||||
*ppData = pcre2_match_data_create_from_pattern(i->pCode, NULL);
|
||||
|
||||
if (!*ppData)
|
||||
{
|
||||
success = false;
|
||||
MXS_ERROR("PCRE2 match data creation failed.");
|
||||
}
|
||||
|
||||
++i;
|
||||
++ppData;
|
||||
}
|
||||
|
||||
if (!success)
|
||||
{
|
||||
free_thread_data();
|
||||
}
|
||||
|
||||
return success;
|
||||
}
|
||||
|
||||
void free_thread_data()
|
||||
{
|
||||
pcre2_match_data** begin = this_thread.match_datas;
|
||||
pcre2_match_data** i = begin + N_REGEXES;
|
||||
|
||||
while (i > begin)
|
||||
{
|
||||
--i;
|
||||
|
||||
if (*i)
|
||||
{
|
||||
pcre2_match_data_free(*i);
|
||||
*i = NULL;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
namespace maxscale
|
||||
{
|
||||
|
||||
//static
|
||||
bool TrxBoundaryMatcher::process_init()
|
||||
{
|
||||
bool rc = compile_regexes();
|
||||
|
||||
if (rc)
|
||||
{
|
||||
rc = thread_init();
|
||||
|
||||
if (!rc)
|
||||
{
|
||||
free_regexes();
|
||||
}
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
//static
|
||||
void TrxBoundaryMatcher::process_end()
|
||||
{
|
||||
thread_end();
|
||||
free_regexes();
|
||||
}
|
||||
|
||||
//static
|
||||
bool TrxBoundaryMatcher::thread_init()
|
||||
{
|
||||
return create_thread_data();
|
||||
}
|
||||
|
||||
//static
|
||||
void TrxBoundaryMatcher::thread_end()
|
||||
{
|
||||
free_thread_data();
|
||||
}
|
||||
|
||||
//static
|
||||
uint32_t TrxBoundaryMatcher::type_mask_of(const char* pSql, size_t len)
|
||||
{
|
||||
uint32_t type_mask = 0;
|
||||
|
||||
REGEX_DATA* i = this_unit.pRegexes;
|
||||
REGEX_DATA* end = i + N_REGEXES;
|
||||
pcre2_match_data** ppData = this_thread.match_datas;
|
||||
|
||||
while ((type_mask == 0) && (i < end))
|
||||
{
|
||||
if (pcre2_match(i->pCode, (PCRE2_SPTR)pSql, len, 0, 0, *ppData, NULL) >= 0)
|
||||
{
|
||||
type_mask = i->type_mask;
|
||||
}
|
||||
|
||||
++i;
|
||||
++ppData;
|
||||
}
|
||||
|
||||
return type_mask;
|
||||
}
|
||||
|
||||
//static
|
||||
uint32_t TrxBoundaryMatcher::type_mask_of(GWBUF* pBuf)
|
||||
{
|
||||
uint32_t type_mask = 0;
|
||||
|
||||
char* pSql;
|
||||
int len;
|
||||
|
||||
// This will exclude prepared statement but we are fine with that.
|
||||
if (modutil_extract_SQL(pBuf, &pSql, &len))
|
||||
{
|
||||
type_mask = type_mask_of(pSql, len);
|
||||
}
|
||||
|
||||
return type_mask;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user