use limit just for object classes which have "person" in them

[virtual-ldap] / lib / LDAP / Koha.pm

diff --git a/lib/LDAP/Koha.pm b/lib/LDAP/Koha.pm
index fccc38f..7c2791c 100644 (file)
--- a/lib/LDAP/Koha.pm
+++ b/lib/LDAP/Koha.pm
@@ -25,9 +25,10 @@ our $database = 'koha';
 our $user     = 'unconfigured-user';
 our $passwd   = 'unconfigured-password';
 
-our $max_results = 1500; # 100; # FIXME
+our $max_results = $ENV{MAX_RESULTS} || 3000; # FIXME must be enough for all users
+our $objectclass_default = 'hrEduPerson';
 
-our $objectclass = 'HrEduPerson';
+our $objectclass;
 
 $SIG{__DIE__} = sub {
        warn "!!! DIE ", @_;
@@ -156,6 +157,7 @@ sub search {
 
                my $sql_where = '';
                @values = ();
+               $objectclass = '';
 
                foreach my $filter ( keys %{ $reqData->{'filter'} } ) {
 
@@ -177,22 +179,34 @@ sub search {
                                        }
                                }
 
-                               $sql_where .= ' ' . join( " $filter ", @limits );
-
                        } else {
                                __ldap_search_to_sql( $filter, $reqData->{'filter'}->{$filter} );
                        }
 
+                       $sql_where .= ' ' . join( " $filter ", @limits ) if @limits;
+
                }
 
+               $objectclass ||= $objectclass_default;
+
+               my $sql_select = read_file( lc "sql/$objectclass.sql" );
                if ( $sql_where ) {
-                       $sql_where = " where $sql_where";
+                       if ( $sql_select !~ m{where}i ) {
+                               $sql_where = " where $sql_where";
+                       } else {
+                               $sql_where = " and $sql_where";
+                       }
                }
 
-               my $sql_select = read_file( lc "sql/$objectclass.sql" );
 
-               warn "# SQL:\n$sql_select\n", $sql_where ? $sql_where : '-- no where', "\n# DATA: ",dump( @values );
-               my $sth = $dbh->prepare( $sql_select . $sql_where . " LIMIT $max_results" ); # XXX remove limit?
+               my $sql
+                       = $sql_select
+                       . $sql_where
+                       . ( $objectclass =~ m{person}i ? " LIMIT $max_results" : '' ) # add limit just for persons
+                       ;
+
+               warn "# SQL:\n$sql\n# DATA: ",dump( @values );
+               my $sth = $dbh->prepare( $sql );
                $sth->execute( @values );
 
                warn "# ", $sth->rows, " results for ",dump( $reqData->{'filter'} );