Python signal(信号量)的使用

1. 什么是信号量

信号量（Signal）是操作系统中的一种进程间通信方式，用来通知某个进程发生了某种事件。在多线程编程中，信号量也可以用来实现线程之间的通信。

在Python中，使用signal模块可以处理信号量。

2. 常用的信号量

2.1 SIGINT

SIGINT是程序终止信号，通常是用户在终端按下Ctrl+C触发。

在Python中，我们可以通过注册SIGINT信号并设置处理函数来捕获用户按下Ctrl+C的信号。

import signal
import sys

def handle_sigint(signum, frame):
    print('收到SIGINT信号，程序将退出')
    sys.exit(0)

signal.signal(signal.SIGINT, handle_sigint)

# 主循环
while True:
    pass

上述代码中，我们定义了一个handle_sigint函数来处理SIGINT信号，当收到SIGINT信号时，打印一条消息并退出程序。

2.2 SIGTERM

SIGTERM信号是程序终止信号，通常由kill命令或操作系统发出。

我们可以使用相同的方法来处理SIGTERM信号，请看下面的示例代码：

import signal
import sys

def handle_sigterm(signum, frame):
    print('收到SIGTERM信号，程序将退出')
    sys.exit(0)

signal.signal(signal.SIGTERM, handle_sigterm)

# 主循环
while True:
    pass

2.3 SIGALRM

SIGALRM信号是程序定时信号，可以通过设置闹钟来触发。

下面的示例代码演示了如何使用SIGALRM信号来实现定时任务：

import signal

def handle_alarm(signum, frame):
    print('定时任务执行')

signal.signal(signal.SIGALRM, handle_alarm)
signal.alarm(5)  # 设置5秒的闹钟

# 主循环
while True:
    pass

上述代码中，我们定义了一个handle_alarm函数来处理SIGALRM信号，在主循环中，我们通过调用signal.alarm(5)来设置一个5秒的闹钟，当闹钟时间到达时，会触发SIGALRM信号，然后执行定时任务。

3. 自定义信号量

除了系统内置的信号量外，我们还可以自定义一些信号量用于进程或线程间通信。

下面的示例代码演示了如何定义和使用自定义信号量：

import signal

# 定义自定义信号量
SIG_CUSTOM = signal.SIGUSR1

def handle_custom(signum, frame):
    print('收到自定义信号量SIGUSR1')

signal.signal(SIG_CUSTOM, handle_custom)

# 发送自定义信号量
os.kill(os.getpid(), SIG_CUSTOM)

上述代码中，我们先定义了一个自定义信号量SIG_CUSTOM，然后定义了一个handle_custom函数来处理SIG_CUSTOM信号。最后通过调用os.kill函数来发送SIG_CUSTOM信号。