hystrix.py

#!/usr/bin/python
# Filename: hystrix.py

# This start/stop the hystrix test demos
# inlucde 2 web app, jumper -> echo

import os
import sys
import subprocess
import time

def get_sysinfo():
    sys = platform.system()
    return os.getpid(),sys
    
def get_path():
    p=os.path.split(os.path.realpath(__file__))  # ('D:\\workspace\\python\\src\\mysql', 'dao.py')
    p=os.path.split(p[0])
    if not p:
        os.mkdir(p)
    return p[0]


def get_pid_path():
    return get_path() +'/tmp/yqs.pid'


def check_pid(pid = 0,osname=''):
    if pid is None or pid == 0:
        return False
    wincmd = 'tasklist /FI "PID eq %s"  /FI "IMAGENAME eq python.exe "' % str(pid)
    lincmd = 'ps ax |grep %s |grep python' % str(pid)
    cmd,size = (wincmd,150) if osname=='Windows' else (lincmd,20)
    returnstr=subprocess.Popen(cmd,stdout=subprocess.PIPE,stderr=subprocess.PIPE, shell=True)
    data = returnstr.stdout.read()
    return len(data) > size
    
def read_pid():
    if os.path.exists(get_pid_path()):
        try:
            with open(get_pid_path(),'r') as f:
                strpid = f.readline().strip()
                return int(strpid)
        except Exception :
            return None
    return None


def rm_pid():
    if os.path.exists(get_pid_path()):
        os.remove(get_pid_path())
        
def kill(pid):
    """kill function for Win32"""
    kernel32 = ctypes.windll.kernel32
    handle = kernel32.OpenProcess(1, 0, pid)
    return (0 != kernel32.TerminateProcess(handle, 0))

def check_run():
    pid,osname = get_sysinfo()
    if not os.path.exists(get_pid_path()):
        with open(get_pid_path(),'w') as f: f.write(str(pid))
        return False
    
    # start checking
    rs = check_pid(read_pid(),osname)
    if not rs : 
        with open(get_pid_path(),'w') as f: f.write(str(pid))
    return rs

class Control :
    def start(self):
        if check_run():
            print 'pro has run'
        else :
            print 'pro start...'
            time.sleep(1000)
    
    def stop(self):
        filePid = read_pid()
        if filePid is not None and filePid > 0:
            print 'pro has kill %s' % filePid
            kill(filePid)
            rm_pid()
        else :
            print 'Process has closed'
            
    def check(self):
        filePid = read_pid()
        if not filePid or not check_run() :
            message = "Process has closed\n"
            sys.stderr.write(message)
        else :
            message = "The process has been run, the process id:%d\n"
            sys.stderr.write(message % filePid)
                
    def test(self):
        print "usage: start|stop|check|test|help"

    def helpInfo(self):
        print """usage: demo_name start|stop|check|test
  list    show demo list
  help    show this"""

def start_hystrix():
    print "startting demo hystrix ..."
    p=subprocess.Popen("MAVEN_OPTS='-javaagent:./pinpoint-agent/pinpoint-bootstrap-1.5.2-SNAPSHOT.jar -Dpinpoint.agentId=echo1 -Dpinpoint.applicationName=EchoService' mvn -f echowebsvr/pom.xml tomcat:run -Dmaven.tomcat.port=8099 &> log/echosvr.log & echo $! > log/echosvr.pid", shell=True)
    p=subprocess.Popen("MAVEN_OPTS='-javaagent:./pinpoint-agent/pinpoint-bootstrap-1.5.2-SNAPSHOT.jar -Dpinpoint.agentId=jumper1 -Dpinpoint.applicationName=Jumper' mvn -f jumperwebapp/pom.xml tomcat:run -Dmaven.tomcat.port=8098 &> log/jumper.log & echo $! > log/jumper.pid", shell=True)
    for i in range(0,10):
        if check_demo("hystrix"):
            print("demo hystrix started")
            return
        time.sleep(1)
    print("demo hystrix fail to start")

def stop_hystrix():
    print "stopping demo hystrix ..."
    p=subprocess.call("if [ -f log/echosvr.pid ]; then kill `cat log/echosvr.pid` || rm log/echosvr.pid; fi", shell=True)
    p=subprocess.call("if [ -f log/jumper.pid ]; then kill `cat log/jumper.pid` || rm log/jumper.pid; fi", shell=True)

def port_inuse(port):
    import socket
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.bind(("",port))
    except:
        return True
    finally:
        s.close()
    return False
    

def check_demo(demo):
    result = 0
    if port_inuse(8099):
        print("echosvr is running, port=8099")
        result += 1
    if port_inuse(8098):
        print("jumper is running, port=8098")
        result += 1
    if (result == 2):
        return True
    return False
    
def check_hystrix():
    print "checking demo hystrix ..."
    if check_demo("hystrix"):
        print("demo hystrix are running")
    else:
        print("demo are not running well")

def test_hystrix():
    p=subprocess.call("curl http://localhost:8098/jumper-webapp/jumper", shell=True)

def list_demos():
    print("hystrix")

####################################################

if __name__ == "__main__":
    contr=Control()
    if len(sys.argv) >= 2:
        param = sys.argv[1]
        if 'list' == param:
            list_demos()
        elif 'help' == param:
            contr.helpInfo()
        else:
            if len(sys.argv) == 3:
                demo_name = sys.argv[1]
                param = sys.argv[2]
                print("params: %s %s " % (demo_name, param))
                if 'start' == param:
                    start_hystrix()
                elif 'stop' == param:
                    stop_hystrix()
                elif 'check' == param:
                    check_hystrix()
                elif 'test' == param:
                    test_hystrix()
            else:
                contr.helpInfo()
    else:
        contr.helpInfo()

#        print "usage: %s module start|stop|check|test|help" % sys.argv[0]