add Close method to interface, fix closing of StaticAutoNAT

libp2p · Aug 29, 2021 · 1be3293 · 1be3293
1 parent e6e029f
commit 1be3293
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/autonat.go b/autonat.go
@@ -457,3 +457,8 @@ func (s *StaticAutoNAT) PublicAddr() (ma.Multiaddr, error) {
 	}
 	return nil, errors.New("no available address")
 }
+
+func (s *StaticAutoNAT) Close() error {
+	s.service.Disable()
+	return nil
+}
diff --git a/interface.go b/interface.go
@@ -2,6 +2,7 @@ package autonat
 
 import (
 	"context"
+	"io"
 
 	"github.com/libp2p/go-libp2p-core/network"
 	"github.com/libp2p/go-libp2p-core/peer"
@@ -16,6 +17,7 @@ type AutoNAT interface {
 	// PublicAddr returns the public dial address when NAT status is public and an
 	// error otherwise
 	PublicAddr() (ma.Multiaddr, error)
+	io.Closer
 }
 
 // Client is a stateless client interface to AutoNAT peers