Debbugs/DB/Load.pm

   1 # This module is part of debbugs, and is released
   2 # under the terms of the GPL version 2, or any later version. See the
   3 # file README and COPYING for more information.
   4 # Copyright 2013 by Don Armstrong <don@donarmstrong.com>.
   5
   6 package Debbugs::DB::Load;
   7
   8 =head1 NAME
   9
  10 Debbugs::DB::Load -- Utility routines for loading the database
  11
  12 =head1 SYNOPSIS
  13
  14
  15 =head1 DESCRIPTION
  16
  17
  18 =head1 BUGS
  19
  20 None known.
  21
  22 =cut
  23
  24 use warnings;
  25 use strict;
  26 use vars qw($VERSION $DEBUG %EXPORT_TAGS @EXPORT_OK @EXPORT);
  27 use base qw(Exporter);
  28
  29 BEGIN{
  30      ($VERSION) = q$Revision$ =~ /^Revision:\s+([^\s+])/;
  31      $DEBUG = 0 unless defined $DEBUG;
  32
  33      @EXPORT = ();
  34      %EXPORT_TAGS = (load_bug    => [qw(load_bug handle_load_bug_queue)],
  35                     );
  36      @EXPORT_OK = ();
  37      Exporter::export_ok_tags(keys %EXPORT_TAGS);
  38      $EXPORT_TAGS{all} = [@EXPORT_OK];
  39 }
  40
  41 use Params::Validate qw(validate_with :types);
  42
  43 use Debbugs::Status qw(read_bug split_status_fields);
  44 use Debbugs::DB;
  45 use DateTime;
  46 use Debbugs::Common qw(make_list);
  47 use Debbugs::Config qw(:config);
  48
  49 =head2 Bug loading
  50
  51 Routines to load bug; exported with :load_bug
  52
  53 =over
  54
  55 =item load_bug
  56
  57      load_bug(db => $schema,
  58               data => split_status_fields($data),
  59               tags => \%tags,
  60               queue => \%queue);
  61
  62 Loads a bug's metadata into the database. (Does not load any messages)
  63
  64 =over
  65
  66 =item db -- Debbugs::DB object
  67
  68 =item data -- Bug data (from read_bug) which has been split with split_status_fields
  69
  70 =item tags -- tag cache (hashref); optional
  71
  72 =item queue -- queue of operations to perform after bug is loaded; optional.
  73
  74 =back
  75
  76 =cut
  77
  78 sub load_bug {
  79      my %param = validate_with(params => \@_,
  80                               spec => {db => {type => OBJECT,
  81                                              },
  82                                        data => {type => HASHREF},
  83                                        tags => {type => HASHREF,
  84                                                 default => sub {return {}},
  85                                                 optional => 1},
  86                                        queue => {type => HASHREF,
  87                                                  optional => 1},
  88                                       });
  89     my $s = $param{db};
  90     my $data = $param{data};
  91     my $tags = $param{tags};
  92     my $queue = $param{queue};
  93     my $can_queue = 1;
  94     if (not defined $queue) {
  95         $can_queue = 0;
  96         $queue = {};
  97     }
  98     my $s_data = split_status_fields($data);
  99     my @tags;
 100     for my $tag (make_list($s_data->{keywords})) {
 101         next unless defined $tag and length $tag;
 102         # this allows for invalid tags. But we'll use this to try to
 103         # find those bugs and clean them up
 104         if (not exists $tags->{$tag}) {
 105             $tags->{$tag} = $s->resultset('Tag')->find_or_create({tag => $tag});
 106         }
 107         push @tags, $tags->{$tag};
 108     }
 109     my $bug = {id => $data->{bug_num},
 110                creation => DateTime->from_epoch(epoch => $data->{date}),
 111                log_modified => DateTime->from_epoch(epoch => $data->{log_modified}),
 112                last_modified => DateTime->from_epoch(epoch => $data->{last_modified}),
 113                archived => $data->{archived},
 114                (defined $data->{unarchived} and length($data->{unarchived}))?(unarchived => DateTime->from_epoch(epoch => $data->{unarchived})):(),
 115                forwarded => $data->{forwarded} // '',
 116                summary => $data->{summary} // '',
 117                outlook => $data->{outlook} // '',
 118                subject => $data->{subject} // '',
 119                done => $data->{done} // '',
 120                owner => $data->{owner} // '',
 121                severity => length($data->{severity}) ? $data->{severity} : $config{default_severity},
 122               };
 123     $s->resultset('Bug')->update_or_create($bug);
 124     $s->txn_do(sub {
 125                    for my $ff (qw(found fixed)) {
 126                        my @elements = $s->resultset('BugVer')->search({bug_id => $data->{bug_num},
 127                                                                        found  => $ff eq 'found'?1:0,
 128                                                                       });
 129                        my %elements_to_delete = map {($elements[$_]->ver_string(),$_)} 0..$#elements;
 130                        my @elements_to_add;
 131                        for my $version (@{$data->{"${ff}_versions"}}) {
 132                            if (exists $elements_to_delete{$version}) {
 133                                delete $elements_to_delete{$version};
 134                            } else {
 135                                push @elements_to_add,$version;
 136                            }
 137                        }
 138                        for my $element (keys %elements_to_delete) {
 139                            $elements_to_delete{$element}->delete();
 140                        }
 141                        for my $element (@elements_to_add) {
 142                            # find source package and source version id
 143                            my $ne = $s->resultset('BugVer')->new_result({bug_id => $data->{bug_num},
 144                                                                          ver_string => $element,
 145                                                                          found => $ff eq 'found'?1:0,
 146                                                                         }
 147                                                                        );
 148                            if (my ($src_pkg,$src_ver) = $element =~ m{^([^\/]+)/(.+)$}) {
 149                                my $src_pkg_e = $s->resultset('SrcPkg')->single({pkg => $src_pkg});
 150                                if (defined $src_pkg_e) {
 151                                    $ne->src_pkg_id($src_pkg_e->id());
 152                                    my $src_ver_e = $s->resultset('SrcVer')->single({src_pkg_id => $src_pkg_e->id(),
 153                                                                                     ver => $src_ver
 154                                                                                    });
 155                                    $ne->src_ver_id($src_ver_e->id()) if defined $src_ver_e;
 156                                }
 157                            }
 158                            $ne->insert();
 159                        }
 160                    }
 161                });
 162     $s->txn_do(sub {
 163                    $s->resultset('BugTag')->search({bug_id => $data->{bug_num}})->delete();
 164                    $s->populate(BugTag => [[qw(bug_id tag_id)], map {[$data->{bug_num}, $_->id()]} @tags]);
 165                });
 166     # because these bugs reference other bugs which might not exist
 167     # yet, we can't handle them until we've loaded all bugs. queue
 168     # them up.
 169     $queue->{merged}{$data->{bug_num}} = [@{$data->{mergedwith}}];
 170     $queue->{blocks}{$data->{bug_num}} = [@{$data->{blocks}}];
 171
 172     if (not $can_queue) {
 173         handle_load_bug_queue(db => $s,queue => $queue);
 174     }
 175
 176     print STDERR "Handled $data->{bug_num}\n";
 177     # still need to handle merges, versions, etc.
 178 }
 179
 180 =item handle_load_bug_queue
 181
 182      handle_load_bug_queue(db => $schema,queue => $queue);
 183
 184 Handles a queue of operations created by load bug. [These operations
 185 are used to handle cases where a bug referenced by a loaded bug may
 186 not exist yet. In cases where the bugs should exist, the queue is
 187 cleared automatically by load_bug if queue is undefined.
 188
 189 =cut
 190
 191 sub handle_load_bug_queue{
 192     my %param = validate_with(params => \@_,
 193                               spec => {db => {type => OBJECT,
 194                                              },
 195                                        queue => {type => HASHREF,
 196                                                 },
 197                                       });
 198     my $s = $param{db};
 199     my $queue = $param{queue};
 200     my %queue_types =
 201         (merged => {set => 'BugMerged',
 202                     columns => [qw(bug_id merged)],
 203                     bug_id => 'bug_id',
 204                    },
 205          blocks => {set => 'BugBlock',
 206                     columns => [qw(bug_id blocks)],
 207                     bug_id => 'bug_id',
 208                    },
 209         );
 210     for my $queue_type (keys %queue_types) {
 211         for my $bug (%{$queue->{$queue_type}}) {
 212             my $qt = $queue_types{$queue_type};
 213             $s->txn_do(sub {
 214                            $s->resultset($qt->{set})->search({$qt->{bug_id},$bug})->delete();
 215                            $s->populate($qt->{set},[[@{$qt->{columns}}],map {[$bug,$_]} @{$queue->{$queue_type}{$bug}}]) if
 216                                @{$queue->{$queue_type}{$bug}};
 217                        }
 218                       );
 219         }
 220     }
 221 }
 222
 223 =back
 224
 225 =head2 Debinfo
 226
 227 Commands to handle src and package version loading from debinfo files
 228
 229 =item load_debinfo
 230
 231      load_debinfo($schema,$binname, $binver, $binarch, $srcname, $srcver);
 232
 233
 234
 235 =cut
 236
 237 sub load_debinfo {
 238     my ($schema,$binname, $binver, $binarch, $srcname, $srcver) = @_;
 239     my $sp = $schema->resultset('SrcPkg')->find_or_create({pkg => $srcname});
 240     my $sv = $schema->resultset('SrcVer')->find_or_create({src_pkg_id=>$sp->id(),
 241                                                            ver => $srcver});
 242     my $arch = $schema->resultset('Arch')->find_or_create({arch => $binarch});
 243     my $bp = $schema->resultset('BinPkg')->find_or_create({pkg => $binname});
 244     $schema->resultset('BinVer')->find_or_create({bin_pkg_id => $bp->id(),
 245                                                   src_ver_id => $sv->id(),
 246                                                   arch_id    => $arch->id(),
 247                                                   ver        => $binver,
 248                                                  });
 249 }
 250
 251 1;
 252
 253
 254 __END__