[Go]TCP服务中读写进行协程分离
2023-02-18 15:41:12 时间
读写两部分进行一下分离,中间通过chan进行传递数据 ,这样可以方便的在write中进行一些业务处理
single/snet/tcpconn.go
package snet import ( "bufio" "fmt" "log" "net" ) type Conn struct { IP string Port uint32 TCPConn *net.TCPConn MsgChan chan []byte ExitChan chan bool Closed bool } func NewConn(IP string, Port uint32) *Conn { s := &Conn{ IP: IP, Port: Port, MsgChan:make(chan []byte), ExitChan: make(chan bool), } return s } func (c *Conn) Start() { log.Printf("%s:%d start...\n", c.IP, c.Port) go func() { addr, err := net.ResolveTCPAddr("tcp4", fmt.Sprintf("%s:%d", c.IP, c.Port)) if err != nil { log.Println("resolve tcp addr err ", err) return } listener, err := net.ListenTCP("tcp4", addr) if err != nil { log.Println("listen tcp err ", err) return } var connid uint32 connid = 0 for { conn, err := listener.AcceptTCP() if err != nil { log.Println("accept tcp err ", err) continue } c.TCPConn = conn go c.StartRead() go c.StartWrite() connid++ } }() select {} } func (c *Conn) StartRead() { log.Println("read groutine is waiting") defer c.Stop() defer log.Println("read groutine exit") reader := bufio.NewReader(c.TCPConn) for { lineBytes, err := reader.ReadBytes('\n') if err != nil { log.Println("startread read bytes error ", err) break } len:=len(lineBytes) line:=lineBytes[:len-1] log.Println("start read from client ",string(line)) go c.HandleMsg(line) } } func (c *Conn) StartWrite() { log.Println("write groutine is waiting") defer log.Println("write groutine exit") for { select { case data := <-c.MsgChan: if _, err := c.TCPConn.Write(data); err != nil { log.Println("startwrite conn write error ", err) return } log.Println("start write from server ",string(data)) case <-c.ExitChan: return } } } func (c *Conn) HandleMsg(data []byte) { res := fmt.Sprintf("res:%s", string(data)) c.MsgChan <- []byte(res) } func (c *Conn) Stop() { if c.Closed { return } c.Closed = true c.ExitChan <- true c.TCPConn.Close() close(c.ExitChan) close(c.MsgChan) }
测试代码,使用上面的包:
server.go
package main import "single/snet" func main(){ s:=snet.NewConn("0.0.0.0",8777) s.Start() }
相关文章
- 一个故事看懂Linux文件权限管理
- 闯荡Linux帝国:nginx的创业故事
- 实践GoF的设计模式:单例模式
- Go 1.18 新特性:多模块工作区模式
- 用过 mongodb 吧, 这三个大坑踩过吗?
- 教你配置windows上的windbg,linux上的lldb,打入clr内部这一篇就够了
- 使用mongodb作为Quartz.Net下的JobStore实现底层的持久化机制
- mongodb之使用explain和hint性能分析和优化
- mongodb 3.x 之实用新功能窥看[2] ——使用$lookup做多表关联处理
- mongodb 3.x 之实用新功能窥看[1] ——使用TTLIndex做Cache处理
- 双十一来了,别让你的mongodb宕机了
- 分享几个你可能不知道的交互式Git 命令
- 实践GoF的23的设计模式:SOLID原则(下)
- 实践GoF的23种设计模式:SOLID原则(上)
- k8s源码Client-go中Reflector解析
- HTTP流量神器Goreplay核心源码详解
- 手把手带你基于嵌入式Linux移植samba服务
- 8天学通MongoDB——第八天 驱动实践
- 8天学通MongoDB——第七天 运维技术
- 8天学通MongoDB——第六天 分片技术