专栏首页陶士涵的菜地[Go]TCP服务中读写进行协程分离

[Go]TCP服务中读写进行协程分离

读写两部分进行一下分离,中间通过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()
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • [Go] 轻量服务器框架基础TCP服务模块

    陶士涵
  • [Go] 轻量服务器框架基础TCP连接的抽象和封装

    陶士涵
  • [日常] Go语言圣经--示例: 并发的Echo服务

    1.io.Copy()方法 func Copy(dst Writer, src Reader) (written int64, err error)

    陶士涵
  • [Go] 轻量服务器框架基础TCP服务模块

    陶士涵
  • PL/SQL 下SQL结果集以html形式发送邮件

          在运维的过程中,有时候需要定时将SQL查询的数据结果集以html表格形式发送邮件,因此需要将SQL查询得到的结果集拼接成html代码。对于这种情形通...

    Leshami
  • Go实现海量日志收集系统(四)

    coders
  • [Go]TCP服务中增加消息队列与工作池

    之前的处理中每一个连接都会创建一个主groutine , 每个连接中的主groutine中创建出读groutine 和写groutine

    陶士涵
  • 如何优雅地处理Async/Await的异常?

    async/await 中的异常处理很让人混乱。尽管有很多种方式来应对async 函数的异常,但是连经验丰富的开发者有时候也会搞错。

    Fundebug
  • Django+Bootstrap+Mysql 搭建个人博客(三)

    zhang_derek
  • JS创建类的两种常用方式

    IT故事会

扫码关注云+社区

领取腾讯云代金券