Websockets as a drop in TCP replacement

dialer.go

package wsnet

import (
	"net"
	"time"

	"golang.org/x/net/websocket"
)

// Dial gives you a net.Conn that talks to a WS destination.
// Addr should be like "ws://localhost:8080/"
func Dial(addr string, timeout time.Duration) (net.Conn, error) {
	return websocket.Dial(addr, "", "http://localhost/")
}

listen.go

package wsnet

import (
	"net"
	"net/http"

	"golang.org/x/net/websocket"
)

type WSServer struct {
	listener net.Listener
	conns    chan net.Conn
}

func Listen(laddr string) (net.Listener, error) {
	listener, err := net.Listen("tcp", laddr)
	if err != nil {
		return nil, err
	}
	wss := &WSServer{
		listener: listener,
		conns:    make(chan net.Conn),
	}
	http.Handle("/", websocket.Handler(wss.wsHandler))
	err = http.Serve(listener, nil)
	if err != nil {
		listener.Close()
		return nil, err
	}
	return wss, nil
}

func (w *WSServer) Accept() (net.Conn, error) {
	return <-w.conns, nil
}

func (w *WSServer) Close() error {
	return w.listener.Close()
}

func (w *WSServer) Addr() net.Addr {
	// This is still legit enough? Maybe?
	return w.listener.Addr()
}

func (w *WSServer) wsHandler(ws *websocket.Conn) {
	w.conns <- ws
}