libp2p/go-libp2p-peerstore
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-peerstore.git synced 2025-03-13 11:30:09 +08:00
Code Issues Projects Releases Wiki Activity

pstore.Close(): return errors from children.

Browse Source
This commit is contained in:
Raúl Kripalani 2019-02-22 14:39:51 +00:00
parent 4806c742b5
commit 8db09c59cf
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
peerstore.go
View File

@ -33,14 +33,21 @@ func NewPeerstore(kb KeyBook, ab AddrBook, md PeerMetadata) Peerstore {
}
func (ps *peerstore) Close() (err error) {
if cl, ok := ps.KeyBook.(io.Closer); ok {
cl.Close()
var errs []error
weakClose := func(name string, c interface{}) {
if cl, ok := c.(io.Closer); ok {
if err = cl.Close(); err != nil {
errs = append(errs, fmt.Errorf("%s error: %s", name, err))
}
}
}
if cl, ok := ps.AddrBook.(io.Closer); ok {
cl.Close()
}
if cl, ok := ps.PeerMetadata.(io.Closer); ok {
cl.Close()
weakClose("keybook", ps.KeyBook)
weakClose("addressbook", ps.AddrBook)
weakClose("peermetadata", ps.PeerMetadata)
if len(errs) > 0 {
return fmt.Errorf("failed while closing peerstore; err(s): %q", errs)
}
return nil
}