wehub-resource-sync
498b235461
Build and test / Build and test AMD64 Ubuntu 22.04 (push) Failing after 0s
Publish Builder / amazonlinux2023 (push) Failing after 1s
Build and test / UT for Go (push) Has been skipped
Publish KRTE Images / KRTE (push) Failing after 1s
Build and test / Integration Test (push) Has been skipped
Build and test / Upload Code Coverage (push) Has been skipped
Publish Builder / rockylinux9 (push) Failing after 1s
Publish Builder / ubuntu22.04 (push) Failing after 0s
Publish Builder / ubuntu24.04 (push) Failing after 0s
Publish Gpu Builder / publish-gpu-builder (push) Failing after 1s
Publish Test Images / PyTest (push) Failing after 0s
Build and test / UT for Cpp (push) Has been cancelled
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import psutil
|
|
import time
|
|
from loguru import logger
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser(description='config for rolling update process')
|
|
parser.add_argument('--wait_time', type=int, default=60, help='wait time after rolling update started')
|
|
args = parser.parse_args()
|
|
wait_time = args.wait_time
|
|
logger.info("start to watch rolling update process")
|
|
start_time = time.time()
|
|
end_time = time.time()
|
|
flag = False
|
|
while not flag and end_time - start_time < 360:
|
|
process_list = [p.info for p in psutil.process_iter(attrs=['pid', 'name','cmdline'])]
|
|
for process in process_list:
|
|
logger.debug(process)
|
|
logger.debug("##"*30)
|
|
for process in process_list:
|
|
if isinstance(process.get("cmdline", []), list):
|
|
cmdline_list = process.get("cmdline", [])
|
|
for cmdline in cmdline_list:
|
|
if "rollingUpdate.sh" in cmdline:
|
|
logger.info(f"rolling update process: {process} started")
|
|
flag = True
|
|
break
|
|
if flag:
|
|
break
|
|
time.sleep(0.5)
|
|
end_time = time.time()
|
|
if not flag:
|
|
logger.info(f"rolling update process not found, wait for {end_time - start_time} seconds")
|
|
else:
|
|
logger.info(f"rolling update process {process} found, wait for {end_time - start_time} seconds")
|
|
if flag:
|
|
logger.info(f"wait {wait_time}s to kill rolling update process")
|
|
time.sleep(wait_time)
|
|
logger.info("start to kill rolling update process")
|
|
try:
|
|
p = psutil.Process(process["pid"])
|
|
p.terminate()
|
|
logger.info(f"rolling update process: {process} killed")
|
|
except Exception as e:
|
|
logger.error(f"rolling update process: {process} kill failed, {e}")
|
|
else:
|
|
logger.info("all process info")
|
|
for process in process_list:
|
|
logger.info(process)
|
|
|
|
|
|
|