如果在關(guān)機前保存數(shù)據(jù)？

zhoulifa

有了上面這個例子，我們就可以完成一個關(guān)機前保存數(shù)據(jù)的程序了。

首先，C程序代碼如下：

1. 
   
2. #include <sys/types.h>
   
3. #include <time.h>
   
4. #include <signal.h>
   
5. #include <stdio.h>
   
6. #include <stdlib.h>
   
7. #include <unistd.h>
   
8. 
   
9. #define SIGTOCAP SIGTERM
   
10. 
    
11. void sighandler(int);
    
12. char status;
    
13. time_t t;
    
14. 
    
15. /************關(guān)于本文檔********************************************
    
16. *filename: SaveData.c
    
17. *purpose: 展示如何在關(guān)機前保存數(shù)據(jù)
    
18. *wrote by: zhoulifa(zhoulifa@163.com) 周立發(fā)(http://zhoulifa.bokee.com)
    
19. Linux愛好者 Linux知識傳播者 SOHO族 開發(fā)者 最擅長C語言
    
20. *date time:2006-09-02 22:11:54
    
21. *Note: 任何人可以任意復(fù)制代碼并運用這些文檔，當(dāng)然包括你的商業(yè)用途
    
22. * 但請遵循GPL
    
23. *Hope:希望越來越多的人貢獻自己的力量，為科學(xué)技術(shù)發(fā)展出力
    
24. *         科技站在巨人的肩膀上進步更快！感謝有開源前輩的貢獻！
    
25. *********************************************************************/
    
26. int main(int argc, char ** argv)
    
27. {
    
28.   FILE * fp = 0;
    
29. 
    
30.   if(fork()) exit(0);
    
31.   if(fork()) exit(0); /* 變成 daemon 精靈程序進行后臺服務(wù) */
    
32. 
    
33.   signal(SIGTOCAP, sighandler); /* 捕捉退出信號 */
    
34.   if(!(fp = fopen("/var/run/save-data.pid", "w"))) {
    
35.     fprintf(stdout, "cannot open file '/var/run/save-data.pid'\n");
    
36.     exit(1);
    
37.   }
    
38.   fprintf(fp, "%d\n", getpid()); /* 保存運行程序的 pid */
    
39.   fclose(fp);
    
40. 
    
41.   if(!(fp = fopen("/home/test/running.data", "a+"))) {
    
42.     fprintf(stdout, "cannot open file '/home/test/running.data'\n");
    
43.     exit(1);
    
44.   }
    
45.   status = 0;
    
46.   while(1) {
    
47.     t = time(0);
    
48.     fprintf(fp, "now running at %s\n", ctime(&t)); /* 每隔一定時間寫一些數(shù)據(jù)，表明程序在運行 */
    
49.     if(status) break; /* 如果收到退出信號，并保存完數(shù)據(jù)， 則退出程序運行 */
    
50.     usleep(1000000);
    
51.   }
    
52.   fclose(fp);
    
53.   remove("/var/run/save-data.pid"); /* 退出前刪除 pid 文件 */
    
54.   return 0;
    
55. }
    
56. 
    
57. void sighandler(int sig)
    
58. {
    
59.   FILE * fp = 0;
    
60.   int i = 0;
    
61.   if(sig == SIGTOCAP) { /* 捕捉到退出信號后開始保存數(shù)據(jù) */
    
62.     if(!(fp = fopen("/home/test/saving.data", "a+"))) {
    
63.       fprintf(stdout, "cannot open file '/home/test/saving.data'\n");
    
64.       exit(-1);
    
65.     }
    
66.     for(i = 0; i < 10000; i++) {
    
67.       t = time(0);
    
68.       fprintf(fp, "now get ready to exit at %s\n", ctime(&t));
    
69.     }
    
70.     fclose(fp);
    
71.     status= 1; /* 數(shù)據(jù)保存完畢，設(shè)置退出標志 */
    
72.   }
    
73. }
    

復(fù)制代碼

然后編譯此程序：

gcc SaveData.c -o save-data

再把這個程序放到/bin目錄下去：

mv save-data /bin

然后編寫一個腳本/etc/init.d/save-data，內(nèi)容如下：

1. 
   
2. #!/bin/sh
   
3. #
   
4. # $Id: save-data.init.d,v 1.0.0.0 2006/09/02 22:11:54
   
5. #
   
6. 
   
7. test -x /bin/save-data || exit 0
   
8. 
   
9. MYPID=/var/run/save-data.pid
   
10. 
    
11. case "$1" in
    
12.         start)
    
13.                 echo -n "Starting save-data daemon: "
    
14.                 echo -n ""
    
15.                 sleep 10 # Just for test , you'd better delete this line for use
    
16.                 start-stop-daemon --start --quiet --pidfile $MYPID --exec /bin/save-data
    
17.                 if [ -f "$MYPID" ] && ps h `cat "$MYPID"` >/dev/null; then
    
18.                         echo " OK."
    
19.                 else
    
20.                         echo " Failed."
    
21.                 fi
    
22.                 ;;
    
23.         stop)
    
24.                 echo -n "Stopping save-data daemon: save-data"
    
25.                 echo -n ""
    
26.                 sleep 10 # Just for test , you'd better delete this line for use
    
27.                 start-stop-daemon --stop --quiet --pidfile $MYPID
    
28.                 echo "."
    
29.                 ;;
    
30.         restart | force-reload)
    
31.                 $0 stop
    
32.                 sleep 2
    
33.                 $0 start
    
34.                 ;;
    
35.         *)
    
36.                 echo "Usage: /etc/init.d/save-data {start|stop|restart}"
    
37.                 exit 1
    
38. esac
    
39. 
    
40. exit 0
    

復(fù)制代碼

再創(chuàng)建鏈接：

ln -s /etc/init.d/save-data /etc/rc2.d/S70save-data
ln -s /etc/init.d/save-data /etc/rc0.d/K10save-data

然后重啟機器試試，會看到數(shù)據(jù)被保存到saving.data文件里了

zhoulifa

關(guān)于“ daemon 精靈程序 ” ， 我這里只是簡單的 fork 了兩次，事實上還有其它許多工作要做，比如關(guān)閉終端， google 一下很多的，就不贅述了