83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
import psutil
|
|
import re
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
class PySPDK(object):
|
|
|
|
def __init__(self, pname):
|
|
super(PySPDK, self).__init__()
|
|
self.pid = None
|
|
self.pname = pname
|
|
|
|
def start_server(self, spdk_dir, server_name):
|
|
if not self.is_alive():
|
|
self.init_hugepages(spdk_dir)
|
|
server_dir = os.path.join(spdk_dir, 'app/')
|
|
file_dir = self._search_file(server_dir, server_name)
|
|
print file_dir
|
|
os.chdir(file_dir)
|
|
p = subprocess.Popen(
|
|
'sudo ./%s' % server_name,
|
|
shell=True, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
out, err = p.communicate()
|
|
return out
|
|
|
|
def init_hugepages(self, spdk_dir):
|
|
huge_dir = os.path.join(spdk_dir, 'scripts/')
|
|
file_dir = self._search_file(huge_dir, 'setup.sh')
|
|
print file_dir
|
|
os.chdir(file_dir)
|
|
p = subprocess.Popen(
|
|
'sudo ./setup.sh',
|
|
shell=True, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
out, err = p.communicate()
|
|
return out
|
|
|
|
@staticmethod
|
|
def _search_file(spdk_dir, file_name):
|
|
for dirpath, dirnames, filenames in os.walk(spdk_dir):
|
|
for filename in filenames:
|
|
if filename == file_name:
|
|
return dirpath
|
|
|
|
def _get_process_id(self):
|
|
for proc in psutil.process_iter():
|
|
try:
|
|
pinfo = proc.as_dict(attrs=['pid', 'cmdline'])
|
|
if re.search(self.pname, str(pinfo.get('cmdline'))):
|
|
self.pid = pinfo.get('pid')
|
|
return self.pid
|
|
except psutil.NoSuchProcess:
|
|
print "NoSuchProcess:%s" % self.pname
|
|
print "NoSuchProcess:%s" % self.pname
|
|
return self.pid
|
|
|
|
def is_alive(self):
|
|
self.pid = self._get_process_id()
|
|
if self.pid:
|
|
p = psutil.Process(self.pid)
|
|
if p.is_running():
|
|
return True
|
|
return False
|
|
|
|
@staticmethod
|
|
def exec_rpc(method, server='127.0.0.1', port=5260, sub_args=None):
|
|
exec_cmd = ["./rpc.py", "-s", "-p"]
|
|
exec_cmd.insert(2, server)
|
|
exec_cmd.insert(4, str(port))
|
|
exec_cmd.insert(5, method)
|
|
if sub_args is None:
|
|
sub_args = []
|
|
exec_cmd.extend(sub_args)
|
|
p = subprocess.Popen(
|
|
exec_cmd,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE
|
|
)
|
|
out, err = p.communicate()
|
|
return out
|