python 进程单例,起停功能

python文件锁实现进程单例

通过参数控制进程起停, 类似 httpd -k <start | stop>

#!/usr/bin/python
import os
import getopt
import fcntl
import time
import sys

def main():
    while(True):
        print os.getpid()
        time.sleep(3)

def print_usage():
    print 'usage'

if __name__ == "__main__":
    pid_file = '/opt/nsfocus/mp-wafsl/tmp/daemon.pid'
    optarg = 'null'
    try:
        opts, args = getopt.getopt(sys.argv[1:], "hk:", ['help', 'kk='])
        for name,value in opts:
            if name in ('-h'):
                print_usage()
                sys.exit(0)
            if name in ('-k'):
                optarg = value
    except getopt.GetoptError:
        print_usage()
        sys.exit(2)
    # Must use a+ here
    fp = open(pid_file, 'a+')
    is_running = False
    try:
        fcntl.lockf(fp, fcntl.LOCK_EX | fcntl.LOCK_NB)
    except IOError:
        is_running = True
    
    if is_running and optarg == 'stop':
        # stop running process
        content = fp.read()
        fp.close()
        pid = int(content)
        os.kill(pid, 9)
        sys.exit(0)
    elif optarg == 'start' and is_running is False:
        fp.truncate()
        fp.write(str(os.getpid()))
        fp.flush()
        main()
    else:
        sys.exit(0)


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值