fumiama/go-registry
1
0
Fork 0
mirror of https://github.com/fumiama/go-registry.git synced 2026-06-23 12:00:32 +08:00
Code Activity

add ConnectIn api

Browse Source
This commit is contained in:
fumiama
2021-12-13 21:11:14 +08:00
parent 6ffb6c1f4c
commit caa589d23f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
reg.go
View File

@@ -3,6 +3,7 @@ package registry
import ( import (
"errors" "errors"
"net" "net"
"time"
tea "github.com/fumiama/gofastTEA" tea "github.com/fumiama/gofastTEA"
"github.com/wdvxdr1123/ZeroBot/utils/helper" "github.com/wdvxdr1123/ZeroBot/utils/helper"
@@ -45,6 +46,11 @@ func (r *Regedit) Connect() (err error) {
return 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) { func (r *Regedit) Close() (err error) {
p := NewCmdPacket(CMDEND, []byte("fill"), &r.tp) p := NewCmdPacket(CMDEND, []byte("fill"), &r.tp)
r.conn.Write(p.Encrypt(r.seq)) r.conn.Write(p.Encrypt(r.seq))