Hanfodion gweithio gyda zmq mewn python, creu storfa allwedd/gwerth syml

Cyflwyniad

Edrychwn ar enghraifft o storfa allwedd / gwerth syml, fel memcache. Fe'i cynlluniwyd yn syml - mae'r data'n cael ei storio yn y cof, mewn strwythur hashmap. Ceir mynediad iddynt trwy soced tcp. Yn Python, mae hashmap yn ddit rheolaidd. Ar gyfer mynediad byddwn yn defnyddio zeromq.

addasiad

I osod y pecyn hwn yn debian / ubuntu, nodwch yn y consol
sudo apt-get install libzmq-dev
sudo pip install zmq

Cod

Gadewch i ni ysgrifennu dosbarth i weithio gyda'n gweinydd:
Y math o soced zmq a ddefnyddir yw REQ(REQuest, request), rydym yn anfon cais ac yn aros am ymateb.
I storio a throsglwyddo unrhyw fath o ddata dros y rhwydwaith, rydym yn defnyddio'r modiwl picl safonol. Mae'r β€œprotocol” gwaith yn tuple o dri gwerth: (gorchymyn, allwedd, data)

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

cache = SuperCacher()
cache.set('allwedd', 'gwerth')
cache.get ('allwedd')

Fel gwaith cartref, gwella'r gweithrediad trwy ychwanegu'r gallu i nodi cyfeiriad/porth wrth greu enghraifft o ddosbarth.

Nawr gadewch i ni ysgrifennu'r gweinydd ei hun.
Y tro hwn defnyddir y soced REP(REPly, response) - rydym yn aros am y cais, yn anfon ymateb. Rydym yn dosrannu'r cais ac yn ymateb naill ai gyda 'iawn' yn achos ysgrifennu, neu gyda data / Dim yn achos darllen.

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

I brofi popeth gyda'n gilydd, rydym yn cychwyn y gweinydd gyda'r gorchymyn
python daemon.py

Yn y tab nesaf, lansiwch python yn y modd rhyngweithiol.

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

O wyrth, mae'n gweithio! Nawr gallwch chi ysgrifennu'n ddiogel yn eich ailddechrau "datblygu storfa gwerth allweddol gan ddefnyddio'r protocol zmq"

Ffynhonnell: hab.com

Ychwanegu sylw