view release on metacpan or  search on metacpan

lib/GrowthForecast/Data.pm  view on Meta::CPAN

            else {
                $subtract = 0;
            }
        }
        $dbh->commit;
        $data->{subtract_short} = $subtract;
    }
    $self->inflate_row($data);
}

sub get_by_id_for_rrdupdate {
    my ($self, $id) = @_;
    my $dbh = $self->dbh;

    my $data = $dbh->select_row(
        'SELECT * FROM graphs WHERE id = ?',
        $id
    );
    return if !$data;

    unless ( $self->{disable_subtract} ) {
        $dbh->begin_work;
        my $subtract;

        my $for_update = ( $dbh->connect_info->[0] =~ /^(?i:dbi):mysql:/ ) ? ' FOR UPDATE' : '';
        my $prev = $dbh->select_row(
            'SELECT * FROM prev_graphs WHERE graph_id = ?' . $for_update,
            $data->{id}
        );

        if ( !$prev ) {
            $subtract = 'U';
            $dbh->query(
                'INSERT INTO prev_graphs (graph_id, number, subtract, updated_at)
                VALUES (?,?,?,?)',
                $data->{id}, $data->{number}, undef, $data->{updated_at});
        }
        elsif ( $data->{updated_at} != $prev->{updated_at} ) {
            $subtract = $data->{number} - $prev->{number};
            $dbh->query(
                'UPDATE prev_graphs SET number=?, subtract=?, updated_at=? WHERE graph_id = ?',
                $data->{number}, $subtract, $data->{updated_at}, $data->{id}
            );
        }
        else {
            if ( $data->{mode} eq 'gauge' || $data->{mode} eq 'modified' ) {
                $subtract = $prev->{subtract};
                $subtract = 'U' if ! defined $subtract;
            }
            else {
                $subtract = 0;
            }
        }

        $dbh->commit;
        $data->{subtract} = $subtract;
    }
    $self->inflate_row($data);
}

sub update {
    my ($self, $service, $section, $graph, $number, $mode, $color, $timestamp ) = @_;
    my $dbh = $self->dbh;
    $dbh->begin_work;

    my $for_update = ( $dbh->connect_info->[0] =~ /^(?i:dbi):mysql:/ ) ? ' FOR UPDATE' : '';
    my $data = $self->dbh->select_row(
        'SELECT * FROM graphs WHERE service_name = ? AND section_name = ? AND graph_name = ?' . $for_update,
        $service, $section, $graph
    );

    if ( defined $data ) {
        if ( $mode eq 'count' ) {
            $number += $data->{number};
        }
        if ( $mode ne 'modified' || ($mode eq 'modified' && $data->{number} != $number) ) {
            $color ||= $data->{color};
            $dbh->query(
                'UPDATE graphs SET number=?, mode=?, color=?, updated_at=?, timestamp=? WHERE id = ?',
                $number, $mode, $color, time, $timestamp, $data->{id}
            );
        }
    }
    else {
        my @colors = List::Util::shuffle(qw/33 66 99 cc/);
        $color ||= '#' . join('', splice(@colors,0,3));
        $dbh->query(
            'INSERT INTO graphs (service_name, section_name, graph_name, number, mode, color, llimit, sllimit, created_at, updated_at, timestamp)
                         VALUES (?,?,?,?,?,?,?,?,?,?,?)',
            $service, $section, $graph, $number, $mode, $color, -1000000000, -100000 , time, time, $timestamp
        );
    }

    my $row = $self->dbh->select_row(
        'SELECT * FROM graphs WHERE service_name = ? AND section_name = ? AND graph_name = ?',
        $service, $section, $graph
    );

    $dbh->commit;

    $self->inflate_row($row);
}

sub update_graph {
    my ($self, $id, $args) = @_;
    my @update = map { delete $args->{$_} } qw/service_name section_name graph_name description sort gmode color type stype llimit ulimit sllimit sulimit/;
    my $meta = encode_json($args);
    my $dbh = $self->dbh;
    $dbh->query(
        'UPDATE graphs SET service_name=?, section_name=?, graph_name=?, description=?, sort=?, gmode=?, color=?, type=?, stype=?,
         llimit=?, ulimit=?, sllimit=?, sulimit=?, meta=? WHERE id = ?',
        @update, $meta, $id
    );
    return 1;
}

sub update_graph_description {
    my ($self, $id, $description) = @_;
    my $dbh = $self->dbh;
    $dbh->query(
        'UPDATE graphs SET description=? WHERE id = ?',
        $description, $id
    );
    return 1;    
}

sub get_services {
    my $self = shift;
    my $rows = $self->dbh->select_all(
        'SELECT DISTINCT service_name FROM graphs ORDER BY service_name',
    );
    my $complex_rows = $self->dbh->select_all(
        'SELECT DISTINCT service_name FROM complex_graphs ORDER BY service_name',
    );
    my @names = uniq map { $_->{service_name} } (@$rows,@$complex_rows);
    \@names
}

sub get_sections {
    my $self = shift;
    my $service_name = shift;
    my $rows = $self->dbh->select_all(
        'SELECT DISTINCT section_name FROM graphs WHERE service_name = ? ORDER BY section_name',
        $service_name,
    );
    my $complex_rows = $self->dbh->select_all(
        'SELECT DISTINCT section_name FROM complex_graphs WHERE service_name = ? ORDER BY section_name',
        $service_name,
    );
    my @names = uniq map { $_->{section_name} } (@$rows,@$complex_rows);
    \@names;
} 

sub get_graphs {
   my $self = shift;
   my ($service_name, $section_name) = @_;
   my $rows = $self->dbh->select_all(
       'SELECT * FROM graphs WHERE service_name = ? AND section_name = ? ORDER BY sort DESC',
       $service_name, $section_name
   );
   my $complex_rows = $self->dbh->select_all(
       'SELECT * FROM complex_graphs WHERE service_name = ? AND section_name = ? ORDER BY sort DESC',
       $service_name, $section_name
   );
   my @ret;
   for my $row ( @$rows ) {
       push @ret, $self->inflate_row($row); 
   }
   for my $row ( @$complex_rows ) {
       push @ret, $self->inflate_complex_row($row); 
   }
   @ret = sort { $b->{sort} <=> $a->{sort} } @ret;
   \@ret;
}

sub get_all_graph_id {
   my $self = shift;

lib/GrowthForecast/Data.pm  view on Meta::CPAN

    $uri .= ":0"; #stack

    if ( !ref $ref->{'type-2'} ) {
        $ref->{$_} = [$ref->{$_}] for qw /type-2 path-2 gmode-2 stack-2/;
    }
    my $num = scalar @{$ref->{'type-2'}};
    my @ret;
    for ( my $i = 0; $i < $num; $i++ ) {
        $uri .= ':' . join ":", map { $ref->{$_}->[$i] } qw /type-2 path-2 gmode-2 stack-2/;
        push @ret, {
            type => $ref->{'type-2'}->[$i],
            path => $ref->{'path-2'}->[$i],
            gmode => $ref->{'gmode-2'}->[$i],
            stack => $ref->{'stack-2'}->[$i],
            graph => $self->get_by_id($ref->{'path-2'}->[$i]),
        };        
    }

    $ref->{sumup} = 0 if ! exists $ref->{sumup};
    $ref->{data_rows} = \@ret;
    $ref->{complex_graph} = $uri;
    my %result = (
        %$ref,
        %$row
    );
    \%result
}

sub get_complex {
    my ($self, $service, $section, $graph) = @_;
    my $row = $self->dbh->select_row(
        'SELECT * FROM complex_graphs WHERE service_name = ? AND section_name = ? AND graph_name = ?',
        $service, $section, $graph
    );
    return unless $row;
    $self->inflate_complex_row($row);
}

sub get_complex_by_id {
    my ($self, $id) = @_;
    my $row = $self->dbh->select_row(
        'SELECT * FROM complex_graphs WHERE id = ?',
        $id
    );
    return unless $row;
    $self->inflate_complex_row($row);
}

sub create_complex {
    my ($self, $service, $section, $graph, $args) = @_;
    my @update = map { delete $args->{$_} } qw/description sort/;
    my $meta = encode_json($args);
    $self->dbh->query(
        'INSERT INTO complex_graphs (service_name, section_name, graph_name, description, sort, meta,  created_at, updated_at) 
                         VALUES (?,?,?,?,?,?,?,?)',
        $service, $section, $graph, @update, $meta, time, time
    ); 
    $self->get_complex($service, $section, $graph);
}

sub update_complex {
    my ($self, $id, $args) = @_;
    my @update = map { delete $args->{$_} } qw/service_name section_name graph_name description sort/;
    my $meta = encode_json($args);
    $self->dbh->query(
        'UPDATE complex_graphs SET service_name = ?, section_name = ?, graph_name = ? , 
                                   description = ?, sort = ?, meta = ?, updated_at = ?
                             WHERE id=?',
        @update, $meta, time, $id        
    );
    $self->get_complex_by_id($id);
}

sub remove_complex {
    my ($self, $id ) = @_;
    my $dbh = $self->dbh;
    $dbh->query(
        'DELETE FROM complex_graphs WHERE id = ?',
        $id
    );
}

sub get_all_complex_graph_id {
   my $self = shift;
   $self->dbh->select_all(
       'SELECT id FROM complex_graphs',
   );
}

sub get_all_complex_graph_name {
   my $self = shift;
   $self->dbh->select_all(
       'SELECT id,service_name,section_name,graph_name FROM complex_graphs ORDER BY service_name, section_name, graph_name DESC',
   );
}

sub get_all_complex_graph_all {
    my $self = shift;
    my $list = $self->dbh->select_all(
        'SELECT * FROM complex_graphs ORDER BY service_name, section_name, graph_name DESC',
    );
    return [] unless $list;
    my @ret = map { $self->inflate_complex_row($_) } @$list;
    \@ret;
}

sub update_vrule {
    my ($self, $graph_path, $time, $color, $desc, $dashes) = @_;

    $self->dbh->query(
        'INSERT INTO vrules (graph_path,time,color,description,dashes) values (?,?,?,?,?)',
        $graph_path, $time, $color, $desc, $dashes,
    );

   my $row = $self->dbh->select_row(
        'SELECT * FROM vrules WHERE graph_path = ? AND time = ? AND color = ? AND description = ? AND dashes = ?',
        $graph_path, $time, $color, $desc, $dashes,
    );

    return $row;
}

# "$span" is a parameter named "t",
# "$from" and "$to" are paramters same nameed.
sub get_vrule {
    my ($self, $span, $from, $to, $graph_path) = @_;

    my($from_time, $to_time) = (0, time);
    # same rule as GrowthForecast::RRD#calc_period
    if ( $span eq 'all' ) {
        $from_time = 0;
        $to_time   = 4294967295; # unsigned int max
    } elsif ( $span eq 'c' || $span eq 'sc' ) {
        if ($from =~ /\A[1-9][0-9]*\z/) {
            $from_time = $from;
        } else {
            $from_time = HTTP::Date::str2time($from);
            die "invalid from date: $from" unless $from_time;
        }
        if ($to && $to =~ /\A[1-9][0-9]*\z/) {
            $to_time = $to;
        } else {
            $to_time = $to ? HTTP::Date::str2time($to) : time;
            die "invalid to date: $to" unless $to_time;
        }
        die "from($from) is newer than to($to)" if $from_time > $to_time;
    } elsif ( $span eq 'h' || $span eq 'sh' ) {
        $from_time = time -1 * 60 * 60 * 2;
    } elsif ( $span eq 'n' || $span eq 'sn' ) {
        $from_time = time -1 * 60 * 60 * 14;
    } elsif ( $span eq 'w' ) {
        $from_time = time -1 * 60 * 60 * 24 * 8;
    } elsif ( $span eq 'm' ) {
        $from_time = time -1 * 60 * 60 * 24 * 35;
    } elsif ( $span eq 'y' ) {
        $from_time = time -1 * 60 * 60 * 24 * 400;
    } elsif ( $span eq '3d' ) {
        $from_time = time -1 * 60 * 60 * 24 * 3;
    } elsif ( $span eq '8h' ) {
        $from_time = time -1 * 8 * 60 * 60;
    } elsif ( $span eq '4h' ) {
        $from_time = time -1 * 4 * 60 * 60;
    } else {
        $from_time = time -1 * 60 * 60 * 33; # 33 hours
    }

    my @vrules = ();