forked from mirror/gin
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
|
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)
|
||
|
|
}
|