Added connection dict object, where is stored all connections

This commit is contained in:
andrei 2022-02-11 16:31:56 +02:00
parent 94d82a3e68
commit 38db28b92a

View File

@ -4,11 +4,26 @@ from planetmint.backend.tarantool.utils import run
class TarantoolDB: class TarantoolDB:
def __init__(self, host, port, username, password): def __init__(self, host: str, port: int, username: str, password: str):
self.conn = tarantool.connect(host=host, port=port, user=username, password=password) self.db_connect = tarantool.connect(host=host, port=port, user=username, password=password)
self._spaces = {
"abci_chains": self.db_connect.space("abci_chains"),
"assets": self.db_connect.space("assets"),
"blocks": {"blocks": self.db_connect.space("blocks"), "blocks_tx": self.db_connect.space("blocks_tx")},
"elections": self.db_connect.space("elections"),
"meta_data": self.db_connect.space("meta_data"),
"pre_commits": self.db_connect.space("pre_commits"),
"validators": self.db_connect.space("validators"),
"transactions": {
"transactions": self.db_connect.space("transactions"),
"inputs": self.db_connect.space("inputs"),
"outputs": self.db_connect.space("outputs"),
"keys": self.db_connect.space("keys")
}
}
def connect_to_sapce(self, spacename): def get_space(self, spacename: str):
self.conn.space(spacename) return self._spaces[spacename]
def init_tarantool(): def init_tarantool():