Compare commits
16 Commits
dev-0.1.2
...
dev-0.1.14
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
263d4df300
|
||
|
|
029deead75
|
||
|
|
da9fb5925d
|
||
|
|
cbf00281c6
|
||
|
|
01ad409077
|
||
|
|
3a8ec01588
|
||
|
|
b63793fd9b
|
||
|
|
84b8eb10e6
|
||
|
|
532ff230b4
|
||
|
|
f7c09acfe2
|
||
|
|
04d9901f0d
|
||
|
|
b8c2b1b86a
|
||
|
|
c94b291d39
|
||
|
|
6c360ca2e5
|
||
|
|
ff74679de8
|
||
|
|
94bd5c8cdf
|
26
CHANGELOG
26
CHANGELOG
@@ -1,3 +1,29 @@
|
|||||||
|
- 0.1.14
|
||||||
|
* Upgrade shep to handle exception in filestore list
|
||||||
|
- 0.1.13
|
||||||
|
* Remove sync on each get
|
||||||
|
* Upgrade shep to guarantee atomic state lock state
|
||||||
|
- 0.1.12
|
||||||
|
* Raise correct exception from index store exists check
|
||||||
|
- 0.1.11
|
||||||
|
* Allow for sync skip in store instantiation
|
||||||
|
- 0.1.10
|
||||||
|
* Improve logging
|
||||||
|
- 0.1.9
|
||||||
|
* Upgrade deps
|
||||||
|
- 0.1.8
|
||||||
|
* Upgrade deps
|
||||||
|
- 0.1.7
|
||||||
|
* Improve logging
|
||||||
|
- 0.1.6
|
||||||
|
* Sort upcoming queue item chronologically
|
||||||
|
* Add unit testing for upcoming query method
|
||||||
|
- 0.1.5
|
||||||
|
* Add reserved state check method
|
||||||
|
- 0.1.4
|
||||||
|
* Dependency cleanups
|
||||||
|
- 0.1.3
|
||||||
|
* Add CLI args and config handling, settings object
|
||||||
- 0.1.2
|
- 0.1.2
|
||||||
* Add CLI inspection tools
|
* Add CLI inspection tools
|
||||||
- 0.1.1
|
- 0.1.1
|
||||||
|
|||||||
11
chainqueue/cli/__init__.py
Normal file
11
chainqueue/cli/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
# standard imports
|
||||||
|
import os
|
||||||
|
|
||||||
|
# local imports
|
||||||
|
from .arg import process_flags
|
||||||
|
from .config import process_config
|
||||||
|
|
||||||
|
|
||||||
|
__script_dir = os.path.dirname(os.path.realpath(__file__))
|
||||||
|
data_dir = os.path.join(os.path.dirname(__script_dir), 'data')
|
||||||
|
config_dir = os.path.join(data_dir, 'config')
|
||||||
2
chainqueue/cli/arg.py
Normal file
2
chainqueue/cli/arg.py
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
def process_flags(argparser, flags):
|
||||||
|
argparser.add_argument('--backend', type=str, help='Backend to use for state store')
|
||||||
8
chainqueue/cli/config.py
Normal file
8
chainqueue/cli/config.py
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
def process_config(config, args, flags):
|
||||||
|
args_override = {}
|
||||||
|
|
||||||
|
args_override['QUEUE_BACKEND'] = getattr(args, 'backend')
|
||||||
|
|
||||||
|
config.dict_override(args_override, 'local cli args')
|
||||||
|
|
||||||
|
return config
|
||||||
@@ -1,9 +1,2 @@
|
|||||||
[database]
|
[queue]
|
||||||
name =
|
backend = mem
|
||||||
engine =
|
|
||||||
driver =
|
|
||||||
host =
|
|
||||||
port =
|
|
||||||
user =
|
|
||||||
password =
|
|
||||||
debug = 0
|
|
||||||
|
|||||||
@@ -134,6 +134,10 @@ class QueueEntry:
|
|||||||
self.store.cache.set_block(self.tx_hash, block, tx)
|
self.store.cache.set_block(self.tx_hash, block, tx)
|
||||||
|
|
||||||
|
|
||||||
|
def test(self, state):
|
||||||
|
return self.__match_state(state)
|
||||||
|
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
v = self.store.get(self.tx_hash)
|
v = self.store.get(self.tx_hash)
|
||||||
n = self.store.state(v[0])
|
n = self.store.state(v[0])
|
||||||
|
|||||||
@@ -24,12 +24,6 @@ class CacheIntegrityError(ChainQueueException):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class BackendIntegrityError(ChainQueueException):
|
|
||||||
"""Raised when queue backend has invalid state
|
|
||||||
"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class DuplicateTxError(ChainQueueException):
|
class DuplicateTxError(ChainQueueException):
|
||||||
"""Backend already knows transaction
|
"""Backend already knows transaction
|
||||||
"""
|
"""
|
||||||
|
|||||||
8
chainqueue/settings.py
Normal file
8
chainqueue/settings.py
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
# external imports
|
||||||
|
from chainlib.settings import ChainSettings
|
||||||
|
|
||||||
|
|
||||||
|
class ChainqueueSettings(ChainSettings):
|
||||||
|
|
||||||
|
def process_queue_backend(self, config):
|
||||||
|
self.o['QUEUE_BACKEND'] = config.get('QUEUE_BACKEND')
|
||||||
@@ -106,10 +106,10 @@ class Verify:
|
|||||||
|
|
||||||
class Status(shep.persist.PersistedState):
|
class Status(shep.persist.PersistedState):
|
||||||
|
|
||||||
def __init__(self, store_factory):
|
def __init__(self, store_factory, allow_invalid=False, event_callback=None):
|
||||||
verify = Verify().verify
|
verify = Verify().verify
|
||||||
self.set_default_state('PENDING')
|
self.set_default_state('PENDING')
|
||||||
super(Status, self).__init__(store_factory, 12, verifier=verify)
|
super(Status, self).__init__(store_factory, 12, verifier=verify, check_alias=not allow_invalid, event_callback=event_callback)
|
||||||
self.add('QUEUED')
|
self.add('QUEUED')
|
||||||
self.add('RESERVED')
|
self.add('RESERVED')
|
||||||
self.add('IN_NETWORK')
|
self.add('IN_NETWORK')
|
||||||
|
|||||||
@@ -2,12 +2,15 @@
|
|||||||
import re
|
import re
|
||||||
import datetime
|
import datetime
|
||||||
import logging
|
import logging
|
||||||
|
import time
|
||||||
|
|
||||||
# local imports
|
# local imports
|
||||||
from chainqueue.cache import CacheTx
|
from chainqueue.cache import CacheTx
|
||||||
from chainqueue.entry import QueueEntry
|
from chainqueue.entry import QueueEntry
|
||||||
from chainqueue.error import (
|
from chainqueue.error import NotLocalTxError
|
||||||
NotLocalTxError,
|
from chainqueue.enum import (
|
||||||
|
StatusBits,
|
||||||
|
all_errors,
|
||||||
)
|
)
|
||||||
|
|
||||||
logg = logging.getLogger(__name__)
|
logg = logging.getLogger(__name__)
|
||||||
@@ -21,11 +24,12 @@ def from_key(k):
|
|||||||
(ts_str, seq_str, tx_hash) = k.split('_')
|
(ts_str, seq_str, tx_hash) = k.split('_')
|
||||||
return (float(ts_str), int(seq_str), tx_hash, )
|
return (float(ts_str), int(seq_str), tx_hash, )
|
||||||
|
|
||||||
|
all_local_errors = all_errors() - StatusBits.NETWORK_ERROR
|
||||||
|
|
||||||
re_u = r'^[^_][_A-Z]+$'
|
re_u = r'^[^_][_A-Z]+$'
|
||||||
class Store:
|
class Store:
|
||||||
|
|
||||||
def __init__(self, chain_spec, state_store, index_store, counter, cache=None):
|
def __init__(self, chain_spec, state_store, index_store, counter, cache=None, sync=True):
|
||||||
self.chain_spec = chain_spec
|
self.chain_spec = chain_spec
|
||||||
self.cache = cache
|
self.cache = cache
|
||||||
self.state_store = state_store
|
self.state_store = state_store
|
||||||
@@ -43,9 +47,21 @@ class Store:
|
|||||||
'unset',
|
'unset',
|
||||||
'name',
|
'name',
|
||||||
'modified',
|
'modified',
|
||||||
|
'purge',
|
||||||
]:
|
]:
|
||||||
setattr(self, v, getattr(self.state_store, v))
|
setattr(self, v, getattr(self.state_store, v))
|
||||||
self.state_store.sync()
|
|
||||||
|
if not sync:
|
||||||
|
return
|
||||||
|
|
||||||
|
sync_err = None
|
||||||
|
try:
|
||||||
|
self.state_store.sync()
|
||||||
|
except Exception as e:
|
||||||
|
sync_err = e
|
||||||
|
|
||||||
|
if sync_err != None:
|
||||||
|
raise FileNotFoundError(sync_err)
|
||||||
|
|
||||||
|
|
||||||
def put(self, v, cache_adapter=CacheTx):
|
def put(self, v, cache_adapter=CacheTx):
|
||||||
@@ -63,30 +79,41 @@ class Store:
|
|||||||
|
|
||||||
|
|
||||||
def get(self, k):
|
def get(self, k):
|
||||||
|
v = None
|
||||||
|
s = self.index_store.get(k)
|
||||||
|
err = None
|
||||||
try:
|
try:
|
||||||
s = self.index_store.get(k)
|
v = self.state_store.get(s)
|
||||||
except FileNotFoundError:
|
except FileNotFoundError as e:
|
||||||
raise NotLocalTxError(k)
|
err = e
|
||||||
self.state_store.sync()
|
if v == None:
|
||||||
v = self.state_store.get(s)
|
raise NotLocalTxError('could not find tx {}: {}'.format(k, err))
|
||||||
return (s, v,)
|
return (s, v,)
|
||||||
|
|
||||||
|
|
||||||
def by_state(self, state=0, limit=4096, strict=False, threshold=None):
|
def by_state(self, state=0, not_state=0, limit=4096, strict=False, threshold=None):
|
||||||
hashes = []
|
hashes = []
|
||||||
i = 0
|
i = 0
|
||||||
|
|
||||||
refs_state = self.state_store.list(state)
|
refs_state = self.state_store.list(state)
|
||||||
|
refs_state.sort()
|
||||||
|
|
||||||
for ref in refs_state:
|
for ref in refs_state:
|
||||||
v = from_key(ref)
|
v = from_key(ref)
|
||||||
hsh = v[2]
|
hsh = v[2]
|
||||||
|
|
||||||
|
item_state = self.state_store.state(ref)
|
||||||
|
|
||||||
if strict:
|
if strict:
|
||||||
item_state = self.state_store.state(ref)
|
|
||||||
if item_state & state != item_state:
|
if item_state & state != item_state:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if item_state & not_state > 0:
|
||||||
|
continue
|
||||||
|
|
||||||
|
item_state_str = self.state_store.name(item_state)
|
||||||
|
logg.info('state {} {} ({})'.format(ref, item_state_str, item_state))
|
||||||
|
|
||||||
if threshold != None:
|
if threshold != None:
|
||||||
v = self.state_store.modified(ref)
|
v = self.state_store.modified(ref)
|
||||||
if v > threshold:
|
if v > threshold:
|
||||||
@@ -94,7 +121,9 @@ class Store:
|
|||||||
|
|
||||||
hashes.append(hsh)
|
hashes.append(hsh)
|
||||||
|
|
||||||
|
i += 1
|
||||||
|
if limit > 0 and i == limit:
|
||||||
|
break
|
||||||
|
|
||||||
hashes.sort()
|
hashes.sort()
|
||||||
return hashes
|
return hashes
|
||||||
@@ -108,6 +137,17 @@ class Store:
|
|||||||
return self.by_state(state=self.DEFERRED, limit=limit, threshold=threshold)
|
return self.by_state(state=self.DEFERRED, limit=limit, threshold=threshold)
|
||||||
|
|
||||||
|
|
||||||
|
def failed(self, limit=4096):
|
||||||
|
#return self.by_state(state=all_local_errors, limit=limit)
|
||||||
|
r = []
|
||||||
|
r += self.by_state(state=self.LOCAL_ERROR, limit=limit)
|
||||||
|
r += self.by_state(state=self.NODE_ERROR, limit=limit)
|
||||||
|
r.sort()
|
||||||
|
if len(r) > limit:
|
||||||
|
r = r[:limit]
|
||||||
|
return r
|
||||||
|
|
||||||
|
|
||||||
def pending(self, limit=4096):
|
def pending(self, limit=4096):
|
||||||
return self.by_state(state=0, limit=limit, strict=True)
|
return self.by_state(state=0, limit=limit, strict=True)
|
||||||
|
|
||||||
@@ -130,6 +170,7 @@ class Store:
|
|||||||
def fail(self, k):
|
def fail(self, k):
|
||||||
entry = QueueEntry(self, k)
|
entry = QueueEntry(self, k)
|
||||||
entry.load()
|
entry.load()
|
||||||
|
logg.debug('fail {}'.format(k))
|
||||||
entry.sendfail()
|
entry.sendfail()
|
||||||
|
|
||||||
|
|
||||||
@@ -153,3 +194,13 @@ class Store:
|
|||||||
entry = QueueEntry(self, k)
|
entry = QueueEntry(self, k)
|
||||||
entry.load()
|
entry.load()
|
||||||
entry.sent()
|
entry.sent()
|
||||||
|
|
||||||
|
|
||||||
|
def is_reserved(self, k):
|
||||||
|
entry = QueueEntry(self, k)
|
||||||
|
entry.load()
|
||||||
|
return entry.test(self.RESERVED)
|
||||||
|
|
||||||
|
|
||||||
|
def sync(self):
|
||||||
|
self.state_store.sync()
|
||||||
|
|||||||
@@ -6,7 +6,10 @@ import logging
|
|||||||
from leveldir.hex import HexDir
|
from leveldir.hex import HexDir
|
||||||
|
|
||||||
# local imports
|
# local imports
|
||||||
from chainqueue.error import DuplicateTxError
|
from chainqueue.error import (
|
||||||
|
DuplicateTxError,
|
||||||
|
NotLocalTxError,
|
||||||
|
)
|
||||||
|
|
||||||
logg = logging.getLogger(__name__)
|
logg = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -22,7 +25,7 @@ class IndexStore(HexDir):
|
|||||||
existing = None
|
existing = None
|
||||||
try:
|
try:
|
||||||
existing = self.get(k)
|
existing = self.get(k)
|
||||||
except FileNotFoundError:
|
except NotLocalTxError:
|
||||||
pass
|
pass
|
||||||
return existing != None
|
return existing != None
|
||||||
|
|
||||||
@@ -37,7 +40,14 @@ class IndexStore(HexDir):
|
|||||||
|
|
||||||
def get(self, k):
|
def get(self, k):
|
||||||
fp = self.store.to_filepath(k)
|
fp = self.store.to_filepath(k)
|
||||||
f = open(fp, 'rb')
|
f = None
|
||||||
|
err = None
|
||||||
|
try:
|
||||||
|
f = open(fp, 'rb')
|
||||||
|
except FileNotFoundError as e:
|
||||||
|
err = e
|
||||||
|
if err != None:
|
||||||
|
raise NotLocalTxError(err)
|
||||||
v = f.read()
|
v = f.read()
|
||||||
f.close()
|
f.close()
|
||||||
return v.decode('utf-8')
|
return v.decode('utf-8')
|
||||||
@@ -64,7 +74,7 @@ class CounterStore:
|
|||||||
|
|
||||||
v = f.read(8)
|
v = f.read(8)
|
||||||
self.count = int.from_bytes(v, byteorder='big')
|
self.count = int.from_bytes(v, byteorder='big')
|
||||||
logg.info('counter starts at {}'.format(self.count))
|
logg.debug('counter starts at {}'.format(self.count))
|
||||||
|
|
||||||
f.seek(0)
|
f.seek(0)
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
pysha3==1.0.2
|
#pysha3==1.0.2
|
||||||
hexathon~=0.1.5
|
hexathon~=0.1.5
|
||||||
leveldir~=0.3.0
|
leveldir~=0.3.0
|
||||||
alembic==1.4.2
|
#alembic==1.4.2
|
||||||
SQLAlchemy==1.3.20
|
#SQLAlchemy==1.3.20
|
||||||
confini~=0.6.0
|
confini~=0.6.0
|
||||||
pyxdg~=0.27
|
#pyxdg~=0.27
|
||||||
chainlib>=0.1.0b1,<=0.1.0
|
chainlib~=0.1.1
|
||||||
shep>=0.1.1rc1,<=0.3.0
|
shep~=0.2.7
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[metadata]
|
[metadata]
|
||||||
name = chainqueue
|
name = chainqueue
|
||||||
version = 0.1.2
|
version = 0.1.14
|
||||||
description = Generic blockchain transaction queue control
|
description = Generic blockchain transaction queue control
|
||||||
author = Louis Holbrook
|
author = Louis Holbrook
|
||||||
author_email = dev@holbrook.no
|
author_email = dev@holbrook.no
|
||||||
@@ -25,7 +25,7 @@ licence_files =
|
|||||||
LICENSE.txt
|
LICENSE.txt
|
||||||
|
|
||||||
[options]
|
[options]
|
||||||
python_requires = >= 3.6
|
python_requires = >= 3.7
|
||||||
include_package_data = True
|
include_package_data = True
|
||||||
packages =
|
packages =
|
||||||
chainqueue
|
chainqueue
|
||||||
@@ -33,6 +33,7 @@ packages =
|
|||||||
chainqueue.unittest
|
chainqueue.unittest
|
||||||
chainqueue.store
|
chainqueue.store
|
||||||
chainqueue.runnable
|
chainqueue.runnable
|
||||||
|
chainqueue.cli
|
||||||
|
|
||||||
#[options.entry_points]
|
#[options.entry_points]
|
||||||
#console_scripts =
|
#console_scripts =
|
||||||
|
|||||||
@@ -6,14 +6,23 @@ import logging
|
|||||||
import shutil
|
import shutil
|
||||||
|
|
||||||
# external imports
|
# external imports
|
||||||
|
from chainlib.chain import ChainSpec
|
||||||
|
from shep.store.noop import NoopStoreFactory
|
||||||
|
|
||||||
# local imports
|
# local imports
|
||||||
from chainqueue.store.fs import (
|
from chainqueue.store.fs import (
|
||||||
IndexStore,
|
IndexStore,
|
||||||
CounterStore,
|
CounterStore,
|
||||||
)
|
)
|
||||||
|
from chainqueue.store.base import Store
|
||||||
from chainqueue.error import DuplicateTxError
|
from chainqueue.error import DuplicateTxError
|
||||||
|
from chainqueue.state import Status
|
||||||
|
|
||||||
|
# tests imports
|
||||||
|
from tests.common import (
|
||||||
|
MockTokenCache,
|
||||||
|
MockCacheTokenTx,
|
||||||
|
)
|
||||||
|
|
||||||
logging.basicConfig(level=logging.DEBUG)
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
logg = logging.getLogger()
|
logg = logging.getLogger()
|
||||||
@@ -58,5 +67,38 @@ class TestStoreImplementations(unittest.TestCase):
|
|||||||
store.put(hx, data)
|
store.put(hx, data)
|
||||||
|
|
||||||
|
|
||||||
|
def test_upcoming_limit(self):
|
||||||
|
index_store = IndexStore(self.path)
|
||||||
|
counter_store = CounterStore(self.path)
|
||||||
|
chain_spec = ChainSpec('foo', 'bar', 42, 'baz')
|
||||||
|
factory = NoopStoreFactory().add
|
||||||
|
state_store = Status(factory)
|
||||||
|
cache_store = MockTokenCache()
|
||||||
|
queue_store = Store(chain_spec, state_store, index_store, counter_store, cache=cache_store)
|
||||||
|
|
||||||
|
txs = []
|
||||||
|
for i in range(3):
|
||||||
|
tx_src = os.urandom(128).hex()
|
||||||
|
tx = queue_store.put(tx_src, cache_adapter=MockCacheTokenTx)
|
||||||
|
txs.append(tx)
|
||||||
|
|
||||||
|
r = queue_store.upcoming(limit=3)
|
||||||
|
self.assertEqual(len(r), 0)
|
||||||
|
|
||||||
|
for tx in txs:
|
||||||
|
queue_store.enqueue(tx[1])
|
||||||
|
|
||||||
|
r = queue_store.upcoming(limit=3)
|
||||||
|
self.assertEqual(len(r), 3)
|
||||||
|
|
||||||
|
queue_store.send_start(txs[0][1])
|
||||||
|
r = queue_store.upcoming(limit=3)
|
||||||
|
self.assertEqual(len(r), 2)
|
||||||
|
|
||||||
|
queue_store.send_end(txs[0][1])
|
||||||
|
r = queue_store.upcoming(limit=3)
|
||||||
|
self.assertEqual(len(r), 2)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|||||||
Reference in New Issue
Block a user