script 中使用 trap_trap script-CSDN博客

最近在跑分散式的程式，每次測試都要開好幾隻程式在跑。就算可以寫 script 自動開啟，還是會遇到一個很討厭的問題，沒辦法自動關閉那些屬於 worker 的程式。好像解釋的不太清楚....就是我希望在我程式跑完的時候，或是我把程式強迫終止的時候，我可以同時把好幾個 process 也 kill 掉。特別是強迫終止的時候，因為無論我在最後寫好什麼動作，一旦強迫終止就執行不到了。

於是上網尋找解法，找到了 trap 可以用。Trap 這東西，也就是陷阱...我想他的意思是，會讓你指定的那些 signal 掉入陷阱吧...(笑)。不曉得 signal 是什麼的請自行閱讀 manpage 或是上網找。

這裡提供兩個範例，第一個是測試 trap 功能的範例：

 
           #!/bin/bash

           trap 
           
           "echo TRAP!!; exit" SIGTERM SIGINT SIGHUP

           for 
           
           (( i=0; i<5; i=i+1 ))

           do

           echo 
           
           $i

               sleep 1

           done

view raw trap_test1.sh hosted with ❤ by GitHub

這個範例中，我指定 SIGTERM SIGINT SIGHUP 這三個 signal 會被 trap 抓住，而 trap 抓到後所要執行的指令為 "echo TRAP!!; exit"。注意這邊我加上了 exit 讓這個 script 會在接到 signal 的時候終止，否則它只會 echo 出 TRAP!! 後繼續執行它要執行的動作。如果你沒有另外指定別的離開方法，就必須乖乖的等待 script 執行完畢，或是用 kill -9 (送 SIGKILL) 才能終止它了...

再來是第二個範例：

 
           #!/bin/bash

           function 
           
           clean_up {

           echo CLEAN UP
           
           !!

           exit

           }

           trap clean_up SIGTERM SIGINT SIGHUP

           for 
           
           (( i=0; i<5; i=i+1 ))

           do

           echo 
           
           $i

           	sleep 1

           done

           clean_up

view raw trap_test2.sh hosted with ❤ by GitHub

這裡差不多是一個實際使用的 use case 了，參考自：這篇文章。把需要做清理的動作寫進 function 裡，以我最上面提到的情況就是把該關掉的 worker 程式都關掉，在這裡我就印出個 CLEAN UP!! 代替就好XD。然後在 trap 裡設定它負責清理，另外也在 script 正常結束前清理，這樣就可以保證(大概吧?)清理動作一定會被執行，除非是被 SIGKILL 強迫中斷。

對了...我實際在用的時候其實是用 zsh 啦，不過我試過了，用 bash 也可以正常執行沒有問題。我指的是 script 第一行指定的 shell，不是執行 script 的 shell 唷！

另外還有，我認為 clean_up 應該只要負責清理就好，所以最後小小修改，變成了這樣：

 
           #!/bin/bash

           function 
           
           clean_up {

           echo CLEAN UP
           
           !!

           }

           trap 
           
           "clean_up;exit" SIGTERM SIGINT SIGHUP

           for 
           
           (( i=0; i<5; i=i+1 ))

           do

           echo 
           
           $i

           	sleep 1

           done

           clean_up