d1b94da4a4
This patch adds a python script which can be used as a demo client. The script is interactive and will allow the user to register, request statistics, and unregister. To run the script, an argument for the client file path must be passed in: "python telemetry_client.py <file_path>". This script is useful to see how the Telemetry API for DPDK is used, and how to make the initial connection. Signed-off-by: Ciara Power <ciara.power@intel.com> Signed-off-by: Brian Archbold <brian.archbold@intel.com> Signed-off-by: Kevin Laatz <kevin.laatz@intel.com> Acked-by: Harry van Haaren <harry.van.haaren@intel.com>
117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
# SPDK-License-Identifier: BSD-3-Clause
|
|
# Copyright(c) 2018 Intel Corporation
|
|
|
|
import socket
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
BUFFER_SIZE = 200000
|
|
|
|
METRICS_REQ = "{\"action\":0,\"command\":\"ports_all_stat_values\",\"data\":null}"
|
|
API_REG = "{\"action\":1,\"command\":\"clients\",\"data\":{\"client_path\":\""
|
|
API_UNREG = "{\"action\":2,\"command\":\"clients\",\"data\":{\"client_path\":\""
|
|
DEFAULT_FP = "/var/run/dpdk/default_client"
|
|
|
|
class Socket:
|
|
|
|
def __init__(self):
|
|
self.send_fd = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
|
self.recv_fd = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
|
|
self.client_fd = None
|
|
|
|
def __del__(self):
|
|
try:
|
|
self.send_fd.close()
|
|
self.recv_fd.close()
|
|
self.client_fd.close()
|
|
except:
|
|
print("Error - Sockets could not be closed")
|
|
|
|
class Client:
|
|
|
|
def __init__(self): # Creates a client instance
|
|
self.socket = Socket()
|
|
self.file_path = None
|
|
self.choice = None
|
|
self.unregistered = 0
|
|
|
|
def __del__(self):
|
|
try:
|
|
if self.unregistered == 0:
|
|
self.unregister();
|
|
except:
|
|
print("Error - Client could not be destroyed")
|
|
|
|
def getFilepath(self, file_path): # Gets arguments from Command-Line and assigns to instance of client
|
|
self.file_path = file_path
|
|
|
|
def register(self): # Connects a client to DPDK-instance
|
|
if os.path.exists(self.file_path):
|
|
os.unlink(self.file_path)
|
|
try:
|
|
self.socket.recv_fd.bind(self.file_path)
|
|
except socket.error as msg:
|
|
print ("Error - Socket binding error: " + str(msg) + "\n")
|
|
self.socket.recv_fd.settimeout(2)
|
|
self.socket.send_fd.connect("/var/run/dpdk/rte/telemetry")
|
|
JSON = (API_REG + self.file_path + "\"}}")
|
|
self.socket.send_fd.sendall(JSON)
|
|
self.socket.recv_fd.listen(1)
|
|
self.socket.client_fd = self.socket.recv_fd.accept()[0]
|
|
|
|
def unregister(self): # Unregister a given client
|
|
self.socket.client_fd.send(API_UNREG + self.file_path + "\"}}")
|
|
self.socket.client_fd.close()
|
|
|
|
def requestMetrics(self): # Requests metrics for given client
|
|
self.socket.client_fd.send(METRICS_REQ)
|
|
data = self.socket.client_fd.recv(BUFFER_SIZE)
|
|
print "\nResponse: \n", str(data)
|
|
|
|
def repeatedlyRequestMetrics(self, sleep_time): # Recursively requests metrics for given client
|
|
print("\nPlease enter the number of times you'd like to continuously request Metrics:")
|
|
n_requests = int(input("\n:"))
|
|
print("\033[F") #Removes the user input from screen, cleans it up
|
|
print("\033[K")
|
|
for i in range(n_requests):
|
|
self.requestMetrics()
|
|
time.sleep(sleep_time)
|
|
|
|
def interactiveMenu(self, sleep_time): # Creates Interactive menu within the script
|
|
while self.choice != 3:
|
|
print("\nOptions Menu")
|
|
print("[1] Send for Metrics for all ports")
|
|
print("[2] Send for Metrics for all ports recursively")
|
|
print("[3] Unregister client")
|
|
|
|
try:
|
|
self.choice = int(input("\n:"))
|
|
print("\033[F") #Removes the user input for screen, cleans it up
|
|
print("\033[K")
|
|
if self.choice == 1:
|
|
self.requestMetrics()
|
|
elif self.choice == 2:
|
|
self.repeatedlyRequestMetrics(sleep_time)
|
|
elif self.choice == 3:
|
|
self.unregister()
|
|
self.unregistered = 1
|
|
else:
|
|
print("Error - Invalid request choice")
|
|
except:
|
|
pass
|
|
|
|
if __name__ == "__main__":
|
|
|
|
sleep_time = 1
|
|
file_path = ""
|
|
if (len(sys.argv) == 2):
|
|
file_path = sys.argv[1]
|
|
else:
|
|
print("Warning - No filepath passed, using default (" + DEFAULT_FP + ").")
|
|
file_path = DEFAULT_FP
|
|
client = Client()
|
|
client.getFilepath(file_path)
|
|
client.register()
|
|
client.interactiveMenu(sleep_time)
|