fumiama/go-registry
1
0
Fork 0
mirror of https://github.com/fumiama/go-registry.git synced 2026-06-27 14:20:26 +08:00
Code Activity

fix: Close

Browse Source
This commit is contained in:
源文雨
2022-03-28 17:06:52 +08:00
parent 5fd402268a
commit b75079a9ca
1 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
reg.go
View File

@@ -55,22 +55,26 @@ func NewRegReader(addr, pwd string) *Regedit {
} }
func (r *Regedit) Connect() (err error) { func (r *Regedit) Connect() (err error) {
r.Lock() if !r.isopen {
r.conn, err = net.Dial("tcp", r.addr) r.Lock()
if err != nil { r.conn, err = net.Dial("tcp", r.addr)
r.isopen = true if err == nil {
r.isopen = true
}
r.Unlock()
} }
r.Unlock()
return return
} }
func (r *Regedit) ConnectIn(wait time.Duration) (err error) { func (r *Regedit) ConnectIn(wait time.Duration) (err error) {
r.Lock() if !r.isopen {
r.conn, err = net.DialTimeout("tcp", r.addr, wait) r.Lock()
if err != nil { r.conn, err = net.DialTimeout("tcp", r.addr, wait)
r.isopen = true if err == nil {
r.isopen = true
}
r.Unlock()
} }
r.Unlock()
return return
} }
@@ -83,7 +87,9 @@ func (r *Regedit) Close() (err error) {
p.Put() p.Put()
r.seq = 0 r.seq = 0
r.isopen = false r.isopen = false
return r.conn.Close() err = r.conn.Close()
r.conn = nil
return
} }
return return
} }