41 lines
742 B
Go
41 lines
742 B
Go
|
package decorators
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Dialer interface {
|
||
|
|
DialTimeout(network, address string, timeout time.Duration) (net.Conn, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type NetDialer struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d NetDialer) DialTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
|
||
|
|
return net.DialTimeout(network, address, timeout)
|
||
|
|
}
|
||
|
|
|
||
|
|
type DialerDecorator interface {
|
||
|
|
DialPort(network, addr string, p int) int
|
||
|
|
}
|
||
|
|
|
||
|
|
type NetDecorator struct {
|
||
|
|
Dialer Dialer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d NetDecorator) DialPort(network, addr string, p int) int {
|
||
|
|
timeout := time.Second
|
||
|
|
c, err := d.Dialer.DialTimeout(network, addr+":"+strconv.Itoa(p), timeout)
|
||
|
|
if err != nil {
|
||
|
|
return 0
|
||
|
|
} else {
|
||
|
|
err = c.Close()
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return p
|
||
|
|
}
|
||
|
|
}
|