Use new connection class in pipeline

This commit is contained in:
vrde 2016-09-07 00:54:19 +02:00
parent 7097efaa33
commit 52243e1271
No known key found for this signature in database
GPG Key ID: 6581C7C39B3D397D
2 changed files with 16 additions and 13 deletions

View File

@ -53,10 +53,10 @@ class Block:
Returns:
The transaction.
"""
r.table('backlog')\
.get(tx['id'])\
.delete(durability='hard')\
.run(self.bigchain.conn)
self.bigchain.connection.run(
r.table('backlog')
.get(tx['id'])
.delete(durability='hard'))
return tx
@ -118,12 +118,14 @@ def initial():
b = Bigchain()
rs = r.table('backlog')\
.between([b.me, r.minval],
[b.me, r.maxval],
index='assignee__transaction_timestamp')\
.order_by(index=r.asc('assignee__transaction_timestamp'))\
.run(b.conn)
rs = b.connection.run(
r.table('backlog')
.between(
[b.me, r.minval],
[b.me, r.maxval],
index='assignee__transaction_timestamp')
.order_by(index=r.asc('assignee__transaction_timestamp')))
return rs

View File

@ -25,9 +25,10 @@ class Election:
"""
Checks if block has enough invalid votes to make a decision
"""
next_block = r.table('bigchain')\
.get(next_vote['vote']['voting_for_block'])\
.run(self.bigchain.conn)
next_block = self.bigchain.connection.run(
r.table('bigchain')
.get(next_vote['vote']['voting_for_block']))
if self.bigchain.block_election_status(next_block) == self.bigchain.BLOCK_INVALID:
return next_block