Riak Ruby Client (riak-client)
riak-client
is a rich Ruby client/toolkit for Riak, Basho's
distributed database that contains a basic wrapper around typical
operations, including bucket manipulation, object CRUD, link-walking,
and map-reduce.
Exhaustive documentation is available at
http://basho.github.io/riak-ruby-client/ , and API documentation can be
read at http://www.rubydoc.info/gems/riak-client/frames .
Build Status

Dependencies
Ruby 1.9.3, 2.0, 2.1, and 2.2 are supported. JRuby in 2.0 mode is also supported.
Travis CI is configured to run unit tests on 1.9.3, 2.X versions of Ruby and JRuby.
NOTE: official support for the Ruby 1.9 series ended February of 2015.
In JRuby 1.7.13, OCSP validation is absent, and CRL validation always
fails. This issue is being tracked and this document will be updated when
it is fixed. Additionally, client certificate authentication doesn't work in
JRuby. This issue is also being tracked, and this document will be updated
when it works.
riak-client
requires beefcake, cert_validator, i18n, innertube, and
multi_json.
Development dependencies are handled with bundler. Install bundler
(gem install bundler
) and run this command to get started:
$ bundle install
Run the RSpec suite using bundle exec
:
$ bundle exec rake
Basic Example
require 'riak'
client = Riak::Client.new
client = Riak::Client.new(authentication: {
ca_file: '/home/zedo/ca.crt',
user: 'zedo',
password: 'catnip',
client_ca: '/home/zedo/ca.crt',
cert: File.read '/home/zedo/zedo.crt',
key: OpenSSL::PKey::RSA.new(File.read '/home/zedo/zedo.key')
})
client = Riak::Client.new(:nodes => [
{:host => '10.0.0.1'},
{:host => '10.0.0.2', :pb_port => 1234},
{:host => '10.0.0.3', :pb_port => 5678}
])
bucket = client.bucket("doc")
object = bucket.get_or_new("index.html")
object.raw_data = "<html><body>Hello, world!</body></html>"
object.content_type = "text/html"
object.store
object.reload
object.reload :force => true
client['doc']['index.html']
new_one = Riak::RObject.new(bucket, "application.js")
new_one.content_type = "application/javascript"
new_one.raw_data = "alert('Hello, World!')"
new_one.store
Bucket Types
Riak 2 uses bucket types to enable groups of similar buckets to share
properties, configuration, and to namespace values within those buckets. Bucket
type support is integral to how CRDTs work.
In versions 2.2.0 and newer of this client, bucket types have a first-class
representation, and can be used to create BucketTyped::Bucket
objects that are
namespaced differently from regular Riak::Bucket
objects.
beverages = client.bucket_type 'beverages'
coffees = beverages.bucket 'coffees'
untyped_coffees = client.bucket 'coffees'
chapadao = coffees.new 'chapadao'
chapadao.data = "Chapadao de Ferro"
chapadao.store
untyped_coffees.get 'chapadao'
coffees.get 'chapadao'
chapadao.reload
untyped_coffees.delete 'chapadao'
chapadao.delete
coffees.delete 'chapadao'
Client 2.0 and 2.1 code that uses the type
argument to methods still works:
coffees = client.bucket 'coffees'
chapadao = coffees.new 'chapadao'
chapadao.data = "Chapadao de Ferro"
chapadao.store type: 'beverages'
coffees.get 'chapadao'
coffees.get 'chapadao', type: 'beverages'
chapadao.reload
chapadao.reload type: 'beverages'
chapadao.delete
coffees.delete 'chapadao'
chapadao.delete type: 'beverages'
coffees.delete 'chapadao', type: 'beverages'
Map-Reduce Example
results = Riak::MapReduce.new(client).
add("artists","Beatles").
link(:bucket => "albums").
map("function(v){ return [JSON.parse(v.values[0].data).title]; }", :keep => true).run
p results
Riak Search Examples
This client supports the new Riak Search 2 (codenamed "Yokozuna"). For more information about Riak Search, see the Riak documentation.
This documentation assumes there's a yokozuna
bucket type created and activated.
client = Riak::Client.new
bucket_type = client.bucket_type 'yokozuna'
bucket = bucket_type.bucket 'pizzas'
index = Riak::Search::Index.new client, 'pizzas'
index.exist?
index.create!
client.set_bucket_props bucket, {search_index: 'pizzas'}, 'yokozuna'
meat = bucket.new 'meat'
meat.data = {toppings_ss: %w{pepperoni ham sausage}}
meat.store type: 'yokozuna'
hawaiian = bucket.new 'hawaiian'
hawaiian.data = {toppings_ss: %w{ham pineapple}}
hawaiian.store type: 'yokozuna'
query = Riak::Search::Query.new client, index, 'toppings_ss:ham'
query.rows = 5
result = query.results
result.length
result.num_found
pizza_result = result.first
pizza_result.score
pizza_result.key
pizza = pizza_result.robject
Secondary Index Examples
Riak supports secondary indexes. Secondary indexing, or "2i," gives you the
ability to tag objects with multiple queryable values at write time, and then
query them later.
Tagging Objects
Objects are tagged with a hash kept behind the indexes
method. Secondary index
storage logic is in lib/riak/rcontent.rb
.
object = bucket.get_or_new 'cobb.salad'
object.indexes['ingredients_bin'] = %w{lettuce tomato bacon egg chives}
object.indexes['calories_int'] = [220]
object.store
Finding Objects
Secondary index queries return a list of keys exactly matching a scalar or
within a range.
bucket.get_index 'calories_int', 220
bucket.get_index 'calories_int', 100..300
bucket.get_index 'ingredients_bin', 'tomata'..'tomatz'
c = bucket.get_index 'ingredients_bin', 'lettuce', max_results: 5
c.length
c.continuation
c2 = bucket.get_index 'ingredients_bin', 'lettuce', max_results: 5, continuation: c.continuation
q = Riak::SecondaryIndex.new bucket, 'ingredients_bin', 'lettuce', max_results: 5
q.keys
q.values
q.has_next_page?
q2 = q.next_page
Riak 2 Data Types
Riak 2 features new distributed data structures: counters, sets, and maps
(containing counters, flags, maps, registers, and sets). These are implemented
by the Riak database as Convergent Replicated Data Types.
Riak data type support requires bucket types to be configured to support each
top-level data type. If you're just playing around, use the
Riak Ruby Vagrant setup to
get started with the appropriate configuration and bucket types quickly.
The examples below presume that the appropriate bucket types are named
counters
, maps
, and sets
; these bucket type names are the client's defaults.
Viewing and changing the defaults is easy:
Riak::Crdt::DEFAULT_BUCKET_TYPES[:set]
Riak::Crdt::DEFAULT_BUCKET_TYPES[:set] = "a_cooler_set"
The top-level CRDT types have both immediate and batch mode. If you're doing
multiple writes to a single top-level counter or set, or updating multiple map
entries, batch mode will make fewer round-trips to Riak.
Top-level CRDT types accept nil
as a key. This allows Riak to assign a random
key for them.
Deleting CRDTs requires you to use the key-value API for the time being.
brews = Riak::Crdt::Set.new bucket, 'brews'
brews.add 'espresso'
brews.add 'aeropress'
bucket.delete brews.key, type: brews.bucket_type
Counters
Riak 2 integer counters have one operation: increment by an integer.
counter = Riak::Crdt::Counter.new bucket, key
counter.value
counter.increment
counter.value
counter.increment 3
counter.value
counter.decrement
counter.value
Counter operations can be batched:
counter.batch do |c|
c.increment
c.increment 5
end
Maps
Riak 2 maps can contain counters, flags (booleans), registers (strings), sets, and
other maps.
Maps are similar but distinct from the standard Ruby Hash
. Entries are
segregated by both name and type, so you can have counters, registers, and sets inside a map that all have the same name.
map = Riak::Crdt::Map.new bucket, key
map.counters['potatoes'].value
map.sets['potatoes'].include? 'yukon gold'
map.sets['cacti'].value
map.sets['cacti'].remove 'prickly pear'
map.registers['favorite butterfly'] = 'the mighty monarch'
map.flags['pet cat'] = true
map.maps['atlantis'].registers['location']
map.counters.delete 'thermometers'
Maps are a prime candidate for batched operations:
map.batch do |m|
m.counters['hits'].increment
m.sets['followers'].add 'basho_elevator'
end
Frequently, you might want a map with a Riak-assigned name instead of one you
come up with yourself:
map = Riak::Crdt::Map.new bucket, nil
map.registers['coat_pattern'] = 'tabby'
map.key
Sets
Sets are an unordered collection of entries.
PROTIP: Ruby and Riak Ruby Client both have classes called Set
. Be careful
to refer to the Ruby version as ::Set
and the Riak client version as
Riak::Crdt::Set
.
set = Riak::Crdt::Set.new bucket, key
set.members
set.add "Newcastle"
set.remove "London"
set.include? "Leeds"
Sets support batched operations:
set.batch do |s|
s.add "York"
s.add "Aberdeen"
s.remove "Newcastle"
end
Client Implementation Notes
The client code for these types is in the Riak::Crdt
namespace, and mostly
in the lib/riak/crdt
directory.
Riak 1.4 Counters
For more information about 1.4-style counters in Riak, see the Basho documentation.
Counter records are automatically persisted on increment or decrement. The initial default value is 0.
bucket = client.bucket "counters"
bucket.allow_mult = true
counter = bucket.counter("counter-key-here")
counter.increment
=> nil
p counter.value
1
=> 1
counter.increment
persisted_counter = Riak::Counter.new(bucket, "counter-key-here")
p persisted_counter.value
2
=> 2
persisted_counter.increment(20)
p persisted_counter.value
22
=> 22
persisted_counter.decrement
persisted_counter.value
=> 21
persisted_counter.decrement(6)
persisted_counter.value
=> 15
persisted_counter.increment "nonsense"
ArgumentError: Counters can only be incremented or decremented by integers.
That's about it. PN Counters in Riak are distributed, so each node will receive the proper increment/decrement operation. Enjoy using them.
Instrumentation
The Riak client has built-in event hooks for all major over-the-wire operations including:
- riak.list_buckets
- riak.list_keys
- riak.set_bucket_props
- riak.get_bucket_props
- riak.clear_bucket_props
- riak.get_index
- riak.store_object
- riak.get_object
- riak.reload_object
- riak.delete_object
- riak.map_reduce
- riak.ping
Events are propogated using ActiveSupport::Notifications, provided by the Instrumentable gem.
Enabling
Instrumentation is opt-in. If instrumentable
is not available, instrumentation will not be available. To turn on instrumentation, simply require the instrumentable
gem in your app's Gemfile:
gem 'instrumentable', '~> 1.1.0'
Then, to subscribe to events:
ActiveSupport::Notifications.subscribe(/^riak\.*/) do |name, start, finish, id, payload|
name
start
finish
id
payload
end
The payload for each event contains the following keys:
:client_id
: The client_id of the Riak client
:_method_args
: The array of method arguments for the instrumented method. For instance, for riak.get_object
, this value would resemble [<Riak::Bucket ...>, 'key', {}]
Running Specs
We aim to have a comprehensive and fast set of tests, implemented using a modern,
well-supported version of RSpec. These tests include both unit specs for
individual classes, and integration specs that ensure the client works properly
with an actual Riak instance.
The Riak Ruby Vagrant virtual machine's Riak configuration is normally
used to test this client in development. Once it's up and running, configure
the Ruby test_client.yml
on the host machine to connect to pb_port: 17017
and test away.
Configuring the Riak node the tests connect to is done via the
spec/support/test_client.yml
file, which is loaded into a Ruby hash with
symbolized keys, and passed to Riak::Client.new
.
pb_port: 10017
Spec dependencies
Specs depend on the following Riak configurations:
- The LevelDB backend is necessary for testing secondary indexes.
- allow_mult is required for many features: conflict resolution, and legacy
counters among them.
- Riak Search 2.0 ("Yokozuna") must be configured for testing full-text
search.
The following bucket types are used during testing:
riak-admin bucket-type create counters '{"props":{"datatype":"counter", "allow_mult":true}}'
riak-admin bucket-type create other_counters '{"props":{"datatype":"counter", "allow_mult":true}}'
riak-admin bucket-type create maps '{"props":{"datatype":"map", "allow_mult":true}}'
riak-admin bucket-type create sets '{"props":{"datatype":"set", "allow_mult":true}}'
riak-admin bucket-type create yokozuna '{"props":{}}'
riak-admin bucket-type activate other_counters
riak-admin bucket-type activate counters
riak-admin bucket-type activate maps
riak-admin bucket-type activate sets
riak-admin bucket-type activate yokozuna
Client tests run both with and without security enabled, as we have to test
several positive and negative paths. The tests broadly depend on these users
and roles:
riak-admin security add-user user password=password
riak-admin security add-user certuser
riak-admin security add-source user 0.0.0.0/0 password
riak-admin security add-source certuser 0.0.0.0/0 certificate
riak-admin security grant riak_kv.get,riak_kv.put,riak_kv.delete,\
riak_kv.index,riak_kv.list_keys,riak_kv.list_buckets,\
riak_core.get_bucket,riak_core.set_bucket,\
riak_core.get_bucket_type,riak_core.set_bucket_type,\
search.admin,search.query,riak_kv.mapreduce on any to user
How to Contribute
-
Fork the project on Github. If you have already forked, use git pull --rebase
to reapply your changes on top of the mainline. Example:
$ git checkout master
$ git pull --rebase basho master
-
Copy spec/support/test_server.yml.example to spec/support/test_server.yml and change that file according to your local installation of riak.
-
Create a topic branch. If you've already created a topic branch, rebase it on top of changes from the mainline "master" branch. Examples:
-
Write an RSpec example or set of examples that demonstrate the necessity and validity of your changes. Patches without specs will most often be ignored. Just do it, you'll thank me later. Documentation patches need no specs, of course.
-
Make your feature addition or bug fix. Make your specs and stories pass (green).
-
Run the suite using multiruby or rvm to ensure cross-version compatibility.
-
Cleanup any trailing whitespace in your code (try @whitespace-mode@ in Emacs, or "Remove Trailing Spaces in Document" in the "Text" bundle in Textmate). You can use the clean_whitespace
Rake task if you like.
-
Commit, do not mess with Rakefile. If related to an existing issue in the tracker, include "Closes #X" in the commit message (where X is the issue number).
-
Send a pull request to the Basho repository.
License & Copyright
Copyright ©2010-2016 Sean Cribbs and Basho Technologies, Inc.
Licensed 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.
Auxillary Licenses
The included photo (spec/fixtures/cat.jpg) is Copyright ©2009 Sean Cribbs, and is licensed under the Creative Commons Attribution Non-Commercial 3.0 license.

gzip
encoding support is copied from the ActiveSupport project.
Contributors
Thank you to all of our contributors!