#!/usr/bin/env perl

BEGIN {
   die "The PERCONA_TOOLKIT_BRANCH environment variable is not set.\n"
      unless $ENV{PERCONA_TOOLKIT_BRANCH} && -d $ENV{PERCONA_TOOLKIT_BRANCH};
   unshift @INC, "$ENV{PERCONA_TOOLKIT_BRANCH}/lib";
};

use strict;
use warnings FATAL => 'all';
use English qw(-no_match_vars);
use Time::HiRes qw(sleep);
use Test::More;
use threads;

use PerconaTest;
use Sandbox;
require "$trunk/bin/pt-kill";

use Data::Dumper;
$Data::Dumper::Indent    = 1;
$Data::Dumper::Sortkeys  = 1;
$Data::Dumper::Quotekeys = 0;


my $o = new OptionParser(description => 'Diskstats',
   file        => "$trunk/bin/pt-kill",
);
$o->get_specs("$trunk/bin/pt-table-checksum");
$o->get_opts();

my $dp  = new DSNParser(opts=>$dsn_opts);
my $sb  = new Sandbox(basedir => '/tmp', DSNParser => $dp);
my $dbh = $sb->get_dbh_for('master');

if ( !$dbh ) {
   plan skip_all => 'Cannot connect to sandbox master';
}

$sb->load_file('master', "t/pt-kill/samples/pt_167.sql");
my $ps = $dbh->prepare('INSERT INTO test.foo VALUES (?)');
for (my $i=0; $i < 20000; $i++) {
    $ps->execute($i);
}

sub start_thread {
   my ($dp, $o, $dsn, $sleep_time) = @_;

   diag("Thread started");

   my $cxn = new Cxn(DSNParser => $dp, OptionParser => $o, dsn => $dsn);
   $cxn->connect();
   my $dbh = $cxn->dbh();
   my $sth = $dbh->prepare('SELECT COUNT(*) FROM (SELECT a.id AS a_id, b.id AS b_id FROM foo AS a, foo AS b) AS c');
   # Since this query is going to be killed, wrap the execution in an eval to prevent
   # displaying the error message.
   eval {
       $sth->execute();
   };
}

my $dsn = "h=127.1,P=12345,u=msandbox,p=msandbox,D=test;mysql_server_prepare=1";
my $thr = threads->create('start_thread', $dp, $o, $dp->parse($dsn), 1);
$thr->detach();
threads->yield();

sleep(1);

my $rows = $dbh->selectall_hashref('show processlist', 'id');
my $pid = 0;  # reuse, reset
map  { $pid = $_->{id} }
grep { $_->{info} && $_->{info} =~ m/SELECT COUNT\(\*\) FROM \(SELECT a.id/ }
values %$rows;

ok(
   $pid,
   "Got proc id of sleeping query: $pid"
);

$dsn = $sb->dsn_for('master');
my $output = output(
   sub { pt_kill::main($dsn, "--kill-busy-commands","Query,Execute", qw(--run-time 3s --kill --busy-time 2s --print --match-info), "^(select|SELECT)"), },
   stderr => 1,
);

like(
   $output,
   qr/KILL $pid \(Execute/,
   '--kill-query'
) or diag($output);


# #############################################################################
# Done.
# #############################################################################
$sb->wipe_clean($dbh);
ok($sb->ok(), "Sandbox servers") or BAIL_OUT(__FILE__ . " broke the sandbox");
done_testing;