Grep/Search/Estraier.pm

package Grep::Search::Estraier;

use strict;
use warnings;
use base qw( Class::Accessor );
Grep::Search::Estraier->mk_accessors( qw( index_path ) );

use Search::Estraier;
use Encode qw/encode/;

=head1 NAME

Grep::Search::Estraier - full text search using L<Search::Estraier>

=head1 METHODS

=cut

=head2 node

  $search->node

=cut

our $indexes;

sub node {
        my $self = shift;

        my $node = new Search::Estraier::Node(
                url => 'http://localhost:1978/node/grep',
                user => 'admin',
                passwd => 'admin',
                create => 1,
                label => 'Grep index',
                croak_on_error => 1,
#               debug => 1,
        );

        return $node;
}

=head2 add

  $self->add( $doc_hash );

=cut

sub add {
        my $self = shift;
        my $doc = shift;

        my $doc_draft = new Search::Estraier::Document;
        $doc_draft->add_attr('@uri', $doc->{id});
        $doc_draft->add_attr('@title', $doc->{title});
        foreach my $attr ( keys %$doc ) {
                my $v = encode('iso-8859-1', $doc->{$attr});
#               my $v = $doc->{$attr};
                $doc_draft->add_attr( $attr, $v );
                $doc_draft->add_text( $v );
        }

        $self->node->put_doc($doc_draft)
}

=head2 search

  my $fetch_hit_coderef = $self->search('search query');

=cut

sub search {
        my $self = shift;
        my $q = shift;

        my $cond = new Search::Estraier::Condition;

        $cond->set_phrase( $q );

        my $nres = $self->node->search($cond, 0);

        $self->hits( $nres->hits );

        warn "# search '$q' got ", $self->hits, " results\n";

        my $i = 0;

        return sub {
                return if $i++ >= $nres->doc_num;
                warn "# get_doc($i)\n";
                my $rdoc = $nres->get_doc($i);
                return {
                        id => $rdoc->attr('id'),
                        title => $rdoc->attr('title'),
                        score => $rdoc->score,
                };
        };

}


1;
1	package Grep::Search::Estraier;
2
3	use strict;
4	use warnings;
5	use base qw( Class::Accessor );
6	Grep::Search::Estraier->mk_accessors( qw( index_path ) );
7
8	use Search::Estraier;
9	use Encode qw/encode/;
10
11	=head1 NAME
12
13	Grep::Search::Estraier - full text search using L<Search::Estraier>
14
15	=head1 METHODS
16
17	=cut
18
19	=head2 node
20
21	$search->node
22
23	=cut
24
25	our $indexes;
26
27	sub node {
28	my $self = shift;
29
30	my $node = new Search::Estraier::Node(
31	url => 'http://localhost:1978/node/grep',
32	user => 'admin',
33	passwd => 'admin',
34	create => 1,
35	label => 'Grep index',
36	croak_on_error => 1,
37	# debug => 1,
38	);
39
40	return $node;
41	}
42
43	=head2 add
44
45	$self->add( $doc_hash );
46
47	=cut
48
49	sub add {
50	my $self = shift;
51	my $doc = shift;
52
53	my $doc_draft = new Search::Estraier::Document;
54	$doc_draft->add_attr('@uri', $doc->{id});
55	$doc_draft->add_attr('@title', $doc->{title});
56	foreach my $attr ( keys %$doc ) {
57	my $v = encode('iso-8859-1', $doc->{$attr});
58	# my $v = $doc->{$attr};
59	$doc_draft->add_attr( $attr, $v );
60	$doc_draft->add_text( $v );
61	}
62
63	$self->node->put_doc($doc_draft)
64	}
65
66	=head2 search
67
68	my $fetch_hit_coderef = $self->search('search query');
69
70	=cut
71
72	sub search {
73	my $self = shift;
74	my $q = shift;
75
76	my $cond = new Search::Estraier::Condition;
77
78	$cond->set_phrase( $q );
79
80	my $nres = $self->node->search($cond, 0);
81
82	$self->hits( $nres->hits );
83
84	warn "# search '$q' got ", $self->hits, " results\n";
85
86	my $i = 0;
87
88	return sub {
89	return if $i++ >= $nres->doc_num;
90	warn "# get_doc($i)\n";
91	my $rdoc = $nres->get_doc($i);
92	return {
93	id => $rdoc->attr('id'),
94	title => $rdoc->attr('title'),
95	score => $rdoc->score,
96	};
97	};
98
99	}
100
101
102	1;