Dump EXTRA_ARGS in ztest.pl
[simpleserver-moved-to-github.git] / ztest.pl
index 5640d93..40a9101 100755 (executable)
--- a/ztest.pl
+++ b/ztest.pl
 #!/usr/bin/perl -w
+
+## This file is part of simpleserver
+## Copyright (C) 2000-2013 Index Data.
+## All rights reserved.
+## Redistribution and use in source and binary forms, with or without
+## modification, are permitted provided that the following conditions are met:
+##
+##     * Redistributions of source code must retain the above copyright
+##       notice, this list of conditions and the following disclaimer.
+##     * Redistributions in binary form must reproduce the above copyright
+##       notice, this list of conditions and the following disclaimer in the
+##       documentation and/or other materials provided with the distribution.
+##     * Neither the name of Index Data nor the names of its contributors
+##       may be used to endorse or promote products derived from this
+##       software without specific prior written permission.
+##
+## THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
+## EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+## WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+## DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
+## DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+## (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+## LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+## ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+## THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
 use ExtUtils::testlib;
+use Data::Dumper;
 use Net::Z3950::SimpleServer;
 use Net::Z3950::OID;
+use strict;
 
-
-sub udskriv_hash {
-
+sub dump_hash {
        my $href = shift;
        my $key;
-       my $item;
-
-       foreach $key (keys %{ $href }) {
-               print "$key = ";
-               if ($key eq "DATABASES") {
-                       foreach $item ( @{ $href->{DATABASES} }) {
-                               print "$item  ";
-                       }
-                       print "\n";
-               } elsif ($key eq "HANDLE") {
-                       foreach $item ( keys %{ $href->{HANDLE} }) {
-                               print "        $item  => ";
-                               print ${ $href->{HANDLE}}{$item};
-                               print "\n";
-                       }
-               } else {
-                       print $href->{$key};
-                       print "\n";
-               }
+
+       foreach $key (keys %$href) {
+               printf("%10s    =>      %s\n", $key, $href->{$key});
        }
 }
 
 
-
 sub my_init_handler {
+       my $args = shift;
+       my $session = {};
+
+       $args->{IMP_NAME} = "DemoServer";
+       $args->{IMP_ID} = "81";
+       $args->{IMP_VER} = "3.14159";
+       $args->{ERR_CODE} = 0;
+       $args->{HANDLE} = $session;
+       if (defined($args->{PASS}) && defined($args->{USER})) {
+           printf("Received USER/PASS=%s/%s\n", $args->{USER},$args->{PASS});
+       }
 
-       my $href = shift;
-       my $hash = {};
-
-       $hash->{Anders} = "Sønderberg Mortensen";
-       $hash->{Birgit} = "Stenhøj Andersen";
-       $href->{IMP_NAME} = "MyServer";
-       $href->{IMP_VER} = "3.14159";
-       $href->{ERR_CODE} = 0;
-       $href->{HANDLE} = $hash;
-       print "\n";
-       print "---------------------------------------------------------------\n";
-       print "Connection established\n";
-       print "\n";
-       udskriv_hash($href);
-       print "---------------------------------------------------------------\n";
 }
 
-sub my_search_handler { 
 
-       my $href = shift;
-       my $key;
-       my $hash = $href->{HANDLE};
-#      my $hash = {};
-
-       $href->{HITS} = 1;
-       $href->{ERR_STR} = "A";
-       $hash->{Search} = "Search Handler er besøgt";
-#      $href->{HANDLE} = $hash;
-       print "\n";
-       print "---------------------------------------------------------------\n";
-       print "Search handler\n";
-       print "\n";
-       udskriv_hash($href);
-       print "---------------------------------------------------------------\n";
+sub my_sort_handler {
+    my ($args) = @_;
+
+    print "Sort handler called\n";
+    print Dumper( $args );
 }
 
+sub my_scan_handler {
+       my $args = shift;
+       my $term = $args->{TERM};
+       my $entries = [
+                               {       TERM            =>      'Number 1',
+                                       DISPLAY_TERM    =>      'Number .1',
+                                       OCCURRENCE      =>      10 },
+                               {       TERM            =>      'Number 2',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 3',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 4',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 5',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 6',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 7',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 8',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 9',
+                                       OCCURRENCE      =>      8 },
+                               {       TERM            =>      'Number 10',
+                                       OCCURRENCE      =>      4 },
+                       ];
+       $args->{NUMBER} = 10;
+       $args->{ENTRIES} = $entries;
+       $args->{STATUS} = Net::Z3950::SimpleServer::ScanPartial;
+       print "Welcome to scan....\n";
+       $args->{EXTRA_RESPONSE_DATA} = '<scanextra>b</scanextra>';
+       print "You scanned for term '$term'\n";
+}
 
-sub my_present_handler {
-       my $href = shift;
 
-       $href->{ERR_CODE} = 0;
+my $_fail_frequency = 0;
+my $_counter = 0;
+
+sub my_search_handler {
+       my $args = shift;
+
+       my $data = [{
+                       name            =>      "Peter Dornan",
+                       title           =>      "Spokesman",
+                       collaboration   =>      "ATLAS"
+                   }, {
+                       name            =>      "Jorn Dines Hansen",
+                       title           =>      "Professor",
+                       collaboration   =>      "HERA-B"
+                   }, {
+                       name            =>      "Alain Blondel",
+                       title           =>      "Head of coll.",
+                       collaboration   =>      "ALEPH"
+                   }];
+
+       my $session = $args->{HANDLE};
+       my $set_id = $args->{SETNAME};
+       my $rpn = $args->{RPN};
+       my @database_list = @{ $args->{DATABASES} };
+       my $query = $args->{QUERY};
+       my $facets = $args->{INPUTFACETS};
+       my $hits = 3;
 
-       $href->{ERR_STR} = "";
-       print "\n";
-       print "--------------------------------------------------------------\n";
-       print "Present handler\n";
-       print "\n";
-       udskriv_hash($href);
-       print "--------------------------------------------------------------\n";
-       return;
-}
-
-sub my_close_handler {
-       my $href = shift;
+       print "------------------------------------------------------------\n";
+       print "Processing query : $query\n";
+       printf("Database set     : %s\n", join(" ", @database_list));
+       print "Setname          : $set_id\n";
+       print " inputfacets:\n";
+       print Dumper($facets);
+        print " extra args:\n";
+        print Dumper($args->{EXTRA_ARGS});
+       print "------------------------------------------------------------\n";
 
-       print "\n";
-       print "-------------------------------------------------------------\n";
-       print "Connection closed\n";
-       print "\n";
-       udskriv_hash($href);
-       print "-------------------------------------------------------------\n";
+       $args->{OUTPUTFACETS} = $facets;
 
+       $args->{EXTRA_RESPONSE_DATA} = '<searchextra>b</searchextra>';
+       $args->{HITS} = $hits;
+       $session->{$set_id} = $data;
+       $session->{__HITS} = $hits;
+       if ($_fail_frequency != 0 && ++$_counter % $_fail_frequency == 0) {
+           print "Exiting to be nasty to client\n";
+           exit(1);
+       }
 }
 
 
 sub my_fetch_handler {
-       my $href = shift;
-       my $hash = $href->{HANDLE};
-
-       $hash->{Fetch} = "Fetch handler er besøgt";
-       ##$href->{RECORD} = "<head>Overskrift</head> <text>Her kommer teksten</text>";
-       $href->{RECORD} = "<xml><head>Overskrift</head><body>Der var engang en mand</body></xml>";
-       $href->{LEN} = 69;
-       $href->{NUMBER} = 1;
-       $href->{BASENAME} = "MS-Gud";
-       $href->{LAST} = 1;
-       ## $href->{HANDLE} = \%hash;
-       print "\n";
-       print "------------------------------------------------------------\n";
-       print "Fetch handler\n";
-       print "\n";
-       udskriv_hash($href);
-       if ($href->{REQ_FORM} eq Net::Z3950::OID::unimarc) {
-               print "Formatet UNIMARC\n";
-       } else {
-               print "Formatet er IKKE unimarc\n";
+       my $args = shift;
+       my $session = $args->{HANDLE};
+       my $set_id = $args->{SETNAME};
+       my $data = $session->{$set_id};
+       my $offset = $args->{OFFSET};
+       my $record = "<xml>";
+       my $field;
+       my $hits = $session->{__HITS};
+       my $href = $data->[$offset - 1];
+
+       $args->{REP_FORM} = Net::Z3950::OID::xml;
+       foreach $field (keys %$href) {
+               $record .= "<" . $field . ">" . $href->{$field} . "</" . $field . ">";
+       }
+
+       $record .= "</xml>";
+       $args->{RECORD} = $record;
+       if ($offset == $session->{__HITS}) {
+               $args->{LAST} = 1;
        }
-       print "------------------------------------------------------------\n";
-       
 }
 
+sub my_start_handler {
+    my $args = shift;
+    my $config = $args->{CONFIG};
+}
 
+Net::Z3950::SimpleServer::yazlog("hello");
 
-my $handler = Net::Z3950::SimpleServer->new({ INIT     =>      \&my_init_handler,
-                               CLOSE   =>      \&my_close_handler,
-                               SEARCH  =>      \&my_search_handler,
-                           FETCH       =>      \&my_fetch_handler
-                         });
+my $handler = new Net::Z3950::SimpleServer(
+                START   =>      "main::my_start_handler",
+               INIT    =>      "main::my_init_handler",
+               SEARCH  =>      "main::my_search_handler",
+               SCAN    =>      "main::my_scan_handler",
+                SORT    =>      "main::my_sort_handler",
+               FETCH   =>      "main::my_fetch_handler" );
 
+if (@ARGV >= 2 && $ARGV[0] eq "-n") {
+    $_fail_frequency = $ARGV[1];
+    shift;
+    shift;
+}
 $handler->launch_server("ztest.pl", @ARGV);
-