42 lines
941 B
Python
42 lines
941 B
Python
"""
|
||
@Time ： 2022/10/9 17:50
|
||
@Auth ： 东
|
||
@File ：RedisMQTool.py
|
||
@IDE ：PyCharm
|
||
@Motto：ABC(Always Be Coding)
|
||
@Desc：
|
||
|
||
"""
|
||
import json
|
||
|
||
import redis
|
||
|
||
|
||
class Task(object):
|
||
|
||
def __init__(self, redis_conn, channel):
|
||
self.rcon = redis_conn
|
||
self.ps = self.rcon.pubsub()
|
||
self.key = 'task:pubsub:%s' % channel
|
||
self.ps.subscribe(self.key)
|
||
|
||
def listen_task(self):
|
||
for i in self.ps.listen():
|
||
if i['type'] == 'message':
|
||
print("Task get ", i["data"])
|
||
|
||
def publish_task(self, data):
|
||
self.rcon.publish(self.key, json.dumps(data))
|
||
|
||
def del_listen(self):
|
||
self.ps.unsubscribe(self.key)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
print("listen task channel")
|
||
|
||
pool = redis.ConnectionPool(host='127.0.0.1',
|
||
port=6379, db=5,)
|
||
|
||
redis_conn = redis.StrictRedis(connection_pool=pool)
|
||
Task(redis_conn, 'channel').listen_task() |