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

package main

import (

    "context"

    "fmt"

    "github.com/spf13/viper"

    "go.uber.org/zap"

    "log"

    "net/http"

    "os"

    "os/signal"

    "syscall"

    "time"

)

func main() {

    //启动服务(优雅关机)

    srv := &http.Server{

        Addr:    fmt.Sprintf(":%d", viper.GetInt("app.port")),

        Handler: r,

    }

    go func() {

        // 开启一个goroutine启动服务

        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {

            log.Fatalf("listen: %s\n", err)

        }

    }()

    // 等待中断信号来优雅地关闭服务器，为关闭服务器操作设置一个5秒的超时

    quit := make(chan os.Signal, 1) // 创建一个接收信号的通道

    // kill 默认会发送 syscall.SIGTERM 信号

    // kill -2 发送 syscall.SIGINT 信号，我们常用的Ctrl+C就是触发系统SIGINT信号

    // kill -9 发送 syscall.SIGKILL 信号，但是不能被捕获，所以不需要添加它

    // signal.Notify把收到的 syscall.SIGINT或syscall.SIGTERM 信号转发给quit

    signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此处不会阻塞

    <-quit                                               // 阻塞在此，当接收到上述两种信号时才会往下执行

    zap.L().Info("Shutdown Server ...")

    // 创建一个5秒超时的context

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

    defer cancel()

    // 5秒内优雅关闭服务（将未处理完的请求处理完再关闭服务），超过5秒就超时退出

    if err := srv.Shutdown(ctx); err != nil {

        zap.L().Fatal("Server Shutdown: ", zap.Error(err))

    }

    zap.L().Info("Server exiting")

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

import (

    "log"

    "net/http"

    "time"

    "github.com/fvbock/endless"

    "github.com/gin-gonic/gin"

)

func main() {

    router := gin.Default()

    router.GET("/", func(c *gin.Context) {

        c.String(http.StatusOK, "hello xiaosheng !")

    })

    // 默认endless服务器会监听下列信号：

    // syscall.SIGHUP，syscall.SIGUSR1，syscall.SIGUSR2，syscall.SIGINT，syscall.SIGTERM和syscall.SIGTSTP

    // 接收到 SIGHUP 信号将触发`fork/restart` 实现优雅重启（kill -1 pid会发送SIGHUP信号）

    // 接收到 syscall.SIGINT或syscall.SIGTERM 信号将触发优雅关机

    // 接收到 SIGUSR2 信号将触发HammerTime

    // SIGUSR1 和 SIGTSTP 被用来触发一些用户自定义的hook函数

    if err := endless.ListenAndServe(":8080", router); err!=nil{

        log.Fatalf("listen: %s\n", err)

    }

    log.Println("Server exiting...")