add native websocket

zing-dev · zing-dev · commit 03c201334e29 · 2021-09-26T20:15:29.000+08:00
diff --git a/modules/websocket/neffos/native-messages/browser/index.html b/modules/websocket/neffos/native-messages/browser/index.html
@@ -0,0 +1,47 @@
+<html>
+
+<body style="padding:10px;">
+    <input type="text" id="messageTxt" />
+    <button type="button" id="sendBtn">Send</button>
+    <div id="messages" style="width: 375px;margin:10px 0 0 0;border-top: 1px solid black;">
+    </div>
+
+    <script type="text/javascript">
+        let messageTxt = document.getElementById("messageTxt");
+        let messages = document.getElementById("messages");
+        let sendBtn = document.getElementById("sendBtn")
+
+        let w = new WebSocket("ws://localhost:9999/endpoint");
+        w.onopen = function () {
+            console.log("Websocket connection enstablished");
+        };
+        w.onclose = function () {
+            appendMessage("<div style='text-align: center'><h3>Disconnected</h3></div>");
+        };
+        w.onmessage = function (message) {
+            appendMessage("<div>" + message.data + "</div>");
+        };
+
+        sendBtn.onclick = function () {
+            let myText = messageTxt.value;
+            messageTxt.value = "";
+
+            appendMessage("<div style='color: red'> me: " + myText + "</div>");
+            w.send(myText);
+        };
+
+        messageTxt.addEventListener("keyup", function (e) {
+            if (e.code === 'Enter') {
+                e.preventDefault();
+
+                sendBtn.click();
+            }
+        });
+
+        function appendMessage(messageDivHTML) {
+            messages.insertAdjacentHTML('afterbegin', messageDivHTML);
+        }
+    </script>
+</body>
+
+</html>
diff --git a/modules/websocket/neffos/native-messages/go.mod b/modules/websocket/neffos/native-messages/go.mod
@@ -0,0 +1,11 @@
+module neffos-native-messages
+
+go 1.17
+
+require github.com/kataras/neffos v0.0.18
+
+require (
+	github.com/gorilla/websocket v1.4.2 // indirect
+	github.com/iris-contrib/go.uuid v2.0.0+incompatible // indirect
+	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect
+)
diff --git a/modules/websocket/neffos/native-messages/native.go b/modules/websocket/neffos/native-messages/native.go
@@ -0,0 +1,85 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/kataras/neffos"
+	"github.com/kataras/neffos/gorilla"
+)
+
+// HOW TO RUN
+//
+// $ go run native.go server
+// #
+// # open a browser tab and navigate to http://localhost:8080 and write something to the text input.
+// # in the same time you can also run the go client
+// # which is created using neffos but the standard go websocket library could be used instead as well.
+// #
+// $ go run native.go client
+func main() {
+	args := os.Args[1:]
+	if len(args) == 0 {
+		panic("expected program to start with 'server' or 'client' argument")
+	}
+	side := args[0]
+
+	switch side {
+	case "server":
+		runServer()
+	case "client":
+		runClient()
+	default:
+		fmt.Printf("unexpected argument, expected 'server' or 'client' but got '%s'", side)
+		os.Exit(1)
+	}
+}
+
+var pongMessage = []byte("pong")
+
+var events = neffos.Events{
+	neffos.OnNativeMessage: func(c *neffos.NSConn, msg neffos.Message) error {
+		log.Printf("Got: %s", string(msg.Body))
+
+		if !c.Conn.IsClient() {
+			return c.Conn.Socket().WriteText(pongMessage, 0)
+		}
+
+		return nil
+	},
+}
+
+func runServer() {
+	websocketServer := neffos.New(
+		gorilla.DefaultUpgrader, events)
+
+	router := http.NewServeMux()
+	router.Handle("/endpoint", websocketServer)
+	router.Handle("/", http.FileServer(http.Dir("./browser")))
+
+	log.Println("Serving websockets on localhost:9999/endpoint")
+	log.Fatal(http.ListenAndServe(":9999", router))
+}
+
+func runClient() {
+	ctx := context.Background()
+
+	client, err := neffos.Dial(ctx, gorilla.DefaultDialer, "ws://localhost:9999/endpoint", events)
+	if err != nil {
+		panic(err)
+	}
+
+	// Connect does nothing at this case,
+	// it just returns a connection to work with its methods.
+	c, err := client.Connect(ctx, "")
+	if err != nil {
+		panic(err)
+	}
+
+	c.Conn.Socket().WriteText([]byte("ping"), 0)
+
+	<-client.NotifyClose
+}
