File Coverage

blib/lib/ETL/Yertl/Command/yts.pm

Criterion	Covered	Total	%
statement	41	49	83.6
branch	11	20	55.0
condition	1	6	16.6
subroutine	5	5	100.0
pod	0	1	0.0
total	58	81	71.6

package ETL::Yertl::Command::yts;
our $VERSION = '0.035';
# ABSTRACT: Read/Write time series data

#pod =head1 SYNOPSIS
#pod
#pod =head1 DESCRIPTION
#pod
#pod =head1 SEE ALSO
#pod
#pod L
#pod
#pod =cut

use ETL::Yertl;
use ETL::Yertl::Util qw( load_module );
use Getopt::Long qw( GetOptionsFromArray :config pass_through );
use IO::Interactive qw( is_interactive );

sub main {
    my $class = shift;

    my %opt;
    if ( ref $_[-1] eq 'HASH' ) {
        %opt = %{ pop @_ };
    }

    my @args = @_;
    GetOptionsFromArray( \@args, \%opt,
        'start=s',
        'end=s',
        'short|s',
        'tags=s%',
    );
    #; use Data::Dumper;
    #; say Dumper \@args;
    #; say Dumper \%opt;

    my ( $db_spec, $metric ) = @args;

    die "Must give a database\n" unless $db_spec;

    my ( $db_type ) = $db_spec =~ m{^([^:]+):};

    my $db = load_module( adapter => $db_type )->new( $db_spec );

    # Write metrics
    if ( !is_interactive( \*STDIN ) ) {
        if ( $opt{short} ) {
            die "Must give a metric\n" unless $metric;
        }

        my $in_fmt = load_module( format => 'default' )->new( input => \*STDIN );
        my $count = 0;

        while ( my @docs = $in_fmt->read ) {
            for my $doc ( @docs ) {
                #; use Data::Dumper
                #; say "Got doc: " . Dumper $doc;
                if ( $opt{short} ) {
                    my @docs;
                    for my $stamp ( sort keys %$doc ) {
                        push @docs, {
                            timestamp => $stamp,
                            metric => $metric,
                            value => $doc->{ $stamp },
                            ( $opt{tags} ? ( tags => $opt{tags} ) : () ),
                        };
                    }
                    #; use Data::Dumper;
                    #; print Dumper \@docs;
                    $db->write_ts( @docs );
                    $count += @docs;
                }
                else {
                    $doc->{metric} ||= $metric;
                    $doc->{tags} ||= $opt{tags} if $opt{tags};
                    $db->write_ts( $doc );
                    $count++;
                }
            }
        }

        #; say "Wrote $count points";
    }
    # Read metrics
    else {
        die "Must give a metric\n" unless $metric;
        my $out_fmt = load_module( format => 'default' )->new;
        my @points = $db->read_ts( {
            metric => $metric,
            tags => $opt{tags},
            start => $opt{start},
            end => $opt{end},
        } );
        if ( $opt{short} ) {
            my %ts = map { $_->{timestamp} => $_->{value} } @points;
            print $out_fmt->write( \%ts );
        }
        else {
            print $out_fmt->write( $_ ) for @points;
        }
    }

    return 0;
}

1;

__END__