refactor(risingwave): port to sqlglot (ibis-project#8171)

Co-authored-by: Kexiang Wang <[email protected]> Co-authored-by: Phillip Cloud <[email protected]> Co-authored-by: Jim Crist-Harif <[email protected]>
cpcloud · Feb 12, 2024 · b89c06d · b89c06d
1 parent 491314f
commit b89c06d
Show file tree

Hide file tree

Showing 50 changed files with 1,026 additions and 1,993 deletions.
diff --git a/.github/workflows/ibis-backends.yml b/.github/workflows/ibis-backends.yml
@@ -125,6 +125,12 @@ jobs:
               - postgres
             sys-deps:
               - libgeos-dev
+          - name: risingwave
+            title: Risingwave
+            services:
+              - risingwave
+            extras:
+              - risingwave
           - name: impala
             title: Impala
             serial: true
@@ -220,6 +226,14 @@ jobs:
                 - postgres
               sys-deps:
                 - libgeos-dev
+          - os: windows-latest
+            backend:
+              name: risingwave
+              title: Risingwave
+              services:
+                - risingwave
+              extras:
+                - risingwave
           - os: windows-latest
             backend:
               name: postgres

diff --git a/ci/schema/risingwave.sql b/ci/schema/risingwave.sql
@@ -1,27 +1,27 @@
 SET RW_IMPLICIT_FLUSH=true;
 
-DROP TABLE IF EXISTS diamonds CASCADE;
-
-CREATE TABLE diamonds (
-    carat FLOAT,
-    cut TEXT,
-    color TEXT,
-    clarity TEXT,
-    depth FLOAT,
+DROP TABLE IF EXISTS "diamonds" CASCADE;
+
+CREATE TABLE "diamonds" (
+    "carat" FLOAT,
+    "cut" TEXT,
+    "color" TEXT,
+    "clarity" TEXT,
+    "depth" FLOAT,
     "table" FLOAT,
-    price BIGINT,
-    x FLOAT,
-    y FLOAT,
-    z FLOAT
+    "price" BIGINT,
+    "x" FLOAT,
+    "y" FLOAT,
+    "z" FLOAT
 ) WITH (
   connector = 'posix_fs',
   match_pattern = 'diamonds.csv',
   posix_fs.root = '/data',
 ) FORMAT PLAIN ENCODE CSV ( without_header = 'false', delimiter = ',' );
 
-DROP TABLE IF EXISTS astronauts CASCADE;
+DROP TABLE IF EXISTS "astronauts" CASCADE;
 
-CREATE TABLE astronauts (
+CREATE TABLE "astronauts" (
     "id" BIGINT,
     "number" BIGINT,
     "nationwide_number" BIGINT,
@@ -52,12 +52,12 @@ CREATE TABLE astronauts (
   posix_fs.root = '/data',
 ) FORMAT PLAIN ENCODE CSV ( without_header = 'false', delimiter = ',' );
 
-DROP TABLE IF EXISTS batting CASCADE;
+DROP TABLE IF EXISTS "batting" CASCADE;
 
-CREATE TABLE batting (
+CREATE TABLE "batting" (
     "playerID" TEXT,
     "yearID" BIGINT,
-    stint BIGINT,
+    "stint" BIGINT,
     "teamID" TEXT,
     "lgID" TEXT,
     "G" BIGINT,
@@ -83,93 +83,93 @@ CREATE TABLE batting (
   posix_fs.root = '/data',
 ) FORMAT PLAIN ENCODE CSV ( without_header = 'false', delimiter = ',' );
 
-DROP TABLE IF EXISTS awards_players CASCADE;
+DROP TABLE IF EXISTS "awards_players" CASCADE;
 
-CREATE TABLE awards_players (
+CREATE TABLE "awards_players" (
     "playerID" TEXT,
     "awardID" TEXT,
     "yearID" BIGINT,
     "lgID" TEXT,
-    tie TEXT,
-    notes TEXT
+    "tie" TEXT,
+    "notes" TEXT
 ) WITH (
   connector = 'posix_fs',
   match_pattern = 'awards_players.csv',
   posix_fs.root = '/data',
 ) FORMAT PLAIN ENCODE CSV ( without_header = 'false', delimiter = ',' );
 
-DROP TABLE IF EXISTS functional_alltypes CASCADE;
-
-CREATE TABLE functional_alltypes (
-    id INTEGER,
-    bool_col BOOLEAN,
-    tinyint_col SMALLINT,
-    smallint_col SMALLINT,
-    int_col INTEGER,
-    bigint_col BIGINT,
-    float_col REAL,
-    double_col DOUBLE PRECISION,
-    date_string_col TEXT,
-    string_col TEXT,
-    timestamp_col TIMESTAMP WITHOUT TIME ZONE,
-    year INTEGER,
-    month INTEGER
+DROP TABLE IF EXISTS "functional_alltypes" CASCADE;
+
+CREATE TABLE "functional_alltypes" (
+    "id" INTEGER,
+    "bool_col" BOOLEAN,
+    "tinyint_col" SMALLINT,
+    "smallint_col" SMALLINT,
+    "int_col" INTEGER,
+    "bigint_col" BIGINT,
+    "float_col" REAL,
+    "double_col" DOUBLE PRECISION,
+    "date_string_col" TEXT,
+    "string_col" TEXT,
+    "timestamp_col" TIMESTAMP WITHOUT TIME ZONE,
+    "year" INTEGER,
+    "month" INTEGER
 ) WITH (
   connector = 'posix_fs',
   match_pattern = 'functional_alltypes.csv',
   posix_fs.root = '/data',
 ) FORMAT PLAIN ENCODE CSV ( without_header = 'false', delimiter = ',' );
 
-DROP TABLE IF EXISTS tzone CASCADE;
+DROP TABLE IF EXISTS "tzone" CASCADE;
 
-CREATE TABLE tzone (
-    ts TIMESTAMP WITH TIME ZONE,
-    key TEXT,
-    value DOUBLE PRECISION
+CREATE TABLE "tzone" (
+    "ts" TIMESTAMP WITH TIME ZONE,
+    "key" TEXT,
+    "value" DOUBLE PRECISION
 );
 
-INSERT INTO tzone
+INSERT INTO "tzone"
     SELECT
 	CAST('2017-05-28 11:01:31.000400' AS TIMESTAMP WITH TIME ZONE) +
-	    t * INTERVAL '1 day 1 second' AS ts,
-	CHR(97 + t) AS key,
-	t + t / 10.0 AS value
-    FROM generate_series(0, 9) AS t;
-
-DROP TABLE IF EXISTS array_types CASCADE;
-
-CREATE TABLE IF NOT EXISTS array_types (
-    x BIGINT[],
-    y TEXT[],
-    z DOUBLE PRECISION[],
-    grouper TEXT,
-    scalar_column DOUBLE PRECISION,
-    multi_dim BIGINT[][]
+	    t * INTERVAL '1 day 1 second' AS "ts",
+	CHR(97 + t) AS "key",
+	t + t / 10.0 AS "value"
+    FROM generate_series(0, 9) AS "t";
+
+DROP TABLE IF EXISTS "array_types" CASCADE;
+
+CREATE TABLE IF NOT EXISTS "array_types" (
+    "x" BIGINT[],
+    "y" TEXT[],
+    "z" DOUBLE PRECISION[],
+    "grouper" TEXT,
+    "scalar_column" DOUBLE PRECISION,
+    "multi_dim" BIGINT[][]
 );
 
-INSERT INTO array_types VALUES
+INSERT INTO "array_types" VALUES
     (ARRAY[1, 2, 3], ARRAY['a', 'b', 'c'], ARRAY[1.0, 2.0, 3.0], 'a', 1.0, ARRAY[ARRAY[NULL::BIGINT, NULL, NULL], ARRAY[1, 2, 3]]),
     (ARRAY[4, 5], ARRAY['d', 'e'], ARRAY[4.0, 5.0], 'a', 2.0, ARRAY[]::BIGINT[][]),
     (ARRAY[6, NULL], ARRAY['f', NULL], ARRAY[6.0, NULL], 'a', 3.0, ARRAY[NULL, ARRAY[]::BIGINT[], NULL]),
     (ARRAY[NULL, 1, NULL], ARRAY[NULL, 'a', NULL], ARRAY[]::DOUBLE PRECISION[], 'b', 4.0, ARRAY[ARRAY[1], ARRAY[2], ARRAY[NULL::BIGINT], ARRAY[3]]),
     (ARRAY[2, NULL, 3], ARRAY['b', NULL, 'c'], NULL, 'b', 5.0, NULL),
     (ARRAY[4, NULL, NULL, 5], ARRAY['d', NULL, NULL, 'e'], ARRAY[4.0, NULL, NULL, 5.0], 'c', 6.0, ARRAY[ARRAY[1, 2, 3]]);
 
-DROP TABLE IF EXISTS json_t CASCADE;
+DROP TABLE IF EXISTS "json_t" CASCADE;
 
-CREATE TABLE IF NOT EXISTS json_t (js JSONB);
+CREATE TABLE IF NOT EXISTS "json_t" ("js" JSONB);
 
-INSERT INTO json_t VALUES
+INSERT INTO "json_t" VALUES
     ('{"a": [1,2,3,4], "b": 1}'),
     ('{"a":null,"b":2}'),
     ('{"a":"foo", "c":null}'),
     ('null'),
     ('[42,47,55]'),
     ('[]');
 
-DROP TABLE IF EXISTS win CASCADE;
-CREATE TABLE win (g TEXT, x BIGINT, y BIGINT);
-INSERT INTO win VALUES
+DROP TABLE IF EXISTS "win" CASCADE;
+CREATE TABLE "win" ("g" TEXT, "x" BIGINT, "y" BIGINT);
+INSERT INTO "win" VALUES
     ('a', 0, 3),
     ('a', 1, 2),
     ('a', 2, 0),