support websocket #20

add example chat
2025-06-11 06:50:39 +00:00 · 2013-04-08 00:28:32 +08:00
parent 2573696860
commit 3ad639e739
4 changed files with 223 additions and 4 deletions
--- a/example/chat/chat.go
+++ b/example/chat/chat.go
@ -0,0 +1,42 @@
+package main
+
+import (
+	"fmt"
+	"github.com/astaxie/beego"
+	"github.com/fzzy/sockjs-go/sockjs"
+	"strings"
+)
+
+var users *sockjs.SessionPool = sockjs.NewSessionPool()
+
+func chatHandler(s sockjs.Session) {
+	users.Add(s)
+	defer users.Remove(s)
+
+	for {
+		m := s.Receive()
+		if m == nil {
+			break
+		}
+		fullAddr := s.Info().RemoteAddr
+		addr := fullAddr[:strings.LastIndex(fullAddr, ":")]
+		m = []byte(fmt.Sprintf("%s: %s", addr, m))
+		users.Broadcast(m)
+	}
+}
+
+type MainController struct {
+	beego.Controller
+}
+
+func (m *MainController) Get() {
+	m.TplNames = "index.html"
+}
+
+func main() {
+	conf := sockjs.NewConfig()
+	sockjshandler := sockjs.NewHandler("/chat", chatHandler, conf)
+	beego.Router("/", &MainController{})
+	beego.RouterHandler("/chat/:info(.*)", sockjshandler)
+	beego.Run()
+}
--- a/example/chat/views/index.html
+++ b/example/chat/views/index.html
@ -0,0 +1,81 @@
+<!DOCTYPE html>
+<html>
+  <head>
+	<script src="http://ajax.googleapis.com/ajax/libs/jquery/1.4.2/jquery.min.js"></script>
+	<script src="http://cdn.sockjs.org/sockjs-0.3.4.min.js"></script>
+	<script>
+    $(function() {
+		var conn = null;
+
+		function log(msg) {
+			var control = $('#log');
+			control.html(control.html() + msg + '<br/>');
+			control.scrollTop(control.scrollTop() + 1000);
+		}
+
+		function disconnect() {
+			if (conn != null) {
+				log('Disconnecting...');
+
+				conn.close();
+				conn = null;
+
+				updateUi();
+			}
+		}
+
+		function updateUi() {
+			if (conn == null || conn.readyState != SockJS.OPEN) {
+				$('#status').text('disconnected');
+				$('#connect').text('Connect');
+			} else {
+				$('#status').text('connected (' + conn.protocol + ')');
+				$('#connect').text('Disconnect');
+			}
+		}
+
+		$('form').submit(function() {
+			var text = $('#message').val();
+			conn.send(text);
+			$('#message').val('').focus();
+			return false;
+		});
+
+		conn = new SockJS('http://' + window.location.host + '/chat');
+		log('Connecting...');
+
+		conn.onopen = function() {
+			log('Connected.');
+			updateUi();
+		};
+
+		conn.onmessage = function(e) {
+			log(e.data);
+		};
+
+		conn.onclose = function() {
+			log('Disconnected.');
+			conn = null;
+			updateUi();
+		};
+
+		$('#message').val('').focus();
+    });
+	</script>
+	<title>Sockjs-go chat</title>
+  </head>
+  <body>
+	<h1>Sockjs-go chat</h1>
+
+	<div>
+	  Status: <span id="status">disconnected</span>
+	</div>
+	<div id="log" style="width: 60em; height: 20em; overflow:auto; border: 1px solid black">
+	</div>
+	<form id="chatform">
+	  <label for="message">Message:</label>
+	  <input id="message" type="text" />
+	  <input type="submit" value="Send" />
+	</form>
+  </body>
+</html>