rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-17 05:20:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:make version() show greptime info. (#1749)

Browse Source 
* feat:impl get_version() to return greptime info.

* fix: refactor test case.
This commit is contained in:
Zou Wei
2023-06-09 11:38:52 +08:00
committed by GitHub
parent 115d9eea8d
commit f91cd250f8
1 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/servers/src/mysql/federated.rs
View File

@@ -16,6 +16,7 @@
//! Inspired by Databend's "[mysql_federated.rs](https://github.com/datafuselabs/databend/blob/ac706bf65845e6895141c96c0a10bad6fdc2d367/src/query/service/src/servers/mysql/mysql_federated.rs)".
use std::collections::HashMap;
use std::env;
use std::sync::Arc;
use common_query::Output;
@@ -30,8 +31,6 @@ use regex::bytes::RegexSet;
use regex::Regex;
use session::context::QueryContextRef;
const MYSQL_VERSION: &str = "8.0.26";
static SELECT_VAR_PATTERN: Lazy<Regex> = Lazy::new(|| Regex::new("(?i)^(SELECT @@(.*))").unwrap());
static MYSQL_CONN_JAVA_PATTERN: Lazy<Regex> =
Lazy::new(|| Regex::new("(?i)^(/\\* mysql-connector-j(.*))").unwrap());
@@ -284,7 +283,7 @@ fn check_others(query: &str, query_ctx: QueryContextRef) -> Option<Output> {
}
let recordbatches = if SELECT_VERSION_PATTERN.is_match(query) {
Some(select_function("version()", MYSQL_VERSION))
Some(select_function("version()", &get_version()))
} else if SELECT_DATABASE_PATTERN.is_match(query) {
let schema = query_ctx.current_schema();
Some(select_function("database()", &schema))
@@ -317,8 +316,16 @@ pub(crate) fn check(query: &str, query_ctx: QueryContextRef) -> Option<Output> {
.or_else(|| check_others(query, query_ctx))
}
// get GreptimeDB's version.
fn get_version() -> String {
format!(
"{}-greptime",
env::var("CARGO_PKG_VERSION").unwrap_or_else(|_| "unknown".to_string()),
)
}
#[cfg(test)]
mod test {
use session::context::QueryContext;
use super::*;
@@ -344,13 +351,15 @@ mod test {
}
let query = "select version()";
let expected = "\
+-----------+
| version() |
+-----------+
| 8.0.26 |
+-----------+";
test(query, expected);
let expected = format!(
r#"+----------------+
| version() |
+----------------+
| {}-greptime |
+----------------+"#,
env::var("CARGO_PKG_VERSION").unwrap_or_else(|_| "unknown".to_string())
);
test(query, &expected);
let query = "SELECT @@version_comment LIMIT 1";
let expected = "\