trunk/bin/lorry.pl

#!/usr/bin/perl

use warnings;
use strict;

use Data::Dump qw(dump);
use File::Slurp;

use lib 'lib';
use Sack::From;
use Sack::Lorry;
use Sack::Color;
use Getopt::Long;
use Time::HiRes qw(time);


my $view   = 'views/00.demo.pl';
my $offset = 0;
my $limit  = 5000;
my $cloud_file;


GetOptions(
        'view=s'  => \$view,
        'limit=i' => \$limit,
        'cloud=s' => \$cloud_file,
) or die $!;

my $t = time;
sub duration {
        my $now = time;
        my $d = $now - $t;
        warn sprintf "%.4fs %s\n", $d, join(' ',@_);
        $t = $now;
}


my $lorry = Sack::Lorry->new;

my @cloud = -e $cloud_file ? read_file $cloud_file : ( 'localhost' );
@cloud = map { chomp $_; $_; } @cloud;
warn "bring up cloud: ",join(' ', @cloud),$/;

my $from  = Sack::From->new(
        path => '/data/isi/full.txt',
        limit => $limit * scalar @cloud,
);

duration 'load finished';

my $info;

my $port = 4000;

foreach my $host ( @cloud ) {
        chomp $host;

        if ( $lorry->start_node_port( $host, $port ) ) {

                warn "started [$port] on $host\n";

                my $data = $from->shard( $limit );
                $lorry->send_to( $port, { data => $data } ) || die "can't send to $port: $!";

                $info->{$port} = {
                        host => $host,
                        offset => $offset,
                        limit => $limit,
                };

                $port++;
                $offset += $limit;

        }
}

duration 'cloud up';

warn "info ",dump $info;

warn "load status ", dump( $lorry->get_from_all ),$/;

our $out;
our $repl = 1;

while ( $repl ) {
        print "sack> ";
        my $cmd = <STDIN>;
        last unless defined $cmd; # CTRL+D
        duration 'repl wait';
        chomp($cmd);
        if ( $cmd =~ m{^v} ) {
                $out = $lorry->view( $view );
                duration 'view';
        } elsif ( $cmd =~ m{^d} ) {
                warn dump $out;
                duration 'dump';
        } elsif ( $cmd =~ m{^x} ) {
                $repl = 0;
        } elsif ( $cmd =~ m{^r} ) {
                $lorry->restart_nodes;
        } elsif ( $cmd =~ m{^i} ) {
                $lorry->send_to_all({ info => 1 });
                my $info = $lorry->get_from_all;
                foreach my $port ( $lorry->connected ) {
                        warn "[$port] $lorry->{port_on_host}->{$port} ", dump( $info->{$port} ), "\n";
                }
        } elsif ( $cmd =~ m{^u} ) {
                my $updated;
                foreach my $host ( @cloud ) {
                        next if $updated->{$host}++;
                        warn "update $host\n";
                        system "find /srv/Sack/ | cpio --create | ssh -F etc/lib.ssh $host cpio --extract --make-directories --unconditional";
                }
        } elsif ( $cmd =~ m{^sh\s+(.+)} ) {
                $lorry->send_to_all({ sh => $1 });
                my $sh = $lorry->get_from_all;
                foreach my $port ( $lorry->connected ) {
                        warn "[$port]# $1\n$sh->{$port}->{sh}";
                }
        } else {
                warn "UNKNOWN $cmd\n" if $cmd;
        }
}

warn "exit all nodes\n";
$lorry->send_to_all( { exit => 1 } );

duration 'total usage';

1	dpavlin	92	#!/usr/bin/perl
2
3			use warnings;
4			use strict;
5
6			use Data::Dump qw(dump);
7	dpavlin	93	use File::Slurp;
8	dpavlin	92
9			use lib 'lib';
10			use Sack::From;
11			use Sack::Lorry;
12	dpavlin	102	use Sack::Color;
13	dpavlin	107	use Getopt::Long;
14			use Time::HiRes qw(time);
15	dpavlin	92
16	dpavlin	107
17	dpavlin	93	my $view = 'views/00.demo.pl';
18	dpavlin	92	my $offset = 0;
19	dpavlin	100	my $limit = 5000;
20	dpavlin	108	my $cloud_file;
21	dpavlin	92
22
23	dpavlin	107	GetOptions(
24			'view=s' => \$view,
25			'limit=i' => \$limit,
26			'cloud=s' => \$cloud_file,
27			) or die $!;
28
29			my $t = time;
30			sub duration {
31			my $now = time;
32			my $d = $now - $t;
33			warn sprintf "%.4fs %s\n", $d, join(' ',@_);
34			$t = $now;
35			}
36
37
38	dpavlin	92	my $lorry = Sack::Lorry->new;
39
40	dpavlin	108	my @cloud = -e $cloud_file ? read_file $cloud_file : ( 'localhost' );
41	dpavlin	127	@cloud = map { chomp $_; $_; } @cloud;
42			warn "bring up cloud: ",join(' ', @cloud),$/;
43	dpavlin	107
44	dpavlin	95	my $from = Sack::From->new(
45			path => '/data/isi/full.txt',
46			limit => $limit * scalar @cloud,
47			);
48
49	dpavlin	107	duration 'load finished';
50
51	dpavlin	92	my $info;
52
53	dpavlin	127	my $port = 4000;
54
55	dpavlin	92	foreach my $host ( @cloud ) {
56	dpavlin	113	chomp $host;
57	dpavlin	101
58	dpavlin	127	if ( $lorry->start_node_port( $host, $port ) ) {
59	dpavlin	92
60			warn "started [$port] on $host\n";
61
62	dpavlin	95	my $data = $from->shard( $limit );
63			$lorry->send_to( $port, { data => $data } ) \|\| die "can't send to $port: $!";
64	dpavlin	92
65			$info->{$port} = {
66			host => $host,
67			offset => $offset,
68			limit => $limit,
69			};
70
71	dpavlin	127	$port++;
72	dpavlin	92	$offset += $limit;
73
74			}
75			}
76
77	dpavlin	107	duration 'cloud up';
78
79	dpavlin	92	warn "info ",dump $info;
80
81	dpavlin	119	warn "load status ", dump( $lorry->get_from_all ),$/;
82	dpavlin	92
83	dpavlin	110	our $out;
84	dpavlin	113	our $repl = 1;
85	dpavlin	110
86	dpavlin	113	while ( $repl ) {
87	dpavlin	100	print "sack> ";
88			my $cmd = <STDIN>;
89			last unless defined $cmd; # CTRL+D
90	dpavlin	110	duration 'repl wait';
91	dpavlin	100	chomp($cmd);
92	dpavlin	110	if ( $cmd =~ m{^v} ) {
93			$out = $lorry->view( $view );
94			duration 'view';
95			} elsif ( $cmd =~ m{^d} ) {
96			warn dump $out;
97			duration 'dump';
98	dpavlin	113	} elsif ( $cmd =~ m{^x} ) {
99			$repl = 0;
100	dpavlin	119	} elsif ( $cmd =~ m{^r} ) {
101			$lorry->restart_nodes;
102			} elsif ( $cmd =~ m{^i} ) {
103			$lorry->send_to_all({ info => 1 });
104			my $info = $lorry->get_from_all;
105	dpavlin	127	foreach my $port ( $lorry->connected ) {
106	dpavlin	119	warn "[$port] $lorry->{port_on_host}->{$port} ", dump( $info->{$port} ), "\n";
107			}
108	dpavlin	127	} elsif ( $cmd =~ m{^u} ) {
109			my $updated;
110			foreach my $host ( @cloud ) {
111			next if $updated->{$host}++;
112			warn "update $host\n";
113			system "find /srv/Sack/ \| cpio --create \| ssh -F etc/lib.ssh $host cpio --extract --make-directories --unconditional";
114			}
115	dpavlin	119	} elsif ( $cmd =~ m{^sh\s+(.+)} ) {
116			$lorry->send_to_all({ sh => $1 });
117			my $sh = $lorry->get_from_all;
118	dpavlin	127	foreach my $port ( $lorry->connected ) {
119			warn "[$port]# $1\n$sh->{$port}->{sh}";
120	dpavlin	119	}
121	dpavlin	110	} else {
122	dpavlin	119	warn "UNKNOWN $cmd\n" if $cmd;
123	dpavlin	110	}
124	dpavlin	100	}
125	dpavlin	93
126	dpavlin	92	warn "exit all nodes\n";
127	dpavlin	93	$lorry->send_to_all( { exit => 1 } );
128
129	dpavlin	107	duration 'total usage';
130