Bazoj pri laborado kun zmq en python, kreante simplan ŝlosil-/valoran vendejon

Enkonduko

Ni rigardu ekzemplon de simpla ŝlosilo/valora stokado, kiel memcache. Ĝi estas aranĝita simple - la datumoj estas konservitaj en memoro, en la hashmap-strukturo. Aliro al ili estas farita per tcp-ingo. En python, hashmap estas nur dikto. Por aliro ni uzos zeromq.

alĝustigo

Por instali ĉi tiun pakaĵon en debian/ubuntu, simple eniru en la konzolon
sudo apt-get install libzmq-dev
sudo pip install zmq

Kodo

Ni skribu klason por labori kun nia servilo:
La tipo de zmq-ingo uzata estas REQ(REQuest, request), ni sendas peton - ni atendas respondon.
Por stoki kaj transdoni ajnan tipon de datumoj tra la reto, ni uzas la norman piklan modulon. "Protokolo" de laboro - opo de tri valoroj: (komando, ŝlosilo, datumoj)

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'
Uzo

kaŝmemoro = SuperCacher ()
cache.set('ŝlosilo', 'valoro')
cache.get('ŝlosilo')

Kiel hejmtasko, plibonigu la efektivigon aldonante la kapablon specifi adreson/havenon dum kreado de ekzemplo de la klaso.

Nun ni skribu la servilon mem.
Ĉi-foje, la ingo REP(REPly, response) estas uzata - ni atendas la peton, ni sendas la respondon. Ni analizas la peton, respondas aŭ 'ok' en kazo de skribo, aŭ datumoj / Neniu en kazo de legado.

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()

Por provi ĉion kune, lanĉu la servilon per la komando
python daemon.py

En la sekva langeto, rulu python en interaga reĝimo.

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

Ve, ĝi funkcias! Nun vi povas sekure skribi en via vivresumo "disvolviĝo de ŝlosilvalora stokado uzante la protokolon zmq"

fonto: www.habr.com

Aldoni komenton