[Go]TCP服务中读写进行协程分离
2023-02-18 15:41:26 时间
读写两部分进行一下分离,中间通过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() }
相关文章
- git tag 命令详解
- PHP 中的魔术方法
- PHP 拼手气红包分配算法
- git rm 从暂存区中删除内容
- git checkout 命令详解
- git add 命令详解
- git status 查看仓库文件状态
- git commit 命令详解
- PHP 获取手机号运营商
- git init 初始化仓库
- Linux NTP工具的基本使用
- Linux 系统环境监测
- Linux I/O模型
- Linux Lvs的防火墙标记
- linux LVS的DR模式多网段
- linux LVS的DR模式实现
- Linux LVS-NAT模式的实现
- linux IPtables的匹配条件
- linux IPtables 规则的管理
- Linux IPtables相关概念