collabora-online/scripts/test-ha.pl

112 lines
2.4 KiB
Perl
Executable File

#!/usr/bin/perl -w
use strict;
use Time::HiRes qw( time );
my $uri = shift @ARGV || die 'pass in URI of server or file to fetch';
my $tmp = '/tmp/downloaded';
my %hitcount;
my %slowcount;
my %timecount;
# Try to detect bad WOPI hosts by bucketing latency.
sub testWOPIServer()
{
for (my $i = 0; $i < 50; $i++)
{
unlink $tmp;
my $begin_time = time();
my $pipe;
open ($pipe, "curl --insecure --stderr - -i -v -m 300 -o $tmp $uri |") || die "Can't launch curl";
my $node = 'unknown';
my $size = 0;
while (<$pipe>) {
my $line = $_;
$line =~ m/^< Set-Cookie:/ || next;
if ($line =~ m/BIGipServerIP[^=]+=([0-9\.]+);/) {
$node = $1;
}
if ($line =~ m/Content-Length: ([0-9]+)/) {
$size = $1;
}
}
close ($pipe);
my $end_time = time();
my $time_taken = $end_time - $begin_time;
if (!defined $slowcount{$node}) {
$hitcount{$node} = 0;
$slowcount{$node} = 0;
$timecount{$node} = 0.0;
}
$hitcount{$node}++;
$timecount{$node} += $time_taken;
my $slow = '';
if ($time_taken > 1.0) {
$slow = ' slow';
$slowcount{$node}++;
}
printf("%.2fs from $node size: $size $slow\n", $time_taken);
}
print "hits\t#slow\ttotal\tNode\n";
for my $node (keys %slowcount) {
printf ("%s\t%s\t%.2fs\t%s\n", $hitcount{$node}, $slowcount{$node}, $timecount{$node}, $node);
}
}
sub randstr()
{
my @hex = ('0' ..'9', 'A' .. 'F');
return join '' => map $hex[rand @hex], 1 .. 8;
}
sub testCoolCluster()
{
my @ids;
my %serverId_by_src;
my $i;
my $size = 0;
my $iters = 50; # number of checks
my $tests = 10; # number of src strings
my $uri_base = "$uri/hosting/capabilities?WOPISrc=";
for ($i = 0; $i < $tests; $i++) {
push(@ids,randstr());
}
printf("Touching $uri_base $iters times\n");
for ($i = 0; $i < $iters; $i++)
{
my $pipe;
my $src = $ids[rand(@ids)];
my $id = 0;
open ($pipe, "curl --insecure -m 500 -s $uri_base$src |") || die "Can't launch curl";
while (<$pipe>) {
my $line = $_;
# printf ("$line\n");
if ($line =~ m/\"serverId\":\"([^"]+)\"/) {
$id = $1;
}
}
close ($pipe);
# printf("id: $id\n");
$serverId_by_src{$src} = $id if (!defined $serverId_by_src{$src});
if ($serverId_by_src{$src} ne $id)
{
die("ERROR: ID mismatch for $id vs. $serverId_by_src{$src} for WOPISrc '$src'\n");
}
}
print ("WOPISrc check with $iters iterations and $tests keys passed cleanly.\n");
}
testCoolCluster();