未配置关闭端口通过os信号关闭服务器,读文章 《优雅关闭的 Go Web 服务器》

[读 "优雅关闭的 Go Web 服务器" ](https://github.com/stayfoo/stayfoo-hub/blob/master/docs/golang/read-article/1、read-优雅关闭的GoWeb服务器.md)

GitHub 仓库：https://github.com/stayfoo/stayfoo-hub

> 文章原文链接：[Go语言中文网](https://mp.weixin.qq.com/s/rA_oh472ZhfcAsAkWyyXFA) : https://mp.weixin.qq.com/s/rA_oh472ZhfcAsAkWyyXFA

文章《优雅的关闭 `Go Web` 服务器》写到：可以通过开启一个单独的 `goroutine` 拦截关闭信号，这样，当服务器真正关闭之前，可以做一些任务，做完任务再发出执行完毕的关闭号。

一些任务比如：清理某些资源；完成数据库事务；一些其他长时间的操作；退出服务的那一刻，刚好收到一个响应，为了保证所有请求完成，就可以在这里，在最大超时时间内去处理这个响应；dump进程当前状态；记录日志的动作。

启动应用，`Ctrl + C` 中断程序，中断信号被拦截，do something.....

```bash

go run main.go -listen-addr=5001

http: 2019/08/31 11:34:30 Server is ready to handle requests at :5001

^Chttp: 2019/08/31 11:34:32 Server is shutting down...

do something start ..... 2019-08-31 11:34:32.594668 +0800 CST m=+2.337451148

do something end ..... 2019-08-31 11:34:37.598248 +0800 CST m=+7.340881516

http: 2019/08/31 11:34:37 Server stopped

```

对文中代码做了改造，代码如下：

```go

var listenAddr string

func init() {

//接收端口号，默认端口号：5000

flag.StringVar(&listenAddr, "listen-addr", ":5000", "server listen address")

}

func main() {

flag.Parse() //外部参数解析

listenAddr = fmt.Sprintf(":%s",listenAddr)

logger := log.New(os.Stdout, "http: ", log.LstdFlags)

//创建 server：

server := newWebServer(logger)

done := make(chan struct{}, 1)

quit := make(chan os.Signal, 1)

//os.Interrupt: syscall.SIGINT

signal.Notify(quit, os.Interrupt)

//启动另一个 goroutine ，监听将要关闭信号：

go shutdown(server, logger, quit, done)

//启动 server：

logger.Println("Server is ready to handle requests at",listenAddr)

err := server.ListenAndServe()

if err != nil && err != http.ErrServerClosed {

logger.Fatalf("Could not listen on %s: %v \n", listenAddr, err)

}

//等待已经关闭的信号：

logger.Println("Server stopped")

}

//初始化 server

func newWebServer(logger *log.Logger) *http.Server {

//路由:

router := http.NewServeMux()

router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {

w.WriteHeader(http.StatusOK)

})

//http 服务配置:

server := &http.Server{

Addr: listenAddr,

Handler: router,

ErrorLog: logger,

ReadTimeout: 5 * time.Second,

WriteTimeout: 10 * time.Second,

IdleTimeout: 15 * time.Second,

}

return server

}

//关闭 server

//quit: 接收关闭信号

//done: 发出已经关闭信号

func shutdown(server *http.Server, logger *log.Logger, quit

//等待接收到退出信号：

logger.Println("Server is shutting down...")

ctx, cancel := context.WithTimeout(context.Background(), 30 * time.Second)

defer cancel()

server.SetKeepAlivesEnabled(false)

err := server.Shutdown(ctx)

if err != nil {

logger.Fatalf("Could not gracefully shutdown the server: %v \n", err)

}

//do Something ：

fmt.Println("do something start ..... ", time.Now())

time.Sleep(5 * time.Second)

fmt.Println("do something end ..... ", time.Now())

close(done)

}

```

## 改造的地方

原文代码：

```go

done := make(chan bool, 1)

```

改造后代码：

```go

done := make(chan struct{}, 1)

```

可以看到，把关闭信号从 `chan bool` 改成了 `chan struct{}`。

好处：空 `struct{}` 不占用空间，`chan bool` 占1字节。空 `struct{}` 更节省资源。

由下面测试代码可以知道：

```go

s := struct {}{}

b := true

fmt.Println("struct: ", unsafe.Sizeof(s), " bool: ", unsafe.Sizeof(b))

//打印结果：struct: 0 bool: 1

```

## 一些其他

### Go中的信号

`os/signal` 包对信号处理。

- 监听收到的信号：

```go

func Notify(c chan

```

- 取消监听：

```go

func Stop(c chan

```

- 监听 Interrupt 信号，用户 `Ctrl+C` 触发：

```go

quit := make(chan os.Signal, 1)

//os.Interrupt: syscall.SIGINT

signal.Notify(quit, os.Interrupt)

```

- 监听所有信号：

```go

c := make(chan os.Signal)

signal.Notify(c)

```

- 监听多个指定信号：

```go

c := make(chan os.Signal)

signal.Notify(c,os.Interrupt, os.Kill, syscall.SIGQUIT)

```

- 一些信号：

```go

//操作系统收到信号后的动作:

//Term: 表明默认动作为终止进程

//Ign: 表明默认动作为忽略该信号

//Core: 表明默认动作为终止进程同时输出core dump

//Stop: 表明默认动作为停止进程

// Signals

const (

SIGABRT = Signal(0x6) //调用abort函数触发，十进制值：6, Core

SIGALRM = Signal(0xe) //时钟定时信号，十进制值：14, Term

SIGBUS = Signal(0xa) //非法地址(内存地址对齐错误)，十进制值：10 Core

SIGCHLD = Signal(0x14)//子进程结束(由父进程接收)，十进制值：20 Ign

SIGCONT = Signal(0x13)//继续执行已经停止的进程(不能被阻塞)，十进制：19 Cont

SIGEMT = Signal(0x7)

SIGFPE = Signal(0x8)//算术运行错误(浮点运算错误、除数为零等)，十进制：8 Core

SIGHUP = Signal(0x1)//终端控制进程结束(终端连接断开),十进制：1 Term

SIGILL = Signal(0x4)//非法指令(程序错误、试图执行数据段、栈溢出等) Core

SIGINFO = Signal(0x1d)

SIGINT = Signal(0x2)//用户发送INTR字符(Ctrl+C)触发，十进制值：2

SIGIO = Signal(0x17)

SIGIOT = Signal(0x6)

SIGKILL = Signal(0x9)//无条件结束程序(不能被捕获、阻塞或忽略)十进制：9

SIGPIPE = Signal(0xd)//消息管道损坏(FIFO/Socket通信时，管道未打开而进行写操作)

SIGPROF = Signal(0x1b)

SIGQUIT = Signal(0x3)//用户发送QUIT字符(Ctrl+/)触发，十进制值：3

SIGSEGV = Signal(0xb)//无效内存引用(试图访问不属于自己的内存空间、对只读内存空间进行写操作)

SIGSTOP = Signal(0x11)//停止进程(不能被捕获、阻塞或忽略)

SIGSYS = Signal(0xc)

SIGTERM = Signal(0xf)

SIGTRAP = Signal(0x5)

SIGTSTP = Signal(0x12)//停止进程(可以被捕获、阻塞或忽略)

SIGTTIN = Signal(0x15)//后台程序从终端中读取数据时触发

SIGTTOU = Signal(0x16)//后台程序向终端中写数据时触发

SIGURG = Signal(0x10)

SIGUSR1 = Signal(0x1e)

SIGUSR2 = Signal(0x1f)

SIGVTALRM = Signal(0x1a)

SIGWINCH = Signal(0x1c)

SIGXCPU = Signal(0x18)//超过CPU时间资源限制(4.2BSD)

SIGXFSZ = Signal(0x19)//超过文件大小资源限制(4.2BSD)

)

```

有疑问加站长微信联系(非本文作者))