วิธีการประมวลผลสัญญาณ SIGTERM อย่างสง่างาม?


198

สมมติว่าเรามีดีมอนตัวเล็ก ๆ เขียนในไพ ธ อน:

def mainloop():
    while True:
        # 1. do
        # 2. some
        # 3. important
        # 4. job
        # 5. sleep

mainloop()

และเรา daemonize โดยใช้start-stop-daemonซึ่งเริ่มต้นโดยส่งSIGTERM( TERM) --stopสัญญาณ

#2สมมติว่าขั้นตอนการดำเนินการในปัจจุบันมี และในขณะนี้เรากำลังส่งTERMสัญญาณ

สิ่งที่เกิดขึ้นคือการกระทำนั้นสิ้นสุดลงทันที

ฉันพบว่าฉันสามารถจัดการเหตุการณ์สัญญาณใช้แต่สิ่งที่เป็นว่ามันยังคงขัดจังหวะการดำเนินการในปัจจุบันและผ่านการควบคุมไปsignal.signal(signal.SIGTERM, handler)handler

ดังนั้นคำถามของฉันคือ - เป็นไปได้หรือไม่ที่จะไม่ขัดจังหวะการทำงานปัจจุบัน แต่จัดการTERMสัญญาณในเธรดที่แยกจากกัน (?) เพื่อให้ฉันสามารถตั้งค่าshutdown_flag = Trueเพื่อให้mainloop()มีโอกาสหยุดอย่างสง่างาม?


2
ฉันทำสิ่งที่คุณขอมาก่อนโดยใช้signalfdและปิดบังการส่งมอบSIGTERMกระบวนการ
Eric Urban

คำตอบ:


280

โซลูชันที่ใช้งานง่าย

import signal
import time

class GracefulKiller:
  kill_now = False
  def __init__(self):
    signal.signal(signal.SIGINT, self.exit_gracefully)
    signal.signal(signal.SIGTERM, self.exit_gracefully)

  def exit_gracefully(self,signum, frame):
    self.kill_now = True

if __name__ == '__main__':
  killer = GracefulKiller()
  while not killer.kill_now:
    time.sleep(1)
    print("doing something in a loop ...")

  print("End of the program. I was killed gracefully :)")

1
ขอบคุณสำหรับความคิด! ฉันใช้วิธีการแก้ไขใน reboot-Guard github.com/ryran/reboot-guard/blob/master/rguard#L284:L304
rsaw

7
นี่คือคำตอบที่ดีที่สุด (ไม่จำเป็นต้องมีเธรด) และควรเป็นวิธีการทดลองแรกที่ต้องการ
jose.angel.jimenez

2
@ Mausy5043 Python อนุญาตให้คุณไม่มีวงเล็บสำหรับการกำหนดคลาส แม้ว่ามันจะใช้ได้ดีสำหรับ python 3.x แต่สำหรับ python 2.x แนวทางที่ดีที่สุดคือใช้ "class XYZ (object):" เหตุผลคือ: docs.python.org/2/reference/datamodel.html#newstyle
Mayank Jaiswal

2
ติดตามเพื่อให้คุณมีแรงบันดาลใจขอบคุณ ฉันใช้สิ่งนี้ตลอดเวลา
chrisfauerbach

2
ในกรณีที่แย่กว่านั้นก็หมายถึงการทำซ้ำอีกครั้งก่อนที่จะปิดตัวลงอย่างสง่างาม Falseถูกตั้งค่าเพียงครั้งเดียวแล้วก็สามารถไปจาก false เป็น true เพื่อให้การเข้าถึงหลายไม่เป็นปัญหา
Alceste_

52

shutdown_flagครั้งแรกที่ฉันไม่แน่ใจว่าคุณต้องมีหัวข้อที่สองการตั้งค่า
ทำไมไม่ตั้งค่าโดยตรงในตัวจัดการ SIGTERM

อีกทางเลือกหนึ่งคือยกข้อยกเว้นจากSIGTERMตัวจัดการซึ่งจะแพร่กระจายขึ้นสแต็ก สมมติว่าคุณได้รับการจัดการข้อยกเว้นที่เหมาะสม (เช่นกับwith/ contextmanagerและtry: ... finally:บล็อก) นี่ควรเป็นการปิดระบบที่สวยงามพอสมควรคล้ายกับถ้าคุณเข้าสู่Ctrl+Cโปรแกรมของคุณ

โปรแกรมตัวอย่างsignals-test.py:

#!/usr/bin/python

from time import sleep
import signal
import sys


def sigterm_handler(_signo, _stack_frame):
    # Raises SystemExit(0):
    sys.exit(0)

if sys.argv[1] == "handle_signal":
    signal.signal(signal.SIGTERM, sigterm_handler)

try:
    print "Hello"
    i = 0
    while True:
        i += 1
        print "Iteration #%i" % i
        sleep(1)
finally:
    print "Goodbye"

ตอนนี้ดูCtrl+Cพฤติกรรม:

$ ./signals-test.py default
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
^CGoodbye
Traceback (most recent call last):
  File "./signals-test.py", line 21, in <module>
    sleep(1)
KeyboardInterrupt
$ echo $?
1

ครั้งนี้ฉันจะส่งมันSIGTERMหลังจากทำซ้ำ 4 ครั้งด้วยkill $(ps aux | grep signals-test | awk '/python/ {print $2}'):

$ ./signals-test.py default
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
Terminated
$ echo $?
143

ครั้งนี้ฉันเปิดใช้งานSIGTERMตัวจัดการที่กำหนดเองของฉันและส่งมันSIGTERM:

$ ./signals-test.py handle_signal
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
Goodbye
$ echo $?
0

3
"ทำไมไม่ตั้งค่าโดยตรงในตัวจัดการ SIGTERM" --- เนื่องจากเธรดผู้ปฏิบัติงานจะขัดจังหวะในสถานที่แบบสุ่ม หากคุณใส่คำสั่งหลาย ๆ คำลงในลูปผู้ปฏิบัติงานของคุณคุณจะเห็นว่าโซลูชันของคุณยุติพนักงานในตำแหน่งสุ่มซึ่งทำให้งานอยู่ในสถานะที่ไม่รู้จัก
zerkms

ทำงานได้ดีสำหรับฉันและในบริบทของนักเทียบท่า ขอบคุณ!
แมเรียน

4
หากคุณเพิ่งตั้งค่าสถานะและไม่ยกข้อยกเว้นมันจะเหมือนกับในเธรด ดังนั้นการใช้เธรดจึงไม่จำเป็นที่นี่
Suor

28

ฉันคิดว่าคุณอยู่ใกล้กับทางออกที่เป็นไปได้

ดำเนินการในหัวข้อที่แยกต่างหากและขยายมันกับทรัพย์สินmainloop shutdown_flagสัญญาณสามารถจับได้signal.signal(signal.SIGTERM, handler)ในเธรดหลัก (ไม่ได้อยู่ในเธรดแยกต่างหาก) ตัวจัดการสัญญาณควรตั้งค่าshutdown_flagเป็น True และรอให้เธรดสิ้นสุดthread.join()


4
ใช่แล้วเธรดที่แยกออกมาเป็นวิธีที่ฉันแก้ไขได้ในที่สุดขอบคุณ
zerkms

7
ไม่จำเป็นต้องมีเธรดที่นี่ ในโปรแกรมเธรดเดี่ยวคุณสามารถลงทะเบียนตัวจัดการสัญญาณก่อน (การลงทะเบียนตัวจัดการสัญญาณไม่ใช่การปิดกั้น) จากนั้นเขียน mainloop ฟังก์ชันตัวจัดการสัญญาณควรตั้งค่าสถานะเมื่อใดและวนซ้ำควรตรวจสอบค่าสถานะนี้ ฉันได้วางวิธีการแก้ปัญหาตามระดับเดียวกันที่นี่
Mayank Jaiswal

2
ไม่จำเป็นต้องมีเธรดที่สอง ลงทะเบียนตัวจัดการสัญญาณ
oneloop

หน้ามีประโยชน์: g-loaded.eu/2016/11/24
Kamil Sindi

26

นี่เป็นตัวอย่างง่ายๆที่ไม่มีเธรดหรือคลาส

import signal

run = True

def handler_stop_signals(signum, frame):
    global run
    run = False

signal.signal(signal.SIGINT, handler_stop_signals)
signal.signal(signal.SIGTERM, handler_stop_signals)

while run:
    pass # do stuff including other IO stuff

11

จากคำตอบก่อนหน้านี้ฉันได้สร้างตัวจัดการบริบทซึ่งป้องกันจาก sigint และ sigterm

import logging
import signal
import sys


class TerminateProtected:
    """ Protect a piece of code from being killed by SIGINT or SIGTERM.
    It can still be killed by a force kill.

    Example:
        with TerminateProtected():
            run_func_1()
            run_func_2()

    Both functions will be executed even if a sigterm or sigkill has been received.
    """
    killed = False

    def _handler(self, signum, frame):
        logging.error("Received SIGINT or SIGTERM! Finishing this block, then exiting.")
        self.killed = True

    def __enter__(self):
        self.old_sigint = signal.signal(signal.SIGINT, self._handler)
        self.old_sigterm = signal.signal(signal.SIGTERM, self._handler)

    def __exit__(self, type, value, traceback):
        if self.killed:
            sys.exit(0)
        signal.signal(signal.SIGINT, self.old_sigint)
        signal.signal(signal.SIGTERM, self.old_sigterm)


if __name__ == '__main__':
    print("Try pressing ctrl+c while the sleep is running!")
    from time import sleep
    with TerminateProtected():
        sleep(10)
        print("Finished anyway!")
    print("This only prints if there was no sigint or sigterm")

4

พบวิธีที่ง่ายที่สุดสำหรับฉัน นี่คือตัวอย่างพร้อม fork เพื่อความชัดเจนว่าวิธีนี้มีประโยชน์สำหรับการควบคุมการไหล

import signal
import time
import sys
import os

def handle_exit(sig, frame):
    raise(SystemExit)

def main():
    time.sleep(120)

signal.signal(signal.SIGTERM, handle_exit)

p = os.fork()
if p == 0:
    main()
    os._exit()

try:
    os.waitpid(p, 0)
except (KeyboardInterrupt, SystemExit):
    print('exit handled')
    os.kill(p, 15)
    os.waitpid(p, 0)

0

ทางออกที่ง่ายที่สุดที่ฉันค้นพบโดยได้แรงบันดาลใจจากคำตอบข้างต้นคือ

class SignalHandler:

    def __init__(self):

        # register signal handlers
        signal.signal(signal.SIGINT, self.exit_gracefully)
        signal.signal(signal.SIGTERM, self.exit_gracefully)

        self.logger = Logger(level=ERROR)

    def exit_gracefully(self, signum, frame):
        self.logger.info('captured signal %d' % signum)
        traceback.print_stack(frame)

        ###### do your resources clean up here! ####

        raise(SystemExit)
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.