def main(): options, args = option_parser.parse_args() # Set logging level if options.log_level.upper() == 'SILENT': log.silence() else: log.set_level(options.log_level.upper()) # Build address split_addr = options.address.split(':') if len(split_addr) == 1: host, port = split_addr[0], 3000 elif len(split_addr) == 2: host, port = split_addr[0], int(split_addr[1]) else: print 'Invalid address specified; defaulting to 127.0.0.1:3000' host, port = '127.0.0.1', 3000 # Build message unit message = options.message.decode('utf-8') unit = [message] * options.unit_size message_count = options.num_messages # Configure mode (sync/async). mode = 'async' if options.synchronous: mode = 'sync' # Configure method (native/http) method = 'native' if options.http: method = 'http' client = QueueClient(mode=mode, method=method, host=host, port=port) start_time = time.time() while message_count > 0: client.push_many(*unit) assert (client.pull_many(options.unit_size, timeout=0) == unit) message_count -= options.unit_size end_time = time.time() if method == 'native': client.close() time_taken = end_time - start_time print '%d messages transferred in chunks of %s' % (options.num_messages, options.unit_size) print 'Time taken: %0.4f seconds' % (time_taken,) print 'Average speed: %0.4f messages/second' % ( options.num_messages / time_taken,)
def main(): options, args = option_parser.parse_args() # Set logging level if options.log_level.upper() == 'SILENT': log.silence() else: log.set_level(options.log_level.upper()) # Build address split_addr = options.address.split(':') if len(split_addr) == 1: host, port = split_addr[0], 3000 elif len(split_addr) == 2: host, port = split_addr[0], int(split_addr[1]) else: print 'Invalid address specified; defaulting to 127.0.0.1:3000' host, port = '127.0.0.1', 3000 # Build message unit message = options.message.decode('utf-8') unit = [message] * options.unit_size message_count = options.num_messages # Configure mode (sync/async). mode = 'async' if options.synchronous: mode = 'sync' # Configure method (native/http) method = 'native' if options.http: method = 'http' client = QueueClient(mode=mode, method=method, host=host, port=port) start_time = time.time() while message_count > 0: client.push_many(*unit) assert (client.pull_many(options.unit_size, timeout=0) == unit) message_count -= options.unit_size end_time = time.time() if method == 'native': client.close() time_taken = end_time - start_time print '%d messages transferred in chunks of %s' % (options.num_messages, options.unit_size) print 'Time taken: %0.4f seconds' % (time_taken, ) print 'Average speed: %0.4f messages/second' % (options.num_messages / time_taken, )
def _main(): options, args = OPTION_PARSER.parse_args() # Handle log level. log_level = options.log_level if log_level.upper() == 'SILENT': # Completely disables logging output. log.silence() elif log_level.upper() not in log.LOG_LEVELS: log.ROOT_LOGGER.warning( 'Invalid log level supplied, defaulting to INFO') log.ROOT_LOGGER.setLevel(log.INFO) else: log.ROOT_LOGGER.setLevel(getattr(log, log_level.upper())) # Instantiate and start server. server = NativeQueueServer(max_size=options.max_size) server.serve(interface=options.interface, port=options.port)
def _main(): options, args = OPTION_PARSER.parse_args() # Handle log level. log_level = options.log_level if log_level.upper() == 'SILENT': # Completely disables logging output. log.silence() elif log_level.upper() not in log.LOG_LEVELS: log.ROOT_LOGGER.warning( 'Invalid log level supplied, defaulting to INFO') log.ROOT_LOGGER.setLevel(log.INFO) else: log.ROOT_LOGGER.setLevel(getattr(log, log_level.upper())) # Instantiate and start server. server = HTTPQueueServer() server.serve(interface=options.interface, port=options.port, max_size=options.max_size)