67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import pika
|
||
|
|
import json
|
||
|
|
import threading
|
||
|
|
|
||
|
|
|
||
|
|
buffer = []
|
||
|
|
lock = threading.Lock()
|
||
|
|
|
||
|
|
print('pika version: %s' % pika.__version__)
|
||
|
|
|
||
|
|
|
||
|
|
connection = pika.BlockingConnection(pika.ConnectionParameters(host='localhost'))
|
||
|
|
|
||
|
|
main_channel = connection.channel()
|
||
|
|
consumer_channel = connection.channel()
|
||
|
|
bind_channel = connection.channel()
|
||
|
|
|
||
|
|
if pika.__version__=='0.9.5':
|
||
|
|
main_channel.exchange_declare(exchange='com.micex.sten', type='direct')
|
||
|
|
main_channel.exchange_declare(exchange='com.micex.lasttrades', type='direct')
|
||
|
|
else:
|
||
|
|
main_channel.exchange_declare(exchange='com.micex.sten', exchange_type='direct')
|
||
|
|
main_channel.exchange_declare(exchange='com.micex.lasttrades', exchange_type='direct')
|
||
|
|
|
||
|
|
queue = main_channel.queue_declare(exclusive=True).method.queue
|
||
|
|
queue_tickers = main_channel.queue_declare(exclusive=True).method.queue
|
||
|
|
|
||
|
|
main_channel.queue_bind(exchange='com.micex.sten', queue=queue, routing_key='order.stop.create')
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def process_buffer():
|
||
|
|
if not lock.acquire(False):
|
||
|
|
print('locked!')
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
while len(buffer):
|
||
|
|
body = buffer.pop(0)
|
||
|
|
|
||
|
|
ticker = None
|
||
|
|
if 'ticker' in body['data']['params']['condition']: ticker = body['data']['params']['condition']['ticker']
|
||
|
|
if not ticker: continue
|
||
|
|
|
||
|
|
print('got ticker %s, gonna bind it...' % ticker)
|
||
|
|
bind_channel.queue_bind(exchange='com.micex.lasttrades', queue=queue_tickers, routing_key=str(ticker))
|
||
|
|
print('ticker %s binded ok' % ticker)
|
||
|
|
finally:
|
||
|
|
lock.release()
|
||
|
|
|
||
|
|
|
||
|
|
def callback(ch, method, properties, body):
|
||
|
|
body = json.loads(body)['order.stop.create']
|
||
|
|
buffer.append(body)
|
||
|
|
process_buffer()
|
||
|
|
|
||
|
|
|
||
|
|
consumer_channel.basic_consume(callback,
|
||
|
|
queue=queue, no_ack=True)
|
||
|
|
|
||
|
|
try:
|
||
|
|
consumer_channel.start_consuming()
|
||
|
|
finally:
|
||
|
|
connection.close()
|