diff --git a/README.md b/README.md
index 13f95c0..07ece6a 100644
--- a/README.md
+++ b/README.md
@@ -21,19 +21,16 @@ your way so you can write real-time apps. Features include:
go get github.com/olahol/melody
```
-## [Example](https://github.com/olahol/melody/tree/master/examples)
+## [Example: chat](https://github.com/olahol/melody/tree/master/examples/chat)
-[Multi channel chat server](https://github.com/olahol/melody/tree/master/examples/multichat),
-error handling left as en exercise for the developer.
-
-[](https://github.com/olahol/melody/tree/master/examples/multichat)
+[](https://github.com/olahol/melody/tree/master/examples/chat)
```go
package main
import (
- "github.com/olahol/melody"
"github.com/gin-gonic/gin"
+ "github.com/olahol/melody"
"net/http"
)
@@ -45,24 +42,87 @@ func main() {
http.ServeFile(c.Writer, c.Request, "index.html")
})
- r.GET("/channel/:name", func(c *gin.Context) {
- http.ServeFile(c.Writer, c.Request, "chan.html")
- })
-
- r.GET("/channel/:name/ws", func(c *gin.Context) {
+ r.GET("/ws", func(c *gin.Context) {
m.HandleRequest(c.Writer, c.Request)
})
m.HandleMessage(func(s *melody.Session, msg []byte) {
- m.BroadcastFilter(msg, func(q *melody.Session) bool {
- return q.Request.URL.Path == s.Request.URL.Path
- })
+ m.Broadcast(msg)
})
r.Run(":5000")
}
```
+## [Example: gophers](https://github.com/olahol/melody/tree/master/examples/gophers)
+
+[](https://github.com/olahol/melody/tree/master/examples/gophers)
+
+```go
+package main
+
+import (
+ "github.com/gin-gonic/gin"
+ "github.com/olahol/melody"
+ "net/http"
+ "strconv"
+ "strings"
+ "sync"
+)
+
+type GopherInfo struct {
+ ID, X, Y string
+}
+
+func main() {
+ router := gin.Default()
+ mrouter := melody.New()
+ gophers := make(map[*melody.Session]*GopherInfo)
+ lock := new(sync.Mutex)
+ counter := 0
+
+ router.GET("/", func(c *gin.Context) {
+ http.ServeFile(c.Writer, c.Request, "index.html")
+ })
+
+ router.GET("/ws", func(c *gin.Context) {
+ mrouter.HandleRequest(c.Writer, c.Request)
+ })
+
+ mrouter.HandleConnect(func(s *melody.Session) {
+ lock.Lock()
+ for _, info := range gophers {
+ s.Write([]byte("set " + info.ID + " " + info.X + " " + info.Y))
+ }
+ gophers[s] = &GopherInfo{strconv.Itoa(counter), "0", "0"}
+ s.Write([]byte("iam " + gophers[s].ID))
+ counter += 1
+ lock.Unlock()
+ })
+
+ mrouter.HandleDisconnect(func(s *melody.Session) {
+ lock.Lock()
+ mrouter.BroadcastOthers([]byte("dis "+gophers[s].ID), s)
+ delete(gophers, s)
+ lock.Unlock()
+ })
+
+ mrouter.HandleMessage(func(s *melody.Session, msg []byte) {
+ p := strings.Split(string(msg), " ")
+ lock.Lock()
+ info := gophers[s]
+ if len(p) == 2 {
+ info.X = p[0]
+ info.Y = p[1]
+ mrouter.BroadcastOthers([]byte("set "+info.ID+" "+info.X+" "+info.Y), s)
+ }
+ lock.Unlock()
+ })
+
+ router.Run(":5000")
+}
+```
+
### [More examples](https://github.com/olahol/melody/tree/master/examples)
## [Documentation](https://godoc.org/github.com/olahol/melody)
diff --git a/examples/gophers/demo.gif b/examples/gophers/demo.gif
new file mode 100644
index 0000000..23e1b88
Binary files /dev/null and b/examples/gophers/demo.gif differ
diff --git a/examples/gophers/index.html b/examples/gophers/index.html
new file mode 100644
index 0000000..0d2ac88
--- /dev/null
+++ b/examples/gophers/index.html
@@ -0,0 +1,70 @@
+
+
+
+ goofy gophers
+
+
+
+
+
+
diff --git a/examples/gophers/main.go b/examples/gophers/main.go
new file mode 100644
index 0000000..e40e686
--- /dev/null
+++ b/examples/gophers/main.go
@@ -0,0 +1,62 @@
+package main
+
+import (
+ "github.com/gin-gonic/gin"
+ "github.com/olahol/melody"
+ "net/http"
+ "strconv"
+ "strings"
+ "sync"
+)
+
+type GopherInfo struct {
+ ID, X, Y string
+}
+
+func main() {
+ router := gin.Default()
+ mrouter := melody.New()
+ gophers := make(map[*melody.Session]*GopherInfo)
+ lock := new(sync.Mutex)
+ counter := 0
+
+ router.GET("/", func(c *gin.Context) {
+ http.ServeFile(c.Writer, c.Request, "index.html")
+ })
+
+ router.GET("/ws", func(c *gin.Context) {
+ mrouter.HandleRequest(c.Writer, c.Request)
+ })
+
+ mrouter.HandleConnect(func(s *melody.Session) {
+ lock.Lock()
+ for _, info := range gophers {
+ s.Write([]byte("set " + info.ID + " " + info.X + " " + info.Y))
+ }
+ gophers[s] = &GopherInfo{strconv.Itoa(counter), "0", "0"}
+ s.Write([]byte("iam " + gophers[s].ID))
+ counter += 1
+ lock.Unlock()
+ })
+
+ mrouter.HandleDisconnect(func(s *melody.Session) {
+ lock.Lock()
+ mrouter.BroadcastOthers([]byte("dis "+gophers[s].ID), s)
+ delete(gophers, s)
+ lock.Unlock()
+ })
+
+ mrouter.HandleMessage(func(s *melody.Session, msg []byte) {
+ p := strings.Split(string(msg), " ")
+ lock.Lock()
+ info := gophers[s]
+ if len(p) == 2 {
+ info.X = p[0]
+ info.Y = p[1]
+ mrouter.BroadcastOthers([]byte("set "+info.ID+" "+info.X+" "+info.Y), s)
+ }
+ lock.Unlock()
+ })
+
+ router.Run(":5000")
+}