python使用signal模块编写超时处理

我下面写了一个超时处理脚本,主要用来超时处理nfs服务挂死的情况,如果检测到挂载超时,就会执行sys.exit(1)报错退出:

#!/usr/bin/python
# -*- coding: utf-8 -*-
import os,os.path,sys,commands,signal,time

class MountExcepition(Exception):
    def __init__(self,value):
        self.value = value
    def __str__(self):
        return repr(self.value)

#设置命令超时
class TimeOutException(Exception):
    pass
 
def setTimeout(num, callback):
    def wrape(func):
        def handle(signum, frame):
            raise TimeOutException("运行超时!")
        def toDo(*args, **kwargs):
            try:
                signal.signal(signal.SIGALRM, handle)
                signal.alarm(num)#开启闹钟信号
                rs = func(*args, **kwargs)
                signal.alarm(0)#关闭闹钟信号
                return rs
            except TimeOutException, e:
                callback()          
        return toDo
    return wrape

        
if __name__ == '__main__':
    def doSome():
        sys.exit(1)    
    @setTimeout(8,doSome)
    #nfs服务检查函数
    def check_task():
        try:
            cmd = "ls -l /nfs_5580_check"
            (status,output) = commands.getstatusoutput("%s" %cmd)
            if status != 0:
                raise MountExcepition(output)
            
        except MountExcepition as e:
            print "fail"
            sys.exit(1)       
        else:
            print "success"
    
    check_task()

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值