External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-16 20:30:46 -03:30
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #453 from matburt/fix_proot_cp_issue

Browse Source 
Fix proot controlpersist cleanup issue
This commit is contained in:
Matthew Jones 2015-10-01 09:38:22 -04:00
commit 4ca29c4adf
1 changed files with 5 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
awx/plugins/callback/job_event_callback.py
View File

@ -37,7 +37,6 @@ import logging
import os
import pwd
import urlparse
import time
# Requests
import requests
@ -45,12 +44,7 @@ import requests
# ZeroMQ
import zmq
# PSUtil
try:
import psutil
except ImportError:
psutil = None
import psutil
class TokenAuth(requests.auth.AuthBase):
@ -249,19 +243,14 @@ class BaseCallbackModule(object):
if not cp_files:
return
# HACK: If psutil isn't available, sleep and allow the control master
# processes to timeout and die.
if not psutil:
time.sleep(60)
# Attempt to find any running control master processes.
username = pwd.getpwuid(os.getuid())[0]
ssh_cm_procs = []
for proc in psutil.process_iter():
try:
pname = proc.name
pcmdline = proc.cmdline
pusername = proc.username
pname = proc.name()
pcmdline = proc.cmdline()
pusername = proc.username()
except psutil.NoSuchProcess:
continue
if pusername != username:
@ -277,19 +266,7 @@ class BaseCallbackModule(object):
# version of psutil that may not have wait_procs implemented.
for proc in ssh_cm_procs:
proc.terminate()
if hasattr(psutil, 'wait_procs'):
procs_gone, procs_alive = psutil.wait_procs(ssh_cm_procs, timeout=5)
else:
procs_gone = []
procs_alive = ssh_cm_procs[:]
for x in xrange(5):
for proc in procs_alive[:]:
if not proc.is_running():
procs_alive.remove(proc)
procs_gone.append(proc)
if not procs_alive:
break
time.sleep(1)
procs_gone, procs_alive = psutil.wait_procs(ssh_cm_procs, timeout=5)
for proc in procs_alive:
proc.kill()