4080e46c80
telemetry has support for fetching port based stats from metrics library. Metrics library also has global stats which are not fetched by telemetry, so extend telemetry to fetch the global metrics. Signed-off-by: Reshma Pattan <reshma.pattan@intel.com> Acked-by: Kevin Laatz <kevin.laatz@intel.com>
127 lines
4.5 KiB
Python
Executable File
127 lines
4.5 KiB
Python
Executable File
#! /usr/bin/env 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\":\""
|
|
GLOBAL_METRICS_REQ = "{\"action\":0,\"command\":\"global_stat_values\",\"data\":null}"
|
|
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 requestGlobalMetrics(self): #Requests global metrics for given client
|
|
self.socket.client_fd.send(GLOBAL_METRICS_REQ)
|
|
data = self.socket.client_fd.recv(BUFFER_SIZE)
|
|
print "\nResponse: \n", str(data)
|
|
|
|
def interactiveMenu(self, sleep_time): # Creates Interactive menu within the script
|
|
while self.choice != 4:
|
|
print("\nOptions Menu")
|
|
print("[1] Send for Metrics for all ports")
|
|
print("[2] Send for Metrics for all ports recursively")
|
|
print("[3] Send for global Metrics")
|
|
print("[4] 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.requestGlobalMetrics()
|
|
elif self.choice == 4:
|
|
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)
|