mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-06-10 11:45:45 +03:00

This changes the build to always enable JIT - but to only enable it at run-time, when required. This keeps the runtime closure small without JIT, but allows enabling it without a rebuild. We can do this, because JIT is actually built as a shared module, which is loaded at run-time. We put it into a -jit output and only link it into the environment when requested. Under the hood, this uses withPackages and adds the "JIT package" - thus, to be able to use withPackages on top of that, we also need to be able to apply withPackages repeatedly. This cuts down the number of NixOS tests in half, because we don't need to run it for every version with and without JIT anymore. There really is no point in running everything with llvmjit.so in place, when the queries are not making use of it anyway. Also, we only need to build each extension once and not twice, further reducing the number of rebuilds required for PRs touching postgresql.
97 lines
2.4 KiB
Nix
97 lines
2.4 KiB
Nix
{
|
|
pkgs,
|
|
makeTest,
|
|
genTests,
|
|
}:
|
|
|
|
let
|
|
inherit (pkgs) lib;
|
|
|
|
test-sql = pkgs.writeText "postgresql-test" ''
|
|
CREATE EXTENSION timescaledb;
|
|
CREATE EXTENSION timescaledb_toolkit;
|
|
|
|
CREATE TABLE sth (
|
|
time TIMESTAMPTZ NOT NULL,
|
|
value DOUBLE PRECISION
|
|
);
|
|
|
|
SELECT create_hypertable('sth', 'time');
|
|
|
|
INSERT INTO sth (time, value) VALUES
|
|
('2003-04-12 04:05:06 America/New_York', 1.0),
|
|
('2003-04-12 04:05:07 America/New_York', 2.0),
|
|
('2003-04-12 04:05:08 America/New_York', 3.0),
|
|
('2003-04-12 04:05:09 America/New_York', 4.0),
|
|
('2003-04-12 04:05:10 America/New_York', 5.0)
|
|
;
|
|
|
|
WITH t AS (
|
|
SELECT
|
|
time_bucket('1 day'::interval, time) AS dt,
|
|
stats_agg(value) AS stats
|
|
FROM sth
|
|
GROUP BY time_bucket('1 day'::interval, time)
|
|
)
|
|
SELECT
|
|
average(stats)
|
|
FROM t;
|
|
|
|
SELECT * FROM sth;
|
|
'';
|
|
|
|
makeTestFor =
|
|
package:
|
|
makeTest {
|
|
name = "timescaledb-${package.name}";
|
|
meta = with lib.maintainers; {
|
|
maintainers = [ typetetris ];
|
|
};
|
|
|
|
nodes.machine =
|
|
{ ... }:
|
|
{
|
|
services.postgresql = {
|
|
inherit package;
|
|
enable = true;
|
|
extensions =
|
|
ps: with ps; [
|
|
timescaledb
|
|
timescaledb_toolkit
|
|
];
|
|
settings = {
|
|
shared_preload_libraries = "timescaledb, timescaledb_toolkit";
|
|
};
|
|
};
|
|
};
|
|
|
|
testScript = ''
|
|
def check_count(statement, lines):
|
|
return 'test $(sudo -u postgres psql postgres -tAc "{}"|wc -l) -eq {}'.format(
|
|
statement, lines
|
|
)
|
|
|
|
|
|
machine.start()
|
|
machine.wait_for_unit("postgresql")
|
|
|
|
with subtest("Postgresql with extensions timescaledb and timescaledb_toolkit is available just after unit start"):
|
|
machine.succeed(
|
|
"sudo -u postgres psql -f ${test-sql}"
|
|
)
|
|
|
|
machine.fail(check_count("SELECT * FROM sth;", 3))
|
|
machine.succeed(check_count("SELECT * FROM sth;", 5))
|
|
machine.fail(check_count("SELECT * FROM sth;", 4))
|
|
|
|
machine.shutdown()
|
|
'';
|
|
};
|
|
in
|
|
# Not run by default, because this requires allowUnfree.
|
|
# To run these tests:
|
|
# NIXPKGS_ALLOW_UNFREE=1 nix-build -A nixosTests.postgresql.timescaledb
|
|
lib.dontRecurseIntoAttrs (genTests {
|
|
inherit makeTestFor;
|
|
filter = _: p: !p.pkgs.timescaledb.meta.broken;
|
|
})
|