More examples and auto ID (#187)

* More examples and auto id

* Fix test

* woops

Author: levkk

Cargo.lock

@@ -1,6 +1,6 @@
 [workspace]
 
-members = [
+members = [ "examples/demo",
     "examples/routing-plugin", "integration/rust",
     "pgdog",
     "pgdog-plugin",

Cargo.toml

@@ -0,0 +1,5 @@
+CREATE TABLE kv (
+    id BIGINT PRIMARY KEY,
+    data JSONB,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW ()
+);

examples/demo/1_setup.sql

@@ -0,0 +1,11 @@
+/* pgdog_shard: 0 */
+SELECT
+    pgdog.install_next_id ('public', 'kv', 'id', 3, 0);
+
+/* pgdog_shard: 1 */
+SELECT
+    pgdog.install_next_id ('public', 'kv', 'id', 3, 1);
+
+/* pgdog_shard: 2 */
+SELECT
+    pgdog.install_next_id ('public', 'kv', 'id', 3, 2);

examples/demo/2_schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE kv (
+    id BIGINT PRIMARY KEY,
+    data JSONB,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW ()
+);

examples/demo/3_ids.sql

@@ -0,0 +1,13 @@
+INSERT INTO
+    kv (data)
+VALUES
+    '{"fruit": "orange"}' RETURNING *;
+
+
+SELECT * FROM kv WHERE id = 1;
+
+SELECT * FROM kv;
+
+SELECT * FROM kv ORDER BY id;
+
+SELECT data->>'fruit', count(*) FROM kv GROUP BY 1 ORDER BY 2 DESC;

examples/demo/3_kv.sql

@@ -0,0 +1,17 @@
+CREATE TABLE kv_meta (
+    kv_id BIGINT NOT NULL REFERENCES kv(id),
+    created_by VARCHAR NOT NULL,
+    admin BOOL NOT NULL DEFAULT false
+);
+
+INSERT INTO kv_meta (data_id, created_by) VALUES (1, 'lev') RETURNING *;
+INSERT INTO kv_meta (data_id, created_by) VALUES (1, 'bob') RETURNING *;
+INSERT INTO kv_meta (data_id, created_by) VALUES (1, 'alice') RETURNING *;
+
+
+SELECT * FROM data INNER JOIN kv_meta
+ON data.id = kv_meta.data_id
+WHERE data_id = 1;
+
+SELECT * FROM data INNER JOIN kv_meta
+ON data.id = kv_meta.data_id;

examples/demo/4_kv_ops.sql

@@ -0,0 +1,9 @@
+[package]
+name = "demo"
+version = "0.1.0"
+edition = "2024"
+
+[dependencies]
+colored = "3.0"
+tokio = { version = "1", features = ["full"]}
+which = "7"

examples/demo/5_joins.sql

@@ -0,0 +1,30 @@
+services:
+  db_0:
+    image: postgres:16
+    environment:
+      POSTGRES_PASSWORD: postgres
+    ports:
+      - 6000:5432
+    volumes:
+      - shard_0:/var/lib/postgresql/data
+  db_1:
+    image: postgres:16
+    environment:
+      POSTGRES_PASSWORD: postgres
+    ports:
+      - 6001:5432
+    volumes:
+      - shard_1:/var/lib/postgresql/data
+  db_2:
+    image: postgres:16
+    environment:
+      POSTGRES_PASSWORD: postgres
+    ports:
+      - 6002:5432
+    volumes:
+      - shard_2:/var/lib/postgresql/data
+
+volumes:
+  shard_0:
+  shard_1:
+  shard_2:

examples/demo/Cargo.toml

@@ -0,0 +1,2 @@
+select
+    1;

examples/demo/docker-compose.yml

@@ -0,0 +1,35 @@
+#
+# PgDog configuration.
+#
+
+[general]
+openmetrics_port = 9090
+
+[[databases]]
+name = "postgres"
+host = "127.0.0.1"
+port = 6000
+shard = 0
+
+[[databases]]
+name = "postgres"
+host = "127.0.0.1"
+port = 6001
+shard = 1
+
+[[databases]]
+name = "postgres"
+host = "127.0.0.1"
+port = 6002
+shard = 2
+
+[[sharded_tables]]
+name = "kv"
+column = "id"
+data_type = "bigint"
+database = "postgres"
+
+[[sharded_tables]]
+column = "kv_id"
+data_type = "bigint"
+database = "postgres"

examples/demo/pgbench.sql

@@ -0,0 +1,4 @@
+DROP TABLE IF EXISTS kv CASCADE;
+DROP TABLE IF EXISTS kv_meta CASCADE;
+
+DROP SCHEMA IF EXISTS pgdog CASCADE;