יסודות העבודה עם zmq ב-python, יצירת חנות מפתח/ערך פשוטה

מבוא

בואו נסתכל על דוגמה לאחסון מפתח/ערך פשוט, כגון memcache. הוא מתוכנן בפשטות - הנתונים מאוחסנים בזיכרון, במבנה hashmap. הגישה אליהם מתבצעת באמצעות שקע tcp. ב-Python, hashmap הוא dict רגיל. לגישה נשתמש ב- zeromq.

התאמה

כדי להתקין חבילה זו ב-debian/ubuntu, פשוט היכנס לקונסולה
sudo apt-get install libzmq-dev
sudo pip install zmq

קוד

בואו נכתוב מחלקה לעבודה עם השרת שלנו:
סוג שקע zmq המשמש הוא REQ(REQuest, 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'

להשתמש

cache = SuperCacher()
cache.set('key', 'value')
cache.get('key')

כשיעורי בית, שפר את היישום על ידי הוספת היכולת לציין כתובת/יציאה בעת יצירת מופע של מחלקה.

עכשיו בואו נכתוב את השרת עצמו.
הפעם נעשה שימוש בשקע REP(REPly, תגובה) - אנו ממתינים לבקשה, שולחים תגובה. אנו מנתחים את הבקשה ומגיבים או ב'אישור' במקרה של כתיבה, או עם נתונים / אין במקרה של קריאה.

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