Skip to content
Snippets Groups Projects

Draft: ref: output: Rewrite sink functionality

Files
35
+ 14
15
#! /usr/bin/python2
#! /usr/bin/python3
import socket
import socket
import time
import time
MAX_RUNS = 3
MAX_RUNS = 3
EXEC_TIME_SECS = 5
EXEC_TIME_SECS = 20
STOP_DELAY_SECS = 10
STOP_DELAY_SECS = 20
def send_command(sock, cmd, run_no):
def send_command(sock, cmd, run_no):
message = cmd + " " + str(run_no)
message = cmd + " " + str(run_no)
print "Sending message: ", message
print("Sending message: ", message)
try:
try:
sock.sendall(message)
sock.sendall(message.encode('utf-8'))
data = sock.recv(256)
data = sock.recv(256).decode('utf-8')
print "Server response: ", data
if "ok" in data:
if "ok" in data:
print "Command successful"
return cmd + " command successful: received " + data
return True
else:
else:
return False
return cmd + " command FAILED: received " + data
except Exception as e:
except Exception as e:
print "ERROR! ", e
print("ERROR! ", e)
return False
return False
if __name__ == "__main__":
if __name__ == "__main__":
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_address = ('localhost', 8000)
server_address = ('localhost', 8000)
sock.connect(server_address)
sock.connect(server_address)
 
time.sleep(5)
 
do_start = True
do_start = True
for current_run in range(0, MAX_RUNS):
for current_run in range(1, MAX_RUNS + 1):
if do_start:
if do_start:
print "Starting run ", current_run
print("Starting run ", current_run)
send_command(sock, "start", current_run)
send_command(sock, "start", current_run)
time.sleep(EXEC_TIME_SECS)
time.sleep(EXEC_TIME_SECS)
print "Halting run ", current_run
print("Halting run ", current_run)
send_command(sock, "stop", current_run)
send_command(sock, "stop", current_run)
time.sleep(STOP_DELAY_SECS)
time.sleep(STOP_DELAY_SECS)
Loading