#!/usr/bin/perl

use strict;

use warnings;

use Test::More;

use FindBin qw($Bin);

use lib "$Bin/lib";

use MemcachedTest;

use Data::Dumper qw/Dumper/;

my $ext_path;

if (!supports_extstore()) {

    plan skip_all => 'extstore not enabled';

    exit 0;

}

$ext_path = "/tmp/extstore.$$";

my $server = new_memcached("-m 64 -U 0 -o ext_page_size=8,ext_wbuf_size=2,ext_threads=1,ext_io_depth=2,ext_item_size=512,ext_item_age=2,ext_recache_rate=10000,ext_max_frag=0.9,ext_path=$ext_path:64m,slab_automove=0,ext_compact_under=1");

my $sock = $server->sock;

# Wait until all items have flushed

sub wait_for_ext {

    my $target = shift || 0;

    my $sum = $target + 1;

    while ($sum > $target) {

        my $s = mem_stats($sock, "items");

        $sum = 0;

        for my $key (keys %$s) {

            if ($key =~ m/items:(\d+):number/) {

                # Ignore classes which can contain extstore items

                next if $1 < 3;

                $sum += $s->{$key};

            }

        }

        sleep 1 if $sum > $target;

    }

}

my $value;

{

    my @chars = ("C".."Z");

    for (1 .. 20000) {

        $value .= $chars[rand @chars];

    }

}

# fill a small object

print $sock "set foo 0 0 2\r\nhi\r\n";

is(scalar <$sock>, "STORED\r\n", "stored small value");

# fetch

mem_get_is($sock, "foo", "hi");

# check extstore counters

{

    my $stats = mem_stats($sock);

    is($stats->{extstore_objects_written}, 0);

}

# fill some larger objects

{

    # set one canary value for later

    print $sock "set canary 0 0 20000 noreply\r\n$value\r\n";

    my $keycount = 1000;

    for (1 .. $keycount) {

        print $sock "set nfoo$_ 0 0 20000 noreply\r\n$value\r\n";

    }

    # wait for a flush

    wait_for_ext();

    # fetch

    # TODO: Fetch back all values

    mem_get_is($sock, "nfoo1", $value);

    # check extstore counters

    my $stats = mem_stats($sock);

    cmp_ok($stats->{extstore_page_allocs}, '>', 0, 'at least one page allocated');

    cmp_ok($stats->{extstore_objects_written}, '>', $keycount / 2, 'some objects written');

    cmp_ok($stats->{extstore_bytes_written}, '>', length($value) * 2, 'some bytes written');

    cmp_ok($stats->{get_extstore}, '>', 0, 'one object was fetched');

    cmp_ok($stats->{extstore_objects_read}, '>', 0, 'one object read');

    cmp_ok($stats->{extstore_bytes_read}, '>', length($value), 'some bytes read');

    # Remove half of the keys for the next test.

    for (1 .. $keycount) {

        next unless $_ % 2 == 0;

        print $sock "delete nfoo$_ noreply\r\n";

    }

    my $stats2 = mem_stats($sock);

    cmp_ok($stats->{extstore_bytes_used}, '>', $stats2->{extstore_bytes_used},

        'bytes used dropped after deletions');

    cmp_ok($stats->{extstore_objects_used}, '>', $stats2->{extstore_objects_used},

        'objects used dropped after deletions');

    is($stats2->{badcrc_from_extstore}, 0, 'CRC checks successful');

    is($stats2->{miss_from_extstore}, 0, 'no misses');

    # delete the rest

    for (1 .. $keycount) {

        next unless $_ % 2 == 1;

        print $sock "delete nfoo$_ noreply\r\n";

    }

}

# fill to eviction

{

    my $keycount = 4000;

    for (1 .. $keycount) {

        print $sock "set mfoo$_ 0 0 20000 noreply\r\n$value\r\n";

        # wait to avoid evictions

        wait_for_ext(500) if ($_ % 2000 == 0);

    }

    # because item_age is set to 2s

    wait_for_ext();

    my $stats = mem_stats($sock);

    is($stats->{evictions}, 0, 'no evictions');

    is($stats->{miss_from_extstore}, 0, 'no misses');

    # FIXME: test is flaky; something can rescue the canary because of a race

    # condition. might need to roundtrip twice or disable compaction?

    #mem_get_is($sock, "canary", undef);

    # check counters

    $stats = mem_stats($sock);

    cmp_ok($stats->{extstore_page_evictions}, '>', 0, 'at least one page evicted');

    cmp_ok($stats->{extstore_objects_evicted}, '>', 0, 'at least one object evicted');

    cmp_ok($stats->{extstore_bytes_evicted}, '>', 0, 'some bytes evicted');

    cmp_ok($stats->{extstore_pages_free}, '<', 2, 'few pages are free');

    #is($stats->{miss_from_extstore}, 1, 'exactly one miss');

    # refresh some keys so rescues happen while drop_unread == 1.

    for (1 .. $keycount / 2) {

        next unless $_ % 2 == 1;

        # Need to be fetched twice in order to bump

        print $sock "touch mfoo$_ 0 noreply\r\n";

        print $sock "touch mfoo$_ 0 noreply\r\n";

    }

    print $sock "extstore drop_unread 1\r\n";

    my $res = <$sock>;

    print $sock "extstore max_frag 0\r\n";

    $res = <$sock>;

    print $sock "extstore compact_under 4\r\n";

    $res = <$sock>;

    print $sock "extstore drop_under 3\r\n";

    $res = <$sock>;

    for (1 .. $keycount) {

        next unless $_ % 2 == 0;

        print $sock "delete mfoo$_ noreply\r\n";

    }

    sleep 4;

    $stats = mem_stats($sock);

    cmp_ok($stats->{extstore_pages_free}, '>', 0, 'some pages now free');

    cmp_ok($stats->{extstore_compact_rescues}, '>', 0, 'some compaction rescues happened');

    cmp_ok($stats->{extstore_compact_skipped}, '>', 0, 'some compaction skips happened');

    print $sock "extstore drop_unread 0\r\n";

    $res = <$sock>;

}

# attempt to incr/decr/append/prepend or chunk objects that were sent to disk.

{

    my $keycount = 100;

    for (1 .. $keycount) {

        print $sock "set bfoo$_ 0 0 20000 noreply\r\n$value\r\n";

    }

    wait_for_ext();

    # incr should be blocked.

    print $sock "incr bfoo1 1\r\n";

    is(scalar <$sock>, "CLIENT_ERROR cannot increment or decrement non-numeric value\r\n", 'incr fails');

    # append/prepend *could* work, but it would require pulling the item back in.

    print $sock "append bfoo1 0 0 2\r\nhi\r\n";

    is(scalar <$sock>, "NOT_STORED\r\n", 'append fails');

    print $sock "prepend bfoo1 0 0 2\r\nhi\r\n";

    is(scalar <$sock>, "NOT_STORED\r\n", 'prepend fails');

}

done_testing();

END {

    unlink $ext_path if $ext_path;

}