memory_storage_zz.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import redis
  2. class MemoryStorage():
  3. def __init__(self, config):
  4. self.ip = config['ip']
  5. self.port = config['port']
  6. self.conn = redis.StrictRedis(host=self.ip, port=self.port, db=0, decode_responses=True)
  7. def connected(self):
  8. pass
  9. def set_value(self, data_dict):
  10. try:
  11. pipe = self.conn.pipeline(transaction=True)
  12. for key_name in data_dict.keys():
  13. pipe.set(key_name, data_dict[key_name], ex=120)
  14. pipe.execute()
  15. except Exception as e:
  16. print(e)
  17. return e
  18. else:
  19. return True
  20. def get_value(self, keys):
  21. dict = {}
  22. try:
  23. pipe = self.conn.pipeline(transaction=True)
  24. for index in range(len(keys)):
  25. pipe.get(keys[index])
  26. result = pipe.execute()
  27. for index in range(len(keys)):
  28. dict[keys[index]] = result[index]
  29. return dict
  30. except Exception as e:
  31. return e
  32. def is_connected(self):
  33. pass
  34. def re_connected(self):
  35. pass