1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
use strict;
use warnings;
package EvensOnlyCollector;
use base qw( Lucy::Search::Collector );
our %doc_ids;
sub new {
my $self = shift->SUPER::new;
$doc_ids{$$self} = [];
return $self;
}
sub collect {
my ( $self, $doc_id ) = @_;
if ( $doc_id % 2 == 0 ) {
push @{ $doc_ids{$$self} }, $doc_id;
}
}
sub get_doc_ids { $doc_ids{ ${ +shift } } }
sub DESTROY {
my $self = shift;
delete $doc_ids{$$self};
$self->SUPER::DESTROY;
}
package main;
use Test::More tests => 1;
use Lucy::Test;
use LucyX::Search::MockMatcher;
my $collector = EvensOnlyCollector->new;
my $matcher
= LucyX::Search::MockMatcher->new( doc_ids => [ 1, 5, 10, 1000 ], );
$collector->set_matcher($matcher);
while ( my $doc_id = $matcher->next ) {
$collector->collect($doc_id);
}
is_deeply(
$collector->get_doc_ids,
[ 10, 1000 ],
"Collector can be subclassed"
);
|
