summaryrefslogtreecommitdiff
path: root/lib/Net/Riak
diff options
context:
space:
mode:
Diffstat (limited to 'lib/Net/Riak')
-rw-r--r--lib/Net/Riak/Client.pm5
-rw-r--r--lib/Net/Riak/Role/REST.pm1
-rw-r--r--lib/Net/Riak/Role/REST/Search.pm74
-rw-r--r--lib/Net/Riak/Search.pm18
4 files changed, 98 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..642964b
--- /dev/null
+++ b/lib/Net/Riak/Role/REST/Search.pm
@@ -0,0 +1,74 @@
+package Net::Riak::Role::REST::Search;
+use Moose::Role;
+use JSON;
+
+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..8cf42b7
--- /dev/null
+++ b/lib/Net/Riak/Search.pm
@@ -0,0 +1,18 @@
+package Net::Riak::Search;
+
+use Moose;
+
+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;