package Foo::Inbox2;

use strict;
use warnings;

our $VERSION = '0.003';

use MCE::Shared;

# $inbox = Foo::Inbox->new();

sub new {
    my ( $class, @names ) = @_;
    my %self = map { $_ => MCE::Shared->queue( fast => 1 ) } @names;

    MCE::Shared->start() unless ( exists $INC{'IO/FDPass.pm'} );

    bless \%self, $class;
}

# $scalar = $inbox->size( [ $key ] );
# %pairs  = $inbox->size();

sub size {
    my ( $self, $key ) = @_;

    if ( defined $key ) {
        exists $self->{$key} ? $self->{$key}->pending() : 0;
    }
    elsif ( wantarray ) {
        local $_;
        map { $_ => $self->{$_}->pending() } keys %{ $self };
    }
    else {
        my $size = 0;
        foreach my $key ( keys %{ $self } ) {
            $size += $self->{$key}->pending();
        }
        $size;
    }
}

# $inbox->send( $from, $to, $arg1, ... );
# $inbox->send( $from, \@list, $arg1, ... );

sub send {
    my ( $self, $from, $to ) = ( shift, shift, shift );
    my $mesg = [ $from, [ @_ ] ];

    if ( ref $to eq 'ARRAY' ) {
        $self->{$_ }->enqueue($mesg) for @{ $to };
    } else {
        $self->{$to}->enqueue($mesg);
    }

    return;
}

# $inbox->recv( $from );

sub recv {
    my ( $self, $from ) = @_;
    return () unless exists $self->{$from};

    @{ $self->{$from}->dequeue() // [] };
}

# $inbox->end();

sub end {
    my ( $self ) = @_;

    foreach my $from ( values %{ $self } ) {
        $from->end();
    }

    return;
}

1;