mirror of
https://github.com/SleepingBag945/dddd.git
synced 2025-06-08 22:16:43 +00:00
24 lines
448 B
Go
24 lines
448 B
Go
|
package common
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func WrapperTcpWithTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
|
||
|
|
d := &net.Dialer{Timeout: timeout}
|
||
|
|
return WrapperTCP(network, address, d)
|
||
|
|
}
|
||
|
|
|
||
|
|
func WrapperTCP(network, address string, forward *net.Dialer) (net.Conn, error) {
|
||
|
|
//get conn
|
||
|
|
var conn net.Conn
|
||
|
|
var err error
|
||
|
|
conn, err = forward.Dial(network, address)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return conn, nil
|
||
|
|
|
||
|
|
}
|