user_dependencies/user-dependencies.rs
1//! $ cargo run --release --example user-dependencies
2//!
3//! Computes the percentage of crates.io which depends directly on at least one
4//! crate by the specified user.
5
6use std::collections::btree_map::Entry;
7use std::collections::{BTreeMap as Map, BTreeSet as Set};
8
9const USER: &str = "dtolnay";
10
11fn main() -> db_dump::Result<()> {
12 // Map of crate id to the most recently published version of that crate.
13 let mut most_recent = Map::new();
14
15 let mut user_id = None;
16 let mut crates = 0;
17 let mut crate_owners = Vec::new();
18 let mut dependencies = Vec::new();
19 db_dump::Loader::new()
20 .users(|row| {
21 if row.gh_login == USER {
22 user_id = Some(row.id);
23 }
24 })
25 .crates(|_row| crates += 1)
26 .crate_owners(|row| crate_owners.push(row))
27 .dependencies(|row| dependencies.push(row))
28 .versions(|row| match most_recent.entry(row.crate_id) {
29 Entry::Vacant(entry) => {
30 entry.insert(row);
31 }
32 Entry::Occupied(mut entry) => {
33 if row.created_at > entry.get().created_at {
34 entry.insert(row);
35 }
36 }
37 })
38 .load("./db-dump.tar.gz")?;
39
40 // User id of the crate author we care about.
41 let user_id = user_id.expect("no such user");
42
43 // Set of crate ids currently owned by that user.
44 let mut their_crates = Set::new();
45 for crate_owner in crate_owners {
46 if crate_owner.owner_id == user_id {
47 their_crates.insert(crate_owner.crate_id);
48 }
49 }
50
51 // Set of version ids which are the most recently published of their crate.
52 let most_recent = Set::from_iter(most_recent.values().map(|version| version.id));
53
54 // Set of version ids which depend directly on at least one crate by the
55 // user.
56 let mut dep_on_them = Set::new();
57 for dep in dependencies {
58 if their_crates.contains(&dep.crate_id) {
59 dep_on_them.insert(dep.version_id);
60 }
61 }
62
63 // Number of crates whose most recent version depends on at least one crate
64 // by the user.
65 let result = dep_on_them.intersection(&most_recent).count();
66
67 println!(
68 "{} / {} = {:.1}%",
69 result,
70 crates,
71 100.0 * result as f64 / crates as f64,
72 );
73
74 Ok(())
75}