-- 基础表示例
CREATE TABLE metrics (
    id BIGSERIAL, -- 可选，某些场景可能需要全局唯一ID
    ts TIMESTAMPTZ NOT NULL, -- 时间戳，核心字段
    metric_name VARCHAR(255) NOT NULL, -- 指标名称，如“cpu_usage”
    tags JSONB, -- 标签集合，用于多维查询，如 `{"host": "server-01", "region": "us-east-1"}`
    value DOUBLE PRECISION NOT NULL, -- 指标数值
    created_at TIMESTAMPTZ DEFAULT NOW()
);

-- 1. 创建主表（分区表）
CREATE TABLE metrics (
    ts TIMESTAMPTZ NOT NULL,
    metric_name VARCHAR(255) NOT NULL,
    tags JSONB,
    value DOUBLE PRECISION NOT NULL,
    created_at TIMESTAMPTZ DEFAULT NOW()
) PARTITION BY RANGE (ts);

-- 2. 创建分区。例如，创建2024年5月1日的分区
CREATE TABLE metrics_y2024m05d01 PARTITION OF metrics
FOR VALUES FROM ('2024-05-01 00:00:00') TO ('2024-05-02 00:00:00');

-- 3. （可选）为分区表单独创建索引。在分区上创建索引比在父表上创建更高效。
CREATE INDEX ON metrics_y2024m05d01 (ts);
CREATE INDEX ON metrics_y2024m05d01 USING GIN (tags); -- 支持JSONB的标签查询

CREATE OR REPLACE FUNCTION create_metrics_partition()
RETURNS void AS $$
DECLARE
    partition_name TEXT;
    start_date TIMESTAMPTZ;
    end_date TIMESTAMPTZ;
BEGIN
    -- 例如，提前创建未来3天的分区
    FOR i IN 0..2 LOOP
        start_date := date_trunc('day', NOW() + (i || ' days')::INTERVAL);
        end_date := start_date + INTERVAL '1 day';
        partition_name := 'metrics_y' || to_char(start_date, 'YYYYmMMdDD');

        -- 仅当分区不存在时才创建
        IF NOT EXISTS (SELECT 1 FROM pg_class WHERE relname = partition_name) THEN
            EXECUTE format(
                'CREATE TABLE %I PARTITION OF metrics FOR VALUES FROM (%L) TO (%L)',
                partition_name, start_date, end_date
            );
            -- 为新分区创建索引
            EXECUTE format('CREATE INDEX ON %I (ts)', partition_name);
            EXECUTE format('CREATE INDEX ON %I USING GIN (tags)', partition_name);
            RAISE NOTICE '分区 % 创建成功。', partition_name;
        END IF;
    END LOOP;
END;
$$ LANGUAGE plpgsql;

-- 使用pg_cron每天执行一次此函数
-- SELECT cron.schedule('0 1 * * *', $$SELECT create_metrics_partition();$$);

-- 在父表或分区上创建BRIN索引
CREATE INDEX metrics_ts_brin_idx ON metrics USING BRIN (ts);

-- 也可以包含多个列，如果value也随时间有序增长
-- CREATE INDEX metrics_ts_value_brin_idx ON metrics USING BRIN (ts, value);

-- 主表存储核心字段
CREATE TABLE metrics_core (
    ts TIMESTAMPTZ NOT NULL,
    metric_name VARCHAR(255) NOT NULL,
    value DOUBLE PRISION NOT NULL
) PARTITION BY RANGE (ts);

-- 详情表存储标签，通过外键关联（这里用ts+metric_name作为逻辑主键）
CREATE TABLE metrics_tags (
    ts TIMESTAMPTZ NOT NULL,
    metric_name VARCHAR(255) NOT NULL,
    tags JSONB NOT NULL,
    PRIMARY KEY (ts, metric_name)
) PARTITION BY RANGE (ts);
-- 需要确保两个表的分区策略完全一致，以便联动管理。

-- 假设已经创建了一个指向慢速HDD的表空间 `slow_disk`
CREATE TABLESPACE slow_disk LOCATION '/mnt/big_slow_hdd/pg_data';

-- 将旧分区移动到慢速存储上
ALTER TABLE metrics_y2023m01d01 SET TABLESPACE slow_disk;

-- 创建主表
CREATE TABLE apm_metrics (
    ts TIMESTAMPTZ NOT NULL,
    service_name VARCHAR(100) NOT NULL,
    endpoint VARCHAR(255),
    response_time_ms DOUBLE PRECISION NOT NULL,
    status_code INT,
    tags JSONB,
    PRIMARY KEY (ts, service_name, endpoint) -- 复合主键
) PARTITION BY RANGE (ts);

-- 创建当前月份的分区
CREATE TABLE apm_metrics_202405 PARTITION OF apm_metrics
FOR VALUES FROM ('2024-05-01') TO ('2024-06-01');
CREATE INDEX ON apm_metrics_202405 USING BRIN (ts);
CREATE INDEX ON apm_metrics_202405 USING GIN (tags);

-- 查询某服务最近一小时的响应时间P95
SELECT
    service_name,
    percentile_cont(0.95) WITHIN GROUP (ORDER BY response_time_ms) as p95_rt
FROM apm_metrics
WHERE ts >= NOW() - INTERVAL '1 hour'
    AND service_name = "user-service"
GROUP BY service_name;

-- 查询特定标签的主机在过去一天的错误数
SELECT COUNT(*)
FROM apm_metrics
WHERE ts >= NOW() - INTERVAL '1 day'
    AND tags @> '{"host": "web-01"}'
    AND status_code >= 500;