support RisingWave

xxchan · xxchan · commit 76ea9c892e43 · 2023-11-02T17:04:37.000+08:00
diff --git a/Makefile b/Makefile
@@ -1,5 +1,5 @@
 SOURCE ?= file go_bindata github github_ee bitbucket aws_s3 google_cloud_storage godoc_vfs gitlab
-DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb yugabytedb clickhouse mongodb sqlserver firebird neo4j pgx pgx5
+DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb yugabytedb clickhouse mongodb sqlserver firebird neo4j pgx pgx5 risingwave
 DATABASE_TEST ?= $(DATABASE) sqlite sqlite3 sqlcipher
 VERSION ?= $(shell git describe --tags 2>/dev/null | cut -c 2-)
 TEST_FLAGS ?=
diff --git a/database/risingwave/README.md b/database/risingwave/README.md
@@ -0,0 +1,25 @@
+# RisingWave
+
+`risingwave://user:password@host:port/dbname?query`
+
+| URL Query  | WithInstance Config | Description                                                                                                                                                                                                           |
+|------------|---------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| `x-migrations-table` | `MigrationsTable` | Name of the migrations table                                                                                                                                                                                          |
+| `x-migrations-table-quoted` | `MigrationsTableQuoted` | By default, migrate quotes the migration table for SQL injection safety reasons. This option disable quoting and naively checks that you have quoted the migration table name. e.g. `"my_schema"."schema_migrations"` |
+| `x-statement-timeout` | `StatementTimeout` | Abort any statement that takes more than the specified number of milliseconds                                                                                                                                         |
+| `dbname` | `DatabaseName` | The name of the database to connect to                                                                                                                                                                                |
+| `user` | | The user to sign in as                                                                                                                                                                                                |
+| `password` | | The user's password                                                                                                                                                                                                   | 
+| `host` | | The host to connect to. Values that start with / are for unix domain sockets. (default is localhost)                                                                                                                  |
+| `port` | | The port to bind to. (default is 4566)                                                                                                                                                                                |
+| `connect_timeout` | | Maximum wait for connection, in seconds. Zero or not specified means wait indefinitely.                                                                                                                               |
+| `sslcert` | | Cert file location. The file must contain PEM encoded data.                                                                                                                                                           |
+| `sslkey` | | Key file location. The file must contain PEM encoded data.                                                                                                                                                            |
+| `sslrootcert` | | The location of the root certificate file. The file must contain PEM encoded data.                                                                                                                                    | 
+| `sslmode` | | Whether or not to use SSL (disable\|require\|verify-ca\|verify-full)                                                                                                                                                  |
+
+RisingWave is PostgreSQL compatible but has some specific features (or lack thereof) that require slightly different behavior.
+
+## Notes
+
+* RisingWave does not support transaction yet. When running multiple SQL statements in one migration, the queries are not executed in any sort of transaction/batch, meaning you are responsible for fixing partial migrations.
diff --git a/database/risingwave/examples/migrations/1085649617_create_users_table.down.sql b/database/risingwave/examples/migrations/1085649617_create_users_table.down.sql
@@ -0,0 +1 @@
+DROP TABLE IF EXISTS users;
diff --git a/database/risingwave/examples/migrations/1085649617_create_users_table.up.sql b/database/risingwave/examples/migrations/1085649617_create_users_table.up.sql
@@ -0,0 +1,5 @@
+CREATE TABLE users (
+  user_id integer,
+  name    varchar,
+  email   varchar
+);
diff --git a/database/risingwave/examples/migrations/1185749658_add_city_to_users.down.sql b/database/risingwave/examples/migrations/1185749658_add_city_to_users.down.sql
@@ -0,0 +1 @@
+ALTER TABLE users DROP COLUMN IF EXISTS city;
diff --git a/database/risingwave/examples/migrations/1185749658_add_city_to_users.up.sql b/database/risingwave/examples/migrations/1185749658_add_city_to_users.up.sql
@@ -0,0 +1,3 @@
+ALTER TABLE users ADD COLUMN city varchar;
+
+
diff --git a/database/risingwave/examples/migrations/1285849751_add_index_on_user_emails.down.sql b/database/risingwave/examples/migrations/1285849751_add_index_on_user_emails.down.sql
@@ -0,0 +1 @@
+DROP INDEX IF EXISTS users_email_index;
diff --git a/database/risingwave/examples/migrations/1285849751_add_index_on_user_emails.up.sql b/database/risingwave/examples/migrations/1285849751_add_index_on_user_emails.up.sql
@@ -0,0 +1,3 @@
+CREATE INDEX users_email_index ON users (email);
+
+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.
diff --git a/database/risingwave/examples/migrations/1385949617_create_books_table.down.sql b/database/risingwave/examples/migrations/1385949617_create_books_table.down.sql
@@ -0,0 +1 @@
+DROP TABLE IF EXISTS books;
diff --git a/database/risingwave/examples/migrations/1385949617_create_books_table.up.sql b/database/risingwave/examples/migrations/1385949617_create_books_table.up.sql
@@ -0,0 +1,5 @@
+CREATE TABLE books (
+  user_id integer,
+  name    varchar,
+  author  varchar
+);
diff --git a/database/risingwave/examples/migrations/1485949617_create_movies_table.down.sql b/database/risingwave/examples/migrations/1485949617_create_movies_table.down.sql
@@ -0,0 +1 @@
+DROP TABLE IF EXISTS movies;
diff --git a/database/risingwave/examples/migrations/1485949617_create_movies_table.up.sql b/database/risingwave/examples/migrations/1485949617_create_movies_table.up.sql
@@ -0,0 +1,5 @@
+CREATE TABLE movies (
+  user_id   integer,
+  name      varchar,
+  director  varchar
+);
diff --git a/database/risingwave/examples/migrations/1585849751_just_a_comment.up.sql b/database/risingwave/examples/migrations/1585849751_just_a_comment.up.sql
@@ -0,0 +1 @@
+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.
diff --git a/database/risingwave/examples/migrations/1685849751_another_comment.up.sql b/database/risingwave/examples/migrations/1685849751_another_comment.up.sql
@@ -0,0 +1 @@
+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.
diff --git a/database/risingwave/examples/migrations/1785849751_another_comment.up.sql b/database/risingwave/examples/migrations/1785849751_another_comment.up.sql
@@ -0,0 +1 @@
+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.
diff --git a/database/risingwave/examples/migrations/1885849751_another_comment.up.sql b/database/risingwave/examples/migrations/1885849751_another_comment.up.sql
@@ -0,0 +1 @@
+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.
diff --git a/database/risingwave/risingwave.go b/database/risingwave/risingwave.go
diff --git a/database/risingwave/risingwave_test.go b/database/risingwave/risingwave_test.go
diff --git a/go.mod b/go.mod
diff --git a/go.sum b/go.sum
diff --git a/internal/cli/build_risingwave.go b/internal/cli/build_risingwave.go

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+ALTER TABLE users DROP COLUMN IF EXISTS city;`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+ALTER TABLE users ADD COLUMN city varchar;`
	`2`	`+`
	`3`	`+`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+DROP INDEX IF EXISTS users_email_index;`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+CREATE INDEX users_email_index ON users (email);`
	`2`	`+`
	`3`	`+-- Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean sed interdum velit, tristique iaculis justo. Pellentesque ut porttitor dolor. Donec sit amet pharetra elit. Cras vel ligula ex. Phasellus posuere.`