mirror of
https://github.com/vacp2p/mvds.git
synced 2026-01-09 12:07:55 -05:00
9e1addbac0
This commits changes the behavior so that we don't rely on the presence of GroupID in the payload. Adds some tests to verify the basic functionalities of BATCH and INTERACTIVE mode, but more are required to ensure the correct functioning. Error handling and state management is probably also to be inspected more carefully.
55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package transport
|
|
|
|
import (
|
|
"errors"
|
|
math "math/rand"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/vacp2p/mvds/protobuf"
|
|
"github.com/vacp2p/mvds/state"
|
|
)
|
|
|
|
// ChannelTransport implements a basic MVDS transport using channels for basic testing purposes.
|
|
type ChannelTransport struct {
|
|
sync.Mutex
|
|
|
|
offline int
|
|
|
|
in <-chan Packet
|
|
out map[state.PeerID]chan<- Packet
|
|
}
|
|
|
|
func NewChannelTransport(offline int, in <-chan Packet) *ChannelTransport {
|
|
return &ChannelTransport{
|
|
offline: offline,
|
|
in: in,
|
|
out: make(map[state.PeerID]chan<- Packet),
|
|
}
|
|
}
|
|
|
|
func (t *ChannelTransport) AddOutput(id state.PeerID, c chan<- Packet) {
|
|
t.out[id] = c
|
|
}
|
|
|
|
func (t *ChannelTransport) Watch() Packet {
|
|
return <-t.in
|
|
}
|
|
|
|
func (t *ChannelTransport) Send(sender state.PeerID, peer state.PeerID, payload protobuf.Payload) error {
|
|
// @todo we can do this better, we put node onlineness into a goroutine where we just stop the nodes for x seconds
|
|
// outside of this class
|
|
math.Seed(time.Now().UnixNano())
|
|
if math.Intn(100) < t.offline {
|
|
return nil
|
|
}
|
|
|
|
c, ok := t.out[peer]
|
|
if !ok {
|
|
return errors.New("peer unknown")
|
|
}
|
|
|
|
c <- Packet{Sender: sender, Payload: payload}
|
|
return nil
|
|
}
|