func (s *PtreeSuite) SetUpTest(c *gc.C) {
s.config = recon.DefaultSettings().PTreeConfig
s.path = filepath.Join(c.MkDir(), "db")
ptree, err := New(s.config, s.path)
c.Assert(err, gc.IsNil)
err = ptree.Create()
c.Assert(err, gc.IsNil)
s.ptree = ptree
}
func (s *ReconSuite) newPeer(listenPort, partnerPort int, mode recon.PeerMode, ptree recon.PrefixTree) *recon.Peer {
settings := recon.DefaultSettings()
settings.ReconAddr = fmt.Sprintf(":%d", listenPort)
partnerAddr := fmt.Sprintf("localhost:%d", partnerPort)
settings.Partners[partnerAddr] = recon.Partner{
ReconAddr: partnerAddr,
}
settings.AllowCIDRs = []string{"0.0.0.0/0"}
settings.GossipIntervalSecs = 2
peer := recon.NewPeer(settings, ptree)
peer.StartMode(mode)
return peer
}
func (s *LeveldbReconSuite) SetUpTest(c *gc.C) {
s.ReconSuite = &testing.ReconSuite{
Factory: func() (recon.PrefixTree, testing.Cleanup, error) {
path := filepath.Join(c.MkDir(), "db")
ptree, err := New(recon.DefaultSettings().PTreeConfig, path)
c.Assert(err, gc.IsNil)
err = ptree.Create()
c.Assert(err, gc.IsNil)
return ptree, func() {
ptree.Drop()
}, nil
},
}
}
func main() {
var dbDir string
if len(os.Args) < 2 {
fmt.Println("Usage: <leveldb path>")
os.Exit(1)
}
dbDir = os.Args[1]
settings := recon.DefaultSettings()
ptree, err := leveldb.New(settings.PTreeConfig, dbDir)
if err != nil {
die(err)
}
err = ptree.Create()
if err != nil {
die(err)
}
walk(ptree)
}
