golang chan 使用例子

最新推荐文章于 2024-05-24 00:18:16 发布

wshelly

最新推荐文章于 2024-05-24 00:18:16 发布

阅读量752

点赞数

分类专栏：总结文章标签： golang

总结专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一个自定义的TCP监听器实现方法，包括初始化监听器、监听TCP连接、接受客户端连接及关闭监听器等功能。该监听器支持多个地址监听，并通过内部协程处理连接请求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

type TcpListeners struct {
    conns  chan *net.TCPConn
    closing chan bool
    wait *sync.WaitGroup
}

func NewTcpListeners(addrs []string) (v *TcpListeners, err error) {
    v = &TcpListeners{
        addrs:     addrs,
        conns:     make(chan *net.TCPConn),
        closing:   make(chan bool, 1),
        wait:      &sync.WaitGroup{},
    }

    return
}

// Listen at addrs format as netowrk://laddr, for example,
// tcp://:1935, tcp4://:1935, tcp6://1935, tcp://0.0.0.0:1935
func (v *TcpListeners) ListenTCP() (err error) {
    for _, addr := range v.addrs {
        vs := strings.Split(addr, "://")
        network, laddr := vs[0], vs[1]

        if l, err := net.Listen(network, laddr); err != nil {
            return nil,err
        } else {
            v.listeners = append(v.listeners, l.(*net.TCPListener))
        }
    }

    v.wait.Add(len(v.listeners))
    for i, l := range v.listeners {
        addr := v.addrs[i]
        go func(l *net.TCPListener, addr string) {
            defer v.wait.Done()
            for {
                var conn *net.TCPConn
                if conn, err = l.AcceptTCP(); err != nil {
                    return
                }

                select {
                case v.conns <- conn:
                case c := <-v.closing:
                    v.closing <- c
                    conn.Close()
                }
            }
        }(l, addr)
    }

    return
}

func (v *TcpListeners) AcceptTCP() (c *net.TCPConn, err error) {
    var ok bool
    if c,ok = <- v.conns; !ok {
        return nil, ListenerDisposed
    }
    return
}

func (v *TcpListeners) Close() (err error) {
    // unblock all listener internal goroutines
    v.closing <- true

    // interrupt all listeners.
    for _, v := range v.listeners {
        if r := v.Close(); r != nil {
            err = r
        }
    }

    // wait for all listener internal goroutines to quit.
    v.wait.Wait()

    // clear the closing signal.
    _ = <-v.closing

    // close channels to unblock the user goroutine to AcceptTCP()
    close(v.conns)

    return
}