Python에서 zmq 작업의 기본 사항, 간단한 키/값 저장소 만들기

소개

Memcache와 같은 간단한 키/값 저장소의 예를 살펴보겠습니다. 이는 간단하게 설계되었습니다. 데이터는 해시맵 구조로 메모리에 저장됩니다. TCP 소켓을 통해 액세스됩니다. Python에서 hashmap은 일반 사전입니다. 액세스를 위해 zeromq를 사용합니다.

조정

이 패키지를 debian/ubuntu에 설치하려면 콘솔에 다음을 입력하세요.
sudo apt-get install libzmq-dev
sudo pip install zmq

암호

서버에서 작동하는 클래스를 작성해 보겠습니다.
사용되는 zmq 소켓 유형은 REQ(REQuest, 요청)이며, 요청을 보내고 응답을 기다립니다.
네트워크를 통해 모든 유형의 데이터를 저장하고 전송하기 위해 표준 피클 모듈을 사용합니다. 작업 "프로토콜"은 세 가지 값(명령, 키, 데이터)으로 구성된 튜플입니다.

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()
캐시.세트('키', '값')
캐시.get('키')

숙제로 클래스 인스턴스를 생성할 때 주소/포트를 지정하는 기능을 추가하여 구현을 개선하세요.

이제 서버 자체를 작성해 보겠습니다.
이번에는 소켓 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 프로토콜을 사용한 키-값 저장소 개발"을 안전하게 작성할 수 있습니다.

출처 : habr.com