mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-06-28 00:27:45 +02:00
e4e528e5c4
Solution: Allow nodes to add, update, or remove validators at runtime using a new command. Implements BEP3.
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
from functools import singledispatch
|
|
|
|
import rethinkdb as r
|
|
|
|
from bigchaindb.backend.mongodb.connection import MongoDBConnection
|
|
from bigchaindb.backend.rethinkdb.connection import RethinkDBConnection
|
|
from bigchaindb.backend.localmongodb.connection import LocalMongoDBConnection
|
|
|
|
|
|
@singledispatch
|
|
def list_dbs(connection):
|
|
raise NotImplementedError
|
|
|
|
|
|
@list_dbs.register(RethinkDBConnection)
|
|
def list_rethink_dbs(connection):
|
|
return connection.run(r.db_list())
|
|
|
|
|
|
@list_dbs.register(MongoDBConnection)
|
|
def list_mongo_dbs(connection):
|
|
raise NotImplementedError
|
|
|
|
|
|
@singledispatch
|
|
def flush_db(connection, dbname):
|
|
raise NotImplementedError
|
|
|
|
|
|
@flush_db.register(RethinkDBConnection)
|
|
def flush_rethink_db(connection, dbname):
|
|
try:
|
|
connection.run(r.db(dbname).table('bigchain').delete())
|
|
connection.run(r.db(dbname).table('backlog').delete())
|
|
connection.run(r.db(dbname).table('votes').delete())
|
|
connection.run(r.db(dbname).table('assets').delete())
|
|
connection.run(r.db(dbname).table('metadata').delete())
|
|
except r.ReqlOpFailedError:
|
|
pass
|
|
|
|
|
|
@flush_db.register(MongoDBConnection)
|
|
def flush_mongo_db(connection, dbname):
|
|
connection.conn[dbname].bigchain.delete_many({})
|
|
connection.conn[dbname].backlog.delete_many({})
|
|
connection.conn[dbname].votes.delete_many({})
|
|
connection.conn[dbname].assets.delete_many({})
|
|
connection.conn[dbname].metadata.delete_many({})
|
|
connection.conn[dbname].utxos.delete_many({})
|
|
|
|
|
|
@flush_db.register(LocalMongoDBConnection)
|
|
def flush_localmongo_db(connection, dbname):
|
|
connection.conn[dbname].bigchain.delete_many({})
|
|
connection.conn[dbname].blocks.delete_many({})
|
|
connection.conn[dbname].transactions.delete_many({})
|
|
connection.conn[dbname].assets.delete_many({})
|
|
connection.conn[dbname].metadata.delete_many({})
|
|
connection.conn[dbname].utxos.delete_many({})
|
|
connection.conn[dbname].validators.delete_many({})
|
|
|
|
|
|
@singledispatch
|
|
def update_table_config(connection, table, **kwrgas):
|
|
raise NotImplementedError
|
|
|
|
|
|
@update_table_config.register(RethinkDBConnection)
|
|
def update_table_config(connection, table, **kwargs):
|
|
return connection.run(r.table(table).config().update(dict(**kwargs)))
|