在 python 中使用 zmq 的基礎知識，創建簡單的鍵/值存儲

介紹

讓我們看一個簡單的鍵/值儲存的範例，例如記憶體快取。 它的設計很簡單——資料以哈希圖結構儲存在記憶體中。 它們是透過 TCP 套接字存取的。 在 Python 中，hashmap 是一個常規字典。 對於訪問，我們將使用 Zeromq。

調整

要在 debian/ubuntu 中安裝此包，只需在控制台中輸入
sudo apt-get install libzmq-dev
sudo pip install zmq

代碼

讓我們編寫一個類別來與我們的伺服器一起使用：
使用的zmq套接字類型是REQ(REQuest, request)，我們發送請求並等待回應。
為了透過網路儲存和傳輸任何類型的數據，我們使用標準 pickle 模組。 工作「協定」是三個值的元組：（指令、鍵、資料）

import zmq
import pickle

class SuperCacher:
    def __init__(self):
        context = zmq.Context()
        self.socket = context.socket(zmq.REQ)
        self.socket.connect('tcp://127.0.0.1:43000')

    def get(self, key):
        self.socket.send(pickle.dumps(('get', key, None)))
        return pickle.loads(self.socket.recv())

    def set(self, key, data):
        self.socket.send(pickle.dumps(('set', key, data)))
        return self.socket.recv() == b'ok'

使用

快取 = SuperCacher()
快取.set('鍵', '值')
快取.get('key')

作為家庭作業，透過在建立類別的實例時新增指定位址/連接埠的功能來改善實作。

現在讓我們編寫伺服器本身。
這次使用了套接字 REP(REPly, response) - 我們正在等待請求，發送回應。 我們解析請求並在寫入時使用“ok”進行回應，在讀取時使用 data/None 進行回應。

import pickle
import json
import zmq

def run_daemon():
    memory = {}

    context = zmq.Context()
    socket = context.socket(zmq.REP)
    socket.bind('tcp://127.0.0.1:43000')

    while True:
        try:
            command, key, data = pickle.loads(socket.recv())
            if command == 'set':
                memory[key] = data
                socket.send(b'ok')
            elif command == 'get':
                result = memory.get(key, None)
                socket.send(pickle.dumps(result))
        except Exception as e:
            print(e)

if __name__ == '__main__':
    run_daemon()

為了一起測試所有內容，我們使用以下命令啟動伺服器
python daemon.py

在下一個選項卡中，以互動模式啟動 python。

>>> from lib import SuperCacher
>>> cache=SuperCacher()
>>> cache.set('key', 'value')
True
>>> cache.get('key')
'value'

哦奇蹟，它有效！ 現在你可以放心地在簡歷中寫下“使用zmq協議進行鍵值存儲的開發”

來源： www.habr.com