本文整理汇总了Golang中net.UDPConn.CloseRead方法的典型用法代码示例。如果您正苦于以下问题:Golang UDPConn.CloseRead方法的具体用法?Golang UDPConn.CloseRead怎么用?Golang UDPConn.CloseRead使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类net.UDPConn
的用法示例。
在下文中一共展示了UDPConn.CloseRead方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Golang代码示例。
示例1: proxyUDPStream
func (p *Proxy) proxyUDPStream(ctx context.Context, src *net.UDPConn) {
srcRemoteAddr := src.RemoteAddr().(*net.UDPAddr)
srcLocalAddr := src.LocalAddr().(*net.UDPAddr)
route := p.routes.GetTable().Lookup(protocols.UDP,
srcRemoteAddr.IP, srcLocalAddr.IP,
uint16(srcRemoteAddr.Port), uint16(srcLocalAddr.Port))
if route == nil {
src.Close()
return
}
go func() {
dstAddr := net.UDPAddr{
IP: route.Outbound.DstIP,
Port: int(route.Outbound.DstPort),
}
dst, err := net.DialUDP("udp", nil, &dstAddr)
if err != nil {
src.Close()
return
}
dst.SetKeepAlivePeriod(10 * time.Second)
src.SetKeepAlivePeriod(10 * time.Second)
go func() {
<-ctx.Done()
src.Close()
dst.Close()
}()
go func() {
defer dst.CloseWrite()
defer src.CloseRead()
io.Copy(dst, src)
}()
go func() {
defer src.CloseWrite()
defer dst.CloseRead()
io.Copy(src, dst)
}()
}()
}