gin/examples/realtime-advanced/routes.go

97 lines
1.8 KiB
Go
Raw Normal View History

2015-05-13 18:54:54 +00:00
package main
import (
2015-05-14 18:50:06 +00:00
"fmt"
2015-05-13 18:54:54 +00:00
"html"
"io"
2015-05-14 15:01:02 +00:00
"strings"
2015-05-13 18:54:54 +00:00
"time"
"github.com/gin-gonic/gin"
)
2015-05-14 18:25:55 +00:00
func rateLimit(c *gin.Context) {
2015-05-14 20:49:42 +00:00
2015-05-14 18:25:55 +00:00
ip := c.ClientIP()
2015-05-14 20:49:42 +00:00
value := int(ips.Add(ip, 1))
2015-05-14 20:52:37 +00:00
if value%50 == 0 {
2015-05-14 20:49:42 +00:00
fmt.Printf("ip: %s, count: %d\n", ip, value)
2015-05-14 18:52:45 +00:00
}
2015-05-14 20:52:37 +00:00
if value >= 200 {
if value%200 == 0 {
2015-05-14 18:50:06 +00:00
fmt.Println("ip blocked")
2015-05-14 18:25:55 +00:00
}
2015-05-14 20:49:42 +00:00
c.Abort()
c.String(503, "you were automatically banned :)")
2015-05-14 18:25:55 +00:00
}
}
2015-05-13 18:54:54 +00:00
func index(c *gin.Context) {
c.Redirect(301, "/room/hn")
}
func roomGET(c *gin.Context) {
roomid := c.ParamValue("roomid")
nick := c.FormValue("nick")
if len(nick) < 2 {
nick = ""
}
if len(nick) > 13 {
nick = nick[0:12] + "..."
}
c.HTML(200, "room_login.templ.html", gin.H{
"roomid": roomid,
"nick": nick,
"timestamp": time.Now().Unix(),
})
}
func roomPOST(c *gin.Context) {
roomid := c.ParamValue("roomid")
nick := c.FormValue("nick")
message := c.PostFormValue("message")
2015-05-14 15:01:02 +00:00
message = strings.TrimSpace(message)
2015-05-13 18:54:54 +00:00
validMessage := len(message) > 1 && len(message) < 200
validNick := len(nick) > 1 && len(nick) < 14
if !validMessage || !validNick {
c.JSON(400, gin.H{
"status": "failed",
"error": "the message or nickname is too long",
})
return
}
post := gin.H{
"nick": html.EscapeString(nick),
"message": html.EscapeString(message),
}
messages.Add("inbound", 1)
room(roomid).Submit(post)
c.JSON(200, post)
}
func streamRoom(c *gin.Context) {
roomid := c.ParamValue("roomid")
listener := openListener(roomid)
ticker := time.NewTicker(1 * time.Second)
2015-05-14 16:16:00 +00:00
users.Add("connected", 1)
defer func() {
closeListener(roomid, listener)
ticker.Stop()
users.Add("disconnected", 1)
}()
2015-05-13 18:54:54 +00:00
c.Stream(func(w io.Writer) bool {
select {
case msg := <-listener:
messages.Add("outbound", 1)
c.SSEvent("message", msg)
case <-ticker.C:
c.SSEvent("stats", Stats())
}
return true
})
}