golang实现守护进程

概览:

守护进程介绍

守护进程(daemon)是一类在后台运行的特殊进程,用于执行特定的系统任务。很多守护进程在系统引导的时候启动,并且一直运行直到系统关闭。另一些只在需要的时候才启动,完成任务后就自动结束。

守护进程是一个在后台运行并且不受任何终端控制的进程。Unix操作系统有很多典型的守护进程(其数目根据需要或20—50不等),它们在后台运行,执行不同的管理任务。

用户使守护进程独立于所有终端是因为,在守护进程从一个终端启动的情况下,这同一个终端可能被其他的用户使用。例如,用户从一个终端启动守护进程后退出,然后另外一个人也登录到这个终端。用户不希望后者在使用该终端的过程中,接收到守护进程的任何错误信息。同样,由终端键入的任何信号(例如中断信号)也不应该影响先前在该终端启动的任何守护进程的运行。虽然让服务器后台运行很容易(只要shell命令行以&结尾即可),但用户还应该做些工作,让程序本身能够自动进入后台,且不依赖于任何终端。

守护进程没有控制终端,因此当某些情况发生时,不管是一般的报告性信息,还是需由管理员处理的紧急信息,都需要以某种方式输出。Syslog 函数就是输出这些信息的标准方法,它把信息发送给 syslogd 守护进程。

创建步骤

(1)创建子进程,终止父进程

由于守护进程是脱离控制终端的,因此首先创建子进程,终止父进程,使得程序在shell终端里造成一个已经运行完毕的假象。之后所有的工作都在子进程中完成,而用户在shell终端里则可以执行其他的命令,从而使得程序以僵尸进程形式运行,在形式上做到了与控制终端的脱离。

(2)在子进程中创建新会话

这个步骤是创建守护进程中最重要的一步,在这里使用的是系统函数setsid。

setsid函数用于创建一个新的会话,并担任该会话组的组长。调用setsid仃三个作用:让进程摆脱原会话的控制、让进程摆脱原进程组的控制和让进程摆脱原控制终端的控制。

在调用fork函数时,子进程全盘拷贝父进程的会话期(session,是一个或多个进程组的集合)、进程组、控制终端等,虽然父进程退出了,但原先的会话期、进程组、控制终端等并没有改变,因此,那还不是真正意义上使两者独立开来。setsid函数能够使进程完全独立出来,从而脱离所有其他进程的控制。

(3)改变工作目录

使用fork创建的子进程也继承了父进程的当前工作目录。由于在进程运行过程中,当前目录所在的文件系统不能卸载,因此,把当前工作目录换成其他的路径,如“/”或“/tmp”等。改变工作目录的常见函数是chdir。

(4)重设文件创建掩码

文件创建掩码是指屏蔽掉文件创建时的对应位。由于使用fork函数新建的子进程继承了父进程的文件创建掩码,这就给该子进程使用文件带来了诸多的麻烦。因此,把文件创建掩码设置为0,可以大大增强该守护进程的灵活性。设置文件创建掩码的函数是umask,通常的使用方法为umask(0)。

(5)关闭文件描述符

用fork新建的子进程会从父进程那里继承一些已经打开了的文件。这些被打开的文件可能永远不会被守护进程读或写,但它们一样消耗系统资源,可能导致所在的文件系统无法卸载。

首先,用c语言实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>

void mydaemon(int ischdir,int isclose)
{
// 调用setsid() 的不能是进程组组长,当前程序有可能是进程组组长
pid_t pid = fork();

// 非子进程则退出
if(pid != 0)
exit(-1);

// 父进程退出,留下子进程

// 创建一个新的会话期,从而脱离原有的会话期
// 进程同时与控制终端脱离
setsid();

// 此时子进程成为新会话期的领导和新的进程组长
// 但这样的身份可能会通过fcntl去获到终端
pid = fork();

// 非子进程则退出
if(pid != 0)
exit(-1);

// 此时留下来的是孙子进程,再也不能获取终端

// 通常来讲, 守护进程应该工作在一个系统永远不会删除的目录下
if(ischdir == 0)
{
chdir("/");
}

// 关闭输入输出和错误流 (通过日志来查看状态)
if(isclose == 0)
{
close(0);
close(1);
close(2);
}

//去掩码位
umask((mode_t)0);//sys/stat.h
}

int main(int argc, char *argv[])
{
mydaemon(1,1);
#加载自己的负载任务
exit(EXIT_SUCCESS);
}

golang版本

Golangc代码混编

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package main

/*
// C
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>

void mydaemon(int ischdir,int isclose){
// 调用setsid() 的不能是进程组组长,当前程序有可能是进程组组长
pid_t pid = fork();

// 非子进程则退出
if(pid != 0)
exit(-1);

// 父进程退出,留下子进程

// 创建一个新的会话期,从而脱离原有的会话期
// 进程同时与控制终端脱离
setsid();

// 此时子进程成为新会话期的领导和新的进程组长
// 但这样的身份可能会通过fcntl去获到终端
pid = fork();

// 非子进程则退出
if(pid != 0)
exit(-1);

// 此时留下来的是孙子进程,再也不能获取终端

// 通常来讲, 守护进程应该工作在一个系统永远不会删除的目录下
if(ischdir == 0)
{
chdir("/");
}

// 关闭输入输出和错误流 (通过日志来查看状态)
if(isclose == 0)
{
close(0);
close(1);
close(2);
}

//去掩码位
umask((mode_t)0);//sys/stat.h
}
*/
import "C"

import (
"fmt"
"os"
"time"
)

func main() {
fmt.Println("Run daemon")
C.mydaemon(1, 1)
fmt.Printf("User gid: %v, uid: %v\n", os.Getegid(), os.Geteuid())
fmt.Printf("Precess id: %v %v\n", os.Getpid(), os.Getppid())

//模拟负载
ticker := time.NewTicker(time.Second * 1)
for {
<-ticker.C
fmt.Println(time.Now())
}
}

效果展示

输出时间是因为用了fmt.Print函数,所以会输出到当前终端