mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-08 20:47:44 +00:00
336 lines
10 KiB
Perl
Executable File
336 lines
10 KiB
Perl
Executable File
#!/usr/bin/perl
|
|
|
|
use DBI;
|
|
use LWP::Simple;
|
|
use Template;
|
|
use strict;
|
|
|
|
my $VERBOSE = 1;
|
|
|
|
use Getopt::Long;
|
|
my $VERBOSE = 0;
|
|
GetOptions("verbose" => \$VERBOSE);
|
|
|
|
# Establish a database connection.
|
|
my $dsn = "DBI:mysql:host=mecha.mozilla.org;database=downloadstats;port=3306";
|
|
my $dbh = DBI->connect($dsn,
|
|
"logprocessord",
|
|
"1ssw?w?",
|
|
{ RaiseError => 1,
|
|
PrintError => 0,
|
|
ShowErrorStatement => 1 }
|
|
);
|
|
|
|
################################################################################
|
|
# Stats Configuration
|
|
|
|
# All these variables can be redefined in the stats definition file.
|
|
# These are just the default values.
|
|
|
|
# List of stats to generate.
|
|
our @stats;
|
|
|
|
# Date range to which we limit the query.
|
|
our ($start_date, $end_date);
|
|
|
|
# Whether or not to try to add up partial downloads from the same client
|
|
# to see if they count as a complete download. Doesn't work well without
|
|
# DNS lookups, which we aren't doing for performance reasons.
|
|
our $do_segment_count = 0;
|
|
|
|
# Do a platform-breakout report by default; this can get overridden by the defs file.
|
|
our $report_type = "platform_build";
|
|
|
|
# Parse the definition file and make sure it defines some stats.
|
|
my $stats_defs = $ARGV[0];
|
|
defined($stats_defs) or die "You didn't reference a stats definition file.\n";
|
|
do $stats_defs || die "Couldn't parse stats definition file: $!\n";
|
|
defined(@stats) or die "The stats definition file didn't define any stats.\n";
|
|
|
|
################################################################################
|
|
# Data Validation
|
|
|
|
if ($start_date) {
|
|
$start_date =~ /^\d\d\d\d-\d\d-\d\d( \d\d:\d\d(:\d\d)?)?$/
|
|
or die "Invalid start date $start_date (must be in format yyyy-mm-dd (hh:mm(:ss)?)?).";
|
|
}
|
|
|
|
if ($end_date) {
|
|
$end_date =~ /^\d\d\d\d-\d\d-\d\d( \d\d:\d\d(:\d\d)?)?$/
|
|
or die "Invalid end date $end_date (must be in format yyyy-mm-dd (hh:mm(:ss)?)?).";
|
|
}
|
|
|
|
################################################################################
|
|
# Queries
|
|
|
|
my @date_criteria = ("1=1");
|
|
if ($start_date) { push(@date_criteria, "date_time >= '$start_date'") }
|
|
if ($end_date) { push(@date_criteria, "date_time <= '$end_date'") }
|
|
my $date_clause = join(" AND ", @date_criteria);
|
|
|
|
# Completed downloads.
|
|
my $done =
|
|
$dbh->prepare("SELECT COUNT(*) FROM entries JOIN files ON entries.file_id = files.id " .
|
|
"WHERE $date_clause AND files.path = ? AND files.name = ? AND bytes = ?");
|
|
|
|
# Not completed downloads.
|
|
my $not_done =
|
|
$dbh->prepare("SELECT COUNT(*) FROM entries JOIN files ON entries.file_id = files.id " .
|
|
"WHERE $date_clause AND files.path = ? AND files.name = ? AND bytes != ? " .
|
|
"AND status = 200");
|
|
|
|
# Partial content requests; may or may not be completed.
|
|
my $may_be_done =
|
|
$dbh->prepare("SELECT COUNT(*) FROM entries JOIN files ON entries.file_id = files.id " .
|
|
"WHERE $date_clause AND files.path = ? AND files.name = ? AND status = 206");
|
|
|
|
# A way to get the count of people who altogether completed a download.
|
|
# Only run if $do_segment_count is true. Note that this query is expensive
|
|
# and only ever returns a fraction of the total, so it's not that useful.
|
|
# Also, it probably doesn't work unless we reverse DNS every address
|
|
# in the logs, which we aren't doing at the moment for performance.
|
|
my $done_in_segments =
|
|
$dbh->prepare("SELECT 1 FROM entries JOIN files ON entries.file_id = files.id " .
|
|
"WHERE $date_clause AND files.path = ? AND files.name = ? " .
|
|
"GROUP BY client HAVING SUM(bytes) = ? AND COUNT(bytes) > 1");
|
|
|
|
# Grab a list of files in a given directory, recursing infinitely.
|
|
# The first condition matches the directory itself, while the second
|
|
# matches subdirectories and should be passed the value '<dir>/%'.
|
|
my $files_in_directory =
|
|
$dbh->prepare("SELECT path, name FROM files WHERE files.path = ? OR files.path LIKE ?");
|
|
|
|
|
|
################################################################################
|
|
# Stats Retrieval
|
|
|
|
sub retrieve_by_file {
|
|
foreach my $stat (@stats) {
|
|
next if !$stat->{isactive};
|
|
print STDERR "$stat->{name} $stat->{version}...\n" if $VERBOSE;
|
|
|
|
# Expand directories in the file list.
|
|
my @files;
|
|
foreach my $file (@{$stat->{files}}) {
|
|
if ($file->{name}) {
|
|
print STDERR "$file->{name}\n" if $VERBOSE;
|
|
push(@files, $file);
|
|
}
|
|
elsif (!$file->{path}) {
|
|
print STDERR "file has no name or path\n" if $VERBOSE;
|
|
next;
|
|
}
|
|
else {
|
|
print STDERR "expanding $file->{path}\n" if $VERBOSE;
|
|
$files_in_directory->execute($file->{path}, "$file->{path}/%");
|
|
while (my $file = $files_in_directory->fetchrow_hashref()) {
|
|
print STDERR "expanding to include $file->{path}/$file->{name}\n" if $VERBOSE;
|
|
push(@files, $file);
|
|
}
|
|
}
|
|
}
|
|
$stat->{files} = \@files;
|
|
|
|
foreach my $file (@files) {
|
|
(undef, $file->{size}) = head("http://ftp.mozilla.org$file->{path}/$file->{name}");
|
|
next if !$file->{size};
|
|
$done->execute($file->{path}, $file->{name}, $file->{size});
|
|
$file->{counts} = {};
|
|
($file->{counts}->{done}) = $done->fetchrow_array();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
sub retrieve_by_platform_build {
|
|
foreach my $stat (@stats) {
|
|
next if !$stat->{isactive};
|
|
|
|
print STDERR "$stat->{name} $stat->{version}...\n" if $VERBOSE;
|
|
|
|
my $platforms = $stat->{platforms};
|
|
|
|
foreach my $platform (keys %$platforms) {
|
|
print STDERR " $platform\n" if $VERBOSE;
|
|
|
|
my $files = $platforms->{$platform};
|
|
|
|
foreach my $type (keys %$files) {
|
|
print STDERR " $type: " if $VERBOSE;
|
|
|
|
my $file = $files->{$type};
|
|
|
|
my (undef, $file_size) = head("http://ftp.mozilla.org$stat->{path}/$file->{name}");
|
|
$file_size ||= $file->{size}
|
|
or die "Can't figure out the size of $stat->{path}/$file->{name}.";
|
|
|
|
$done->execute($stat->{path}, $file->{name}, $file_size);
|
|
my ($done_count) = $done->fetchrow_array();
|
|
|
|
my $done_in_segments_count = "N/A";
|
|
my $total_done = $done_count;
|
|
if ($do_segment_count) {
|
|
$done_in_segments->execute($stat->{path}, $file->{name}, $file_size);
|
|
$done_in_segments_count = $done_in_segments->fetchall_arrayref();
|
|
$done_in_segments_count = scalar(@$done_in_segments_count);
|
|
$total_done += $done_in_segments_count;
|
|
}
|
|
|
|
# commenting these out for speed -bryner
|
|
#$not_done->execute($stat->{path}, $file->{name}, $file_size);
|
|
#my ($not_done_count) = $not_done->fetchrow_array();
|
|
my ($not_done_count) = 0;
|
|
|
|
#$may_be_done->execute($stat->{path}, $file->{name});
|
|
#my ($may_be_done_count) = $may_be_done->fetchrow_array();
|
|
my ($may_be_done_count) = 0;
|
|
|
|
$file->{counts} = {
|
|
complete_uni => $done_count,
|
|
#complete_multi => $done_in_segments_count,
|
|
#incomplete => $not_done_count,
|
|
#partial => $may_be_done_count,
|
|
};
|
|
|
|
print STDERR "$done_count / $not_done_count / $may_be_done_count / $done_in_segments_count / $total_done\n" if $VERBOSE;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
################################################################################
|
|
# Output
|
|
|
|
my $platform_build_template = <<'EOF';
|
|
<html>
|
|
<head>
|
|
<title></title>
|
|
<style type="text/css">
|
|
th { text-align: left; }
|
|
th, td { border: solid 1px black; }
|
|
table { border-collapse: collapse;
|
|
border: solid 1px black; }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
[% FOREACH stat = stats %]
|
|
[% NEXT IF !stat.isactive %]
|
|
[% app_total = 0 %]
|
|
<h2>[% stat.name %] [%+ stat.version %] Download Stats</h2>
|
|
|
|
<p>[% start_date || "the beginning of time" %] to [% end_date || "the end of time" %]</p>
|
|
|
|
<table summary="[% stat.name %] [%+ stat.version %] Downloads">
|
|
<tr>
|
|
<th>Build</th>
|
|
<th>Downloads</th>
|
|
</tr>
|
|
[% FOREACH platform = stat.platforms %]
|
|
[% platform_total = 0 %]
|
|
|
|
<tr>
|
|
<td colspan="2"><h3>[% platform.key %]</h3></td>
|
|
</tr>
|
|
|
|
[% files = platform.value %]
|
|
[% FOREACH file = files %]
|
|
[% file_total = file.value.counts.complete_uni + file.value.counts.complete_multi %]
|
|
<tr>
|
|
<td>[% file.key %]:[% file.value.name %]</td>
|
|
<td>[% file_total %]</td>
|
|
</tr>
|
|
[% platform_total = platform_total + file_total %]
|
|
[% END %]
|
|
|
|
<tr>
|
|
<td>total for [% platform.key %]</td>
|
|
<td>[% platform_total %]</td>
|
|
</tr>
|
|
|
|
[% app_total = app_total + platform_total %]
|
|
[% END %]
|
|
<tr>
|
|
<td>grand total</td>
|
|
<td>[% app_total %]</td>
|
|
</tr>
|
|
</table>
|
|
[% END %]
|
|
</body>
|
|
</html>
|
|
EOF
|
|
|
|
my $file_template = <<'EOF';
|
|
<html>
|
|
<head>
|
|
<title></title>
|
|
<style type="text/css">
|
|
th { text-align: left; }
|
|
th, td { border: solid 1px black; }
|
|
table { border-collapse: collapse;
|
|
border: solid 1px black; }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
[% FOREACH stat = stats %]
|
|
[% NEXT IF !stat.isactive %]
|
|
<h2>[% stat.name %] [%+ stat.version %] Download Stats</h2>
|
|
|
|
<p>[% start_date || "the beginning of time" %] to [% end_date || "the end of time" %]</p>
|
|
|
|
<table summary="[% stat.name %] [%+ stat.version %] Downloads">
|
|
<tr>
|
|
<th>File</th>
|
|
<th>Downloads</th>
|
|
</tr>
|
|
[% FOREACH file = stat.files %]
|
|
<tr>
|
|
<td>[% file.path %]/[% file.name %]</td>
|
|
<td>[% file.counts.done %]</td>
|
|
</tr>
|
|
[% END %]
|
|
</table>
|
|
[% END %]
|
|
</body>
|
|
</html>
|
|
EOF
|
|
|
|
my $total_template = <<'EOF';
|
|
[% grand_total = 0 %]
|
|
[% FOREACH stat = stats %]
|
|
[% NEXT IF !stat.isactive %]
|
|
[% app_total = 0 %]
|
|
[% FOREACH platform = stat.platforms %]
|
|
[% platform_total = 0 %]
|
|
[% files = platform.value %]
|
|
[% FOREACH file = files %]
|
|
[% file_total = file.value.counts.complete_uni + file.value.counts.complete_multi %]
|
|
[% platform_total = platform_total + file_total %]
|
|
[% END %]
|
|
[% app_total = app_total + platform_total %]
|
|
[% END %]
|
|
[% grand_total = grand_total + app_total %]
|
|
[% END %]
|
|
[% grand_total %]
|
|
EOF
|
|
|
|
my $template;
|
|
if ($report_type eq "file") {
|
|
retrieve_by_file();
|
|
$template = $file_template;
|
|
}
|
|
elsif ($report_type eq "platform_build") {
|
|
retrieve_by_platform_build();
|
|
$template = $platform_build_template;
|
|
}
|
|
elsif ($report_type eq "total_downloads") {
|
|
retrieve_by_platform_build();
|
|
$template = $total_template;
|
|
}
|
|
|
|
my $tt = new Template({ PRE_CHOMP => 1, POST_CHOMP => 1});
|
|
|
|
$tt->process(\$template, {stats => \@stats,
|
|
start_date => $start_date,
|
|
end_date => $end_date})
|
|
|| die "Template process failed: ", $template->error(), "\n";
|