1use std::collections::HashMap;
19use std::sync::Arc;
20
21use common_query::Output;
22use common_recordbatch::RecordBatches;
23use common_time::timezone::system_timezone_name;
24use datatypes::prelude::ConcreteDataType;
25use datatypes::schema::{ColumnSchema, Schema};
26use datatypes::vectors::StringVector;
27use once_cell::sync::Lazy;
28use regex::bytes::RegexSet;
29use regex::Regex;
30use session::context::QueryContextRef;
31use session::SessionRef;
32
33static SELECT_VAR_PATTERN: Lazy<Regex> = Lazy::new(|| Regex::new("(?i)^(SELECT @@(.*))").unwrap());
34static MYSQL_CONN_JAVA_PATTERN: Lazy<Regex> =
35 Lazy::new(|| Regex::new("(?i)^(/\\* mysql-connector-j(.*))").unwrap());
36static SHOW_LOWER_CASE_PATTERN: Lazy<Regex> =
37 Lazy::new(|| Regex::new("(?i)^(SHOW VARIABLES LIKE 'lower_case_table_names'(.*))").unwrap());
38static SHOW_VARIABLES_LIKE_PATTERN: Lazy<Regex> =
39 Lazy::new(|| Regex::new("(?i)^(SHOW VARIABLES( LIKE (.*))?)").unwrap());
40
41static SELECT_TIME_DIFF_FUNC_PATTERN: Lazy<Regex> =
43 Lazy::new(|| Regex::new("(?i)^(SELECT TIMEDIFF\\(NOW\\(\\), UTC_TIMESTAMP\\(\\)\\))").unwrap());
44
45static SHOW_SQL_MODE_PATTERN: Lazy<Regex> =
47 Lazy::new(|| Regex::new("(?i)^(SHOW VARIABLES LIKE 'sql_mode'(.*))").unwrap());
48
49static OTHER_NOT_SUPPORTED_STMT: Lazy<RegexSet> = Lazy::new(|| {
50 RegexSet::new([
51 "(?i)^(ROLLBACK(.*))",
53 "(?i)^(COMMIT(.*))",
54 "(?i)^(START(.*))",
55
56 "(?i)^(SET NAMES(.*))",
58 "(?i)^(SET character_set_results(.*))",
59 "(?i)^(SET net_write_timeout(.*))",
60 "(?i)^(SET FOREIGN_KEY_CHECKS(.*))",
61 "(?i)^(SET AUTOCOMMIT(.*))",
62 "(?i)^(SET SQL_LOG_BIN(.*))",
63 "(?i)^(SET SESSION TRANSACTION(.*))",
64 "(?i)^(SET TRANSACTION(.*))",
65 "(?i)^(SET sql_mode(.*))",
66 "(?i)^(SET SQL_SELECT_LIMIT(.*))",
67 "(?i)^(SET PROFILING(.*))",
68
69 "(?i)^(SELECT \\$\\$)",
71
72 "(?i)^(SET SQL_QUOTE_SHOW_CREATE(.*))",
74 "(?i)^(LOCK TABLES(.*))",
75 "(?i)^(UNLOCK TABLES(.*))",
76 "(?i)^(SELECT LOGFILE_GROUP_NAME, FILE_NAME, TOTAL_EXTENTS, INITIAL_SIZE, ENGINE, EXTRA FROM INFORMATION_SCHEMA.FILES(.*))",
77
78 "(?i)^(/\\*!80003 SET(.*) \\*/)$",
80 "(?i)^(SHOW MASTER STATUS)",
81 "(?i)^(SHOW ALL SLAVES STATUS)",
82 "(?i)^(LOCK BINLOG FOR BACKUP)",
83 "(?i)^(LOCK TABLES FOR BACKUP)",
84 "(?i)^(UNLOCK BINLOG(.*))",
85 "(?i)^(/\\*!40101 SET(.*) \\*/)$",
86
87 "(?i)^(SHOW WARNINGS)",
89 "(?i)^(/\\* ApplicationName=(.*)SHOW WARNINGS)",
90 "(?i)^(/\\* ApplicationName=(.*)SHOW PLUGINS)",
91 "(?i)^(/\\* ApplicationName=(.*)SHOW ENGINES)",
92 "(?i)^(/\\* ApplicationName=(.*)SELECT @@(.*))",
93 "(?i)^(/\\* ApplicationName=(.*)SHOW @@(.*))",
94 "(?i)^(/\\* ApplicationName=(.*)SET net_write_timeout(.*))",
95 "(?i)^(/\\* ApplicationName=(.*)SET SQL_SELECT_LIMIT(.*))",
96 "(?i)^(/\\* ApplicationName=(.*)SHOW VARIABLES(.*))",
97
98 "(?i)^(/\\*!40101 SET(.*) \\*/)$",
100
101 "(?i)^(/\\*!40100 SET(.*) \\*/)$",
103 "(?i)^(/\\*!40103 SET(.*) \\*/)$",
104 "(?i)^(/\\*!40111 SET(.*) \\*/)$",
105 "(?i)^(/\\*!40101 SET(.*) \\*/)$",
106 "(?i)^(/\\*!40014 SET(.*) \\*/)$",
107 "(?i)^(/\\*!40000 SET(.*) \\*/)$",
108 ]).unwrap()
109});
110
111static VAR_VALUES: Lazy<HashMap<&str, &str>> = Lazy::new(|| {
112 HashMap::from([
113 ("tx_isolation", "REPEATABLE-READ"),
114 ("session.tx_isolation", "REPEATABLE-READ"),
115 ("transaction_isolation", "REPEATABLE-READ"),
116 ("session.transaction_isolation", "REPEATABLE-READ"),
117 ("session.transaction_read_only", "0"),
118 ("max_allowed_packet", "134217728"),
119 ("interactive_timeout", "31536000"),
120 ("wait_timeout", "31536000"),
121 ("net_write_timeout", "31536000"),
122 ("version_comment", "Greptime"),
123 ])
124});
125
126fn select_function(name: &str, value: &str) -> RecordBatches {
131 let schema = Arc::new(Schema::new(vec![ColumnSchema::new(
132 name,
133 ConcreteDataType::string_datatype(),
134 true,
135 )]));
136 let columns = vec![Arc::new(StringVector::from(vec![value])) as _];
137 RecordBatches::try_from_columns(schema, columns)
138 .unwrap()
140}
141
142fn show_variables(name: &str, value: &str) -> RecordBatches {
147 let schema = Arc::new(Schema::new(vec![
148 ColumnSchema::new("Variable_name", ConcreteDataType::string_datatype(), true),
149 ColumnSchema::new("Value", ConcreteDataType::string_datatype(), true),
150 ]));
151 let columns = vec![
152 Arc::new(StringVector::from(vec![name])) as _,
153 Arc::new(StringVector::from(vec![value])) as _,
154 ];
155 RecordBatches::try_from_columns(schema, columns)
156 .unwrap()
158}
159
160fn select_variable(query: &str, query_context: QueryContextRef) -> Option<Output> {
161 let mut fields = vec![];
162 let mut values = vec![];
163
164 let query = query.to_lowercase();
166 let vars: Vec<&str> = query.split("@@").collect();
167 if vars.len() <= 1 {
168 return None;
169 }
170
171 for var in vars.iter().skip(1) {
173 let var = var.trim_matches(|c| c == ' ' || c == ',' || c == ';');
174 let var_as: Vec<&str> = var
175 .split(" as ")
176 .map(|x| {
177 x.trim_matches(|c| c == ' ')
178 .split_whitespace()
179 .next()
180 .unwrap_or("")
181 })
182 .collect();
183
184 let value = match var_as[0] {
186 "session.time_zone" | "time_zone" => query_context.timezone().to_string(),
187 "system_time_zone" => system_timezone_name(),
188 "max_execution_time" | "session.max_execution_time" => {
189 query_context.query_timeout_as_millis().to_string()
190 }
191 _ => VAR_VALUES
192 .get(var_as[0])
193 .map(|v| v.to_string())
194 .unwrap_or_else(|| "0".to_owned()),
195 };
196
197 values.push(Arc::new(StringVector::from(vec![value])) as _);
198 match var_as.len() {
199 1 => {
200 fields.push(ColumnSchema::new(
203 format!("@@{}", var_as[0]),
204 ConcreteDataType::string_datatype(),
205 true,
206 ));
207 }
208 2 => {
209 fields.push(ColumnSchema::new(
213 var_as[1],
214 ConcreteDataType::string_datatype(),
215 true,
216 ));
217 }
218 _ => return None,
219 }
220 }
221
222 let schema = Arc::new(Schema::new(fields));
223 let batches = RecordBatches::try_from_columns(schema, values).unwrap();
225 Some(Output::new_with_record_batches(batches))
226}
227
228fn check_select_variable(query: &str, query_context: QueryContextRef) -> Option<Output> {
229 if [&SELECT_VAR_PATTERN, &MYSQL_CONN_JAVA_PATTERN]
230 .iter()
231 .any(|r| r.is_match(query))
232 {
233 select_variable(query, query_context)
234 } else {
235 None
236 }
237}
238
239fn check_show_variables(query: &str) -> Option<Output> {
240 let recordbatches = if SHOW_SQL_MODE_PATTERN.is_match(query) {
241 Some(show_variables("sql_mode", "ONLY_FULL_GROUP_BY STRICT_TRANS_TABLES NO_ZERO_IN_DATE NO_ZERO_DATE ERROR_FOR_DIVISION_BY_ZERO NO_ENGINE_SUBSTITUTION"))
242 } else if SHOW_LOWER_CASE_PATTERN.is_match(query) {
243 Some(show_variables("lower_case_table_names", "0"))
244 } else if SHOW_VARIABLES_LIKE_PATTERN.is_match(query) {
245 Some(show_variables("", ""))
246 } else {
247 None
248 };
249 recordbatches.map(Output::new_with_record_batches)
250}
251
252fn check_others(query: &str, _query_ctx: QueryContextRef) -> Option<Output> {
254 if OTHER_NOT_SUPPORTED_STMT.is_match(query.as_bytes()) {
255 return Some(Output::new_with_record_batches(RecordBatches::empty()));
256 }
257
258 let recordbatches = if SELECT_TIME_DIFF_FUNC_PATTERN.is_match(query) {
259 Some(select_function(
260 "TIMEDIFF(NOW(), UTC_TIMESTAMP())",
261 "00:00:00",
262 ))
263 } else {
264 None
265 };
266 recordbatches.map(Output::new_with_record_batches)
267}
268
269pub(crate) fn check(
272 query: &str,
273 query_ctx: QueryContextRef,
274 _session: SessionRef,
275) -> Option<Output> {
276 let the_6th_index = query.char_indices().nth(6).map(|(i, _)| i);
279 if let Some(index) = the_6th_index {
280 if query[..index].eq_ignore_ascii_case("INSERT") {
281 return None;
282 }
283 }
284
285 check_select_variable(query, query_ctx.clone())
287 .or_else(|| check_show_variables(query))
289 .or_else(|| check_others(query, query_ctx))
291}
292
293#[cfg(test)]
294mod test {
295
296 use common_query::OutputData;
297 use common_time::timezone::set_default_timezone;
298 use session::context::{Channel, QueryContext};
299 use session::Session;
300
301 use super::*;
302
303 #[test]
304 fn test_check_abnormal() {
305 let session = Arc::new(Session::new(None, Channel::Mysql, Default::default(), 0));
306 let query = "🫣一点不正常的东西🫣";
307 let output = check(query, QueryContext::arc(), session.clone());
308
309 assert!(output.is_none());
310 }
311
312 #[test]
313 fn test_check() {
314 let session = Arc::new(Session::new(None, Channel::Mysql, Default::default(), 0));
315 let query = "select 1";
316 let result = check(query, QueryContext::arc(), session.clone());
317 assert!(result.is_none());
318
319 let query = "select version";
320 let output = check(query, QueryContext::arc(), session.clone());
321 assert!(output.is_none());
322
323 fn test(query: &str, expected: &str) {
324 let session = Arc::new(Session::new(None, Channel::Mysql, Default::default(), 0));
325 let output = check(query, QueryContext::arc(), session.clone());
326 match output.unwrap().data {
327 OutputData::RecordBatches(r) => {
328 assert_eq!(&r.pretty_print().unwrap(), expected)
329 }
330 _ => unreachable!(),
331 }
332 }
333
334 let query = "SELECT @@version_comment LIMIT 1";
335 let expected = "\
336+-------------------+
337| @@version_comment |
338+-------------------+
339| Greptime |
340+-------------------+";
341 test(query, expected);
342
343 let query = "select @@tx_isolation, @@session.tx_isolation";
345 let expected = "\
346+-----------------+------------------------+
347| @@tx_isolation | @@session.tx_isolation |
348+-----------------+------------------------+
349| REPEATABLE-READ | REPEATABLE-READ |
350+-----------------+------------------------+";
351 test(query, expected);
352
353 set_default_timezone(Some("Asia/Shanghai")).unwrap();
355 let query = "/* mysql-connector-java-8.0.17 (Revision: 16a712ddb3f826a1933ab42b0039f7fb9eebc6ec) */SELECT @@session.auto_increment_increment AS auto_increment_increment, @@character_set_client AS character_set_client, @@character_set_connection AS character_set_connection, @@character_set_results AS character_set_results, @@character_set_server AS character_set_server, @@collation_server AS collation_server, @@collation_connection AS collation_connection, @@init_connect AS init_connect, @@interactive_timeout AS interactive_timeout, @@license AS license, @@lower_case_table_names AS lower_case_table_names, @@max_allowed_packet AS max_allowed_packet, @@net_write_timeout AS net_write_timeout, @@performance_schema AS performance_schema, @@sql_mode AS sql_mode, @@system_time_zone AS system_time_zone, @@time_zone AS time_zone, @@transaction_isolation AS transaction_isolation, @@wait_timeout AS wait_timeout;";
357 let expected = "\
358+--------------------------+----------------------+--------------------------+-----------------------+----------------------+------------------+----------------------+--------------+---------------------+---------+------------------------+--------------------+-------------------+--------------------+----------+------------------+---------------+-----------------------+--------------+
359| auto_increment_increment | character_set_client | character_set_connection | character_set_results | character_set_server | collation_server | collation_connection | init_connect | interactive_timeout | license | lower_case_table_names | max_allowed_packet | net_write_timeout | performance_schema | sql_mode | system_time_zone | time_zone | transaction_isolation | wait_timeout |
360+--------------------------+----------------------+--------------------------+-----------------------+----------------------+------------------+----------------------+--------------+---------------------+---------+------------------------+--------------------+-------------------+--------------------+----------+------------------+---------------+-----------------------+--------------+
361| 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 31536000 | 0 | 0 | 134217728 | 31536000 | 0 | 0 | Asia/Shanghai | Asia/Shanghai | REPEATABLE-READ | 31536000 |
362+--------------------------+----------------------+--------------------------+-----------------------+----------------------+------------------+----------------------+--------------+---------------------+---------+------------------------+--------------------+-------------------+--------------------+----------+------------------+---------------+-----------------------+--------------+";
363 test(query, expected);
364
365 let query = "show variables";
366 let expected = "\
367+---------------+-------+
368| Variable_name | Value |
369+---------------+-------+
370| | |
371+---------------+-------+";
372 test(query, expected);
373
374 let query = "show variables like 'lower_case_table_names'";
375 let expected = "\
376+------------------------+-------+
377| Variable_name | Value |
378+------------------------+-------+
379| lower_case_table_names | 0 |
380+------------------------+-------+";
381 test(query, expected);
382
383 let query = "SELECT TIMEDIFF(NOW(), UTC_TIMESTAMP())";
384 let expected = "\
385+----------------------------------+
386| TIMEDIFF(NOW(), UTC_TIMESTAMP()) |
387+----------------------------------+
388| 00:00:00 |
389+----------------------------------+";
390 test(query, expected);
391 }
392}