mirror of
https://github.com/gin-gonic/gin.git
synced 2025-06-19 01:34:38 +08:00
Adds realtime-chat example code
This commit is contained in:
58
examples/realtime-chat/main.go
Normal file
58
examples/realtime-chat/main.go
Normal file
@ -0,0 +1,58 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"math/rand"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func main() {
|
||||
router := gin.Default()
|
||||
router.SetHTMLTemplate(html)
|
||||
|
||||
router.GET("/room/:roomid", roomGET)
|
||||
router.POST("/room/:roomid", roomPOST)
|
||||
router.DELETE("/room/:roomid", roomDELETE)
|
||||
router.GET("/stream/:roomid", stream)
|
||||
|
||||
router.Run(":8080")
|
||||
}
|
||||
|
||||
func stream(c *gin.Context) {
|
||||
roomid := c.ParamValue("roomid")
|
||||
listener := openListener(roomid)
|
||||
defer closeListener(roomid, listener)
|
||||
|
||||
c.Stream(func(w io.Writer) bool {
|
||||
c.SSEvent("message", <-listener)
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
func roomGET(c *gin.Context) {
|
||||
roomid := c.ParamValue("roomid")
|
||||
userid := fmt.Sprint(rand.Int31())
|
||||
c.HTML(200, "chat_room", gin.H{
|
||||
"roomid": roomid,
|
||||
"userid": userid,
|
||||
})
|
||||
}
|
||||
|
||||
func roomPOST(c *gin.Context) {
|
||||
roomid := c.ParamValue("roomid")
|
||||
userid := c.PostFormValue("user")
|
||||
message := c.PostFormValue("message")
|
||||
room(roomid).Submit(userid + ": " + message)
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"status": "success",
|
||||
"message": message,
|
||||
})
|
||||
}
|
||||
|
||||
func roomDELETE(c *gin.Context) {
|
||||
roomid := c.ParamValue("roomid")
|
||||
deleteBroadcast(roomid)
|
||||
}
|
Reference in New Issue
Block a user