LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

include write lock in try/finally

Browse source
This commit is contained in:
Jack Robison 2020-05-01 11:40:52 -04:00
parent b6f6994db4
commit 0a9d4de126
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lbry/wallet/database.py
View file

@ -137,15 +137,15 @@ class AIOSQLite:
async def run(self, fun, *args, **kwargs): async def run(self, fun, *args, **kwargs):
self.writers += 1 self.writers += 1
self.read_ready.clear() self.read_ready.clear()
async with self.write_lock: try:
try: async with self.write_lock:
return await asyncio.get_event_loop().run_in_executor( return await asyncio.get_event_loop().run_in_executor(
self.writer_executor, lambda: self.__run_transaction(fun, *args, **kwargs) self.writer_executor, lambda: self.__run_transaction(fun, *args, **kwargs)
) )
finally: finally:
self.writers -= 1 self.writers -= 1
if not self.writers: if not self.writers:
self.read_ready.set() self.read_ready.set()
def __run_transaction(self, fun: Callable[[sqlite3.Connection, Any, Any], Any], *args, **kwargs): def __run_transaction(self, fun: Callable[[sqlite3.Connection, Any, Any], Any], *args, **kwargs):
self.writer_connection.execute('begin') self.writer_connection.execute('begin')