[python][zeromq] REP-REQのサンプル (REPがサーバ、REQがクライアント)

zmq_python_rep_example.py

import zmq


def main():
    ctx = zmq.Context()
    socket = ctx.socket(zmq.REP)
    socket.bind('tcp://*:5560')  #: https://stackoverflow.com/questions/6024003/why-doesnt-zeromq-work-on-localhost

    while True:
        m = socket.recv()
        if m == b'quit':
            print('shutdown server....')
            socket.send(b'shutdown')
            break
        print(f'{m}')
        socket.send(m.upper())


if __name__ == '__main__':
    main()

zmq_python_req_example.py

import argparse
import zmq


def main():
    ctx = zmq.Context()
    socket = ctx.socket(zmq.REQ)
    socket.connect('tcp://localhost:5560')

    socket.send(b'quit' if args.quit else b'hello')
    m = socket.recv()
    print(f'{m}')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--quit', action='store_true', default=False)
    args = parser.parse_args()
    main()