def post_shutdown(self): lwc = LedgerWebClient(self.Url) msg = shutdown_message.ShutdownMessage({}) msg.SenderID = self.AdminNode.Address msg.sign_from_node(self.AdminNode) try: lwc.post_message(msg) except MessageException as me: print me
def post_shutdown(self): lwc = LedgerWebClient(self.url) msg = shutdown_message.ShutdownMessage({}) msg.SenderID = self._admin_node.Address msg.sign_from_node(self._admin_node) try: lwc.post_message(msg) except MessageException as me: print me
class ClientController(cmd.Cmd): _TxnVerbMap = {'=': 'set', '+=': 'inc', '-=': 'dec'} pformat = 'client> ' def __init__(self, baseurl, keystring=None): cmd.Cmd.__init__(self) self.prompt = 'client> ' self.CurrentState = {} self.LedgerWebClient = LedgerWebClient(baseurl) signingkey = generate_signing_key( wifstr=keystring) if keystring else generate_signing_key() identifier = generate_identifier(signingkey) self.LocalNode = Node(identifier=identifier, signingkey=signingkey, name="txnclient") def postcmd(self, flag, line): return flag def sign_and_post(self, msg): msg.SenderID = self.LocalNode.Identifier msg.sign_from_node(self.LocalNode) try: result = self.LedgerWebClient.post_message(msg) if result: pretty_print_dict(result) except MessageException as me: print me # ================================================================= # COMMANDS # ================================================================= def do_set(self, args): """ set -- Command to set properties of the interpreter set url --url <url> set nodeid --name <name> --keyfile <file> """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand url|nodeid' return try: if pargs[0] == 'url': parser = argparse.ArgumentParser() parser.add_argument('--url', help='url used to connect to a validator', required=True) options = parser.parse_args(pargs) self.BaseURL = options.url print "server URL set to {0}".format(self.BaseURL) return elif pargs[0] == 'nodeid': pargs = args.split() parser = argparse.ArgumentParser() parser.add_argument('--name', help='name to use for the client', default='txnclient') parser.add_argument('--keyfile', help='name of the file that contains ' 'the wif format private key') options = parser.parse_args(pargs[1:]) addr = (socket.gethostbyname("localhost"), 0) name = options.name if options.keyfile: signingkey = generate_signing_key( wifstr=read_key_file(options.keyfile)) else: signingkey = generate_signing_key() identifier = generate_identifier(signingkey) self.LocalNode = Node(address=addr, identifier=identifier, signingkey=signingkey, name=name) print "local id set to {0}".format(self.LocalNode) return else: print "unknown subcommand; {0}".format(pargs[0]) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return def do_state(self, args): """ state -- Command to manipulate the current ledger state state fetch --store <store> state keys state value --path <path> """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: fetch|keys|value' return try: if pargs[0] == 'fetch': parser = argparse.ArgumentParser() parser.add_argument('--store', choices=TransactionTypes.keys(), default='endpoint') options = parser.parse_args(pargs[1:]) self.CurrentState = self.LedgerWebClient.get_store( TransactionTypes.get(options.store), key='*') elif pargs[0] == 'keys': try: print self.CurrentState.keys() except: print '[]' elif pargs[0] == 'value': parser = argparse.ArgumentParser() parser.add_argument('--path', required=True) options = parser.parse_args(pargs[1:]) pathargs = options.path.split('.') value = self.CurrentState while pathargs: value = value.get(pathargs.pop(0)) print value except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return def do_txn(self, args): """ txn -- Command to create IntegerKey transactions txn <expr> [ && <expr> ]* """ txn = integer_key.IntegerKeyTransaction() # pylint: disable=line-too-long pattern = re.compile(r"^\s*(?P<name>[a-zA-Z0-9]+)\s*(?P<verb>[+-]?=)\s*(?P<value>[0-9]+)\s*$") # noqa expressions = args.split('&&') for expression in expressions: match = pattern.match(expression) if not match: print 'unable to parse the transaction; {0}'.format(expression) return update = integer_key.Update() update.Verb = self._TxnVerbMap[match.group('verb')] update.Name = match.group('name') update.Value = long(match.group('value')) txn.Updates.append(update) txn.sign_from_node(self.LocalNode) msg = integer_key.IntegerKeyTransactionMessage() msg.Transaction = txn self.sign_and_post(msg) def do_nodestats(self, args): """ nodestats -- Command to send nodestats messages to validator pool nodestats reset --metrics [<metric>]+ nodestats dump --metrics [<metric>]+ --domains [<domain>]+ """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: reset|dump' return try: if pargs[0] == 'reset': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.ResetStatsMessage( {'MetricList': options.metrics})) return elif pargs[0] == 'dump': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') parser.add_argument('--domains', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.DumpNodeStatsMessage( {'MetricList': options.metrics, 'DomainList': options.domains})) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return print 'unknown nodestats command {0}'.format(args) def do_peerstats(self, args): """ peerstats -- Command to send peerstats messages to validator pool peerstats reset --metrics [<metric>]+ peerstats dump --metrics [<metric>]+ --peers [<nodeid>]+ """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: reset|dump' return try: if pargs[0] == 'reset': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.ResetPeerStatsMessage( {'MetricList': options.metrics})) return elif pargs[0] == 'dump': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') parser.add_argument('--peers', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.DumpPeerStatsMessage( {'MetricList': options.metrics, 'PeerIDList': options.peers})) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return print 'unknown peerstats command {0}'.format(args) def do_ping(self, args): """ ping -- Command to send a ping message to validator pool """ self.sign_and_post(gossip_debug.PingMessage({})) def do_sping(self, args): """ sping -- Command to send a special ping message to validator pool """ parser = argparse.ArgumentParser() parser.add_argument( '--address', default=self.LocalNode.Identifier) parser.add_argument('--count', default=2, type=int) options = parser.parse_args(args.split()) msg = SpecialPingMessage() msg.Address = options.address msg.Count = options.count self.sign_and_post(msg) def do_dumpquorum(self, args): """ dumpquorum -- Command to request quorum consensus node to dump quorum list """ self.sign_and_post(quorum_debug.DumpQuorumMessage({})) def do_dumpcnxs(self, args): """ dumpcnxs -- Command to send to the validator pool a request to dump current connection information """ self.sign_and_post(gossip_debug.DumpConnectionsMessage({})) def do_dumpblks(self, args): """ dumpblks -- Command to send dump blocks request to the validator pool dumpblks --blocks <count> """ parser = argparse.ArgumentParser() parser.add_argument('--blocks', default=0, type=int) options = parser.parse_args(args.split()) self.sign_and_post(gossip_debug.DumpJournalBlocksMessage( {'Count': options.blocks})) def do_val(self, args): """ val -- Command to send a dump value request to the validator pool val <name> [--type <transaction type>] """ pargs = args.split() parser = argparse.ArgumentParser() parser.add_argument('--type', help='Transaction family', default='/IntegerKeyTransaction') options = parser.parse_args(pargs[1:]) tinfo = {'Name': pargs[0], 'TransactionType': options.type} self.sign_and_post(journal_debug.DumpJournalValueMessage(tinfo)) def do_shutdown(self, args): """ shutdown -- Command to send a shutdown message to the validator pool """ self.sign_and_post(shutdown_message.ShutdownMessage({})) return True def do_exit(self, args): """exit Shutdown the simulator and exit the command loop """ return True def do_eof(self, args): return True
class ClientController(cmd.Cmd): _TxnVerbMap = {'=': 'set', '+=': 'inc', '-=': 'dec'} pformat = 'client> ' def __init__(self, baseurl, keystring=None): cmd.Cmd.__init__(self) self.prompt = 'client> ' self.CurrentState = {} self.LedgerWebClient = LedgerWebClient(baseurl) signingkey = generate_signing_key( wifstr=keystring) if keystring else generate_signing_key() identifier = generate_identifier(signingkey) self.LocalNode = Node(identifier=identifier, signingkey=signingkey, name="txnclient") def postcmd(self, flag, line): return flag def sign_and_post(self, msg): msg.SenderID = self.LocalNode.Identifier msg.sign_from_node(self.LocalNode) try: result = self.LedgerWebClient.post_message(msg) if result: pretty_print_dict(result) except MessageException as me: print me # ================================================================= # COMMANDS # ================================================================= def do_set(self, args): """ set -- Command to set properties of the interpreter set url --url <url> set nodeid --name <name> --keyfile <file> """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand url|nodeid' return try: if pargs[0] == 'url': parser = argparse.ArgumentParser() parser.add_argument('--url', help='url used to connect to a validator', required=True) options = parser.parse_args(pargs) self.BaseURL = options.url print "server URL set to {0}".format(self.BaseURL) return elif pargs[0] == 'nodeid': pargs = args.split() parser = argparse.ArgumentParser() parser.add_argument('--name', help='name to use for the client', default='txnclient') parser.add_argument('--keyfile', help='name of the file that contains ' 'the wif format private key') options = parser.parse_args(pargs[1:]) addr = (socket.gethostbyname("localhost"), 0) name = options.name if options.keyfile: signingkey = generate_signing_key( wifstr=read_key_file(options.keyfile)) else: signingkey = generate_signing_key() identifier = generate_identifier(signingkey) self.LocalNode = Node(address=addr, identifier=identifier, signingkey=signingkey, name=name) print "local id set to {0}".format(self.LocalNode) return else: print "unknown subcommand; {0}".format(pargs[0]) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return def do_state(self, args): """ state -- Command to manipulate the current ledger state state fetch --store <store> state keys state value --path <path> """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: fetch|keys|value' return try: if pargs[0] == 'fetch': parser = argparse.ArgumentParser() parser.add_argument('--store', choices=TransactionTypes.keys(), default='endpoint') options = parser.parse_args(pargs[1:]) self.CurrentState = self.LedgerWebClient.get_store( TransactionTypes.get(options.store), key='*') elif pargs[0] == 'keys': try: print self.CurrentState.keys() except: print '[]' elif pargs[0] == 'value': parser = argparse.ArgumentParser() parser.add_argument('--path', required=True) options = parser.parse_args(pargs[1:]) pathargs = options.path.split('.') value = self.CurrentState while pathargs: value = value.get(pathargs.pop(0)) print value except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return def do_txn(self, args): """ txn -- Command to create IntegerKey transactions txn <expr> [ && <expr> ]* """ txn = integer_key.IntegerKeyTransaction() # pylint: disable=line-too-long pattern = re.compile(r"^\s*(?P<name>[a-zA-Z0-9]+)\s*(?P<verb>[+-]?=)\s*(?P<value>[0-9]+)\s*$") # noqa expressions = args.split('&&') for expression in expressions: match = pattern.match(expression) if not match: print 'unable to parse the transaction; {0}'.format(expression) return update = integer_key.Update() update.Verb = self._TxnVerbMap[match.group('verb')] update.Name = match.group('name') update.Value = long(match.group('value')) txn.Updates.append(update) txn.sign_from_node(self.LocalNode) msg = integer_key.IntegerKeyTransactionMessage() msg.Transaction = txn self.sign_and_post(msg) def do_nodestats(self, args): """ nodestats -- Command to send nodestats messages to validator pool nodestats reset --metrics [<metric>]+ nodestats dump --metrics [<metric>]+ --domains [<domain>]+ """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: reset|dump' return try: if pargs[0] == 'reset': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.ResetStatsMessage( {'MetricList': options.metrics})) return elif pargs[0] == 'dump': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') parser.add_argument('--domains', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.DumpNodeStatsMessage( {'MetricList': options.metrics, 'DomainList': options.domains})) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return print 'unknown nodestats command {0}'.format(args) def do_peerstats(self, args): """ peerstats -- Command to send peerstats messages to validator pool peerstats reset --metrics [<metric>]+ peerstats dump --metrics [<metric>]+ --peers [<nodeid>]+ """ pargs = args.split() if len(pargs) == 0: print 'missing subcommand: reset|dump' return try: if pargs[0] == 'reset': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.ResetPeerStatsMessage( {'MetricList': options.metrics})) return elif pargs[0] == 'dump': parser = argparse.ArgumentParser() parser.add_argument('--metrics', default=[], nargs='+') parser.add_argument('--peers', default=[], nargs='+') options = parser.parse_args(pargs[1:]) self.sign_and_post(gossip_debug.DumpPeerStatsMessage( {'MetricList': options.metrics, 'PeerIDList': options.peers})) return except Exception as e: print 'an error occured processing {0}: {1}'.format(args, str(e)) return print 'unknown peerstats command {0}'.format(args) def do_ping(self, args): """ ping -- Command to send a ping message to validator pool """ self.sign_and_post(gossip_debug.PingMessage({})) def do_sping(self, args): """ sping -- Command to send a special ping message to validator pool """ parser = argparse.ArgumentParser() parser.add_argument( '--address', default=self.LocalNode.Identifier) parser.add_argument('--count', default=2, type=int) options = parser.parse_args(args.split()) msg = SpecialPingMessage() msg.Address = options.address msg.Count = options.count self.sign_and_post(msg) def do_dumpquorum(self, args): """ dumpquorum -- Command to request quorum consensus node to dump quorum list """ self.sign_and_post(quorum_debug.DumpQuorumMessage({})) def do_dumpcnxs(self, args): """ dumpcnxs -- Command to send to the validator pool a request to dump current connection information """ self.sign_and_post(gossip_debug.DumpConnectionsMessage({})) def do_dumpblks(self, args): """ dumpblks -- Command to send dump blocks request to the validator pool dumpblks --blocks <count> """ parser = argparse.ArgumentParser() parser.add_argument('--blocks', default=0, type=int) options = parser.parse_args(args.split()) self.sign_and_post(gossip_debug.DumpJournalBlocksMessage( {'Count': options.blocks})) def do_val(self, args): """ val -- Command to send a dump value request to the validator pool val <name> [--type <transaction type>] """ pargs = args.split() parser = argparse.ArgumentParser() parser.add_argument('--type', help='Transaction family', default='/IntegerKeyTransaction') options = parser.parse_args(pargs[1:]) tinfo = {'Name': pargs[0], 'TransactionType': options.type} self.sign_and_post(journal_debug.DumpJournalValueMessage(tinfo)) def do_shutdown(self, args): """ shutdown -- Command to send a shutdown message to the validator pool """ self.sign_and_post(shutdown_message.ShutdownMessage({})) return True def do_exit(self, args): """exit Shutdown the simulator and exit the command loop """ return True def do_eof(self, args): return True