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

wait till timeout added to rethinkdb connection

Browse Source
This commit is contained in:
tdsgit 2017-04-21 22:55:03 +05:30
parent cc66d5aaa5
commit 5edac3f4ee
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
bigchaindb/backend/rethinkdb/connection.py
View File

@ -1,4 +1,6 @@
import rethinkdb as r
import time
import bigchaindb
from bigchaindb.backend.connection import Connection
from bigchaindb.backend.exceptions import ConnectionError, OperationError
@ -38,8 +40,16 @@ class RethinkDBConnection(Connection):
:exc:`rethinkdb.ReqlDriverError`: After
:attr:`~.RethinkDBConnection.max_tries`.
"""
try:
return r.connect(host=self.host, port=self.port, db=self.dbname)
except r.ReqlDriverError as exc:
raise ConnectionError from exc
connected=False
dbconf = bigchaindb.config['database']
timeout = dbconf['connection_timeout']
end_time = time.time()*1000 + timeout
while not connected:
try:
rconn=r.connect(host=self.host, port=self.port, db=self.dbname)
connected=True
except r.ReqlDriverError as exc:
if time.time()*1000 > end_time:
raise ConnectionError from exc
pass
return rconn