opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-07-01 02:30:39 +08:00
Code Issues Projects Releases Wiki Activity

refactor symbol

Browse Source
This commit is contained in:
Juan Batiz-Benet
2014-09-16 00:57:46 -07:00
committed by Brian Tiger Chow
parent 2f2808e9d8
commit e872482aa7
1 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
routing/dht/dht.go
View File

@ -280,7 +280,7 @@ func (dht *IpfsDHT) handleGetValue(p *peer.Peer, pmes *Message) (*Message, error
} }
// Store a value in this peer local storage // Store a value in this peer local storage
func (dht *IpfsDHT) handlePutValue(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handlePutValue(p *peer.Peer, pmes *Message) {
dht.dslock.Lock() dht.dslock.Lock()
defer dht.dslock.Unlock() defer dht.dslock.Unlock()
dskey := ds.NewKey(pmes.GetKey()) dskey := ds.NewKey(pmes.GetKey())
@ -291,7 +291,7 @@ func (dht *IpfsDHT) handlePutValue(p *peer.Peer, pmes *PBDHTMessage) {
} }
} }
func (dht *IpfsDHT) handlePing(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handlePing(p *peer.Peer, pmes *Message) {
u.DOut("[%s] Responding to ping from [%s]!\n", dht.self.ID.Pretty(), p.ID.Pretty()) u.DOut("[%s] Responding to ping from [%s]!\n", dht.self.ID.Pretty(), p.ID.Pretty())
resp := Message{ resp := Message{
Type: pmes.GetType(), Type: pmes.GetType(),
@ -302,7 +302,7 @@ func (dht *IpfsDHT) handlePing(p *peer.Peer, pmes *PBDHTMessage) {
dht.netChan.Outgoing <- swarm.NewMessage(p, resp.ToProtobuf()) dht.netChan.Outgoing <- swarm.NewMessage(p, resp.ToProtobuf())
} }
func (dht *IpfsDHT) handleFindPeer(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handleFindPeer(p *peer.Peer, pmes *Message) {
resp := Message{ resp := Message{
Type: pmes.GetType(), Type: pmes.GetType(),
ID: pmes.GetId(), ID: pmes.GetId(),
@ -335,9 +335,9 @@ func (dht *IpfsDHT) handleFindPeer(p *peer.Peer, pmes *PBDHTMessage) {
resp.Success = true resp.Success = true
} }
func (dht *IpfsDHT) handleGetProviders(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handleGetProviders(p *peer.Peer, pmes *Message) {
resp := Message{ resp := Message{
Type: PBDHTMessage_GET_PROVIDERS, Type: Message_GET_PROVIDERS,
Key: pmes.GetKey(), Key: pmes.GetKey(),
ID: pmes.GetId(), ID: pmes.GetId(),
Response: true, Response: true,
@ -378,7 +378,7 @@ type providerInfo struct {
Value *peer.Peer Value *peer.Peer
} }
func (dht *IpfsDHT) handleAddProvider(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handleAddProvider(p *peer.Peer, pmes *Message) {
key := u.Key(pmes.GetKey()) key := u.Key(pmes.GetKey())
u.DOut("[%s] Adding [%s] as a provider for '%s'\n", dht.self.ID.Pretty(), p.ID.Pretty(), peer.ID(key).Pretty()) u.DOut("[%s] Adding [%s] as a provider for '%s'\n", dht.self.ID.Pretty(), p.ID.Pretty(), peer.ID(key).Pretty())
dht.providers.AddProvider(key, p) dht.providers.AddProvider(key, p)
@ -393,7 +393,7 @@ func (dht *IpfsDHT) Halt() {
} }
// NOTE: not yet finished, low priority // NOTE: not yet finished, low priority
func (dht *IpfsDHT) handleDiagnostic(p *peer.Peer, pmes *PBDHTMessage) { func (dht *IpfsDHT) handleDiagnostic(p *peer.Peer, pmes *Message) {
seq := dht.routingTables[0].NearestPeers(kb.ConvertPeerID(dht.self.ID), 10) seq := dht.routingTables[0].NearestPeers(kb.ConvertPeerID(dht.self.ID), 10)
listenChan := dht.listener.Listen(pmes.GetId(), len(seq), time.Second*30) listenChan := dht.listener.Listen(pmes.GetId(), len(seq), time.Second*30)
@ -415,7 +415,7 @@ func (dht *IpfsDHT) handleDiagnostic(p *peer.Peer, pmes *PBDHTMessage) {
//Timeout, return what we have //Timeout, return what we have
goto out goto out
case reqResp := <-listenChan: case reqResp := <-listenChan:
pmesOut := new(PBDHTMessage) pmesOut := new(Message)
err := proto.Unmarshal(reqResp.Data, pmesOut) err := proto.Unmarshal(reqResp.Data, pmesOut)
if err != nil { if err != nil {
// It broke? eh, whatever, keep going // It broke? eh, whatever, keep going
@ -428,7 +428,7 @@ func (dht *IpfsDHT) handleDiagnostic(p *peer.Peer, pmes *PBDHTMessage) {
out: out:
resp := Message{ resp := Message{
Type: PBDHTMessage_DIAGNOSTIC, Type: Message_DIAGNOSTIC,
ID: pmes.GetId(), ID: pmes.GetId(),
Value: buf.Bytes(), Value: buf.Bytes(),
Response: true, Response: true,
@ -481,9 +481,9 @@ func (dht *IpfsDHT) getValueOrPeers(p *peer.Peer, key u.Key, timeout time.Durati
} }
// getValueSingle simply performs the get value RPC with the given parameters // getValueSingle simply performs the get value RPC with the given parameters
func (dht *IpfsDHT) getValueSingle(p *peer.Peer, key u.Key, timeout time.Duration, level int) (*PBDHTMessage, error) { func (dht *IpfsDHT) getValueSingle(p *peer.Peer, key u.Key, timeout time.Duration, level int) (*Message, error) {
pmes := Message{ pmes := Message{
Type: PBDHTMessage_GET_VALUE, Type: Message_GET_VALUE,
Key: string(key), Key: string(key),
Value: []byte{byte(level)}, Value: []byte{byte(level)},
ID: swarm.GenerateMessageID(), ID: swarm.GenerateMessageID(),
@ -507,7 +507,7 @@ func (dht *IpfsDHT) getValueSingle(p *peer.Peer, key u.Key, timeout time.Duratio
} }
roundtrip := time.Since(t) roundtrip := time.Since(t)
resp.Peer.SetLatency(roundtrip) resp.Peer.SetLatency(roundtrip)
pmesOut := new(PBDHTMessage) pmesOut := new(Message)
err := proto.Unmarshal(resp.Data, pmesOut) err := proto.Unmarshal(resp.Data, pmesOut)
if err != nil { if err != nil {
return nil, err return nil, err
@ -521,7 +521,7 @@ func (dht *IpfsDHT) getValueSingle(p *peer.Peer, key u.Key, timeout time.Duratio
// one to get the value from? Or just connect to one at a time until we get a // one to get the value from? Or just connect to one at a time until we get a
// successful connection and request the value from it? // successful connection and request the value from it?
func (dht *IpfsDHT) getFromPeerList(key u.Key, timeout time.Duration, func (dht *IpfsDHT) getFromPeerList(key u.Key, timeout time.Duration,
peerlist []*PBDHTMessage_PBPeer, level int) ([]byte, error) { peerlist []*Message_PBPeer, level int) ([]byte, error) {
for _, pinfo := range peerlist { for _, pinfo := range peerlist {
p, _ := dht.Find(peer.ID(pinfo.GetId())) p, _ := dht.Find(peer.ID(pinfo.GetId()))
if p == nil { if p == nil {
@ -597,9 +597,9 @@ func (dht *IpfsDHT) Find(id peer.ID) (*peer.Peer, *kb.RoutingTable) {
return nil, nil return nil, nil
} }
func (dht *IpfsDHT) findPeerSingle(p *peer.Peer, id peer.ID, timeout time.Duration, level int) (*PBDHTMessage, error) { func (dht *IpfsDHT) findPeerSingle(p *peer.Peer, id peer.ID, timeout time.Duration, level int) (*Message, error) {
pmes := Message{ pmes := Message{
Type: PBDHTMessage_FIND_NODE, Type: Message_FIND_NODE,
Key: string(id), Key: string(id),
ID: swarm.GenerateMessageID(), ID: swarm.GenerateMessageID(),
Value: []byte{byte(level)}, Value: []byte{byte(level)},
@ -617,7 +617,7 @@ func (dht *IpfsDHT) findPeerSingle(p *peer.Peer, id peer.ID, timeout time.Durati
case resp := <-listenChan: case resp := <-listenChan:
roundtrip := time.Since(t) roundtrip := time.Since(t)
resp.Peer.SetLatency(roundtrip) resp.Peer.SetLatency(roundtrip)
pmesOut := new(PBDHTMessage) pmesOut := new(Message)
err := proto.Unmarshal(resp.Data, pmesOut) err := proto.Unmarshal(resp.Data, pmesOut)
if err != nil { if err != nil {
return nil, err return nil, err
@ -633,9 +633,9 @@ func (dht *IpfsDHT) printTables() {
} }
} }
func (dht *IpfsDHT) findProvidersSingle(p *peer.Peer, key u.Key, level int, timeout time.Duration) (*PBDHTMessage, error) { func (dht *IpfsDHT) findProvidersSingle(p *peer.Peer, key u.Key, level int, timeout time.Duration) (*Message, error) {
pmes := Message{ pmes := Message{
Type: PBDHTMessage_GET_PROVIDERS, Type: Message_GET_PROVIDERS,
Key: string(key), Key: string(key),
ID: swarm.GenerateMessageID(), ID: swarm.GenerateMessageID(),
Value: []byte{byte(level)}, Value: []byte{byte(level)},
@ -652,7 +652,7 @@ func (dht *IpfsDHT) findProvidersSingle(p *peer.Peer, key u.Key, level int, time
return nil, u.ErrTimeout return nil, u.ErrTimeout
case resp := <-listenChan: case resp := <-listenChan:
u.DOut("FindProviders: got response.\n") u.DOut("FindProviders: got response.\n")
pmesOut := new(PBDHTMessage) pmesOut := new(Message)
err := proto.Unmarshal(resp.Data, pmesOut) err := proto.Unmarshal(resp.Data, pmesOut)
if err != nil { if err != nil {
return nil, err return nil, err
@ -663,7 +663,7 @@ func (dht *IpfsDHT) findProvidersSingle(p *peer.Peer, key u.Key, level int, time
} }
// TODO: Could be done async // TODO: Could be done async
func (dht *IpfsDHT) addPeerList(key u.Key, peers []*PBDHTMessage_PBPeer) []*peer.Peer { func (dht *IpfsDHT) addPeerList(key u.Key, peers []*Message_PBPeer) []*peer.Peer {
var provArr []*peer.Peer var provArr []*peer.Peer
for _, prov := range peers { for _, prov := range peers {
// Dont add outselves to the list // Dont add outselves to the list
@ -687,7 +687,7 @@ func (dht *IpfsDHT) addPeerList(key u.Key, peers []*PBDHTMessage_PBPeer) []*peer
return provArr return provArr
} }
func (dht *IpfsDHT) peerFromInfo(pbp *PBDHTMessage_PBPeer) (*peer.Peer, error) { func (dht *IpfsDHT) peerFromInfo(pbp *Message_PBPeer) (*peer.Peer, error) {
maddr, err := ma.NewMultiaddr(pbp.GetAddr()) maddr, err := ma.NewMultiaddr(pbp.GetAddr())
if err != nil { if err != nil {
return nil, err return nil, err