Huangzj
53 lines
1004 B
Go
53 lines
1004 B
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2020/12/8 11:18
|
||
* @Description：
|
||
*/
|
||
|
||
package gin
|
||
|
||
import (
|
||
"context"
|
||
"github.com/gin-gonic/gin"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestGeneralSwitch(t *testing.T) {
|
||
router := gin.Default()
|
||
router.GET("/", func(c *gin.Context) {
|
||
time.Sleep(5 * time.Second)
|
||
c.String(http.StatusOK, "Welcome Gin Server")
|
||
})
|
||
|
||
srv := &http.Server{
|
||
Addr: ":8080",
|
||
Handler: router,
|
||
}
|
||
|
||
go func() {
|
||
// 服务连接
|
||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
log.Fatalf("listen: %s\n", err)
|
||
}
|
||
}()
|
||
|
||
// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
|
||
quit := make(chan os.Signal)
|
||
signal.Notify(quit, os.Interrupt)
|
||
<-quit
|
||
log.Println("Shutdown Server ...")
|
||
|
||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
defer cancel()
|
||
if err := srv.Shutdown(ctx); err != nil {
|
||
log.Fatal("Server Shutdown:", err)
|
||
}
|
||
log.Println("Server exiting")
|
||
|
||
}
|