Example #1
0
def main():
    """ main loop """
    parser = options.ConnectorOptions()
    opts, _ = parser.parse_args()

    ecode = 0
    threads = []
    results = []
    errors = []
    stats = None
    opts.conn_reconnect = "false"

    if opts.log_stats == 'connector':
        stats = []

    if opts.log_lib is not None:
        utils.set_up_client_logging(opts.log_lib)

    try:
        # main loop
        for i in range(opts.count):
            connector = threading.Thread(target=run_connectors,
                                         args=(opts, results, errors, stats))
            threads.append(connector)
            connector.start()

        for i in range(len(threads)):
            threads[i].join()

    except Exception:  # pylint: disable=broad-except
        ecode = 1

    result = [sum(i_res) for i_res in zip(*results)]
    ecode = (len(result) == 3 and result[1] > 0) or ecode
    stdout = ' '.join(str(val) for val in result)

    for err in errors:
        utils.dump_error(err)

    if opts.log_stats == 'connector':
        for stat in stats:
            print("STATS", stat)
    print(stdout)
    sys.exit(ecode)
Example #2
0
def main():
    """ main loop """
    ecode = 0
    parser = options.ReceiverOptions()
    opts, _ = parser.parse_args()

    if opts.log_lib is not None:
        utils.set_up_client_logging(opts.log_lib)

    try:
        # main loop
        if opts.tx_size or opts.tx_endloop_action is not None:
            container = proton.reactor.Container(TxRecv(opts))
        else:
            container = proton.reactor.Container(Recv(opts))
        super(proton.reactor.Container, container).global_handler.add(
            coreclient.ErrorsHandler(opts.conn_reconnect))
        container.run()
    except Exception:  # pylint: disable=broad-except
        raise
    sys.exit(ecode)