feat: cleanup migrations using postgres functions

Author: EvolveArt

pragma-entities/migrations/2023-10-11-223513_create_entries/up.sql

@@ -1,13 +1,43 @@
 -- Your SQL goes here
 CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
 
+-- SPOT entries
+
 CREATE TABLE entries (
   id uuid DEFAULT uuid_generate_v4(),
   pair_id VARCHAR NOT NULL,
-  publisher TEXT NOT NULL,
-  timestamp TIMESTAMPTZ NOT NULL,
   price NUMERIC NOT NULL,
+  timestamp TIMESTAMPTZ NOT NULL,
+  publisher TEXT NOT NULL,
+  publisher_signature TEXT NOT NULL,
+  source VARCHAR NOT NULL,
   PRIMARY KEY (id, timestamp)
 );
 
+CREATE UNIQUE INDEX idx_entries_unique
+  ON entries(pair_id, source, timestamp DESC);
 CREATE INDEX entries_pair_id_timestamp_idx ON entries (pair_id, timestamp DESC);
+
+SELECT
+  create_hypertable('entries', 'timestamp');
+
+-- FUTURE (PERP) entries
+
+CREATE TABLE future_entries (
+  id uuid DEFAULT uuid_generate_v4(),
+  pair_id VARCHAR NOT NULL,
+  price NUMERIC NOT NULL,
+  timestamp TIMESTAMPTZ NOT NULL,
+  expiration_timestamp TIMESTAMPTZ, -- can be NULL for perp contracts
+  publisher TEXT NOT NULL,
+  publisher_signature TEXT NOT NULL,
+  source VARCHAR NOT NULL,
+  PRIMARY KEY (id, timestamp)
+);
+
+CREATE UNIQUE INDEX idx_future_entries_unique ON future_entries(pair_id, source, timestamp, expiration_timestamp);
+CREATE INDEX idx_future_entries_pair_id_timestamp ON future_entries (pair_id, timestamp DESC);
+CREATE INDEX idx_future_entries_pair_id_timestamp_expiration_timestamp ON future_entries (pair_id, expiration_timestamp, timestamp DESC);
+
+SELECT
+  create_hypertable('future_entries', 'timestamp');

pragma-entities/migrations/2023-10-12-005433_add_source/down.sql

@@ -4,6 +4,7 @@ CREATE TABLE publishers (
     name VARCHAR NOT NULL,
     master_key VARCHAR NOT NULL,
     active_key VARCHAR NOT NULL,
-    active BOOLEAN NOT NULL,
+    account_address VARCHAR NOT NULL DEFAULT '',
+    active BOOLEAN NOT NULL DEFAULT true,
     PRIMARY KEY (id)
 );

pragma-entities/migrations/2023-10-12-005433_add_source/up.sql

@@ -1,7 +1,4 @@
 -- Your SQL goes here
-ALTER TABLE publishers
-ADD COLUMN account_address VARCHAR NOT NULL DEFAULT '';
-
 INSERT INTO publishers (name, master_key, active_key, active, account_address) 
 VALUES (
     'PRAGMA',

pragma-entities/migrations/2023-10-12-232125_add_publishers_table/up.sql

@@ -1,4 +1,26 @@
 -- This file should undo anything in `up.sql`
+
+-- Drop spot price aggregates
+DROP MATERIALIZED VIEW IF EXISTS price_100_ms_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_1_s_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_5_s_agg;
 DROP MATERIALIZED VIEW IF EXISTS price_1_min_agg;
 DROP MATERIALIZED VIEW IF EXISTS price_15_min_agg;
-DROP MATERIALIZED VIEW IF EXISTS price_1_h_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_1_h_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_2_h_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_1_day_agg;
+DROP MATERIALIZED VIEW IF EXISTS price_1_week_agg;
+
+-- Drop future price aggregates
+DROP MATERIALIZED VIEW IF EXISTS price_100_ms_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_1_s_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_5_s_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_1_min_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_15_min_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_1_h_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_2_h_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_1_day_agg_future;
+DROP MATERIALIZED VIEW IF EXISTS price_1_week_agg_future;
+
+-- Drop the function
+DROP FUNCTION IF EXISTS create_continuous_aggregate;

pragma-entities/migrations/2023-10-16-233310_add_account_address/up.sql

@@ -1,49 +1,50 @@
--- Your SQL goes here
-CREATE MATERIALIZED VIEW price_1_min_agg
-WITH (timescaledb.continuous, timescaledb.materialized_only = true)
-AS SELECT 
-    pair_id,
-    time_bucket('1 min'::interval, timestamp) as bucket,
-    approx_percentile(0.5, percentile_agg(price))::numeric AS median_price,
-    COUNT(DISTINCT source) as num_sources
-FROM entries
-GROUP BY bucket, pair_id
-WITH NO DATA;
+CREATE OR REPLACE FUNCTION create_continuous_aggregate(
+    p_name text,
+    p_interval interval,
+    p_start_offset interval,
+    p_table_name text
+)
+RETURNS void AS $$
+BEGIN
+    EXECUTE format('
+        CREATE MATERIALIZED VIEW %s
+        WITH (timescaledb.continuous, timescaledb.materialized_only = false)
+        AS SELECT 
+            pair_id,
+            time_bucket($1::interval, timestamp) as bucket,
+            (percentile_cont(0.5) WITHIN GROUP (ORDER BY price))::numeric(1000,0) AS median_price,
+            COUNT(DISTINCT source) as num_sources
+        FROM %I
+        GROUP BY bucket, pair_id
+        WITH NO DATA;', p_name, p_interval, p_table_name);
 
-SELECT add_continuous_aggregate_policy('price_1_min_agg',
-  start_offset => NULL,
-  end_offset => INTERVAL '1 min',
-  schedule_interval => INTERVAL '1 min');
+    EXECUTE format('
+        SELECT add_continuous_aggregate_policy(''%s'',
+            start_offset => $1,
+            end_offset => $2,
+            schedule_interval => $3);', p_name, p_start_offset, p_interval, p_interval);
+END;
+$$ LANGUAGE plpgsql;
 
-CREATE MATERIALIZED VIEW price_15_min_agg
-WITH (timescaledb.continuous, timescaledb.materialized_only = true)
-AS SELECT 
-    pair_id,
-    time_bucket('15 min'::interval, timestamp) as bucket,
-    approx_percentile(0.5, percentile_agg(price))::numeric AS median_price,
-    COUNT(DISTINCT source) as num_sources
-FROM entries
-GROUP BY bucket, pair_id
-WITH NO DATA;
+-- Spot entries aggregates
+SELECT create_continuous_aggregate('price_100_ms_agg', '100 milliseconds'::interval, '300 milliseconds'::interval, 'entries');
+SELECT create_continuous_aggregate('price_1_s_agg', '1 second'::interval, '3 seconds'::interval, 'entries');
+SELECT create_continuous_aggregate('price_5_s_agg', '5 seconds'::interval, '15 seconds'::interval, 'entries');
+SELECT create_continuous_aggregate('price_1_min_agg', '1 minute'::interval, '3 minutes'::interval, 'entries');
+SELECT create_continuous_aggregate('price_15_min_agg', '15 minutes'::interval, '45 minutes'::interval, 'entries');
+SELECT create_continuous_aggregate('price_1_h_agg', '1 hour'::interval, '3 hours'::interval, 'entries');
+SELECT create_continuous_aggregate('price_2_h_agg', '2 hours'::interval, '6 hours'::interval, 'entries');
+SELECT create_continuous_aggregate('price_1_day_agg', '1 day'::interval, '3 days'::interval, 'entries');
+SELECT create_continuous_aggregate('price_1_week_agg', '1 week'::interval, '3 weeks'::interval, 'entries');
 
-SELECT add_continuous_aggregate_policy('price_15_min_agg',
-  start_offset => NULL,
-  end_offset => INTERVAL '15 min',
-  schedule_interval => INTERVAL '15 min');
-
-CREATE MATERIALIZED VIEW price_1_h_agg
-WITH (timescaledb.continuous, timescaledb.materialized_only = true)
-AS SELECT 
-    pair_id,
-    time_bucket('1 hour'::interval, timestamp) as bucket,
-    approx_percentile(0.5, percentile_agg(price))::numeric AS median_price,
-    COUNT(DISTINCT source) as num_sources
-FROM entries
-GROUP BY bucket, pair_id
-WITH NO DATA;
-
-SELECT add_continuous_aggregate_policy('price_1_h_agg',
-  start_offset => NULL,
-  end_offset => INTERVAL '1 hour',
-  schedule_interval => INTERVAL '1 hour');
+-- Future entries aggregates
+SELECT create_continuous_aggregate('price_100_ms_agg_future', '100 milliseconds'::interval, '300 milliseconds'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_1_s_agg_future', '1 second'::interval, '3 seconds'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_5_s_agg_future', '5 seconds'::interval, '15 seconds'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_1_min_agg_future', '1 minute'::interval, '3 minutes'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_15_min_agg_future', '15 minutes'::interval, '45 minutes'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_1_h_agg_future', '1 hour'::interval, '3 hours'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_2_h_agg_future', '2 hours'::interval, '6 hours'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_1_day_agg_future', '1 day'::interval, '3 days'::interval, 'future_entries');
+SELECT create_continuous_aggregate('price_1_week_agg_future', '1 week'::interval, '3 weeks'::interval, 'future_entries');
 

pragma-entities/migrations/2023-10-25-134954_add_currencies_table/up.sql

@@ -1,6 +1,18 @@
 -- This file should undo anything in `up.sql`
-DROP MATERIALIZED VIEW IF EXISTS one_day_candle;
-DROP MATERIALIZED VIEW IF EXISTS one_hour_candle;
-DROP MATERIALIZED VIEW IF EXISTS fifteen_minute_candle;
-DROP MATERIALIZED VIEW IF EXISTS five_minute_candle;
-DROP MATERIALIZED VIEW IF EXISTS one_minute_candle;
+
+-- Drop spot candlestick views
+DROP MATERIALIZED VIEW IF EXISTS candle_1_min;
+DROP MATERIALIZED VIEW IF EXISTS candle_5_min;
+DROP MATERIALIZED VIEW IF EXISTS candle_15_min;
+DROP MATERIALIZED VIEW IF EXISTS candle_1_h;
+DROP MATERIALIZED VIEW IF EXISTS candle_1_day;
+
+-- Drop future candlestick views
+DROP MATERIALIZED VIEW IF EXISTS candle_1_min_future;
+DROP MATERIALIZED VIEW IF EXISTS candle_5_min_future;
+DROP MATERIALIZED VIEW IF EXISTS candle_15_min_future;
+DROP MATERIALIZED VIEW IF EXISTS candle_1_h_future;
+DROP MATERIALIZED VIEW IF EXISTS candle_1_day_future;
+
+-- Drop the function
+DROP FUNCTION IF EXISTS create_candlestick_view;