go-kit
/
gomail
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set a 10 seconds timeout in Dial

This commit is contained in:
Alexandre Cesaro 2016-04-11 23:23:03 +02:00
parent 92eaa13340
commit e4bd87ad6e
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
smtp.go
View File

@ -7,6 +7,7 @@ import (
"net"
"net/smtp"
"strings"
"time"
)
// A Dialer is a dialer to an SMTP server.
@ -57,7 +58,7 @@ func NewPlainDialer(host string, port int, username, password string) *Dialer {
// Dial dials and authenticates to an SMTP server. The returned SendCloser
// should be closed when done using it.
func (d *Dialer) Dial() (SendCloser, error) {
conn, err := netDial("tcp", addr(d.Host, d.Port))
conn, err := netDialTimeout("tcp", addr(d.Host, d.Port), 10*time.Second)
if err != nil {
return nil, err
}
@ -181,9 +182,9 @@ func (c *smtpSender) Close() error {
// Stubbed out for tests.
var (
netDial = net.Dial
tlsClient = tls.Client
smtpNewClient = func(conn net.Conn, host string) (smtpClient, error) {
netDialTimeout = net.DialTimeout
tlsClient = tls.Client
smtpNewClient = func(conn net.Conn, host string) (smtpClient, error) {
return smtp.NewClient(conn, host)
}
)

3
smtp_test.go
View File

@ -8,6 +8,7 @@ import (
"net/smtp"
"reflect"
"testing"
"time"
)
const (
@ -247,7 +248,7 @@ func doTestSendMail(t *testing.T, d *Dialer, want []string, timeout bool) {
timeout: timeout,
}
netDial = func(network, address string) (net.Conn, error) {
netDialTimeout = func(network, address string, d time.Duration) (net.Conn, error) {
if network != "tcp" {
t.Errorf("Invalid network, got %q, want tcp", network)
}