fsblog-20170219-req_server.py

import zmq
import numpy as np
import json

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

# このサーバは、ポート5556で待ちます
socket = context.socket(zmq.REQ)
socket.bind("tcp://*:5556")

# 送信するデータの生成：２行３列の整数配列
M = np.asarray([[0,1,2],[2,1,0]])


# M をシリアライズ
dtype = str(M.dtype).encode('utf-8')
shape = json.dumps(M.shape).encode('utf-8')
data = M.tostring('C')

# マルチパートメッセージとして送信
socket.send_multipart([dtype, shape, data])

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


# 計算結果を受け取る
[dtype2, shape2, data2] = socket.recv_multipart()

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

print("Reply:\n{} received".format(N))