mirror of
https://github.com/kataras/iris.git
synced 2025-01-23 18:51:03 +01:00
c3205dafa1
Read HISTORY.md it contains a breaking change, second parameter of HandleDir should be iris.Dir(...) instead of just a string relative to: https://github.com/kataras/iris/issues/1556#issuecomment-661057446 Former-commit-id: 14b48a06fb3b99287dff543932be2937a64233b9
119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"sync/atomic"
|
|
|
|
"github.com/kataras/iris/v12"
|
|
"github.com/kataras/iris/v12/mvc"
|
|
"github.com/kataras/iris/v12/websocket"
|
|
)
|
|
|
|
func main() {
|
|
app := iris.New()
|
|
app.Logger().SetLevel("debug")
|
|
|
|
// load templates.
|
|
// app.RegisterView(iris.HTML("./views", ".html"))
|
|
|
|
// render the ./browser/index.html.
|
|
app.HandleDir("/", iris.Dir("./browser"))
|
|
|
|
websocketAPI := app.Party("/websocket")
|
|
|
|
m := mvc.New(websocketAPI)
|
|
m.Register(
|
|
&prefixedLogger{prefix: "DEV"},
|
|
)
|
|
m.HandleWebsocket(&websocketController{Namespace: "default", Age: 42, Otherstring: "other string"})
|
|
|
|
websocketServer := websocket.New(websocket.DefaultGorillaUpgrader, m)
|
|
|
|
websocketAPI.Get("/", websocket.Handler(websocketServer))
|
|
// http://localhost:8080
|
|
app.Listen(":8080")
|
|
}
|
|
|
|
var visits uint64
|
|
|
|
func increment() uint64 {
|
|
return atomic.AddUint64(&visits, 1)
|
|
}
|
|
|
|
func decrement() uint64 {
|
|
return atomic.AddUint64(&visits, ^uint64(0))
|
|
}
|
|
|
|
type websocketController struct {
|
|
*websocket.NSConn `stateless:"true"`
|
|
Namespace string
|
|
Age int
|
|
Otherstring string
|
|
|
|
Logger LoggerService
|
|
}
|
|
|
|
// or
|
|
// func (c *websocketController) Namespace() string {
|
|
// return "default"
|
|
// }
|
|
|
|
func (c *websocketController) OnNamespaceDisconnect(msg websocket.Message) error {
|
|
c.Logger.Log("Disconnected")
|
|
// visits--
|
|
newCount := decrement()
|
|
// This will call the "OnVisit" event on all clients, except the current one,
|
|
// (it can't because it's left but for any case use this type of design)
|
|
c.Conn.Server().Broadcast(nil, websocket.Message{
|
|
Namespace: msg.Namespace,
|
|
Event: "OnVisit",
|
|
Body: []byte(fmt.Sprintf("%d", newCount)),
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *websocketController) OnNamespaceConnected(msg websocket.Message) error {
|
|
// println("Broadcast prefix is: " + c.BroadcastPrefix)
|
|
c.Logger.Log("Connected")
|
|
|
|
// visits++
|
|
newCount := increment()
|
|
|
|
// This will call the "OnVisit" event on all clients, including the current one,
|
|
// with the 'newCount' variable.
|
|
//
|
|
// There are many ways that u can do it and faster, for example u can just send a new visitor
|
|
// and client can increment itself, but here we are just "showcasing" the websocket controller.
|
|
c.Conn.Server().Broadcast(nil, websocket.Message{
|
|
Namespace: msg.Namespace,
|
|
Event: "OnVisit",
|
|
Body: []byte(fmt.Sprintf("%d", newCount)),
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *websocketController) OnChat(msg websocket.Message) error {
|
|
ctx := websocket.GetContext(c.Conn)
|
|
|
|
ctx.Application().Logger().Infof("[IP: %s] [ID: %s] broadcast to other clients the message [%s]",
|
|
ctx.RemoteAddr(), c, string(msg.Body))
|
|
|
|
c.Conn.Server().Broadcast(c, msg)
|
|
|
|
return nil
|
|
}
|
|
|
|
type LoggerService interface {
|
|
Log(string)
|
|
}
|
|
|
|
type prefixedLogger struct {
|
|
prefix string
|
|
}
|
|
|
|
func (s *prefixedLogger) Log(msg string) {
|
|
fmt.Printf("%s: %s\n", s.prefix, msg)
|
|
}
|