前端之家收集整理的这篇文章主要介绍了
golang chan 使用例子,
前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
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
}
原文链接:https://www.f2er.com/go/188755.html