#!/usr/bin/perl

use strict;
use warnings;
use Data::Dumper;
use Carp;
use Parse::CSV;
use SNMP;

use SNMP::Query::AsynchMulti;

#---------------------------------------------------------

my $csv_file = shift || die "Please specify a CSV file with SNMP host info!";

my @reqired_fields = qw(HOSTIP COMMUNITY SNMPVER SNMPPORT);
my @hosts = read_hosts_csv($csv_file, @reqired_fields);

# This object encapsulates the desired queries to run.
my $query = SNMP::Query::AsynchMulti->new();

foreach my $host (@hosts) {

    my $sess = SNMP::Session->new(
            DestHost   => $host->{HOSTIP},
            Community  => $host->{COMMUNITY},
            Version    => $host->{SNMPVER},
            RemotePort => $host->{SNMPPORT},
            #Timeout    => $host->{SNMP_TIMEOUT},
            #Retries    => $host->{SNMP_RETRIES},
    );

    my @varbinds = qw(
           ifDescr ifInOctets ifOutOctets ifAlias ifType
           ifName ifInErrors ifOutErrors ifSpeed
           ifAdminStatus ifOperStatus
    );

    my $varlist = SNMP::VarList->new( map { [$_] } @varbinds );

    $query->add({
            Session      => $sess,
            VarBinds     => $varlist,
            QueryType    => 'getbulk', # See POD for explanation...
            MaxRepeaters => 20,        # Additional options depend on
            NonRepeaters => 0,         # the QueryType...
    });
}

$query->shuffle(); # Randomize order of queries...(not  yet implemented)

# Run all the added queries with up to X
# asynchronous operations in-flight at any time.
foreach my $iter (1..10) {
    sleep 30 unless $iter == 1;
    my $results = $query->execute({ InFlight => 30 });
    print Dumper $results;
}


exit;
#---------------------------------------------------------

# Read in the CSV file.
sub read_hosts_csv {
    my $file = shift;
    my @required_fields = @_;

    # Parse entries from a CSV file into hashes hash
    my $csv_parser = Parse::CSV->new(
        file   => $file,
        fields => 'auto',  # Use the first line as column headers,
                           # which become the hash keys.
    );

    my @node_cfg; # Return a reference to this
    my $line_num = 0;
    while ( my $line = $csv_parser->fetch() ) {
        $line_num++;
        my $error_flag = 0;
        foreach my $field (@required_fields) {
            if ( ! exists $line->{$field} ) {
                $error_flag = 1;
                carp "Missing field [$field] on line [$line_num] in CSV file [$file]";
            }
        }
        croak "Terminating due to errors on line [$line_num] in CSV file [$file]"
            if $error_flag;

        push @node_cfg, $line;
    }

    if ( $csv_parser->errstr() ) {
        croak "Fatal error parsing [$file]: " . $csv_parser->errstr();
    }
    return @node_cfg;
}


1;
__END__