Postgres SQL for循环统计每月现有用户

-- 创建一个存储过程
CREATE OR REPLACE FUNCTION monthly_user_count() RETURNS TABLE(month date, user_count integer) AS $$
DECLARE
    start_date date := '2022-01-01'; -- 开始日期
    end_date date := '2022-12-31'; -- 结束日期
    current_date date := start_date;
BEGIN
    -- 创建一个临时表用于存储结果
    CREATE TEMPORARY TABLE temp_user_count (month date, user_count integer);

    -- 循环处理每个月份
    WHILE current_date <= end_date LOOP
        -- 统计当前月份的用户数量
        INSERT INTO temp_user_count
        SELECT DATE_TRUNC('month', created_at), COUNT(DISTINCT user_id)
        FROM users
        WHERE created_at >= current_date AND created_at < current_date + INTERVAL '1 month';

        -- 增加一个月份
        current_date := current_date + INTERVAL '1 month';
    END LOOP;

    -- 返回结果
    RETURN QUERY SELECT * FROM temp_user_count;
END;
$$ LANGUAGE plpgsql;

-- 调用存储过程获取结果
SELECT * FROM monthly_user_count();