Skip to content

Commit

Permalink
Merge pull request #21 from reinferio/master
Browse files Browse the repository at this point in the history
Updated Riak backend to work with the latest version of the client
  • Loading branch information
didip committed Sep 21, 2013
2 parents 70eed74 + d4e3f4b commit f658f64
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
build
*.py?
*.egg-info
*.swp
10 changes: 5 additions & 5 deletions beaker_extensions/riak_.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,22 +19,22 @@ def __init__(self, namespace, url=None, data_dir=None, lock_dir=None, **params):
NoSqlManager.__init__(self, namespace, url=url, data_dir=data_dir, lock_dir=lock_dir, **params)

def open_connection(self, host, port):
self.db_conn = riak.RiakClient(host=host, port=int(port))
self.db_conn = riak.RiakClient(protocol='pbc', host=host, pb_port=int(port))
self.bucket = self.db_conn.bucket('beaker_cache')

def __contains__(self, key):
return self.bucket.get(self._format_key(key)).exists()
return self.bucket.get(self._format_key(key)).exists

def set_value(self, key, value):
val = self.bucket.get(self._format_key(key))
if not val.exists():
if not val.exists:
self.bucket.new(self._format_key(key), value).store()
else:
val.set_data(value)
val.data = value
val.store()

def __getitem__(self, key):
return self.bucket.get(self._format_key(key)).get_data()
return self.bucket.get(self._format_key(key)).data

def __delitem__(self, key):
self.bucket.get(self._format_key(key)).delete()
Expand Down

0 comments on commit f658f64

Please sign in to comment.