79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
package quic
|
|
|
|
import (
|
|
"context"
|
|
"github.com/anyproto/any-sync/net/transport"
|
|
"github.com/quic-go/quic-go"
|
|
"net"
|
|
)
|
|
|
|
func newConn(cctx context.Context, qconn quic.Connection) transport.MultiConn {
|
|
return &quicMultiConn{
|
|
cctx: cctx,
|
|
Connection: qconn,
|
|
}
|
|
}
|
|
|
|
type quicMultiConn struct {
|
|
cctx context.Context
|
|
quic.Connection
|
|
}
|
|
|
|
func (q *quicMultiConn) Context() context.Context {
|
|
return q.cctx
|
|
}
|
|
|
|
func (q *quicMultiConn) Accept() (conn net.Conn, err error) {
|
|
stream, err := q.Connection.AcceptStream(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return quicNetConn{
|
|
Stream: stream,
|
|
localAddr: q.LocalAddr(),
|
|
remoteAddr: q.RemoteAddr(),
|
|
}, nil
|
|
}
|
|
|
|
func (q *quicMultiConn) Open(ctx context.Context) (conn net.Conn, err error) {
|
|
stream, err := q.OpenStreamSync(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return quicNetConn{
|
|
Stream: stream,
|
|
localAddr: q.LocalAddr(),
|
|
remoteAddr: q.RemoteAddr(),
|
|
}, nil
|
|
}
|
|
|
|
func (q *quicMultiConn) Addr() string {
|
|
return q.RemoteAddr().String()
|
|
}
|
|
|
|
func (q *quicMultiConn) IsClosed() bool {
|
|
select {
|
|
case <-q.Connection.Context().Done():
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (q *quicMultiConn) Close() error {
|
|
return q.Connection.CloseWithError(2, "")
|
|
}
|
|
|
|
type quicNetConn struct {
|
|
quic.Stream
|
|
localAddr, remoteAddr net.Addr
|
|
}
|
|
|
|
func (q quicNetConn) LocalAddr() net.Addr {
|
|
return q.localAddr
|
|
}
|
|
|
|
func (q quicNetConn) RemoteAddr() net.Addr {
|
|
return q.remoteAddr
|
|
}
|