How to use std http for websocket client about nbio

[om26er]

Thanks, I'll close the issue now

[lesismal]

For client side, I think you can use nbhttp/websocket.Client directely.
net/http doesn't include websocket, if we use http.Client to send "upgrade" request and handshake ourselves, then convert it to nbio.Conn, it will be very complex, users need to do more, and we need to make some change for nbhttp.Engine to support the convertion.
Else if we use other websocket dialer, just can save the handshake processes, but still need to convert conn to nbio.Conn and make change for nbhttp.Engine.

If using nbio/nbhttp/websocket.Client directely, here is the full example including both server and client side handled by the same Engine:

package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"net/http"
	"net/url"
	"os"
	"os/signal"
	"time"

	"github.com/lesismal/llib/std/crypto/tls"
	"github.com/lesismal/nbio/nbhttp"
	"github.com/lesismal/nbio/nbhttp/websocket"
)

var (
	addr   = "localhost:8888"
	engine *nbhttp.Engine
)

func newServerUpgrader() *websocket.Upgrader {
	u := websocket.NewUpgrader()
	u.OnMessage(func(c *websocket.Conn, mt websocket.MessageType, data []byte) {
		fmt.Println("server OnMessage:", c.RemoteAddr().String(), mt, string(data))
		c.WriteMessage(mt, data)
	})
	u.OnClose(func(c *websocket.Conn, err error) {
		fmt.Println("server OnClose:", c.RemoteAddr().String(), err)
	})
	return u
}

func onWebsocket(w http.ResponseWriter, r *http.Request) {
	upgrader := newServerUpgrader()
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		panic(err)
	}
	wsConn := conn.(*websocket.Conn)
	fmt.Println("server OnOpen:", wsConn.RemoteAddr().String())
}

func newClientUpgrader() *websocket.Upgrader {
	u := websocket.NewUpgrader()
	u.OnMessage(func(c *websocket.Conn, mt websocket.MessageType, data []byte) {
		fmt.Println("cleint OnMessage:", c.RemoteAddr().String(), mt, string(data))
		time.AfterFunc(time.Second, func() {
			c.WriteMessage(mt, data)
		})
	})
	u.OnClose(func(c *websocket.Conn, err error) {
		fmt.Println("cleint OnClose:", c.RemoteAddr().String(), err)
	})
	return u
}

func client(i int) {
	u := url.URL{Scheme: "ws", Host: addr, Path: "/ws"}
	tlsConfig := &tls.Config{
		InsecureSkipVerify: true,
	}
	dialer := &websocket.Dialer{
		Engine:          engine,
		Upgrader:        newClientUpgrader(),
		DialTimeout:     time.Second * 3,
		TLSClientConfig: tlsConfig,
	}
	c, _, err := dialer.Dial(u.String(), nil)
	if err != nil {
		log.Fatalf("dial failed: %v", err)
	}
	log.Printf("dial success: %v", c.LocalAddr().String())
	c.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("hello %v", i)))
}

func main() {
	flag.Parse()
	mux := &http.ServeMux{}
	mux.HandleFunc("/ws", onWebsocket)

	engine = nbhttp.NewEngine(nbhttp.Config{
		Network: "tcp",
		Addrs:   []string{addr},
		Handler: mux,
		IOMod:   nbhttp.IOModBlocking,
	})

	err := engine.Start()
	if err != nil {
		fmt.Printf("nbio.Start failed: %v\n", err)
		return
	}

	go func() {
		time.Sleep(time.Second / 10)
		for i := 0; i < 3; i++ {
			go client(i)
		}
	}()

	interrupt := make(chan os.Signal, 1)
	signal.Notify(interrupt, os.Interrupt)
	<-interrupt
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
	defer cancel()
	engine.Shutdown(ctx)
}

output:

2023/03/21 12:55:50.077 [INF] NBIO[NB] start
2023/03/21 12:55:50.078 [INF] Serve  NonTLS On: [[email protected]:8888]
server OnOpen: 127.0.0.1:41632
server OnOpen: 127.0.0.1:41634
2023/03/21 12:55:50 dial success: 127.0.0.1:41632
server OnOpen: 127.0.0.1:41636
2023/03/21 12:55:50 dial success: 127.0.0.1:41634
server OnMessage: 127.0.0.1:41634 1 hello 0
server OnMessage: 127.0.0.1:41632 1 hello 1
2023/03/21 12:55:50 dial success: 127.0.0.1:41636
cleint OnMessage: 127.0.0.1:8888 1 hello 0
cleint OnMessage: 127.0.0.1:8888 1 hello 1
server OnMessage: 127.0.0.1:41636 1 hello 2
cleint OnMessage: 127.0.0.1:8888 1 hello 2
server OnMessage: 127.0.0.1:41632 1 hello 1
server OnMessage: 127.0.0.1:41634 1 hello 0
server OnMessage: 127.0.0.1:41636 1 hello 2
cleint OnMessage: 127.0.0.1:8888 1 hello 1
cleint OnMessage: 127.0.0.1:8888 1 hello 2
cleint OnMessage: 127.0.0.1:8888 1 hello 0
......