mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-24 20:07:45 +08:00
c84a714b16
this is a major refactor of the entire codebase it changes the monolithic peer.Peer into using a peer.ID and a peer.Peerstore. Other changes: - removed handshake3. - testutil vastly simplified peer - secio bugfix + debugging logs - testutil: RandKeyPair - backpressure bugfix: w.o.w. - peer: added hex enc/dec - peer: added a PeerInfo struct PeerInfo is a small struct used to pass around a peer with a set of addresses and keys. This is not meant to be a complete view of the system, but rather to model updates to the peerstore. It is used by things like the routing system. - updated peer/queue + peerset - latency metrics - testutil: use crand for PeerID gen RandPeerID generates random "valid" peer IDs. it does not NEED to generate keys because it is as if we lost the key right away. fine to read some randomness and hash it. to generate proper keys and an ID, use: sk, pk, _ := testutil.RandKeyPair() id, _ := peer.IDFromPublicKey(pk) Also added RandPeerIDFatal helper - removed old spipe - updated seccat - core: cleanup initIdentity - removed old getFromPeerList
104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
package bitswap
|
|
|
|
import (
|
|
"time"
|
|
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
ds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore"
|
|
ds_sync "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore/sync"
|
|
blockstore "github.com/jbenet/go-ipfs/blocks/blockstore"
|
|
exchange "github.com/jbenet/go-ipfs/exchange"
|
|
tn "github.com/jbenet/go-ipfs/exchange/bitswap/testnet"
|
|
peer "github.com/jbenet/go-ipfs/peer"
|
|
mockrouting "github.com/jbenet/go-ipfs/routing/mock"
|
|
datastore2 "github.com/jbenet/go-ipfs/util/datastore2"
|
|
delay "github.com/jbenet/go-ipfs/util/delay"
|
|
)
|
|
|
|
func NewSessionGenerator(
|
|
net tn.Network, rs mockrouting.Server) SessionGenerator {
|
|
ctx, cancel := context.WithCancel(context.TODO())
|
|
return SessionGenerator{
|
|
ps: peer.NewPeerstore(),
|
|
net: net,
|
|
rs: rs,
|
|
seq: 0,
|
|
ctx: ctx, // TODO take ctx as param to Next, Instances
|
|
cancel: cancel,
|
|
}
|
|
}
|
|
|
|
type SessionGenerator struct {
|
|
seq int
|
|
net tn.Network
|
|
rs mockrouting.Server
|
|
ps peer.Peerstore
|
|
ctx context.Context
|
|
cancel context.CancelFunc
|
|
}
|
|
|
|
func (g *SessionGenerator) Close() error {
|
|
g.cancel()
|
|
return nil // for Closer interface
|
|
}
|
|
|
|
func (g *SessionGenerator) Next() Instance {
|
|
g.seq++
|
|
return session(g.ctx, g.net, g.rs, g.ps, peer.ID(g.seq))
|
|
}
|
|
|
|
func (g *SessionGenerator) Instances(n int) []Instance {
|
|
instances := make([]Instance, 0)
|
|
for j := 0; j < n; j++ {
|
|
inst := g.Next()
|
|
instances = append(instances, inst)
|
|
}
|
|
return instances
|
|
}
|
|
|
|
type Instance struct {
|
|
Peer peer.ID
|
|
Exchange exchange.Interface
|
|
blockstore blockstore.Blockstore
|
|
|
|
blockstoreDelay delay.D
|
|
}
|
|
|
|
func (i *Instance) Blockstore() blockstore.Blockstore {
|
|
return i.blockstore
|
|
}
|
|
|
|
func (i *Instance) SetBlockstoreLatency(t time.Duration) time.Duration {
|
|
return i.blockstoreDelay.Set(t)
|
|
}
|
|
|
|
// session creates a test bitswap session.
|
|
//
|
|
// NB: It's easy make mistakes by providing the same peer ID to two different
|
|
// sessions. To safeguard, use the SessionGenerator to generate sessions. It's
|
|
// just a much better idea.
|
|
func session(ctx context.Context, net tn.Network, rs mockrouting.Server, ps peer.Peerstore, p peer.ID) Instance {
|
|
|
|
adapter := net.Adapter(p)
|
|
htc := rs.Client(peer.PeerInfo{ID: p})
|
|
|
|
bsdelay := delay.Fixed(0)
|
|
const kWriteCacheElems = 100
|
|
bstore, err := blockstore.WriteCached(blockstore.NewBlockstore(ds_sync.MutexWrap(datastore2.WithDelay(ds.NewMapDatastore(), bsdelay))), kWriteCacheElems)
|
|
if err != nil {
|
|
// FIXME perhaps change signature and return error.
|
|
panic(err.Error())
|
|
}
|
|
|
|
const alwaysSendToPeer = true
|
|
|
|
bs := New(ctx, p, adapter, htc, bstore, alwaysSendToPeer)
|
|
|
|
return Instance{
|
|
Peer: p,
|
|
Exchange: bs,
|
|
blockstore: bstore,
|
|
blockstoreDelay: bsdelay,
|
|
}
|
|
}
|