You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
1.7 KiB
93 lines
1.7 KiB
package p2p
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"sync/atomic"
|
|
|
|
. "github.com/tendermint/tendermint/binary"
|
|
)
|
|
|
|
/* Peer */
|
|
|
|
type Peer struct {
|
|
outbound bool
|
|
mconn *MConnection
|
|
started uint32
|
|
stopped uint32
|
|
|
|
Key string
|
|
}
|
|
|
|
func newPeer(conn net.Conn, outbound bool, chDescs []*ChannelDescriptor, onPeerError func(*Peer, interface{})) *Peer {
|
|
var p *Peer
|
|
onError := func(r interface{}) {
|
|
p.stop()
|
|
onPeerError(p, r)
|
|
}
|
|
mconn := NewMConnection(conn, chDescs, onError)
|
|
p = &Peer{
|
|
outbound: outbound,
|
|
mconn: mconn,
|
|
stopped: 0,
|
|
Key: mconn.RemoteAddress.String(),
|
|
}
|
|
mconn.Peer = p // hacky optimization
|
|
return p
|
|
}
|
|
|
|
func (p *Peer) start() {
|
|
if atomic.CompareAndSwapUint32(&p.started, 0, 1) {
|
|
log.Debug("Starting %v", p)
|
|
p.mconn.Start()
|
|
}
|
|
}
|
|
|
|
func (p *Peer) stop() {
|
|
if atomic.CompareAndSwapUint32(&p.stopped, 0, 1) {
|
|
log.Debug("Stopping %v", p)
|
|
p.mconn.Stop()
|
|
}
|
|
}
|
|
|
|
func (p *Peer) IsOutbound() bool {
|
|
return p.outbound
|
|
}
|
|
|
|
func (p *Peer) Send(chId byte, bytes ByteSlice) bool {
|
|
if atomic.LoadUint32(&p.stopped) == 1 {
|
|
return false
|
|
}
|
|
return p.mconn.Send(chId, bytes)
|
|
}
|
|
|
|
func (p *Peer) TrySend(chId byte, bytes ByteSlice) bool {
|
|
if atomic.LoadUint32(&p.stopped) == 1 {
|
|
return false
|
|
}
|
|
return p.mconn.TrySend(chId, bytes)
|
|
}
|
|
|
|
func (o *Peer) CanSend(chId byte) int {
|
|
if atomic.LoadUint32(&p.stopped) == 1 {
|
|
return 0
|
|
}
|
|
return p.mconn.CanSend(chId)
|
|
}
|
|
|
|
func (p *Peer) WriteTo(w io.Writer) (n int64, err error) {
|
|
return String(p.Key).WriteTo(w)
|
|
}
|
|
|
|
func (p *Peer) String() string {
|
|
if p.outbound {
|
|
return fmt.Sprintf("P(->%v)", p.mconn)
|
|
} else {
|
|
return fmt.Sprintf("P(%v->)", p.mconn)
|
|
}
|
|
}
|
|
|
|
func (p *Peer) Equals(other *Peer) bool {
|
|
return p.Key == other.Key
|
|
}
|