Scaffolding for saving and reading reference results.

2025-10-20 09:40:38 +00:00 · 2013-02-19 19:45:16 -07:00
parent aeb26300cb
commit 4cb51140f1
3 changed files with 646 additions and 15 deletions
--- a/lib/ResultIterator.pm
+++ b/lib/ResultIterator.pm
@@ -0,0 +1,133 @@
+# This program is copyright 2013 Percona Ireland Ltd.
+# Feedback and improvements are welcome.
+#
+# THIS PROGRAM IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED
+# WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
+# MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+#
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU General Public License as published by the Free Software
+# Foundation, version 2; OR the Perl Artistic License.  On UNIX and similar
+# systems, you can issue `man perlgpl' or `man perlartistic' to read these
+# licenses.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program; if not, write to the Free Software Foundation, Inc., 59 Temple
+# Place, Suite 330, Boston, MA  02111-1307  USA.
+# ###########################################################################
+# ResultIterator package
+# ###########################################################################
+{
+package ResultIterator;
+
+use strict;
+use warnings FATAL => 'all';
+use English qw(-no_match_vars);
+use constant PTDEBUG => $ENV{PTDEBUG} || 0;
+
+use Lmo;
+
+has 'dir' => (
+    is       => 'ro',
+    isa      => 'Str',
+    required => 1,
+);
+
+has '_query_fh' => (
+    is       => 'rw',
+    isa      => 'Maybe[FileHandle]',
+    required => 0,
+);
+
+has '_meta_fh' => (
+    is       => 'rw',
+    isa      => 'Maybe[FileHandle]',
+    required => 0,
+);
+
+has '_results_fh' => (
+    is       => 'rw',
+    isa      => 'Maybe[FileHandle]',
+    required => 0,
+);
+
+sub BUILDARGS {
+   my $class = shift;
+   my $args  = $class->SUPER::BUILDARGS(@_);
+
+   my $dir = $args->{dir};
+   die "$dir does not exist\n" unless -d $dir;
+
+   my $query_file = "$dir/query";
+   PTDEBUG && _d('Query file:', $query_file);
+   open my $_query_fh, '<', $query_file
+      or die "Cannot open $query_file for writing: $OS_ERROR";
+
+   my $meta_file = "$dir/meta";
+   PTDEBUG && _d('Meta file:', $meta_file);
+   open my $_meta_fh, '<', $meta_file
+      or die "Cannot open $meta_file for writing: $OS_ERROR";
+
+   my $results_file = "$dir/results";
+   PTDEBUG && _d('Results file:', $results_file);
+   open my $_results_fh, '<', $results_file
+      or die "Cannot open $results_file for writing: $OS_ERROR";
+
+   my $self = {
+      %$args,
+      _query_fh   => $_query_fh,
+      _meta_fh    => $_meta_fh,
+      _results_fh => $_results_fh,
+   };
+
+   return $self;
+}
+
+sub next {
+   my ($self, %args) = @_;
+
+   local $INPUT_RECORD_SEPARATOR = "\n##\n";
+
+   my $_query_fh   = $self->_query_fh;
+   my $_meta_fh    = $self->_meta_fh;
+   my $_results_fh = $self->_results_fh;
+
+   my $query   = <$_query_fh>;
+   my $meta    = <$_meta_fh>;
+   my $results = <$_results_fh>;
+
+   return unless $query;
+   
+   chomp($query);
+
+   if ( $meta ) {
+      chomp($meta);
+      eval $meta;
+   }
+
+   if ( $results ) {
+      chomp($results);
+      eval $results;
+   }
+
+   return {
+      query   => $query,
+      meta    => $meta,
+      results => $results,
+   };
+}
+
+sub _d {
+   my ($package, undef, $line) = caller 0;
+   @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; }
+        map { defined $_ ? $_ : 'undef' }
+        @_;
+   print STDERR "# $package:$line $PID ", join(' ', @_), "\n";
+}
+
+no Lmo;
+1;
+}
+# ###########################################################################
+# End ResultIterator package
+# ###########################################################################