package controller import ( "github.com/gin-gonic/gin" "github.com/taoshihan1991/imaptool/models" "github.com/taoshihan1991/imaptool/tmpl" "github.com/taoshihan1991/imaptool/tools" "github.com/taoshihan1991/imaptool/ws" "net/http" ) func ActionMain(w http.ResponseWriter, r *http.Request) { sessionId := tools.GetCookie(r, "session_id") info := AuthCheck(sessionId) if len(info) == 0 { http.Redirect(w, r, "/login", 302) return } render := tmpl.NewRender(w) render.Display("main", render) } func MainCheckAuth(c *gin.Context) { id, _ := c.Get("kefu_id") userinfo := models.FindUserRole("user.avator,user.name,user.id, role.name role_name", id) c.JSON(200, gin.H{ "code": 200, "msg": "验证成功", "result": gin.H{ "avator": userinfo.Avator, "name": userinfo.Name, "role_name": userinfo.RoleName, }, }) } func GetStatistics(c *gin.Context) { visitors := models.CountVisitors() message := models.CountMessage() session := len(ws.ClientList) c.JSON(200, gin.H{ "code": 200, "msg": "ok", "result": gin.H{ "visitors": visitors, "message": message, "session": session, }, }) }