mirror of
https://github.com/timescale/timescaledb.git
synced 2025-05-23 14:39:15 +08:00
7b94c573ba
- Directory structure now matches common practices - Regression tests now run with pg_regress via the PGXS infrastructure. - Unit tests do not integrate well with pg_regress and have to be run separately. - Docker functionality is separate from main Makefile. Run with `make -f docker.mk` to build and `make -f docker.mk run` to run the database in a container.
83 lines
2.4 KiB
PL/PgSQL
83 lines
2.4 KiB
PL/PgSQL
CREATE OR REPLACE FUNCTION _iobeamdb_data_api.lock_for_chunk_close(
|
|
node_database_name NAME,
|
|
chunk_id INTEGER
|
|
)
|
|
RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
DECLARE
|
|
BEGIN
|
|
PERFORM _iobeamdb_meta.node_transaction_exec_with_return(
|
|
n.database_name, n.server_name,
|
|
format('SELECT * FROM _iobeamdb_internal.lock_for_chunk_close(%L)', chunk_id)
|
|
)
|
|
FROM _iobeamdb_catalog.node n
|
|
WHERE n.database_name = node_database_name;
|
|
END
|
|
$BODY$;
|
|
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_data_api.max_time_for_chunk_close(
|
|
node_database_name NAME,
|
|
schema_name NAME,
|
|
table_name NAME
|
|
)
|
|
RETURNS BIGINT LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
DECLARE
|
|
max_time BIGINT;
|
|
BEGIN
|
|
SELECT _iobeamdb_meta.node_transaction_exec_with_return(
|
|
n.database_name, n.server_name,
|
|
format('SELECT * FROM _iobeamdb_internal.max_time_for_chunk_close(%L, %L)',
|
|
max_time_for_chunk_close.schema_name,
|
|
max_time_for_chunk_close.table_name
|
|
)
|
|
)::bigint
|
|
INTO max_time
|
|
FROM _iobeamdb_catalog.node n
|
|
WHERE n.database_name = node_database_name;
|
|
|
|
RETURN max_time;
|
|
END
|
|
$BODY$;
|
|
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_data_api.set_end_time_for_chunk_close(
|
|
node_database_name NAME,
|
|
chunk_id INTEGER,
|
|
max_time BIGINT
|
|
)
|
|
RETURNS VOID LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
DECLARE
|
|
BEGIN
|
|
PERFORM _iobeamdb_meta.node_transaction_exec_with_return(
|
|
n.database_name, n.server_name,
|
|
format('SELECT * FROM _iobeamdb_internal.set_end_time_for_chunk_close(%L, %L)', chunk_id, max_time)
|
|
)
|
|
FROM _iobeamdb_catalog.node n
|
|
WHERE n.database_name = node_database_name;
|
|
END
|
|
$BODY$;
|
|
|
|
CREATE OR REPLACE FUNCTION _iobeamdb_data_api.get_chunk_size(
|
|
chunk_id INTEGER
|
|
)
|
|
RETURNS BIGINT LANGUAGE PLPGSQL VOLATILE AS
|
|
$BODY$
|
|
DECLARE
|
|
chunk_size BIGINT;
|
|
BEGIN
|
|
SELECT _iobeamdb_internal.node_immediate_commit_exec_with_return(
|
|
n.database_name, n.server_name,
|
|
format('SELECT * FROM _iobeamdb_internal.get_local_chunk_size(%L)', get_chunk_size.chunk_id)
|
|
)::BIGINT
|
|
INTO STRICT chunk_size
|
|
FROM _iobeamdb_catalog.chunk_replica_node crn
|
|
INNER JOIN _iobeamdb_catalog.node n ON (n.database_name = crn.database_name)
|
|
WHERE crn.chunk_id = get_chunk_size.chunk_id
|
|
ORDER BY (crn.database_name = current_database()) DESC --prefer local nodes
|
|
LIMIT 1;
|
|
|
|
RETURN chunk_size;
|
|
END
|
|
$BODY$;
|