diff options
Diffstat (limited to 'lib/Net/Riak')
-rw-r--r-- | lib/Net/Riak/Client.pm | 5 | ||||
-rw-r--r-- | lib/Net/Riak/Role/REST.pm | 1 | ||||
-rw-r--r-- | lib/Net/Riak/Role/REST/Search.pm | 76 | ||||
-rw-r--r-- | lib/Net/Riak/Search.pm | 85 |
4 files changed, 167 insertions, 0 deletions
diff --git a/lib/Net/Riak/Client.pm b/lib/Net/Riak/Client.pm index f38bec6..ad9315e 100644 --- a/lib/Net/Riak/Client.pm +++ b/lib/Net/Riak/Client.pm @@ -15,6 +15,11 @@ has mapred_prefix => ( isa => 'Str', default => 'mapred' ); +has search_prefix => ( + is => 'rw', + isa => 'Str', + default => 'solr' +); has [qw/r w dw/] => ( is => 'rw', isa => 'Int', diff --git a/lib/Net/Riak/Role/REST.pm b/lib/Net/Riak/Role/REST.pm index dfab5a0..261d573 100644 --- a/lib/Net/Riak/Role/REST.pm +++ b/lib/Net/Riak/Role/REST.pm @@ -12,6 +12,7 @@ with qw/Net::Riak::Role::REST::Bucket Net::Riak::Role::REST::Object Net::Riak::Role::REST::Link Net::Riak::Role::REST::MapReduce + Net::Riak::Role::REST::Search /; has http_request => ( diff --git a/lib/Net/Riak/Role/REST/Search.pm b/lib/Net/Riak/Role/REST/Search.pm new file mode 100644 index 0000000..b83e3d4 --- /dev/null +++ b/lib/Net/Riak/Role/REST/Search.pm @@ -0,0 +1,76 @@ +package Net::Riak::Role::REST::Search; +use Moose::Role; +use JSON; + +#ABSTRACT: Search interface + +sub search { + my $self = shift; + my %params = @_; + my $request; + + $request = + $self->new_request( 'GET', + [ $self->search_prefix, "select" ], \%params ) unless $params{index}; + if ( $params{index} ){ + my $index = delete $params{index}; + $request = + $self->new_request( 'GET', + [ $self->search_prefix, $index, "select" ], \%params ); + } + + my $http_response = $self->send_request($request); + + return if (!$http_response); + + my $status = $http_response->code; + if ($status == 404) { + return; + } + + return JSON::decode_json($http_response->content) if $params{wt} =~ /json/i; + $http_response->content; +}; + +sub setup_indexing { + my ( $self, $bucket ) = @_; + my $request = + $self->new_request( 'GET', + [ $self->prefix, $bucket ] ); + + my $http_response = $self->send_request($request); + + return if (!$http_response); + my $status = $http_response->code; + if ($status == 404) { + return; + } + + my $precommits = JSON::decode_json($http_response->content)->{props}->{precommit}; + + for (@$precommits){ + return JSON::decode_json($http_response->content) if $_->{mod} eq "riak_search_kv_hook"; + } + push ( @$precommits, { mod => "riak_search_kv_hook" , fun => "precommit" } ); + + $request = $self->new_request( 'PUT', [ $self->prefix, $bucket ] ); + $request->content( JSON::encode_json({ props => { precommit => $precommits } } ) ); + $request->header('Content-Type' => "application/json" ); + + $http_response = $self->send_request($request); + + return if (!$http_response); + $status = $http_response->code; + if ($status == 404) { + return; + } + $request = + $self->new_request( 'GET', + [ $self->prefix, $bucket ] ); + + $http_response = $self->send_request($request); + + JSON::decode_json($http_response->content); +} + +1; diff --git a/lib/Net/Riak/Search.pm b/lib/Net/Riak/Search.pm new file mode 100644 index 0000000..ee2ed57 --- /dev/null +++ b/lib/Net/Riak/Search.pm @@ -0,0 +1,85 @@ +package +Net::Riak::Search; +use Moose; + +#ABSTRACT: Search interface + +with 'Net::Riak::Role::Base' => {classes => + [{name => 'client', required => 0},]}; + +sub search { + my ($self, $params) = @_; + $self->client->search($params); +}; + +sub setup_indexing { + my ($self, $bucket) = @_; + $self->client->setup_indexing($bucket); +}; + +1; + +=head1 SYNOPSIS + + my $client = Net::Riak->new(...); + my $bucket = $client->bucket('foo'); + + # retrieve an existing object + my $obj1 = $bucket->get('foo'); + + # create/store a new object + my $obj2 = $bucket->new_object('foo2', {...}); + $object->store; + + $bucket->delete_object($key, 3); # optional w val + +=head1 DESCRIPTION + +L<Net::Riak::Search> allows you to enable indexing documents for a given bucket and querying/searching the index. + +=head2 METHODS + +=head3 setup_indexing + + $client->setup_indexing('bucket_name'); + +Does the same as : + + curl -X PUT -H "content-type:application/json" http://localhost:8098/riak/bucket_name -d '{"props":{"precommit":[{"mod":"riak_search_kv_hook","fun":"precommit"}]}' + +but takes in account previouses precommits. + +=head3 search + + my $response = $client->search( + index => 'bucket_name', + q => 'field:value' + ); + # is the same as : + my $response = $client->search( + q => 'bucket_name.field:value' + ); + +Search the index + +=over 4 + +=item wt => 'XML|JSON' + +defines the response format (XML is the default value as for Solr/Lucene) + +=item q + +the query string + +=item index + +is the default index you want to query, if no index is provided you have to add it as a prefix of the fields in the query string + +=item rows + +is the number of documents you want to be returned in the response + +=back + +More parameters are available, just check at L<http://wiki.basho.com/Riak-Search---Querying.html#Querying-via-the-Solr-Interface> |