MVC 结合Websocket

  1. package main
  2. import (
  3. "sync/atomic"
  4. "github.com/kataras/iris"
  5. "github.com/kataras/iris/mvc"
  6. "github.com/kataras/iris/websocket"
  7. )
  8. func main() {
  9. app := iris.New()
  10. // 加载模板
  11. app.RegisterView(iris.HTML("./views", ".html"))
  12. // 渲染视图.
  13. app.Get("/", func(ctx iris.Context) {
  14. ctx.View("index.html")
  15. })
  16. mvc.Configure(app.Party("/websocket"), configureMVC)
  17. // http://localhost:8080
  18. app.Run(iris.Addr(":8080"))
  19. }
  20. func configureMVC(m *mvc.Application) {
  21. ws := websocket.New(websocket.Config{})
  22. // http://localhost:8080/websocket/iris-ws.js
  23. m.Router.Any("/iris-ws.js", websocket.ClientHandler())
  24. //这将绑定ws.Upgrade的结果,这是一个websocket.Connection
  25. //由`m.Handle`服务的控制器。
  26. m.Register(ws.Upgrade)
  27. m.Handle(new(websocketController))
  28. }
  29. var visits uint64
  30. func increment() uint64 {
  31. return atomic.AddUint64(&visits, 1)
  32. }
  33. func decrement() uint64 {
  34. return atomic.AddUint64(&visits, ^uint64(0))
  35. }
  36. type websocketController struct {
  37. //注意你也可以使用匿名字段,无所谓,binder会找到它。
  38. //这是当前的websocket连接,每个客户端都有自己的* websocketController实例。
  39. Conn websocket.Connection
  40. }
  41. func (c *websocketController) onLeave(roomName string) {
  42. //visits--
  43. newCount := decrement()
  44. //这将在所有客户端上调用“visit”事件,当前客户端除外
  45. //(它不能因为它已经离开但是对于任何情况都使用这种类型的设计)
  46. c.Conn.To(websocket.Broadcast).Emit("visit", newCount)
  47. }
  48. func (c *websocketController) update() {
  49. // visits++
  50. newCount := increment()
  51. //这将在所有客户端上调用“visit”事件,包括当前事件
  52. //使用'newCount'变量。
  53. //
  54. //你有很多方法可以做到更快,例如你可以发送一个新的访问者
  55. //并且客户端可以自行增加,但这里我们只是“展示”websocket控制器。
  56. c.Conn.To(websocket.All).Emit("visit", newCount)
  57. }
  58. func (c *websocketController) Get( /* websocket.Connection could be lived here as well, it doesn't matter */ ) {
  59. c.Conn.OnLeave(c.onLeave)
  60. c.Conn.On("visit", c.update)
  61. // 在所有事件回调注册后调用它。
  62. c.Conn.Wait()
  63. }