conn.go 1.72 KB
Newer Older
1 2 3
package swarm

import (
4
	"fmt"
5 6
	"net"

7 8 9 10
	peer "github.com/jbenet/go-ipfs/peer"
	u "github.com/jbenet/go-ipfs/util"
	msgio "github.com/jbenet/go-msgio"
	ma "github.com/jbenet/go-multiaddr"
11 12
)

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
13
// ChanBuffer is the size of the buffer in the Conn Chan
14 15
const ChanBuffer = 10

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
16
// Conn represents a connection to another Peer (IPFS Node).
17
type Conn struct {
18 19 20
	Peer *peer.Peer
	Addr *ma.Multiaddr
	Conn net.Conn
21

22 23 24
	Closed   chan bool
	Outgoing *msgio.Chan
	Incoming *msgio.Chan
25 26
}

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
27
// ConnMap maps Keys (Peer.IDs) to Connections.
28
type ConnMap map[u.Key]*Conn
29

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
30 31
// Dial connects to a particular peer, over a given network
// Example: Dial("udp", peer)
32
func Dial(network string, peer *peer.Peer) (*Conn, error) {
33 34 35 36
	addr := peer.NetAddress(network)
	if addr == nil {
		return nil, fmt.Errorf("No address for network %s", network)
	}
37

38 39 40 41
	network, host, err := addr.DialArgs()
	if err != nil {
		return nil, err
	}
42

43 44 45 46
	nconn, err := net.Dial(network, host)
	if err != nil {
		return nil, err
	}
47

48 49 50 51
	conn := &Conn{
		Peer: peer,
		Addr: addr,
		Conn: nconn,
52
	}
53

54 55 56 57 58 59 60 61
	newConnChans(conn)
	return conn, nil
}

// Construct new channels for given Conn.
func newConnChans(c *Conn) error {
	if c.Outgoing != nil || c.Incoming != nil {
		return fmt.Errorf("Conn already initialized")
62
	}
63

64 65 66
	c.Outgoing = msgio.NewChan(10)
	c.Incoming = msgio.NewChan(10)
	c.Closed = make(chan bool, 1)
67

68 69 70 71
	go c.Outgoing.WriteTo(c.Conn)
	go c.Incoming.ReadFrom(c.Conn, 1<<12)

	return nil
72 73
}

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
74
// Close closes the connection, and associated channels.
75
func (s *Conn) Close() error {
76
	if s.Conn == nil {
77
		return fmt.Errorf("Already closed") // already closed
78
	}
79

80 81 82 83 84 85 86 87
	// closing net connection
	err := s.Conn.Close()
	s.Conn = nil
	// closing channels
	s.Incoming.Close()
	s.Outgoing.Close()
	s.Closed <- true
	return err
88
}