今天小编给大家分享一下GO语言如何实现TCP服务器的相关知识点，内容详细，逻辑清晰，相信大部分人都还太了解这方面的知识，所以分享这篇文章给大家参考一下，希望大家阅读完这篇文章后有所收获，下面我们一起来了解一下吧。
interface/tcp/Handler.go
type Handler interface { Handle(ctx context.Context, conn net.Conn) Close() error}Handler：业务逻辑的处理接口
Handle(ctx context.Context, conn net.Conn) 处理连接
tcp/server.go
type Config struct { Address string}func ListenAndServeWithSignal(cfg *Config, handler tcp.Handler) error { closeChan := make(chan struct{}) listen, err := net.Listen("tcp", cfg.Address) if err != nil { return err } logger.Info("start listen") ListenAndServe(listen, handler, closeChan) return nil}func ListenAndServe(listener net.Listener, handler tcp.Handler, closeChan <-chan struct{}) { ctx := context.Background() var waitDone sync.WaitGroup for true { conn, err := listener.Accept() if err != nil { break } logger.Info("accept link") waitDone.Add(1) go func() { defer func() { waitDone.Done() }() handler.Handler(ctx, conn) }() } waitDone.Wait()}Config：启动tcp服务器的配置
Address：监听地址
ListenAndServe：ctx是上下文，可以传递一些参数。死循环中接收到新连接时，让一个协程去处理连接
如果listener.Accept()出错了就会break跳出来，这时候需要等待已经服务的客户端退出。使用WaitGroup等待客服端退出
func ListenAndServe(listener net.Listener, handler tcp.Handler, closeChan <-chan struct{}) { go func() { <-closeChan logger.Info("shutting down...") _ = listener.Close() _ = handler.Close() }() defer func() { _ = listener.Close() _ = handler.Close() }() ......}listener和handler在退出的时候需要关掉。如果用户直接kill掉了程序，我们也需要关掉listener和handler，这时候要使用closeChan，一旦接收到关闭信号，就执行关闭逻辑
func ListenAndServeWithSignal(cfg *Config, handler tcp.Handler) error { closeChan := make(chan struct{}) sigCh := make(chan os.Signal) signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT) go func() { sig := <-sigCh switch sig { case syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT: closeChan <- struct{}{} } }() listen, err := net.Listen("tcp", cfg.Address) if err != nil { return err } logger.Info("start listen") ListenAndServe(listen, handler, closeChan) return nil}当系统对程序发送信号时，sigCh会接收到信号
tcp/echo.go
type EchoHandler struct { activeConn sync.Map closing atomic.Boolean}EchoHandler：
activeConn：记录连接
closing：是否正在关闭，有并发竞争，使用atomic.Boolean
type EchoClient struct { Conn net.Conn Waiting wait.Wait}func (c *EchoClient) Close() error {c.Waiting.WaitWithTimeout(10 * time.Second)_ = c.Conn.Close()return nil}EchoClient：一个客户端就是一个连接。Close方法关闭客户端连接，超时时间设置为10s
func MakeHandler() *EchoHandler {return &EchoHandler{}}func (h *EchoHandler) Handle(ctx context.Context, conn net.Conn) { // 连接正在关闭，不接收新连接 if h.closing.Get() { _ = conn.Close() } client := &EchoClient{ Conn: conn, } h.activeConn.Store(client, struct{}{}) reader := bufio.NewReader(conn) for { msg, err := reader.ReadString('\n') if err != nil { if err == io.EOF { logger.Info("connection close") h.activeConn.Delete(client) } else { logger.Warn(err) } return } // 正在处理业务，不要关掉 client.Waiting.Add(1) // 将数据原封不动写回去，测试 b := []byte(msg) _, _ = conn.Write(b) client.Waiting.Done() }}func (h *EchoHandler) Close() error { logger.Info("handler shutting down...") h.closing.Set(true) h.activeConn.Range(func(key interface{}, val interface{}) bool { client := key.(*EchoClient) _ = client.Close() return true }) return nil}MakeEchoHandler：创建EchoHandler
Handle：处理客户端的连接。
连接正在关闭时，不接收新连接
存储新连接，value用空结构体
使用缓存区接收用户发来的数据，使用\n作为结束的标志
Close：将所有客户端连接关掉
main.go
const configFile string = "redis.conf"var defaultProperties = &config.ServerProperties{ Bind: "0.0.0.0", Port: 6379,}func fileExists(filename string) bool { info, err := os.Stat(filename) return err == nil && !info.IsDir()}func main() { logger.Setup(&logger.Settings{ Path: "logs", Name: "godis", Ext: "log", TimeFormat: "2022-02-02", }) if fileExists(configFile) { config.SetupConfig(configFile) } else { config.Properties = defaultProperties } err := tcp.ListenAndServeWithSignal( &tcp.Config{ Address: fmt.Sprintf("%s:%d", config.Properties.Bind, config.Properties.Port), }, EchoHandler.MakeHandler()) if err != nil { logger.Error(err) }}以上就是“GO语言如何实现TCP服务器”这篇文章的所有内容，感谢各位的阅读！相信大家阅读完这篇文章都有很大的收获，小编每天都会为大家更新不同的知识，如果还想学习更多的知识，请关注编程网行业资讯频道。
