本文整理匯總了Golang中net.UDPConn.CloseWrite方法的典型用法代碼示例。如果您正苦於以下問題：Golang UDPConn.CloseWrite方法的具體用法？Golang UDPConn.CloseWrite怎麽用？Golang UDPConn.CloseWrite使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類net.UDPConn的用法示例。
在下文中一共展示了UDPConn.CloseWrite方法的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)
}()
}()
}