func dbFunc(c *gin.Context) { if _, err := db.Exec("CREATE TABLE IF NOT EXISTS ticks (tick timestamp)"); err != nil { c.String(http.StatusInternalServerError, fmt.Sprintf("Error creating database table: %q", err)) return } if _, err := db.Exec("INSERT INTO ticks VALUES (now())"); err != nil { c.String(http.StatusInternalServerError, fmt.Sprintf("Error incrementing tick: %q", err)) return } rows, err := db.Query("SELECT tick FROM ticks") if err != nil { c.String(http.StatusInternalServerError, fmt.Sprintf("Error reading ticks: %q", err)) return } defer rows.Close() for rows.Next() { var tick time.Time if err := rows.Scan(&tick); err != nil { c.String(http.StatusInternalServerError, fmt.Sprintf("Error scanning ticks: %q", err)) return } c.String(http.StatusOK, fmt.Sprintf("Read from DB: %s\n", tick.String())) } }
func repeatHandler(c *gin.Context) { var buffer bytes.Buffer for i := 0; i < repeat; i++ { buffer.WriteString("Hello from Go!\n") } c.String(http.StatusOK, buffer.String()) }
func roomGET(c *gin.Context) { roomid := c.Param("roomid") userid := fmt.Sprint(rand.Int31()) c.HTML(200, "chat_room", gin.H{ "roomid": roomid, "userid": userid, }) }
func roomPOST(c *gin.Context) { roomid := c.Param("roomid") userid := c.PostForm("user") message := c.PostForm("message") room(roomid).Submit(userid + ": " + message) c.JSON(200, gin.H{ "status": "success", "message": message, }) }
func roomPOST(c *gin.Context) { roomid := c.Param("roomid") nick := c.Query("nick") message := c.PostForm("message") message = strings.TrimSpace(message) 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 stream(c *gin.Context) { roomid := c.Param("roomid") listener := openListener(roomid) defer closeListener(roomid, listener) c.Stream(func(w io.Writer) bool { c.SSEvent("message", <-listener) return true }) }
func streamRoom(c *gin.Context) { roomid := c.Param("roomid") listener := openListener(roomid) ticker := time.NewTicker(1 * time.Second) users.Add("connected", 1) defer func() { closeListener(roomid, listener) ticker.Stop() users.Add("disconnected", 1) }() 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 }) }
func rateLimit(c *gin.Context) { ip := c.ClientIP() value := int(ips.Add(ip, 1)) if value%50 == 0 { fmt.Printf("ip: %s, count: %d\n", ip, value) } if value >= 200 { if value%200 == 0 { fmt.Println("ip blocked") } c.Abort() c.String(503, "you were automatically banned :)") } }
func roomGET(c *gin.Context) { roomid := c.Param("roomid") nick := c.Query("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 roomDELETE(c *gin.Context) { roomid := c.Param("roomid") deleteBroadcast(roomid) }
func index(c *gin.Context) { c.Redirect(301, "/room/hn") }