64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
|
package net
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net/url"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type TransportPeerConfig struct {
|
||
|
|
Socket interface{}
|
||
|
|
Uris []*url.URL
|
||
|
|
}
|
||
|
|
|
||
|
|
type PeerStatic interface {
|
||
|
|
Connect(uri *url.URL) (interface{}, error) // Returns a connection/socket
|
||
|
|
}
|
||
|
|
|
||
|
|
type PeerFactory interface {
|
||
|
|
NewPeer(options *TransportPeerConfig) (Peer, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
transports sync.Map
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
transports = sync.Map{}
|
||
|
|
//RegisterPeerType("ws", WebSocketPeer)
|
||
|
|
//RegisterPeerType("wss", WebSocketPeer)
|
||
|
|
}
|
||
|
|
func RegisterTransport(peerType string, factory interface{}) {
|
||
|
|
if _, ok := factory.(PeerFactory); !ok {
|
||
|
|
panic("factory must implement PeerFactory")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := factory.(PeerStatic); !ok {
|
||
|
|
panic("factory must implement PeerStatic")
|
||
|
|
}
|
||
|
|
|
||
|
|
transports.Store(peerType, factory)
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateTransportSocket(peerType string, uri *url.URL) (interface{}, error) {
|
||
|
|
static, ok := transports.Load(peerType)
|
||
|
|
if !ok {
|
||
|
|
return nil, errors.New("no static method registered for type: " + peerType)
|
||
|
|
}
|
||
|
|
|
||
|
|
t, err := static.(PeerStatic).Connect(uri)
|
||
|
|
|
||
|
|
return &t, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateTransportPeer(peerType string, options *TransportPeerConfig) (*Peer, error) {
|
||
|
|
factory, ok := transports.Load(peerType)
|
||
|
|
if !ok {
|
||
|
|
return nil, errors.New("no factory registered for type: " + peerType)
|
||
|
|
}
|
||
|
|
|
||
|
|
t, err := factory.(PeerFactory).NewPeer(options)
|
||
|
|
|
||
|
|
return &t, err
|
||
|
|
}
|