queue.sqlite -> queue.sa
This commit is contained in:
parent
e706fabb26
commit
08cb3cd324
@ -188,7 +188,6 @@ class Worker(threading.Thread):
|
|||||||
time.sleep(f * Worker.interval)
|
time.sleep(f * Worker.interval)
|
||||||
|
|
||||||
|
|
||||||
from .sqlite import SQLite3Queue
|
from .sa import SAQueue
|
||||||
|
|
||||||
__all__ = ["Full", "Empty", "Retry", "Timeout", "Message", "Queue",
|
__all__ = ["Full", "Empty", "Retry", "Timeout", "Message", "Queue", "SAQueue"]
|
||||||
"SQLite3Queue"]
|
|
||||||
|
@ -16,12 +16,12 @@ pickle = lambda val: json.dumps(val).encode("utf-8")
|
|||||||
unpickle = lambda val: json.loads(val.decode("utf-8"))
|
unpickle = lambda val: json.loads(val.decode("utf-8"))
|
||||||
|
|
||||||
|
|
||||||
class SQLite3Queue(Queue):
|
class SAQueue(Queue):
|
||||||
"""Implements a shared queue using SQLAlchemy Core
|
"""Implements a shared queue using SQLAlchemy Core
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, db, maxlen=-1, timeout=2**10):
|
def __init__(self, db, maxlen=-1, timeout=2**10):
|
||||||
super(SQLite3Queue, self).__init__(maxlen, timeout)
|
super(SAQueue, self).__init__(maxlen, timeout)
|
||||||
self.metadata = MetaData()
|
self.metadata = MetaData()
|
||||||
self.engine = create_engine(db)
|
self.engine = create_engine(db)
|
||||||
self.queue = Table("queue", self.metadata,
|
self.queue = Table("queue", self.metadata,
|
@ -6,7 +6,6 @@ import unittest
|
|||||||
|
|
||||||
from isso.compat import text_type as str
|
from isso.compat import text_type as str
|
||||||
|
|
||||||
from isso.db import Adapter
|
|
||||||
from isso.cache import Cache, SACache
|
from isso.cache import Cache, SACache
|
||||||
|
|
||||||
ns = "test"
|
ns = "test"
|
||||||
@ -56,4 +55,4 @@ class TestCache(unittest.TestCase):
|
|||||||
class TestSQLite3Cache(TestCache):
|
class TestSQLite3Cache(TestCache):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.cache = SACache(Adapter("sqlite:///:memory:"), threshold=8)
|
self.cache = SACache("sqlite:///:memory:", threshold=8)
|
||||||
|
@ -5,8 +5,7 @@ from __future__ import unicode_literals
|
|||||||
import unittest
|
import unittest
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
from isso.db import SQLite3
|
from isso.queue import Message, Queue, Full, Empty, Timeout, SAQueue
|
||||||
from isso.queue import Message, Queue, Full, Empty, Timeout, SQLite3Queue
|
|
||||||
|
|
||||||
|
|
||||||
class TestMessage(unittest.TestCase):
|
class TestMessage(unittest.TestCase):
|
||||||
@ -100,4 +99,4 @@ class TestQueue(unittest.TestCase):
|
|||||||
class TestSQLite3Queue(TestQueue):
|
class TestSQLite3Queue(TestQueue):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.cls = lambda *x, **z: SQLite3Queue("sqlite:///:memory:", *x, **z)
|
self.cls = lambda *x, **z: SAQueue("sqlite:///:memory:", *x, **z)
|
||||||
|
Loading…
Reference in New Issue
Block a user