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 }