Skip to content

Commit

Permalink
updated riak extension to 1) not conflict with riak package, 2) save …
Browse files Browse the repository at this point in the history
…each session as its own key (using the same _format_key logic as Redis)
  • Loading branch information
andrewzeneski committed Feb 26, 2013
1 parent d0006ca commit 210ab1f
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 5 deletions.
16 changes: 12 additions & 4 deletions beaker_extensions/riak.py → beaker_extensions/riak_.py
Expand Up @@ -23,16 +23,24 @@ def open_connection(self, host, port):
self.bucket = self.db_conn.bucket('beaker_cache')

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

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

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

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

def _format_key(self, key):
return 'beaker:%s:%s' % (self.namespace, key.replace(' ', '\302\267'))

def do_remove(self):
raise Exception("Unimplemented")
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Expand Up @@ -25,7 +25,7 @@
[beaker.backends]
redis = beaker_extensions.redis_:RedisManager
tyrant = beaker_extensions.tyrant_:TokyoTyrantManager
riak = beaker_extensions.riak:RiakManager
riak = beaker_extensions.riak_:RiakManager
dynomite = beaker_extensions.dynomite_:DynomiteManager
ringo = beaker_extensions.ringo:RingoManager
cassandra = beaker_extensions.cassandra:CassandraManager
Expand Down

0 comments on commit 210ab1f

Please sign in to comment.