fsblog-20170219-rep_client.py

import zmq
import numpy as np
import json
import time

# ZeroMQ のバックグラウンド・スレッドのコンテキスト
context = zmq.Context()

# このクライアントは、ポート5556に接続します（バックグラウンドにて）
socket = context.socket(zmq.REP)
socket.connect("tcp://localhost:5556")


# シリアライズされた Numpy Array を受信
[dtype, shape, data] = socket.recv_multipart()

# Numpy Array にデシリアライズ
M = np.frombuffer(data, dtype=dtype.decode('utf-8'))
M.shape = json.loads(shape.decode('utf-8'))

print("Request:\n{} received".format(M))

# なにか計算してみる； 各要素に＋１
M = M + 1

# ちょっと待ってみる
time.sleep(1)

# 計算結果を返信
socket.send_multipart([dtype, shape, M.tostring('C')])

print("Reply:\n{} sent".format(M))