sql query preview

Author: tomas-villagesql

mysql-test/suite/villagesql/extension/vsql-thread-worker/r/thread_worker_sql_query.result

@@ -0,0 +1,13 @@
+SET PERSIST vsql_allow_preview_extensions = ON;
+INSTALL EXTENSION vsql_thread_worker_test;
+SET GLOBAL vsql_thread_worker_test.listener_enabled = ON;
+# Send QUERY SELECT 42
+SELECT vsql_thread_worker_test.active_port() INTO OUTFILE 'MYSQLTEST_VARDIR/tmp/tw_sq_port.txt';
+# last_monitor_result should be 42
+SELECT vsql_thread_worker_test.last_monitor_result();
+vsql_thread_worker_test.last_monitor_result()
+42
+SET GLOBAL vsql_thread_worker_test.listener_enabled = OFF;
+UNINSTALL EXTENSION vsql_thread_worker_test;
+SET PERSIST vsql_allow_preview_extensions = OFF;
+RESET PERSIST vsql_allow_preview_extensions;

mysql-test/suite/villagesql/extension/vsql-thread-worker/t/thread_worker_sql_query.test

@@ -0,0 +1,48 @@
+# Test thread_worker sql_query capability.
+#
+# Verifies:
+# - Worker accepts "QUERY <sql>" commands over TCP
+# - Worker executes the SQL via the sql_query capability
+# - last_monitor_result() VDF returns the first column of the first row
+
+SET PERSIST vsql_allow_preview_extensions = ON;
+
+INSTALL EXTENSION vsql_thread_worker_test;
+
+SET GLOBAL vsql_thread_worker_test.listener_enabled = ON;
+
+--let $port_file = $MYSQLTEST_VARDIR/tmp/tw_sq_port.txt
+
+--echo # Send QUERY SELECT 42
+--replace_result $MYSQLTEST_VARDIR MYSQLTEST_VARDIR
+eval SELECT vsql_thread_worker_test.active_port() INTO OUTFILE '$port_file';
+--perl
+use IO::Socket::INET;
+open(my $fh, '<', $ENV{MYSQLTEST_VARDIR} . '/tmp/tw_sq_port.txt') or die $!;
+my $port = <$fh>; chomp $port;
+close($fh);
+my $sock = IO::Socket::INET->new(
+  PeerAddr => "127.0.0.1",
+  PeerPort => $port,
+  Proto    => "tcp",
+  Timeout  => 5,
+) or die "connect failed: $!";
+$sock->autoflush(1);
+print $sock "QUERY SELECT 42\n";
+shutdown($sock, 1);
+close($sock);
+EOF
+--remove_file $port_file
+
+--let $wait_condition = SELECT vsql_thread_worker_test.last_monitor_result() = '42'
+--source include/wait_condition.inc
+
+--echo # last_monitor_result should be 42
+SELECT vsql_thread_worker_test.last_monitor_result();
+
+SET GLOBAL vsql_thread_worker_test.listener_enabled = OFF;
+
+UNINSTALL EXTENSION vsql_thread_worker_test;
+
+SET PERSIST vsql_allow_preview_extensions = OFF;
+RESET PERSIST vsql_allow_preview_extensions;