You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.8 KiB
83 lines
1.8 KiB
package pex
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/tendermint/tendermint/libs/tempfile"
|
|
)
|
|
|
|
/* Loading & Saving */
|
|
|
|
type addrBookJSON struct {
|
|
Key string `json:"key"`
|
|
Addrs []*knownAddress `json:"addrs"`
|
|
}
|
|
|
|
func (a *addrBook) saveToFile(filePath string) {
|
|
a.mtx.Lock()
|
|
defer a.mtx.Unlock()
|
|
|
|
a.Logger.Info("Saving AddrBook to file", "size", a.size())
|
|
|
|
addrs := make([]*knownAddress, 0, len(a.addrLookup))
|
|
for _, ka := range a.addrLookup {
|
|
addrs = append(addrs, ka)
|
|
}
|
|
aJSON := &addrBookJSON{
|
|
Key: a.key,
|
|
Addrs: addrs,
|
|
}
|
|
|
|
jsonBytes, err := json.MarshalIndent(aJSON, "", "\t")
|
|
if err != nil {
|
|
a.Logger.Error("Failed to save AddrBook to file", "err", err)
|
|
return
|
|
}
|
|
err = tempfile.WriteFileAtomic(filePath, jsonBytes, 0644)
|
|
if err != nil {
|
|
a.Logger.Error("Failed to save AddrBook to file", "file", filePath, "err", err)
|
|
}
|
|
}
|
|
|
|
// Returns false if file does not exist.
|
|
// cmn.Panics if file is corrupt.
|
|
func (a *addrBook) loadFromFile(filePath string) bool {
|
|
// If doesn't exist, do nothing.
|
|
_, err := os.Stat(filePath)
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
|
|
// Load addrBookJSON{}
|
|
r, err := os.Open(filePath)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Error opening file %s: %v", filePath, err))
|
|
}
|
|
defer r.Close()
|
|
aJSON := &addrBookJSON{}
|
|
dec := json.NewDecoder(r)
|
|
err = dec.Decode(aJSON)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Error reading file %s: %v", filePath, err))
|
|
}
|
|
|
|
// Restore all the fields...
|
|
// Restore the key
|
|
a.key = aJSON.Key
|
|
// Restore .bucketsNew & .bucketsOld
|
|
for _, ka := range aJSON.Addrs {
|
|
for _, bucketIndex := range ka.Buckets {
|
|
bucket := a.getBucket(ka.BucketType, bucketIndex)
|
|
bucket[ka.Addr.String()] = ka
|
|
}
|
|
a.addrLookup[ka.ID()] = ka
|
|
if ka.BucketType == bucketTypeNew {
|
|
a.nNew++
|
|
} else {
|
|
a.nOld++
|
|
}
|
|
}
|
|
return true
|
|
}
|