Function read_and_send_socket

examples/multiprocess_access_benchmarks.py:182–208

(
    send_type, array_size, array_bytes, address_func, socket_family
)

def read_and_send_socket(
    send_type, array_size, array_bytes, address_func, socket_family
):
    address = address_func()
    # start the receiving process and pause to allow it to start up
    result_recv, result_send = multiprocessing.Pipe(False)
    recv_process = SocketReceive(
        socket_family, address, result_send, array_bytes
    )
    recv_process.start()
    time.sleep(0.15)
    with tb.open_file("test.h5", "r") as fobj:
        array = fobj.get_node("/", "test")
        start_timestamp = clock()
        # connect to the receiving process' socket
        sock = socket.socket(socket_family, socket.SOCK_STREAM)
        sock.connect(address)
        # read the array from the PyTables file and send its
        # data buffer to the receiving process
        output = array.read(0, array_size, 1)
        sock.send(output.data)
        assert np.all(output + 1 == 2)
        # receive the timestamps from the other process
        recv_timestamp, finish_timestamp = result_recv.recv()
    sock.close()
    print_results(send_type, start_timestamp, recv_timestamp, finish_timestamp)
    recv_process.join()


def print_results(