go-libp2p-webrtc-direct/integration/main.go

89 lines
1.6 KiB
Go
Raw Normal View History

2019-02-14 16:37:30 +00:00
package main
import (
"context"
"flag"
"fmt"
"io/ioutil"
tpt "github.com/libp2p/go-libp2p-transport"
direct "github.com/libp2p/go-libp2p-webrtc-direct"
smux "github.com/libp2p/go-stream-muxer"
ma "github.com/multiformats/go-multiaddr"
"github.com/pions/webrtc"
mplex "github.com/whyrusleeping/go-smux-multiplex"
)
const listenFlag = "listen"
func main() {
listening := flag.Bool(listenFlag, false, "Listen for incoming connections.")
flag.Parse()
maddr, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/9090/http/p2p-webrtc-direct")
check(err)
transport := direct.NewTransport(
webrtc.RTCConfiguration{},
new(mplex.Transport),
)
if *listening {
list, err := transport.Listen(maddr)
check(err)
defer list.Close()
fmt.Println("[listener] Listening")
for {
c, err := list.Accept()
check(err)
fmt.Println("[listener] Got connection")
go handleConn(c)
}
} else {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
c, err := transport.Dial(ctx, maddr, "peerA")
check(err)
defer c.Close()
fmt.Println("[dialer] Opened connection")
s, err := c.OpenStream()
check(err)
fmt.Println("[dialer] Opened stream")
_, err = s.Write([]byte("hey, how is it going. I am the dialer"))
check(err)
err = s.Close()
check(err)
}
}
func handleConn(c tpt.Conn) {
for {
s, err := c.AcceptStream()
if err != nil {
return
}
fmt.Println("[listener] Got stream")
go handleStream(s)
}
}
func handleStream(s smux.Stream) {
b, err := ioutil.ReadAll(s)
check(err)
fmt.Println("[listener] Received:")
fmt.Println(string(b))
}
func check(err error) {
if err != nil {
panic(err)
}
}