• 本文实现一个Echo TCP Server

interface/tcp/Handler.go

  1. type Handler interface {
  2. Handle(ctx context.Context, conn net.Conn)
  3. Close() error
  4. }
  • Handler:业务逻辑的处理接口

    • Handle(ctx context.Context, conn net.Conn) 处理连接

tcp/server.go

  1. type Config struct {
  2. Address string
  3. }
  4. func ListenAndServeWithSignal(cfg *Config, handler tcp.Handler) error {
  5. closeChan := make(chan struct{})
  6. listen, err := net.Listen("tcp", cfg.Address)
  7. if err != nil {
  8. return err
  9. }
  10. logger.Info("start listen")
  11. ListenAndServe(listen, handler, closeChan)
  12. return nil
  13. }
  14. func ListenAndServe(listener net.Listener,
  15. handler tcp.Handler,
  16. closeChan <-chan struct{}) {
  17. ctx := context.Background()
  18. var waitDone sync.WaitGroup
  19. for true {
  20. conn, err := listener.Accept()
  21. if err != nil {
  22. break
  23. }
  24. logger.Info("accept link")
  25. waitDone.Add(1)
  26. go func() {
  27. defer func() {
  28. waitDone.Done()
  29. }()
  30. handler.Handler(ctx, conn)
  31. }()
  32. }
  33. waitDone.Wait()
  34. }
  • Config:启动tcp服务器的配置

    • Address:监听地址
  • ListenAndServe:ctx是上下文,可以传递一些参数。死循环中接收到新连接时,让一个协程去处理连接
  • 如果listener.Accept()出错了就会break跳出来,这时候需要等待已经服务的客户端退出。使用WaitGroup等待客服端退出
  1. func ListenAndServe(listener net.Listener,
  2. handler tcp.Handler,
  3. closeChan <-chan struct{}) {
  4. go func() {
  5. <-closeChan
  6. logger.Info("shutting down...")
  7. _ = listener.Close()
  8. _ = handler.Close()
  9. }()
  10. defer func() {
  11. _ = listener.Close()
  12. _ = handler.Close()
  13. }()
  14. ......
  15. }

listener和handler在退出的时候需要关掉。如果用户直接kill掉了程序,我们也需要关掉listener和handler,这时候要使用closeChan,一旦接收到关闭信号,就执行关闭逻辑

  1. func ListenAndServeWithSignal(cfg *Config, handler tcp.Handler) error {
  2. closeChan := make(chan struct{})
  3. sigCh := make(chan os.Signal)
  4. signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
  5. go func() {
  6. sig := <-sigCh
  7. switch sig {
  8. case syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT:
  9. closeChan <- struct{}{}
  10. }
  11. }()
  12. listen, err := net.Listen("tcp", cfg.Address)
  13. if err != nil {
  14. return err
  15. }
  16. logger.Info("start listen")
  17. ListenAndServe(listen, handler, closeChan)
  18. return nil
  19. }

当系统对程序发送信号时,sigCh会接收到信号

tcp/echo.go

  1. type EchoHandler struct {
  2. activeConn sync.Map
  3. closing atomic.Boolean
  4. }

EchoHandler:

  • activeConn:记录连接
  • closing:是否正在关闭,有并发竞争,使用atomic.Boolean
  1. type EchoClient struct {
  2. Conn net.Conn
  3. Waiting wait.Wait
  4. }
  5. func (c *EchoClient) Close() error {
  6. c.Waiting.WaitWithTimeout(10 * time.Second)
  7. _ = c.Conn.Close()
  8. return nil
  9. }

EchoClient:一个客户端就是一个连接。Close方法关闭客户端连接,超时时间设置为10s

  1. func MakeHandler() *EchoHandler {
  2. return &EchoHandler{}
  3. }
  4. func (h *EchoHandler) Handle(ctx context.Context, conn net.Conn) {
  5. // 连接正在关闭,不接收新连接
  6. if h.closing.Get() {
  7. _ = conn.Close()
  8. }
  9. client := &EchoClient{
  10. Conn: conn,
  11. }
  12. h.activeConn.Store(client, struct{}{})
  13. reader := bufio.NewReader(conn)
  14. for {
  15. msg, err := reader.ReadString('\n')
  16. if err != nil {
  17. if err == io.EOF {
  18. logger.Info("connection close")
  19. h.activeConn.Delete(client)
  20. } else {
  21. logger.Warn(err)
  22. }
  23. return
  24. }
  25. // 正在处理业务,不要关掉
  26. client.Waiting.Add(1)
  27. // 将数据原封不动写回去,测试
  28. b := []byte(msg)
  29. _, _ = conn.Write(b)
  30. client.Waiting.Done()
  31. }
  32. }
  33. func (h *EchoHandler) Close() error {
  34. logger.Info("handler shutting down...")
  35. h.closing.Set(true)
  36. h.activeConn.Range(func(key interface{}, val interface{}) bool {
  37. client := key.(*EchoClient)
  38. _ = client.Close()
  39. return true
  40. })
  41. return nil
  42. }
  • MakeEchoHandler:创建EchoHandler
  • Handle:处理客户端的连接。
    • 1.连接正在关闭时,不接收新连接
    • 2.存储新连接,value用空结构体
    • 3.使用缓存区接收用户发来的数据,使用\n作为结束的标志
  • Close:将所有客户端连接关掉

main.go

  1. const configFile string = "redis.conf"
  2. var defaultProperties = &config.ServerProperties{
  3. Bind: "0.0.0.0",
  4. Port: 6379,
  5. }
  6. func fileExists(filename string) bool {
  7. info, err := os.Stat(filename)
  8. return err == nil && !info.IsDir()
  9. }
  10. func main() {
  11. logger.Setup(&logger.Settings{
  12. Path: "logs",
  13. Name: "godis",
  14. Ext: "log",
  15. TimeFormat: "2022-02-02",
  16. })
  17. if fileExists(configFile) {
  18. config.SetupConfig(configFile)
  19. } else {
  20. config.Properties = defaultProperties
  21. }
  22. err := tcp.ListenAndServeWithSignal(
  23. &tcp.Config{
  24. Address: fmt.Sprintf("%s:%d",
  25. config.Properties.Bind,
  26. config.Properties.Port),
  27. },
  28. EchoHandler.MakeHandler())
  29. if err != nil {
  30. logger.Error(err)
  31. }
  32. }

测试

GO实现Redis:GO实现TCP服务器(1)的更多相关文章

  1. redis来共享各个服务器的session,并同时通过redis来缓存一些常用的资源,加快用户获得请求资源的速度(转)

    时间过得真快,再次登录博客园来写博,才发现距离上次的写博时间已经过去了一个月了,虽然是因为自己找了实习,但这也说明自己对时间的掌控能力还是没那么的强,哈哈,看来还需不断的努力啊!(这里得特别说明一下本 ...

  2. php+redis实现多台服务器内网存储session并读取

    大型网站由于大并发的问题会导致系统出现诡异的崩溃性问题这着实让人很是蛋疼,首先考虑的就是负载均衡服务器来处理这个,当然数据库的性能也是非常非常重要的,今天就说下在负载均衡情况下对于session这个问 ...

  3. 【Swoole】简单安装与创建TCP服务器

    pecl install swoole PHP的异步.并行.高性能网络通信引擎,使用纯C语言编写,提供了php语言的异步多线程服务器,异步TCP/UDP网络客户端,异步MySQL,异步Redis,数据 ...

  4. Socket Server-基于线程池的TCP服务器

    了解线程池 在http://blog.csdn.net/ns_code/article/details/14105457(读书笔记一:TCP Socket)这篇博文中,服务器端采用的实现方式是:一个客 ...

  5. Python TCP服务器

    TCP服务器的创建也比较简单: 1.建立一个socket对象 2.绑定要监听的IP地址和端口 3.当有客户端请求时,启动多线程处理客户端连接 import socket import threadin ...

  6. [安卓] 9、线程、VIEW、消息实现从TCP服务器获取数据动态加载显示

    一.前言: 一般情况下从TCP服务器读取数据是放在一个线程里读的,但是刷新界面又不得不放在线程外面,所以需要用消息传递把线程里从TCP里获得的数据传送出来,然后根据数据对页面进行相应的刷新. 二.业务 ...

  7. Mina、Netty、Twisted一起学(一):实现简单的TCP服务器

    MINA.Netty.Twisted为什么放在一起学习?首先,不妨先分别看一下它们官方网站对其的介绍: MINA: Apache MINA is a network application frame ...

  8. TCP服务器不回复SYN的问题

    个人问题发生环境: 1.TCP服务器是虚拟机,IP地址是192.168.8.12. 2.TCP客户端是宿主机,IP地址是192.168.8.11. 3.从宿主机(192.168.8.11)上启动Soc ...

  9. 高性能、高并发TCP服务器(多线程调用libevent)

    from:http://blog.csdn.net/i_am_jojo/article/details/7587838 本文讲述的TCP服务器是模仿memcache中的TCP网络处理框架,其中是基于l ...

  10. python socket之tcp服务器与客户端demo

    python socket之tcp服务器与客户端demo 作者:vpoet mails:vpoet_sir@163.com server: # -*- coding: cp936 -*- ''' 建立 ...

随机推荐

  1. Pnetlab中锐捷镜像反复重启或telnet无法键入内容

    PNETLab 版本: 5.2.7 或 5.3.3等 锐捷镜像版本: V1.03 故障详情: 基于前文的系统环境和锐捷镜像.替换后的yml文件,更新PnetLAB版本到5.3.3后,设备循环重启,无法 ...

  2. C# DataTable.Select()根据条件筛选数据

    1.前言: 很多时候我们获取到一个表的时候需要根据表的包含的队列去筛选内容,一般来说可能想到的就是遍历整个表的内容进行条件筛选,但是这种方式增加了代码量且易出错,DataTable.Select()就 ...

  3. kafka工具的使用-发送数据

    1.了解推送数据的是哪个topic,选择对应topic下面的partition分区 2.右侧界面选择Data,并点击『+』号: 3.可选择添加单个消息『add Single Message』或者添加多 ...

  4. 修改tomcat启动时,修改默认访问的页面

  5. vue中模块化后mapState的使用

    代码如下: 相当于声明了一个变量name,然后以state入参取得其modules文件夹中user文件里的name属性.因为在模块(如user)中,在抛出时的export default中添加了一句: ...

  6. eclipse项目右键属性java Build Path等丢失

    https://www.cnblogs.com/IDECIDETODOIT/p/3906831.html https://zhidao.baidu.com/question/133415688.htm ...

  7. 自学JavaDay02_class01

    注释 单行注释: //单行注释 多行注释 /** 多行注释* 多行注释* */ 文档注释 /** * 文档注释 * 文档注释 */ 标识符 关键字 标识符 所有的标识符都应该以字母(A-Z 或者 a- ...

  8. FileLocator Pro注册机

  9. android 下载安装包更新app

    1下载 public void download(String url) { Utils.Loge("download",url); Uri uri = Uri.parse(url ...

  10. auto 类型说明符

    编程时常常需要把表达式的值赋给变量,这就要求在声明变量时清楚地知道表达式的类型.然而做到这一点并非那么容易,有时候甚至根本做不到.为了解决这个问题,c++11新标准引入了auto类型说明符,用它就能让 ...