mirror of https://github.com/kianby/stacosys
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
971 B
Python
44 lines
971 B
Python
7 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import zmq
|
||
|
|
from conf import config
|
||
|
|
from threading import Thread
|
||
|
|
import logging
|
||
|
|
import json
|
||
|
|
from app.services import processor
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
context = zmq.Context()
|
||
|
|
|
||
|
|
|
||
|
|
def process(message):
|
||
|
|
data = json.loads(message)
|
||
|
|
if data['topic'] == 'email:newmail':
|
||
|
|
logger.info('newmail => {}'.format(data))
|
||
|
|
processor.enqueue({'request': 'new_mail', 'data': data})
|
||
|
|
|
||
|
|
|
||
|
|
class Consumer(Thread):
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
zsub = context.socket(zmq.SUB)
|
||
|
|
zsub.connect('tcp://127.0.0.1:{}'.format(config.zmq['pub_port']))
|
||
|
|
zsub.setsockopt_string(zmq.SUBSCRIBE, '')
|
||
|
|
self.loop = True
|
||
|
|
while self.loop:
|
||
|
|
message = zsub.recv()
|
||
|
|
try:
|
||
|
|
process(message)
|
||
|
|
except:
|
||
|
|
logger.exception('cannot process broker message')
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.loop = False
|
||
|
|
|
||
|
|
|
||
|
|
def start():
|
||
|
|
c = Consumer()
|
||
|
|
c.start()
|