diff --git a/reg.go b/reg.go index 8879dc0..e911c22 100644 --- a/reg.go +++ b/reg.go @@ -3,6 +3,7 @@ package registry import ( "errors" "net" + "time" tea "github.com/fumiama/gofastTEA" "github.com/wdvxdr1123/ZeroBot/utils/helper" @@ -45,6 +46,11 @@ func (r *Regedit) Connect() (err error) { return } +func (r *Regedit) ConnectIn(wait time.Duration) (err error) { + r.conn, err = net.DialTimeout("tcp", r.addr, wait) + return +} + func (r *Regedit) Close() (err error) { p := NewCmdPacket(CMDEND, []byte("fill"), &r.tp) r.conn.Write(p.Encrypt(r.seq))