mirror of https://github.com/sysown/proxysql
Merge pull request #2762 from sysown/v2.1.0-266_0-3
Implement feature requests: 2660, 2661, 2662, 2663, 1377pull/2779/head
René Cannaò
6 years ago
committed by
GitHub
commit
3af96fb177
@ -0,0 +1,179 @@
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <cstring>
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
|
||||
#include "proxysql.h"
|
||||
|
||||
#include "tap.h"
|
||||
#include "command_line.h"
|
||||
|
||||
#include "utils.h"
|
||||
|
||||
#include <ctype.h>
|
||||
|
||||
__thread int mysql_thread___query_digests_max_query_length = 65000;
|
||||
__thread bool mysql_thread___query_digests_lowercase = false;
|
||||
__thread bool mysql_thread___query_digests_replace_null = false;
|
||||
__thread bool mysql_thread___query_digests_no_digits = false;
|
||||
__thread int mysql_thread___query_digests_grouping_limit = 3;
|
||||
|
||||
const std::vector<std::string> queries {
|
||||
// floats
|
||||
"select 1.1",
|
||||
"select 1192.1102",
|
||||
"select 99.1929",
|
||||
// exponentials
|
||||
"select 1.1e9",
|
||||
"select 1.1e+9",
|
||||
"select 1.1e-9",
|
||||
// operators
|
||||
"select 1 +1",
|
||||
"select 1+ 1",
|
||||
"select 1- 1",
|
||||
"select 1 -1",
|
||||
"select 1* 1",
|
||||
"select 1 *1",
|
||||
"select 1/ 1",
|
||||
"select 1 /1",
|
||||
"select 1% 1",
|
||||
"select 1 %1",
|
||||
// operators and commas
|
||||
"select 1+ 1, 1 -1, 1 * 1 , 1/1 , 100 % 3",
|
||||
"SELECT * FROM t t1, t t2 ,t t3,t t4 LIMIT 0",
|
||||
// strings
|
||||
"select * from t where t = \"foo\"",
|
||||
"select \"1+ 1, 1 -1, 1 * 1 , 1/1 , 100 % 3\"",
|
||||
// not modified
|
||||
"select * fromt t",
|
||||
// test query_digest reduction
|
||||
"SELECT * FROM tablename WHERE id IN (1,2,3,4,5,6,7,8,9,10)",
|
||||
"SELECT * FROM tablename WHERE id IN (1,2,3,4)",
|
||||
// invalid request grouping
|
||||
"SELECT * tablename where id IN (1,2,3,4,5,6,7,8, AND j in (1,2,3,4,5,6 and k=1"
|
||||
};
|
||||
|
||||
const std::vector<std::string> exp_results {
|
||||
// floats
|
||||
"select ?",
|
||||
"select ?",
|
||||
"select ?",
|
||||
// exponentials
|
||||
"select ?",
|
||||
"select ?",
|
||||
"select ?",
|
||||
// operators
|
||||
"select ?+?",
|
||||
"select ?+?",
|
||||
"select ?-?",
|
||||
"select ?-?",
|
||||
"select ?*?",
|
||||
"select ?*?",
|
||||
"select ?/?",
|
||||
"select ?/?",
|
||||
"select ?%?",
|
||||
"select ?%?",
|
||||
// operators and commas
|
||||
"select ?+?,?-?,?*?,?/?,?%?",
|
||||
"SELECT * FROM t t1,t t2,t t3,t t4 LIMIT ?",
|
||||
// strings
|
||||
"select * from t where t = ?",
|
||||
"select ?",
|
||||
// not modified
|
||||
"select * fromt t",
|
||||
// test query_digest reduction
|
||||
"SELECT * FROM tablename WHERE id IN (?,?,?,...)",
|
||||
"SELECT * FROM tablename WHERE id IN (?,?,?,...)",
|
||||
// invalid request grouping
|
||||
"SELECT * tablename where id IN (?,?,?,... AND j in (?,?,?,... and k=?"
|
||||
};
|
||||
|
||||
const std::vector<std::string> queries_grouping {
|
||||
// test query_digest reduction
|
||||
"SELECT * FROM tablename WHERE id IN (1,2, 3,4 ,5 ,6,7,8,9,10)",
|
||||
// invalid request grouping
|
||||
"SELECT * tablename where id IN (1,2,3,4,5 , 6,7,8, AND j in (1, 2,3, 4 ,5,6,7,8,9 and k=1",
|
||||
"SELECT (1.1, 1, 2, 13, 4.81, 12) FROM db.table",
|
||||
"SELECT (1.1, 1.12934 , 21.32 , 91, 91, 12.93 ) FROM db.table2",
|
||||
"SELECT (1.1, 1.12934 , 21.32 , 91.2 , 91, 12 ) FROM db.table7",
|
||||
"SELECT (1.1, 1.12934, 21.32, 391,2381,28.493,1283 ) FROM db.table2",
|
||||
"SELECT (1.1, 1.12934, 21.32 , 91, 91, 12.1 ) FROM db.table3"
|
||||
};
|
||||
|
||||
const std::vector<std::string> exp_queries_grouping {
|
||||
// test query_digest reduction
|
||||
"SELECT * FROM tablename WHERE id IN (?,...)",
|
||||
// invalid request grouping
|
||||
"SELECT * tablename where id IN (?,... AND j in (?,... and k=?",
|
||||
"SELECT (?,...) FROM db.table",
|
||||
"SELECT (?,...) FROM db.table2",
|
||||
"SELECT (?,...) FROM db.table7",
|
||||
"SELECT (?,...) FROM db.table2",
|
||||
"SELECT (?,...) FROM db.table3"
|
||||
};
|
||||
|
||||
std::string replace_str(const std::string& str, const std::string& match, const std::string& repl) {
|
||||
if(match.empty()) {
|
||||
return str;
|
||||
}
|
||||
|
||||
std::string result = str;
|
||||
size_t start_pos = 0;
|
||||
|
||||
while((start_pos = result.find(match, start_pos)) != std::string::npos) {
|
||||
result.replace(start_pos, match.length(), repl);
|
||||
start_pos += repl.length();
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
std::string increase_mark_num(const std::string query, uint32_t num) {
|
||||
std::string result = query;
|
||||
std::string marks = "";
|
||||
|
||||
for (uint32_t i = 0; i < num - 1; i++) {
|
||||
marks += "?,";
|
||||
}
|
||||
marks += "?,...";
|
||||
|
||||
result = replace_str(result, "?,...", marks);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
if (queries.size() != exp_results.size()) {
|
||||
ok(0, "queries and exp_results sizes doesn't match");
|
||||
return exit_status();
|
||||
}
|
||||
|
||||
char buf[QUERY_DIGEST_BUF];
|
||||
|
||||
for (size_t i = 0; i < queries.size(); i++) {
|
||||
const auto& query = queries[i];
|
||||
const auto& exp_res = exp_results[i];
|
||||
|
||||
char* c_res = mysql_query_digest_and_first_comment(const_cast<char*>(query.c_str()), query.length(), NULL, buf);
|
||||
std::string result(c_res);
|
||||
|
||||
ok(result == exp_res, "Digest should be equal to exp result: '%s' == '%s'", result.c_str(), exp_res.c_str());
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < queries_grouping.size(); i++) {
|
||||
for (int j = 1; j <= 5; j++) {
|
||||
mysql_thread___query_digests_grouping_limit = j;
|
||||
|
||||
const auto& query = queries_grouping[i];
|
||||
const auto& exp_res = increase_mark_num(exp_queries_grouping[i], j);
|
||||
|
||||
char* c_res = mysql_query_digest_and_first_comment(const_cast<char*>(query.c_str()), query.length(), NULL, buf);
|
||||
std::string result(c_res);
|
||||
|
||||
ok(result == exp_res, "Grouping digest should be equal to exp result: '%s' == '%s'", result.c_str(), exp_res.c_str());
|
||||
}
|
||||
}
|
||||
|
||||
return exit_status();
|
||||
}
|
||||
Loading…
Reference in new issue