Minor improvement

Author: web-flow

is.go

@@ -103,7 +103,9 @@ func (wbrc *Archiver) Wayback(links []string) (map[string]string, error) {
 	wg.Wait()
 
 	// Close tor connection
-	done <- true
+	defer func() {
+		done <- true
+	}()
 
 	if len(results) == 0 {
 		return results, fmt.Errorf("No results")
@@ -154,6 +156,11 @@ func (wbrc *Archiver) Playback(links []string) (map[string]string, error) {
 	}
 	wg.Wait()
 
+	// Close tor connection
+	defer func() {
+		done <- true
+	}()
+
 	if len(results) == 0 {
 		return results, fmt.Errorf("No results")
 	}

tor.go

@@ -54,18 +54,14 @@ func newTorClient(done <-chan bool) (*http.Client, error) {
 		}
 
 		go func() {
-			// Auto close tor client after 10 min
-			tick := time.NewTicker(10 * time.Minute)
 			for {
 				select {
 				case <-done:
 					logger.Debug("Closed tor client")
-					tick.Stop()
 					t.Close()
 					return
-				case <-tick.C:
+				case <-time.After(10 * time.Minute):
 					logger.Debug("Closed tor client, timeout")
-					tick.Stop()
 					t.Close()
 					return
 				default: