mirror of
https://github.com/timescale/timescaledb.git
synced 2025-05-22 13:40:56 +08:00
00b69ac010
This change is a performance improvement. Previously each insert called a plpgsql function to check if there is a need to close the chunk. This patch implements a c-only fastpath for the case when the table size is less than the configured chunk size.
73 lines
2.5 KiB
PL/PgSQL
73 lines
2.5 KiB
PL/PgSQL
--get the chunk for a given partition and time.
|
|
CREATE OR REPLACE FUNCTION _timescaledb_internal.get_chunk(
|
|
partition_id INT,
|
|
time_point BIGINT
|
|
)
|
|
RETURNS _timescaledb_catalog.chunk LANGUAGE SQL STABLE AS
|
|
$BODY$
|
|
SELECT *
|
|
FROM _timescaledb_catalog.chunk c
|
|
WHERE c.partition_id = get_chunk.partition_id AND
|
|
(c.start_time <= time_point OR c.start_time IS NULL) AND
|
|
(c.end_time >= time_point OR c.end_time IS NULL);
|
|
$BODY$;
|
|
|
|
|
|
--get the chunk for a given partition and time. The function takes a (shared) row lock
|
|
--on the chunk and might therefore block.
|
|
CREATE OR REPLACE FUNCTION _timescaledb_internal.get_chunk_locked(
|
|
partition_id INT,
|
|
time_point BIGINT
|
|
)
|
|
RETURNS _timescaledb_catalog.chunk LANGUAGE SQL VOLATILE AS
|
|
$BODY$
|
|
SELECT *
|
|
FROM _timescaledb_catalog.chunk c
|
|
WHERE c.partition_id = get_chunk_locked.partition_id AND
|
|
(c.start_time <= time_point OR c.start_time IS NULL) AND
|
|
(c.end_time >= time_point OR c.end_time IS NULL)
|
|
FOR SHARE;
|
|
$BODY$;
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION _timescaledb_catalog.local_chunk_size(name, name) RETURNS bigint
|
|
AS '$libdir/timescaledb', 'local_chunk_size' LANGUAGE C IMMUTABLE STRICT;
|
|
|
|
|
|
--returns the current size of a chunk (in bytes) given its ID.
|
|
--The size is typically aligned with the page size in Postgres.
|
|
CREATE OR REPLACE FUNCTION _timescaledb_internal.get_local_chunk_size(
|
|
chunk_id INT
|
|
)
|
|
RETURNS BIGINT LANGUAGE PLPGSQL STABLE AS
|
|
$BODY$
|
|
DECLARE
|
|
chunk_replica_row _timescaledb_catalog.chunk_replica_node;
|
|
BEGIN
|
|
SELECT *
|
|
INTO STRICT chunk_replica_row
|
|
FROM _timescaledb_catalog.chunk_replica_node crn
|
|
WHERE crn.chunk_id = get_local_chunk_size.chunk_id
|
|
AND crn.database_name = current_database();
|
|
|
|
IF chunk_replica_row.database_name != current_database() THEN
|
|
RAISE EXCEPTION 'get_local_chunk_size should only be called locally'
|
|
USING ERRCODE = 'IO501';
|
|
END IF;
|
|
RETURN _timescaledb_catalog.local_chunk_size(chunk_replica_row.schema_name, chunk_replica_row.table_name);
|
|
END
|
|
$BODY$;
|
|
|
|
--returns the max size (in bytes) that a chunk is allowed to grow to.
|
|
CREATE OR REPLACE FUNCTION _timescaledb_internal.get_chunk_max_size(
|
|
chunk_id INT
|
|
)
|
|
RETURNS BIGINT LANGUAGE SQL STABLE AS
|
|
$BODY$
|
|
SELECT h.chunk_size_bytes
|
|
FROM _timescaledb_catalog.chunk_replica_node crn
|
|
INNER JOIN _timescaledb_catalog.partition_replica pr ON (pr.id = crn.partition_replica_id)
|
|
INNER JOIN _timescaledb_catalog.hypertable h ON (h.id = pr.hypertable_id)
|
|
WHERE (crn.chunk_id = get_chunk_max_size.chunk_id);
|
|
$BODY$;
|