Golang Socket Tcp 连接优雅关闭

最新推荐文章于 2024-07-25 23:06:28 发布

____Luke

最新推荐文章于 2024-07-25 23:06:28 发布

阅读量1.7w

点赞数

分类专栏： golang 文章标签： socket golang 优雅关闭

本文链接：https://blog.csdn.net/soekchl521/article/details/66473423

版权

该博客介绍了如何在Golang中实现TCP服务器的优雅关闭，通过监听Interrupt信号，关闭服务器监听并等待所有已建立连接处理完毕后再退出程序。当没有连接时，中断会立即退出系统；若有连接，则在完成处理后退出。

摘要由CSDN通过智能技术生成

主要是利用系统监听Interrupt 信号，并且提前关闭服务器监听，并且等待先前连接处理完毕后在退出程序。

Server 代码：

package main

import (
	"net"
	"os"
	"os/signal"
	"strings"
	"sync"
	"time"

	. "github.com/soekchl/myUtils"
)

var (
	wg sync.WaitGroup = sync.WaitGroup{} // 等待各个socket连接处理
)

func main() {

	stop_chan := make(chan os.Signal) // 接收系统中断信号
	signal.Notify(stop_chan, os.Interrupt)

	listen, err := net.Listen("tcp", ":8080")
	if err != nil {
		Error(err)
		return
	}

	go func() {
		<-stop_chan
		Warn("Get Stop Command. Now Stoping...")
		if err = listen.Close(); err != nil {
			Error(err)
		}
	}()

	Notice("Start listen :8080 ... ")
	for {
		conn, err :&#