src/test/regress/sql/cbdb_db_size_functions.sql

-- start_ignore
DROP TABLE IF EXISTS cbdbheapsizetest;
DROP TABLE IF EXISTS cbdbaosizetest;
DROP EXTERNAL TABLE IF EXISTS cbdbsize_t_ext;
-- end_ignore
-- create heap table
CREATE TABLE cbdbheapsizetest(a int);
INSERT INTO cbdbheapsizetest select generate_series(1, 1000);

-- create ao table
CREATE TABLE cbdbaosizetest (a int) WITH (appendonly=true, orientation=row);
insert into cbdbaosizetest select generate_series(1, 100000);

-- create EXTERNAL table
CREATE EXTERNAL TABLE cbdbsize_t_ext (a integer) LOCATION ('file://127.0.0.1/tmp/foo') FORMAT 'text';

WITH cbdbrelsize AS (
	SELECT *
	FROM cbdb_relation_size((SELECT array['cbdbsize_t_ext'::regclass,'cbdbheapsizetest'::regclass, 'cbdbaosizetest'::regclass]))
), pgrelsize AS (
	SELECT pg_relation_size(oid) as size, relname, oid FROM pg_class where oid in ('cbdbsize_t_ext'::regclass,'cbdbheapsizetest'::regclass, 'cbdbaosizetest'::regclass)
)
SELECT pgrelsize.relname, pgrelsize.size, cbdbrelsize.size
FROM pgrelsize FULL JOIN cbdbrelsize
ON pgrelsize.oid = cbdbrelsize.reloid
WHERE pgrelsize.size != cbdbrelsize.size;

SELECT * FROM cbdb_relation_size(array[]::oid[], 'main');
SELECT size FROM cbdb_relation_size(array['cbdbheapsizetest'::regclass], 'fsm');