conn.go

package swarm

import (
  "fmt"
  "net"
  ma "github.com/jbenet/go-multiaddr"
  peer "github.com/jbenet/go-ipfs/peer"
  msgio "github.com/jbenet/go-msgio"
)

const ChanBuffer = 10

type Conn struct {
  Peer *peer.Peer
  Addr *ma.Multiaddr
  Conn net.Conn

  Closed chan bool
  Outgoing *msgio.Chan
  Incoming *msgio.Chan
}


func Dial(network string, peer *peer.Peer) (*Conn, error) {
  addr := peer.NetAddress(network)
  if addr == nil {
    return nil, fmt.Errorf("No address for network %s", network)
  }

  network, host, err := addr.DialArgs()
  if err != nil {
    return nil, err
  }

  nconn, err := net.Dial(network, host)
  if err != nil {
    return nil, err
  }

  out := msgio.NewChan(10)
  inc := msgio.NewChan(10)

  conn := &Conn{
    Peer: peer,
    Addr: addr,
    Conn: nconn,

    Outgoing: out,
    Incoming: inc,
    Closed: make(chan bool, 1),
  }

  go out.WriteTo(nconn)
  go inc.ReadFrom(nconn, 1 << 12)

  return conn, nil
}

func (s *Conn) Close() error {
  if s.Conn == nil {
    return fmt.Errorf("Already closed.") // already closed
  }

  // closing net connection
  err := s.Conn.Close()
  s.Conn = nil
  // closing channels
  s.Incoming.Close()
  s.Outgoing.Close()
  s.Closed<- true
  return err
}