db: Add default_versions.total column

Author: eth3lbert

crates/crates_io_database/src/schema.rs

@@ -408,6 +408,8 @@ diesel::table! {
         crate_id -> Int4,
         /// Reference to the version in the `versions` table.
         version_id -> Int4,
+        /// The total number of versions.
+        total -> Int4,
     }
 }
 

crates/crates_io_database_dump/src/dump-db.toml

@@ -105,6 +105,7 @@ dependencies = ["crates", "versions"]
 [default_versions.columns]
 crate_id = "public"
 version_id = "public"
+total = "public"
 
 [deleted_crates]
 dependencies = ["users"]

crates/crates_io_database_dump/src/snapshots/[email protected]

@@ -1,6 +1,7 @@
 ---
 source: crates/crates_io_database_dump/src/lib.rs
 expression: content
+snapshot_kind: text
 ---
 BEGIN ISOLATION LEVEL REPEATABLE READ, READ ONLY;
 
@@ -18,7 +19,7 @@ BEGIN ISOLATION LEVEL REPEATABLE READ, READ ONLY;
     \copy (SELECT "crate_id", "created_at", "created_by", "owner_id", "owner_kind" FROM "crate_owners" WHERE NOT deleted) TO 'data/crate_owners.csv' WITH CSV HEADER
 
     \copy "versions" ("bin_names", "categories", "checksum", "crate_id", "crate_size", "created_at", "description", "documentation", "downloads", "edition", "features", "has_lib", "homepage", "id", "keywords", "license", "links", "num", "num_no_build", "published_by", "repository", "rust_version", "updated_at", "yanked") TO 'data/versions.csv' WITH CSV HEADER
-    \copy "default_versions" ("crate_id", "version_id") TO 'data/default_versions.csv' WITH CSV HEADER
+    \copy "default_versions" ("crate_id", "total", "version_id") TO 'data/default_versions.csv' WITH CSV HEADER
     \copy "dependencies" ("crate_id", "default_features", "explicit_name", "features", "id", "kind", "optional", "req", "target", "version_id") TO 'data/dependencies.csv' WITH CSV HEADER
     \copy "version_downloads" ("date", "downloads", "version_id") TO 'data/version_downloads.csv' WITH CSV HEADER
 COMMIT;

crates/crates_io_database_dump/src/snapshots/[email protected]

@@ -1,6 +1,7 @@
 ---
 source: crates/crates_io_database_dump/src/lib.rs
 expression: content
+snapshot_kind: text
 ---
 BEGIN;
     -- Disable triggers on each table.
@@ -60,7 +61,7 @@ BEGIN;
     \copy "crates_keywords" ("crate_id", "keyword_id") FROM 'data/crates_keywords.csv' WITH CSV HEADER
     \copy "crate_owners" ("crate_id", "created_at", "created_by", "owner_id", "owner_kind") FROM 'data/crate_owners.csv' WITH CSV HEADER
     \copy "versions" ("bin_names", "categories", "checksum", "crate_id", "crate_size", "created_at", "description", "documentation", "downloads", "edition", "features", "has_lib", "homepage", "id", "keywords", "license", "links", "num", "num_no_build", "published_by", "repository", "rust_version", "updated_at", "yanked") FROM 'data/versions.csv' WITH CSV HEADER
-    \copy "default_versions" ("crate_id", "version_id") FROM 'data/default_versions.csv' WITH CSV HEADER
+    \copy "default_versions" ("crate_id", "total", "version_id") FROM 'data/default_versions.csv' WITH CSV HEADER
     \copy "dependencies" ("crate_id", "default_features", "explicit_name", "features", "id", "kind", "optional", "req", "target", "version_id") FROM 'data/dependencies.csv' WITH CSV HEADER
     \copy "version_downloads" ("date", "downloads", "version_id") FROM 'data/version_downloads.csv' WITH CSV HEADER
 

migrations/2025-02-05-083109_add-total-column/down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE default_versions
+    DROP COLUMN total;

migrations/2025-02-05-083109_add-total-column/up.sql

@@ -0,0 +1,4 @@
+ALTER TABLE default_versions
+    ADD COLUMN total INTEGER NOT NULL default 0;
+
+COMMENT ON COLUMN default_versions.total IS 'The total number of versions.';