transport

var NewClientStream = func(host, endpoint string) (ClientStreamTransport, error) {

	u := url.URL{Scheme: "ws", Host: host, Path: endpoint}
	h := http.Header{}
	h.Set("Sec-WebSocket-Protocol", "grpc-websockets")
	var conn *websocket.Conn
	conn, _, err := websocket.DefaultDialer.Dial(u.String(), h)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to dial to '%s'", u.String())
	}

	return &webSocketTransport{
		host:     host,
		endpoint: endpoint,
		conn:     conn,
	}, nil
}

var NewUnary = func(host string, opts *ConnectOptions) UnaryTransport {
	cl := http.DefaultClient
	transport := &http.Transport{}
	if opts.WithTLS {
		transport.TLSClientConfig = &tls.Config{}
		if opts.TLSCertificate != nil {
			certPool := x509.NewCertPool()
			decoded, _ := pem.Decode(opts.TLSCertificate)
			if decoded == nil {
				panic("failed to decode cert")
			}
			cert, err := x509.ParseCertificate(decoded.Bytes)
			if err != nil {
				panic(err)
			}
			certPool.AddCert(cert)
			transport.TLSClientConfig.RootCAs = certPool
			transport.TLSClientConfig.ServerName = cert.DNSNames[0]
		}
		transport.TLSClientConfig.InsecureSkipVerify = opts.TlsInsecureSkipVerify
	}
	if opts.Timeout != 0 {
		cl.Timeout = opts.Timeout
	} else {
		cl.Timeout = 5 * time.Second
	}

	if opts.IdleConnTimeout != 0 {
		transport.IdleConnTimeout = opts.IdleConnTimeout
	}

	if opts.TlsHandshakeTimeout != 0 {
		transport.TLSHandshakeTimeout = opts.TlsHandshakeTimeout
	}

	if opts.ExpectContinueTimeout != 0 {
		transport.ExpectContinueTimeout = opts.ExpectContinueTimeout
	}

	cl.Transport = transport
	return &httpTransport{
		host:       host,
		client:     cl,
		opts:       opts,
		header:     make(http.Header),
		clientLock: &sync.RWMutex{},
	}
}