Move objects_lock into the Inventory object

This commit is contained in:
Lee Miller 2024-09-09 00:11:58 +03:00
parent a87bb5ab3b
commit eff18c27bc
Signed by: lee.miller
GPG Key ID: 4F97A5EA88F4AB63
2 changed files with 5 additions and 4 deletions

View File

@ -63,4 +63,3 @@ outgoing_connections = 8
connection_limit = 250
objects = {}
objects_lock = threading.Lock()

View File

@ -2,6 +2,7 @@
import os
import sqlite3
import threading
import time
from . import shared, structure
@ -12,6 +13,7 @@ sqlite3.threadsafety = 3
class Inventory():
"""sqlite inventory"""
def __init__(self):
self._lock = threading.Lock()
self._db = sqlite3.connect(
os.path.join(shared.data_directory, 'objects.dat'),
check_same_thread=False
@ -51,7 +53,7 @@ class Inventory():
tag=tag, vector=vector)
def cleanup(self):
with shared.objects_lock:
with self._lock:
self._db.execute(
'DELETE FROM objects WHERE expires < ?',
(int(time.time()) - 3 * 3600,)
@ -119,13 +121,13 @@ class Inventory():
return item
def __delitem__(self, vector):
with shared.objects_lock: # KeyError
with self._lock: # KeyError
self._db.execute('DELETE FROM objects WHERE vector = ?', (vector,))
self._db.commit()
self.rowid = len(self)
def __setitem__(self, vector, obj):
with shared.objects_lock:
with self._lock:
cur = self._db.execute(
'INSERT INTO objects VALUES (?,?,?,?,?,?,?,?)', (
vector, obj.expires_time, obj.object_type, obj.version,