Close reprovider in the provider system

License: MIT Signed-off-by: Michael Avila <[email protected]>
ipfs · May 6, 2019 · b47b4aa · b47b4aa
1 parent 0e890fc
commit b47b4aa
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/provider/provider.go b/provider/provider.go
@@ -30,4 +30,6 @@ type Reprovider interface {
 	Run()
 	// Trigger a reprovide
 	Trigger(context.Context) error
+	// Close stops the reprovider
+	Close() error
 }
diff --git a/provider/system.go b/provider/system.go
@@ -32,8 +32,20 @@ func (s *system) Run() {
 
 // Close the provider and reprovider
 func (s *system) Close() error {
-	// TODO: Close reprovider here
-	return s.provider.Close()
+	var errs []error
+
+	if err := s.provider.Close(); err != nil {
+		errs = append(errs, err)
+	}
+
+	if err := s.reprovider.Close(); err != nil {
+		errs = append(errs, err)
+	}
+
+	if len(errs) > 0 {
+		return errs[0]
+	}
+	return nil
 }
 
 // Provide a value