use strict;
use warnings;

use MCE;
use Time::HiRes 'time';

my $iterations = 100;
my $chunksize = 50;
my $threads = 5;
my $output = "m.txt";

my %data = ();

foreach ('a'..'z') {
   $data{$_} = $_ x 200;
}

open my $fh, '>', $output or die "open error: $!";

$fh->autoflush(1); test_mce(); close $fh;

sub test_mce {
   my $start = time;

   my $mce = MCE->new(
      max_workers => $threads, chunk_size => $chunksize,
      input_data => input_iter($chunksize, $iterations),
      user_func => \&work,
      init_relay => 0,
   )->run();

   printf STDERR "testa done in %0.02f seconds\n", time - $start;
}

# make an input closure, return iterator

sub input_iter {
   my ($chunk_size, $iterations) = @_;
   my $seq_a = 1;

   return sub {
      return if $seq_a > $iterations;
      my ($chunk_size) = @_;
      my @chunk = ();

      foreach my $seq_b ( 1 .. $chunk_size ) {
         my %retdata = %data;
         $retdata{'.'} = $seq_a * $seq_b;
         push @chunk, \%retdata;
      }

      $seq_a += 1;
      return \@chunk;
   };
}

# MCE task to run in parallel

sub work {
   my ($mce, $chunk_ref, $chunk_id) = @_;
   my $data = $chunk_ref->[0];
   my @ret = ();

   foreach my $chunk (@$data) {
      my %output = ();

      foreach my $key (keys %$chunk) {
         if ($key eq '.') {
            $output{$key} = $$chunk{$key};
            next;
         }
         my $val = $$chunk{$key};
         my $uc = uc($key);
         $val =~ s/$key/$uc/g;
         $output{$key} = $val;
      }

      push(@ret,\%output);
   }

   my $buf = '';

   foreach my $data (@ret) {
      foreach my $key (sort keys %$data) {
         $buf .= $$data{$key};
      }
      $buf .= "\n";
   }

   MCE::relay { print {$fh} $buf };
}