博客
关于我
[日常] Go语言圣经-基于select的多路复用习题
阅读量:664 次
发布时间:2019-03-15

本文共 2374 字,大约阅读时间需要 7 分钟。

Go 服务器超时处理优化示例

本文将介绍如何利用 Go 语言优化一个简单的 TCP 服务器实现,通过增加客户端连接超时机制,实现客户端 10 秒无响应时自动断开连接的功能。

实现概述

以下是在本次改造中的主要内容和目标:

  • 在现有 echo 服务器的基础上增加客户端连接超时处理
  • 当客户端在 10 秒内没有任何交互时,服务器自动断开连接
  • 保持现有功能的同时,优化代码结构和可读性
  • 服务器设计细节

    服务器基于 Go 语言实现,采用标准库 net 包中的 ListenAcceptClose 等函数进行操作。主要代码逻辑包括以下几个部分:

  • 上线监听

    net.Listen("tcp", ":8040") 开启一个 TCP 监听端口 8040

  • 处理连接

    使用 Accept 回调接收新的连接请求,并启动 goroutine 处理每个连接

  • 客户端输入处理

    通过 bufio.Scanner 实现对客户端输入的读取

  • 超时机制实现

    使用 time.After 创建 10 秒超时定时器,实现自动断开连接功能

  • 代码改造解读

    以下是主要改造的代码片段:

    package mainimport (    "bufio"    "fmt"    "log"    "net"    "strings"    "sync"    "time")func main() {    listeners, err := net.Listen("tcp", ":8040")    if err != nil {        log.Fatal("无法绑定监听地址", err)    }    for {        conn, err := listeners.Accept()        if err != nil {            log.Print("连接拒绝:", err) // 例如客户端主动中断连接            continue        }        go handleConn(conn)    }}// 处理每个连接的 goroutinefunc handleConn(c net.Conn) {    input := bufio.NewScanner(c)    wg := sync.WaitGroup    message := make(chan string, 100)    wg.Add(1)    // 定时器退出 goroutine    go func() {        defer wg.Done()        for {            select {            case <-time.After(10 * time.Second):                c.Close() // 10 秒后自动断开连接            case mes := <-message:                // 收到消息后启动新的 goroutine 处理回复                wg.Add(1)                go func(c net.Conn, text string) {                    defer wg.Done()                    // 实现 echo 功能                    fmt.Fprintln(c, "\t", strings.ToUpper(text))                    time.Sleep(1 * time.Second)                    fmt.Fprintln(c, "\t", text)                    time.Sleep(1 * time.Second)                    fmt.Fprintln(c, "\t", strings.ToLower(text))                    // 除非出现错误,否则不使用 Ctrl + w 中断                }(c, mes)            }        }    }()    // 等待 input 中的所有消息处理完成    for input.Scan() {        text := input.Text()        if text != "" {            message <- text        }    }    <-time.After(time.Millisecond) // 确保所有 goroutine 完成等待    c.Close()}

    改造要点总结

  • 定时器与 goroutine 结合使用

    通过 select 语句配合 time.Afterellyً،实现了客户端 10 秒无响应时自动断开连接功能

  • 消息传递与 goroutine 框架

    利用 sync.WaitGroup 进行 goroutine 的同步管理,确保并发操作的顺序性

  • 客户端消息处理

    实现了消息的读取与转发,客户端输入内容将被自动转发给另一 goroutine 进行处理

  • 总结

    本次改造主要通过增加定时器机制,实现了客户端连接的自动断开功能。在代码实现上,主要采用 goroutine 和 select 语句进行并发控制,并通过 sync.WaitGroup 进行 goroutine 的同步。这一设计既保持了原有 echo 服务器的功能,又增加了更加灵活的服务器控制流程,为后续功能扩展提供了很好的基础。

    转载地址:http://stnmz.baihongyu.com/

    你可能感兴趣的文章
    mysql 权限整理记录
    查看>>
    mysql 权限登录问题:ERROR 1045 (28000): Access denied for user ‘root‘@‘localhost‘ (using password: YES)
    查看>>
    MYSQL 查看最大连接数和修改最大连接数
    查看>>
    MySQL 查看有哪些表
    查看>>
    mysql 查看锁_阿里/美团/字节面试官必问的Mysql锁机制,你真的明白吗
    查看>>
    MySql 查询以逗号分隔的字符串的方法(正则)
    查看>>
    MySQL 查询优化:提速查询效率的13大秘籍(避免使用SELECT 、分页查询的优化、合理使用连接、子查询的优化)(上)
    查看>>
    mysql 查询数据库所有表的字段信息
    查看>>
    【Java基础】什么是面向对象?
    查看>>
    mysql 查询,正数降序排序,负数升序排序
    查看>>
    MySQL 树形结构 根据指定节点 获取其下属的所有子节点(包含路径上的枝干节点和叶子节点)...
    查看>>
    mysql 死锁 Deadlock found when trying to get lock; try restarting transaction
    查看>>
    mysql 死锁(先delete 后insert)日志分析
    查看>>
    MySQL 死锁了,怎么办?
    查看>>
    MySQL 深度分页性能急剧下降,该如何优化?
    查看>>
    MySQL 深度分页性能急剧下降,该如何优化?
    查看>>
    MySQL 添加列,修改列,删除列
    查看>>
    mysql 添加索引
    查看>>
    MySQL 添加索引,删除索引及其用法
    查看>>
    mysql 状态检查,备份,修复
    查看>>