Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
ryan 2016-12-06 17:24:20 +01:00 committed by Sylvain Bellemare
parent ca94574150
commit f3cc167edb
1 changed files with 29 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
bigchaindb/backend/mongodb/query.py
View File

@ -1,18 +1,24 @@
"""Query implementation for MongoDB"""
from time import time from time import time
from pymongo import ReturnDocument from pymongo import ReturnDocument
from bigchaindb.backend import query from bigchaindb import backend
from bigchaindb.common.exceptions import CyclicBlockchainError from bigchaindb.common.exceptions import CyclicBlockchainError
from bigchaindb.backend.utils import module_dispatch_registrar
from bigchaindb.backend.mongodb.connection import MongoDBConnection from bigchaindb.backend.mongodb.connection import MongoDBConnection
@query.write_transaction.register(MongoDBConnection) register_query = module_dispatch_registrar(backend.query)
@register_query(MongoDBConnection)
def write_transaction(conn, signed_transaction): def write_transaction(conn, signed_transaction):
return conn.db['backlog'].insert_one(signed_transaction) return conn.db['backlog'].insert_one(signed_transaction)
@query.update_transaction.register(MongoDBConnection) @register_query(MongoDBConnection)
def update_transaction(conn, transaction_id, doc): def update_transaction(conn, transaction_id, doc):
return conn.db['backlog']\ return conn.db['backlog']\
.find_one_and_update({'id': transaction_id}, .find_one_and_update({'id': transaction_id},
@ -20,44 +26,44 @@ def update_transaction(conn, transaction_id, doc):
return_document=ReturnDocument.AFTER) return_document=ReturnDocument.AFTER)
@query.delete_transaction.register(MongoDBConnection) @register_query(MongoDBConnection)
def delete_transaction(conn, *transaction_id): def delete_transaction(conn, *transaction_id):
return conn.db['backlog'].delete_many({'id': {'$in': transaction_id}}) return conn.db['backlog'].delete_many({'id': {'$in': transaction_id}})
@query.get_stale_transactions.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_stale_transactions(conn, reassign_delay): def get_stale_transactions(conn, reassign_delay):
return conn.db['backlog']\ return conn.db['backlog']\
.find({'assignment_timestamp': {'$lt': time() - reassign_delay}}) .find({'assignment_timestamp': {'$lt': time() - reassign_delay}})
@query.get_transaction_from_block.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_transaction_from_block(conn, block_id, tx_id): def get_transaction_from_block(conn, block_id, tx_id):
# this is definitely wrong, but it's something like this # this is definitely wrong, but it's something like this
return conn.db['bigchain'].find_one({'id': block_id, return conn.db['bigchain'].find_one({'id': block_id,
'block.transactions.id': tx_id}) 'block.transactions.id': tx_id})
@query.get_transaction_from_backlog.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_transaction_from_backlog(conn, transaction_id): def get_transaction_from_backlog(conn, transaction_id):
return conn.db['backlog'].find_one({'id': transaction_id}) return conn.db['backlog'].find_one({'id': transaction_id})
@query.get_blocks_status_from_transaction.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_blocks_status_from_transaction(conn, transaction_id): def get_blocks_status_from_transaction(conn, transaction_id):
return conn.db['bigchain']\ return conn.db['bigchain']\
.find({'block.transactions.id': transaction_id}, .find({'block.transactions.id': transaction_id},
projection=['id', 'block.voters']) projection=['id', 'block.voters'])
@query.get_txids_by_asset_id.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_txids_by_asset_id(conn, asset_id): def get_txids_by_asset_id(conn, asset_id):
return conn.db['bigchain']\ return conn.db['bigchain']\
.find({'block.transactions.asset.id': asset_id}, .find({'block.transactions.asset.id': asset_id},
projection=['id']) projection=['id'])
@query.get_asset_by_id.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_asset_by_id(conn, asset_id): def get_asset_by_id(conn, asset_id):
return conn.db['bigchain']\ return conn.db['bigchain']\
.find_one({'block.transactions.asset.id': asset_id, .find_one({'block.transactions.asset.id': asset_id,
@ -65,7 +71,7 @@ def get_asset_by_id(conn, asset_id):
projection=['block.transactions.asset']) projection=['block.transactions.asset'])
@query.get_spent.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_spent(conn, transaction_id, condition_id): def get_spent(conn, transaction_id, condition_id):
return conn.db['bigchain']\ return conn.db['bigchain']\
.find_one({'block.transactions.fulfillments.input.txid': .find_one({'block.transactions.fulfillments.input.txid':
@ -74,64 +80,64 @@ def get_spent(conn, transaction_id, condition_id):
condition_id}) condition_id})
@query.get_owned_ids.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_owned_ids(conn, owner): def get_owned_ids(conn, owner):
return conn.db['bigchain']\ return conn.db['bigchain']\
.find({'block.transactions.transaction.conditions.owners_after': .find({'block.transactions.transaction.conditions.owners_after':
owner}) owner})
@query.get_votes_by_block_id.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_votes_by_block_id(conn, block_id): def get_votes_by_block_id(conn, block_id):
return conn.db['votes']\ return conn.db['votes']\
.find({'vote.voting_for_block': block_id}) .find({'vote.voting_for_block': block_id})
@query.get_votes_by_block_id_and_voter.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_votes_block_id_and_voter(conn, block_id, node_pubkey): def get_votes_block_id_and_voter(conn, block_id, node_pubkey):
return conn.db['votes']\ return conn.db['votes']\
.find({'vote.voting_for_block': block_id, .find({'vote.voting_for_block': block_id,
'node_pubkey': node_pubkey}) 'node_pubkey': node_pubkey})
@query.write_block.register(MongoDBConnection) @register_query(MongoDBConnection)
def write_block(conn, block): def write_block(conn, block):
return conn.db['bigchain'].insert_one(block.to_dict()) return conn.db['bigchain'].insert_one(block.to_dict())
@query.get_block.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_block(conn, block_id): def get_block(conn, block_id):
return conn.db['bigchain'].find_one({'id': block_id}) return conn.db['bigchain'].find_one({'id': block_id})
@query.has_transaction.register(MongoDBConnection) @register_query(MongoDBConnection)
def has_transaction(conn, transaction_id): def has_transaction(conn, transaction_id):
return bool(conn.db['bigchain'] return bool(conn.db['bigchain']
.find_one({'block.transactions.id': transaction_id})) .find_one({'block.transactions.id': transaction_id}))
@query.count_blocks.register(MongoDBConnection) @register_query(MongoDBConnection)
def count_blocks(conn): def count_blocks(conn):
return conn.db['bigchain'].count() return conn.db['bigchain'].count()
@query.count_backlog(MongoDBConnection) @register_query(MongoDBConnection)
def count_backlog(conn): def count_backlog(conn):
return conn.db['backlog'].count() return conn.db['backlog'].count()
@query.write_vote.register(MongoDBConnection) @register_query(MongoDBConnection)
def write_vote(conn, vote): def write_vote(conn, vote):
return conn.db['votes'].insert_one(vote) return conn.db['votes'].insert_one(vote)
@query.get_genesis_block.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_genesis_block(conn): def get_genesis_block(conn):
return conn.db['bigchain'].find_one({'block.transactions.0.operation' == return conn.db['bigchain'].find_one({'block.transactions.0.operation' ==
'GENESIS'}) 'GENESIS'})
@query.get_last_voted_block.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_last_voted_block(conn, node_pubkey): def get_last_voted_block(conn, node_pubkey):
last_voted = conn.db['votes']\ last_voted = conn.db['votes']\
.find({'node_pubkey': node_pubkey}, .find({'node_pubkey': node_pubkey},
@ -158,6 +164,6 @@ def get_last_voted_block(conn, node_pubkey):
return get_block(conn, last_block_id) return get_block(conn, last_block_id)
@query.get_unvoted_blocks.register(MongoDBConnection) @register_query(MongoDBConnection)
def get_unvoted_blocks(conn, node_pubkey): def get_unvoted_blocks(conn, node_pubkey):
pass pass