|
|
#!/usr/bin/python
|
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
|
|
import time
|
|
|
from datetime import datetime, timedelta
|
|
|
import zmq
|
|
|
import json
|
|
|
import numpy as np
|
|
|
from threading import Thread
|
|
|
import argparse
|
|
|
|
|
|
REQUEST_TIMEOUT = 5000
|
|
|
RETRIES = 5
|
|
|
SERVER_ENDPOINT = 'tcp://localhost:4444'
|
|
|
|
|
|
context = zmq.Context()
|
|
|
socket = context.socket(zmq.REQ)
|
|
|
socket.connect (SERVER_ENDPOINT)
|
|
|
poll = zmq.Poller()
|
|
|
poll.register(socket, zmq.POLLIN)
|
|
|
|
|
|
def send(dato):
|
|
|
'''
|
|
|
Function to send data to server
|
|
|
'''
|
|
|
global socket, poll
|
|
|
retries = RETRIES
|
|
|
while True:
|
|
|
socket.send_json(dato)
|
|
|
socks = dict(poll.poll(REQUEST_TIMEOUT))
|
|
|
if socks.get(socket) == zmq.POLLIN:
|
|
|
reply = socket.recv_string()
|
|
|
if reply == 'ok':
|
|
|
print('Server replied (%s)' % reply)
|
|
|
break
|
|
|
else:
|
|
|
print('Malformed reply from server: %s' % reply)
|
|
|
else:
|
|
|
print('No response from server, retries left {}'.format(retries))
|
|
|
socket.setsockopt(zmq.LINGER, 0)
|
|
|
socket.close()
|
|
|
poll.unregister(socket)
|
|
|
retries -= 1
|
|
|
if retries == 0:
|
|
|
print('Server seems to be offline...')
|
|
|
socket = context.socket(zmq.REQ)
|
|
|
socket.connect(SERVER_ENDPOINT)
|
|
|
poll.register(socket, zmq.POLLIN)
|
|
|
break
|
|
|
# Create new connection
|
|
|
socket = context.socket(zmq.REQ)
|
|
|
socket.connect(SERVER_ENDPOINT)
|
|
|
poll.register(socket, zmq.POLLIN)
|
|
|
|
|
|
def main(realtime, code, date=None, interval=30):
|
|
|
'''
|
|
|
Simulate data to be sended to server
|
|
|
'''
|
|
|
|
|
|
if realtime:
|
|
|
dt = datetime.now()
|
|
|
else:
|
|
|
dt = date
|
|
|
|
|
|
while True:
|
|
|
|
|
|
print('Sending {} - {}'.format(code, dt))
|
|
|
|
|
|
dato = {
|
|
|
'time': time.mktime(dt.timetuple()),
|
|
|
'yrange': np.arange(100).tolist(),
|
|
|
'xrange': np.arange(-30, 30).tolist(),
|
|
|
'localtime': True,
|
|
|
'interval': interval,
|
|
|
'exp_code': code,
|
|
|
'data': {
|
|
|
'noise': np.round(np.random.rand(4) + np.array([10,11,12,13]), 2).tolist(),
|
|
|
'rti': np.round(np.random.rand(4, 100)*5 + 10, 2).tolist(),
|
|
|
'spc': np.round(np.random.rand(4, 60, 100)*5 + 10, 2).tolist(),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
dt = dt + timedelta(seconds=interval)
|
|
|
t = Thread(target=send, args=(dato, ))
|
|
|
t.start()
|
|
|
if realtime:
|
|
|
time.sleep(interval)
|
|
|
else:
|
|
|
time.sleep(5)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
parser = argparse.ArgumentParser(description='This is a Client for realtime app')
|
|
|
parser.add_argument('--date', action='store', default=None, help='format: 2018/02/13 12:23:00')
|
|
|
parser.add_argument('-r', action='store_true', dest='realtime', default=None)
|
|
|
parser.add_argument('-c', action='store', dest='code', default='170')
|
|
|
parser.add_argument('-i', action='store', dest='interval', type=int, default=30)
|
|
|
|
|
|
results = parser.parse_args()
|
|
|
if not results.realtime:
|
|
|
try:
|
|
|
results.date = datetime.strptime(results.date, '%Y/%m/%d %H:%M:%S')
|
|
|
except:
|
|
|
raise(NameError('You must specify a date (--date) for non-realtime experiment'))
|
|
|
main(results.realtime, results.code, results.date, results.interval)
|
|
|
|
