guix-data-service/guix-data-service/model/nar.scm
Christopher Baines b430c632b7 Create a table for systems
And use it for the systems in the derivations and package derivations tables.

The primary motivation here is to allow quickly working out what systems the
database contains, and having a small table with just the right data seems a
good way to do that.
2021-04-23 11:14:51 +01:00

512 lines
18 KiB
Scheme

;;; Guix Data Service -- Information about Guix over time
;;; Copyright © 2019 Christopher Baines <mail@cbaines.net>
;;;
;;; This program is free software: you can redistribute it and/or
;;; modify it under the terms of the GNU Affero General Public License
;;; as published by the Free Software Foundation, either version 3 of
;;; the License, or (at your option) any later version.
;;;
;;; This program is distributed in the hope that it will be useful,
;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;;; Affero General Public License for more details.
;;;
;;; You should have received a copy of the GNU Affero General Public
;;; License along with this program. If not, see
;;; <http://www.gnu.org/licenses/>.
(define-module (guix-data-service model nar)
#:use-module (srfi srfi-1)
#:use-module (srfi srfi-19)
#:use-module (ice-9 match)
#:use-module (web uri)
#:use-module (squee)
#:use-module (json)
#:use-module (rnrs bytevectors)
#:use-module (gcrypt pk-crypto)
#:use-module (gcrypt base16)
#:use-module (guix narinfo)
#:use-module (guix-data-service database)
#:use-module (guix-data-service model utils)
#:export (select-outputs-without-known-nar-entries
select-nars-for-output
select-signing-key
select-package-output-availability-for-revision
select-output-consistency-for-revision
record-narinfo-details-and-return-ids))
(define narinfo-contents
(@@ (guix narinfo) narinfo-contents))
(define (record-narinfo-details-and-return-ids conn build-server-id narinfos)
(define data
(map (lambda (narinfo)
(match (string-split
(narinfo-hash narinfo)
#\:)
((hash-algorithm hash)
(list
(narinfo-path narinfo)
hash-algorithm
hash
(narinfo-size narinfo)
(or (narinfo-system narinfo) NULL)
(or (narinfo-deriver narinfo) NULL)))))
narinfos))
(let ((nar-ids
(insert-missing-data-and-return-all-ids
conn
"nars"
'(store_path hash_algorithm hash size system deriver)
data)))
(let ((reference-data
(concatenate
(map (lambda (nar-id narinfo)
(map (lambda (reference)
(simple-format
#f
"(~A, ~A)"
nar-id
(quote-string reference)))
(narinfo-references narinfo)))
nar-ids
narinfos))))
(unless (null? reference-data)
(exec-query
conn
(string-append
"
INSERT INTO nar_references (nar_id, reference)
VALUES "
(string-join reference-data ", ")
"
ON CONFLICT DO NOTHING"))))
(exec-query
conn
(string-append
"
INSERT INTO nar_urls (nar_id, url, compression, file_size)
VALUES "
(string-join
(concatenate
(map (lambda (nar-id narinfo)
(map (lambda (uri compression file-size)
(simple-format
#f
"(~A, ~A, ~A, ~A)"
nar-id
(quote-string
(uri->string uri))
(quote-string compression)
(or file-size "NULL")))
(narinfo-uris narinfo)
(narinfo-compressions narinfo)
(narinfo-file-sizes narinfo)))
nar-ids
narinfos))
", ")
"
ON CONFLICT DO NOTHING"))
(for-each (lambda (nar-id narinfo)
(let ((narinfo-signature-data-id
(narinfo-signature->data-id conn narinfo)))
(exec-query
conn
(string-append
"
INSERT INTO narinfo_signatures (nar_id, narinfo_signature_data_id)
VALUES "
(simple-format
#f
"(~A,~A)"
nar-id
narinfo-signature-data-id)
"
ON CONFLICT DO NOTHING"))
(exec-query
conn
(string-append
"
INSERT INTO narinfo_fetch_records (narinfo_signature_data_id, build_server_id)
VALUES ($1, $2)")
(list (number->string narinfo-signature-data-id)
(number->string build-server-id)))))
nar-ids
narinfos)
nar-ids))
(define (sexp->json-string sexp)
(define (transform x)
(if (list? x)
(list->vector (map transform x))
(if (bytevector? x)
`((base16 . ,(bytevector->base16-string x)))
x)))
(scm->json-string (transform sexp)))
(define (narinfo-signature->data-id conn narinfo)
(let ((public-key-id
(narinfo-signature->public-key-id
conn
(narinfo-signature narinfo)))
(contents
(narinfo-contents narinfo)))
(match (string-contains contents "Signature:")
(#f #f)
(index
(let* ((body (string-take contents index))
(signature-line (string-drop contents index))
(signature-sexp
(canonical-sexp->sexp
(narinfo-signature narinfo))))
(match (string-split (second (string-split signature-line
#\space))
#\;)
((version host-name signature-data)
(first
(insert-missing-data-and-return-all-ids
conn
"narinfo_signature_data"
'(version host_name data_hash data_hash_algorithm
data_json sig_val_json narinfo_signature_public_key_id
narinfo_body narinfo_signature_line)
(list
(append (list (string->number version)
host-name)
(let* ((data-sexp
(find (match-lambda
((component data ...)
(if (eq? component 'data)
data
#f))
(_ #f))
signature-sexp))
(hash-sexp
(third data-sexp))
(hash-algorithm
(second hash-sexp))
(hash
(third hash-sexp)))
(list
(bytevector->base16-string hash)
hash-algorithm
(cons "jsonb"
(sexp->json-string data-sexp))))
(let ((sig-val-sexp
(find (match-lambda
((component data ...)
(if (eq? component 'sig-val)
data
#f))
(_ #f))
signature-sexp)))
(list
(cons "jsonb"
(sexp->json-string sig-val-sexp))))
(list public-key-id
body
signature-line))))))))))))
(define (narinfo-signature->public-key-id conn signature)
(let* ((public-key-sexp
(find (match-lambda
((component data ...)
(if (eq? component 'public-key)
data
#f))
(_ #f))
(canonical-sexp->sexp signature)))
(public-key-json-string
(sexp->json-string public-key-sexp)))
(first
(insert-missing-data-and-return-all-ids
conn
"narinfo_signature_public_keys"
'(sexp_json)
(list (list (cons "jsonb"
public-key-json-string)))))))
(define (select-package-output-availability-for-revision conn revision-commit)
(define query
"
SELECT build_server_id, system, target, substitute_known, COUNT(*)
FROM (
SELECT build_servers.id AS build_server_id,
derivation_output_details.path,
systems.system,
package_derivations.target,
nar_data.build_server_id IS NOT NULL AS substitute_known
FROM derivation_output_details
INNER JOIN derivation_outputs
ON derivation_outputs.derivation_output_details_id =
derivation_output_details.id
INNER JOIN package_derivations
ON derivation_outputs.derivation_id = package_derivations.derivation_id
INNER JOIN systems
ON package_derivations.system_id = systems.id
INNER JOIN guix_revision_package_derivations
ON package_derivations.id =
guix_revision_package_derivations.package_derivation_id
INNER JOIN guix_revisions
ON guix_revision_package_derivations.revision_id = guix_revisions.id
CROSS JOIN build_servers
INNER JOIN build_servers_build_config
ON build_servers.id = build_servers_build_config.build_server_id
AND systems.system = build_servers_build_config.system
AND package_derivations.target = build_servers_build_config.target
LEFT JOIN (
SELECT nars.store_path, narinfo_fetch_records.build_server_id
FROM nars
LEFT JOIN narinfo_signatures
ON narinfo_signatures.nar_id = nars.id
LEFT JOIN narinfo_signature_data
ON narinfo_signatures.narinfo_signature_data_id = narinfo_signature_data.id
LEFT JOIN narinfo_fetch_records
ON narinfo_fetch_records.narinfo_signature_data_id = narinfo_signature_data.id
) AS nar_data
ON nar_data.store_path = derivation_output_details.path
AND nar_data.build_server_id = build_servers.id
WHERE derivation_output_details.hash IS NULL AND
guix_revisions.commit = $1
) data
GROUP BY build_server_id, system, target, substitute_known
ORDER BY build_server_id DESC, system, target, build_server_id, substitute_known")
(map
(match-lambda
((build-server-id . rest)
(cons build-server-id
(group-to-alist
(match-lambda
((system target substitute-known? count)
(cons `((system . ,system)
(target . ,target))
(cons (if substitute-known?
'known
'unknown)
count))))
rest))))
(group-to-alist
(match-lambda
((build-server-id system target substitute-known? count)
(cons build-server-id
(list system target substitute-known? count))))
(map (match-lambda
((build_server_id system target substitutes_known count)
(list (string->number build_server_id)
system
target
(string=? substitutes_known "t")
(string->number count))))
(exec-query conn query (list revision-commit))))))
(define (select-output-consistency-for-revision conn revision-commit)
(define query
"
SELECT system, target, reproducible, COUNT(*)
FROM (
SELECT derivation_output_details.path,
systems.system,
package_derivations.target,
CASE
WHEN (COUNT(DISTINCT nar_data.build_server_id) <= 1) THEN NULL
ELSE (COUNT(DISTINCT nar_data.hash) = 1)
END AS reproducible
FROM derivation_output_details
INNER JOIN derivation_outputs
ON derivation_outputs.derivation_output_details_id =
derivation_output_details.id
INNER JOIN package_derivations
ON derivation_outputs.derivation_id = package_derivations.derivation_id
INNER JOIN systems
ON package_derivations.system_id = systems.id
INNER JOIN guix_revision_package_derivations
ON package_derivations.id =
guix_revision_package_derivations.package_derivation_id
INNER JOIN guix_revisions
ON guix_revision_package_derivations.revision_id = guix_revisions.id
LEFT JOIN (
SELECT nars.store_path, narinfo_fetch_records.build_server_id, nars.hash
FROM nars
INNER JOIN narinfo_signatures
ON narinfo_signatures.nar_id = nars.id
INNER JOIN narinfo_signature_data
ON narinfo_signatures.narinfo_signature_data_id = narinfo_signature_data.id
INNER JOIN narinfo_fetch_records
ON narinfo_fetch_records.narinfo_signature_data_id =
narinfo_signature_data.id
) AS nar_data
ON nar_data.store_path = derivation_output_details.path
WHERE derivation_output_details.hash IS NULL AND
guix_revisions.commit = $1 AND
package_derivations.target = '' -- Exclude cross builds
GROUP BY derivation_output_details.path,
systems.system,
package_derivations.target
) data
GROUP BY system, target, reproducible
ORDER BY COUNT(*) DESC")
(group-to-alist
(match-lambda
((system status count)
(cons system
(cons status count))))
(map (match-lambda
((system target status count)
(list system
(match status
("t" 'matching)
("f" 'not-matching)
(() 'unknown))
(string->number count))))
(exec-query-with-null-handling conn query (list revision-commit)))))
(define* (select-outputs-without-known-nar-entries
conn
build-server-id
guix-revision-commits
#:key
build-success-after)
(define query
(string-append
"
SELECT DISTINCT derivation_output_details.path, derivation_output_details.id
FROM derivation_outputs
INNER JOIN derivation_output_details
ON derivation_outputs.derivation_output_details_id =
derivation_output_details.id"
(if build-success-after
"
INNER JOIN derivation_output_details_sets
ON ARRAY[derivation_output_details.id] <@
derivation_output_details_sets.derivation_output_details_ids
INNER JOIN builds
ON builds.build_server_id = $1
AND builds.derivation_output_details_set_id = derivation_output_details_sets.id
INNER JOIN latest_build_status
ON latest_build_status.build_id = builds.id
AND latest_build_status.status = 'succeeded'
AND latest_build_status.timestamp > $2"
"")
"
WHERE derivation_output_details.path NOT IN (
-- Ignore outputs that have already been fetched
SELECT store_path
FROM nars
INNER JOIN narinfo_signatures
ON nars.id = narinfo_signatures.nar_id
INNER JOIN narinfo_signature_data
ON narinfo_signatures.narinfo_signature_data_id = narinfo_signature_data.id
INNER JOIN narinfo_fetch_records
ON narinfo_signature_data.id = narinfo_fetch_records.narinfo_signature_data_id
WHERE narinfo_fetch_records.build_server_id = $1
)
"
(if (null? guix-revision-commits)
""
(string-append
"
AND derivation_outputs.derivation_id IN (
-- Select outputs that are in the relevant revisions
SELECT derivation_id
FROM package_derivations
INNER JOIN systems
ON package_derivations.system_id = systems.id
INNER JOIN build_servers_build_config
ON build_servers_build_config.build_server_id = $1
AND build_servers_build_config.system = systems.system
AND build_servers_build_config.target = package_derivations.target
INNER JOIN guix_revision_package_derivations
ON guix_revision_package_derivations.package_derivation_id = package_derivations.id
INNER JOIN guix_revisions
ON guix_revisions.id = guix_revision_package_derivations.revision_id
WHERE guix_revisions.commit IN ("
(string-join (map quote-string guix-revision-commits)
",")
")
)"))
"
ORDER BY derivation_output_details.id DESC
LIMIT 100000"))
(map car (exec-query conn
query
`(,(number->string build-server-id)
,@(if build-success-after
(list (date->string build-success-after "~1 ~3"))
'())))))
(define (select-nars-for-output conn output-file-name)
(define query
"
SELECT hash_algorithm, hash, size,
(
SELECT JSON_AGG(
json_build_object('url', url, 'compression', compression, 'size', file_size)
)
FROM nar_urls
WHERE nar_id = nars.id
) AS urls,
(
SELECT JSON_AGG(
json_build_object(
'version', version,
'host_name', host_name,
'data_hash', data_hash,
'data_hash_algorithm', data_hash_algorithm,
'data', data_json,
'sig_val', sig_val_json,
'narinfo_signature_public_key', (
SELECT sexp_json
FROM narinfo_signature_public_keys
WHERE narinfo_signature_public_keys.id = narinfo_signature_public_key_id
),
'body', narinfo_body,
'signature_line', narinfo_signature_line
)
)
FROM narinfo_signature_data
INNER JOIN narinfo_signatures
ON narinfo_signature_data.id = narinfo_signatures.narinfo_signature_data_id
WHERE narinfo_signatures.nar_id = nars.id
)
FROM nars
WHERE store_path = $1")
(map
(match-lambda
((hash-algorithm hash size urls-json signatures-json)
(list hash-algorithm
hash
(string->number size)
(vector->list
(json-string->scm urls-json))
(vector->list
(json-string->scm signatures-json)))))
(exec-query conn query (list output-file-name))))
(define (select-signing-key conn id)
(define query
"
SELECT sexp_json
FROM narinfo_signature_public_keys
WHERE id = $1")
(match (exec-query conn query (list (number->string id)))
(((sexp_json))
(json-string->scm sexp_json))))