From 590efc10404e3bf7795e382012d71b0005f44693 Mon Sep 17 00:00:00 2001
From: Anton Kaliaev <anton.kalyaev@gmail.com>
Date: Mon, 16 Jan 2017 23:36:10 +0400
Subject: [PATCH] call saveToFile OnStop

This is better than waiting because while we wait, anything could happen
(crash, timeout of the code who's using addrbook, ...). If we save
immediately, we have much greater chances of success.
---
 addrbook.go | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/addrbook.go b/addrbook.go
index ace0dba4..7a65cb4b 100644
--- a/addrbook.go
+++ b/addrbook.go
@@ -15,7 +15,6 @@ import (
 	"time"
 
 	. "github.com/tendermint/go-common"
-	"github.com/tendermint/go-crypto"
 )
 
 const (
@@ -86,7 +85,6 @@ type AddrBook struct {
 	addrLookup        map[string]*knownAddress // new & old
 	addrNew           []map[string]*knownAddress
 	addrOld           []map[string]*knownAddress
-	wg                sync.WaitGroup
 	nOld              int
 	nNew              int
 }
@@ -128,7 +126,6 @@ func (a *AddrBook) init() {
 func (a *AddrBook) OnStart() error {
 	a.BaseService.OnStart()
 	a.loadFromFile(a.filePath)
-	a.wg.Add(1)
 	go a.saveRoutine()
 	return nil
 }
@@ -139,6 +136,7 @@ func (a *AddrBook) OnStop() {
 
 func (a *AddrBook) Wait() {
 	a.wg.Wait()
+	a.saveToFile(a.filePath)
 }
 
 func (a *AddrBook) AddOurAddress(addr *NetAddress) {
@@ -309,6 +307,8 @@ type addrBookJSON struct {
 }
 
 func (a *AddrBook) saveToFile(filePath string) {
+	log.Info("Saving AddrBook to file", "size", a.Size())
+
 	// Compile Addrs
 	addrs := []*knownAddress{}
 	for _, ka := range a.addrLookup {
@@ -386,7 +386,6 @@ out:
 	for {
 		select {
 		case <-dumpAddressTicker.C:
-			log.Info("Saving AddrBook to file", "size", a.Size())
 			a.saveToFile(a.filePath)
 		case <-a.Quit:
 			break out
@@ -394,7 +393,6 @@ out:
 	}
 	dumpAddressTicker.Stop()
 	a.saveToFile(a.filePath)
-	a.wg.Done()
 	log.Notice("Address handler done")
 }