simon987/hexlib
1
0
Fork 0
mirror of https://github.com/simon987/hexlib.git synced 2025-04-19 17:56:43 +00:00
Code Issues Projects Releases Wiki Activity

msgpack for queue

Browse Source
This commit is contained in:
simon987 2021-01-20 20:30:57 -05:00
parent c4fca1b754
commit a2cfab55bc
3 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
hexlib/db.py
View File

@ -37,10 +37,12 @@ class VolatileQueue:
self.key = key self.key = key
def put(self, item): def put(self, item):
self.rdb.sadd(self.key, item) self.rdb.sadd(self.key, umsgpack.dumps(item))
def get(self): def get(self):
return self.rdb.spop(self.key) v = self.rdb.spop(self.key)
if v:
return umsgpack.loads(v)
class VolatileBooleanState: class VolatileBooleanState:

2
setup.py
View File

@ -2,7 +2,7 @@ from setuptools import setup
setup( setup(
name="hexlib", name="hexlib",
version="1.25", version="1.26",
description="Misc utility methods", description="Misc utility methods",
author="simon987", author="simon987",
author_email="me@simon987.net", author_email="me@simon987.net",

18
test/test_VolatileState.py
View File

@ -78,4 +78,20 @@ class TestVolatileQueue(TestCase):
s.put("test") s.put("test")
item = s.get() item = s.get()
self.assertTrue(item == b"test") self.assertTrue(item == "test")
def test_dict(self):
s = VolatileQueue(key="test5")
s.put({"a": 1})
item = s.get()
self.assertTrue(item["a"] == 1)
def test_int(self):
s = VolatileQueue(key="test5")
s.put(123)
item = s.get()
self.assertTrue(item == 123)