feat: simple version switch

GreptimeTeam · Dec 9, 2024 · bad3373 · bad3373
1 parent 903da8f
commit bad3373
Show file tree

Hide file tree

Showing 4 changed files with 79 additions and 0 deletions.
diff --git a/tests/runner/src/env.rs b/tests/runner/src/env.rs
@@ -45,6 +45,7 @@ use tokio::sync::Mutex as TokioMutex;
 use tokio_postgres::{Client as PgClient, SimpleQueryMessage as PgRow};
 
 use crate::protocol_interceptor::{MYSQL, PROTOCOL_KEY};
+use crate::util::get_workspace_root;
 use crate::{util, ServerAddr};
 
 const METASRV_ADDR: &str = "127.0.0.1:29302";
@@ -693,8 +694,22 @@ impl GreptimeDB {
 impl Database for GreptimeDB {
     async fn query(&self, ctx: QueryContext, query: String) -> Box<dyn Display> {
         if ctx.context.contains_key("restart") && self.env.server_addrs.server_addr.is_none() {
+            self.env.restart_server(self).await;
+        } else if let Some(version) = ctx.context.get("version") {
+            if version == "latest" {
+                // use default latest by building db now
+                *self.env.bins_dir.lock().unwrap() = Some(util::get_binary_dir("debug"));
+            } else {
+                // use version in dir files
+                let root = get_workspace_root();
+                let new_path = PathBuf::from_iter([&root, version]);
+                println!("DEBUG: {:?}", new_path);
+                *self.env.bins_dir.lock().unwrap() = Some(new_path);
+            }
+
             self.env.restart_server(self).await;
         }
+
         if let Some(protocol) = ctx.context.get(PROTOCOL_KEY) {
             // protocol is bound to be either "mysql" or "postgres"
             if protocol == MYSQL {

diff --git a/tests/runner/src/main.rs b/tests/runner/src/main.rs
@@ -141,6 +141,7 @@ async fn main() {
             args.bins_dir,
         ),
     );
+    println!("Working directory: {:?}", std::env::current_dir());
     runner.run().await.unwrap();
 
     // clean up and exit

diff --git a/tests/upgrade-compat/standalone/test_simple.result b/tests/upgrade-compat/standalone/test_simple.result
@@ -0,0 +1,43 @@
+-- SQLNESS ARG version=v0.9.5
+CREATE TABLE system_metrics (
+    host STRING,
+    idc STRING,
+    cpu_util DOUBLE,
+    memory_util DOUBLE,
+    disk_util DOUBLE,
+    ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP(),
+    PRIMARY KEY(host, idc),
+    TIME INDEX(ts)
+);
+
+Affected Rows: 0
+
+INSERT INTO system_metrics
+VALUES
+    ("host1", "idc_a", 11.8, 10.3, 10.3, 1667446797450),
+    ("host2", "idc_a", 80.0, 70.3, 90.0, 1667446797450),
+    ("host1", "idc_b", 50.0, 66.7, 40.6, 1667446797450);
+
+Affected Rows: 3
+
+-- SQLNESS ARG version=latest
+SHOW CREATE TABLE system_metrics;
+
++----------------+-----------------------------------------------------------+
+| Table          | Create Table                                              |
++----------------+-----------------------------------------------------------+
+| system_metrics | CREATE TABLE IF NOT EXISTS "system_metrics" (             |
+|                |   "host" STRING NULL,                                     |
+|                |   "idc" STRING NULL,                                      |
+|                |   "cpu_util" DOUBLE NULL,                                 |
+|                |   "memory_util" DOUBLE NULL,                              |
+|                |   "disk_util" DOUBLE NULL,                                |
+|                |   "ts" TIMESTAMP(3) NOT NULL DEFAULT current_timestamp(), |
+|                |   TIME INDEX ("ts"),                                      |
+|                |   PRIMARY KEY ("host", "idc")                             |
+|                | )                                                         |
+|                |                                                           |
+|                | ENGINE=mito                                               |
+|                |                                                           |
++----------------+-----------------------------------------------------------+
+
diff --git a/tests/upgrade-compat/standalone/test_simple.sql b/tests/upgrade-compat/standalone/test_simple.sql
@@ -0,0 +1,20 @@
+-- SQLNESS ARG version=v0.9.5
+CREATE TABLE system_metrics (
+    host STRING,
+    idc STRING,
+    cpu_util DOUBLE,
+    memory_util DOUBLE,
+    disk_util DOUBLE,
+    ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP(),
+    PRIMARY KEY(host, idc),
+    TIME INDEX(ts)
+);
+
+INSERT INTO system_metrics
+VALUES
+    ("host1", "idc_a", 11.8, 10.3, 10.3, 1667446797450),
+    ("host2", "idc_a", 80.0, 70.3, 90.0, 1667446797450),
+    ("host1", "idc_b", 50.0, 66.7, 40.6, 1667446797450);
+
+-- SQLNESS ARG version=latest
+SHOW CREATE TABLE system_metrics;