Common-RDBO-Helper/lib/Common/RDBO/Helper/Comment/Manager.pm

   1 package Common::RDBO::Helper::Comment::Manager;
   2
   3 use strict;
   4 use warnings;
   5
   6 use Rose::Object::MixIn();
   7 our @ISA = qw(Rose::Object::MixIn);
   8
   9 use Hash::Merge 'merge';
  10
  11 __PACKAGE__->export_tag(
  12     all => [ qw/ get_list get_recent fill_comment_count / ] );
  13
  14 sub get_list {
  15     my $self = shift;
  16     my %args = ref $_[ 0 ] eq 'HASH' ? %{ $_[ 0 ] } : @_;
  17
  18     use Data::Dumper;
  19
  20     #die Dumper \%args;
  21     return $self->get_objects( %args );
  22 }
  23
  24 sub get_recent {
  25     my $self = shift;
  26     my %args = ref $_[ 0 ] eq 'HASH' ? %{ $_[ 0 ] } : @_;
  27
  28     my $limit = delete $args{ limit } || 10;
  29
  30     use Data::Dumper;
  31
  32     #die Dumper \%args;
  33
  34     my $cond = merge {
  35         sort_by => 'addtime DESC',
  36         limit   => $limit,
  37
  38         #require_objects => [ 'post' ]
  39       },
  40       \%args;
  41
  42     #die Dumper $cond;
  43
  44     return $self->get_list( $cond );
  45 }
  46
  47 sub fill_comment_count {
  48     my ( $self ) = shift;
  49     my %args =
  50         ref $_[ 0 ] eq 'HASH' ? %{ $_[ 0 ] }
  51       : scalar @_ == 1 ? ( objects => $_[ 0 ] )
  52       :                  @_;
  53
  54     my @objects;
  55     if ( ref $args{ objects } eq 'ARRAY' ) {
  56         @objects = @{ delete $args{ objects } };
  57     } else {
  58         $objects[0] = delete $args{ objects };
  59     }
  60
  61     my ( $rel ) =
  62       grep { $_->name eq 'comments' }
  63       @{ $self->object_class->meta->relationships };
  64
  65     if ( scalar @objects ) {
  66         my $cond = merge {
  67             query => [ master_id => [ map { $_->id } @objects ] ],
  68             select   => [ 'master_id', \'COUNT(id) AS comment_count' ],
  69             group_by => 'master_id'
  70           },
  71           \%args;
  72
  73         my $comments = $self->get_objects( %$cond );
  74
  75         my %comments = map { $_->master_id => $_->comment_count } @$comments;
  76
  77         foreach ( @objects ) {
  78             $_->{ comment_count } = $comments{ $_->id }
  79               if $comments{ $_->id };
  80         }
  81     }
  82 }
  83
  84 1;