Los fundamentos de trabajar con zmq en python, creando un almacén simple de clave / valor

Introduccion

Veamos un ejemplo de un almacenamiento simple de clave / valor, como memcache. Simplemente está organizado: los datos se almacenan en la memoria, en la estructura de hashmap. El acceso a ellos es a través de un zócalo TCP. En python, hashmap es un dict regular. Para el acceso usaremos zeromq.

Personalización

Para instalar este paquete en debian / ubuntu simplemente ingrese en la consola
sudo apt-get install libzmq-dev
sudo pip install zmq

Código

Escribamos una clase para trabajar con nuestro servidor:
El tipo de socket zmq utilizado es REQ (REQuest, request), envía la solicitud y espera una respuesta.
Para almacenar y transferir cualquier tipo de datos a través de la red, utilizamos el módulo de pickle estándar. "Protocolo" de trabajo: una tupla de tres valores: (comando, clave, datos)

 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' 

Uso

caché = SuperCacher ()
cache.set ('clave', 'valor')
cache.get ('clave')

Como tarea, mejore la implementación agregando la capacidad de especificar una dirección / puerto al crear una instancia de la clase.

Ahora escribiremos el servidor en sí.
Esta vez, se usa el conector REP (REPly, respuesta): estamos esperando la solicitud, el casco es la respuesta. Analizamos la solicitud, respondemos 'ok' en el caso de la escritura o data / None en el caso de la lectura.

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

Para probar todo junto, inicie el servidor con el comando
python daemon.py

En la siguiente pestaña, inicie Python de forma interactiva.

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

¡Oh milagro, funciona! Ahora puede escribir de forma segura en su currículum "desarrollando almacenamiento de valor-clave utilizando el protocolo zmq"