func (c Admin) SocketConn(token string, ws *websocket.Conn) revel.Result { var access bool user := "" if err := cache.Get(token, &user); err != nil { access = false } else { access = true } cache.Delete(token) if access { subscription := roomevents.Subscribe() defer subscription.Cancel() roomevents.Join(user) defer roomevents.Leave(user) newMessages := make(chan string) go func() { var msg string for { err := websocket.Message.Receive(ws, &msg) if err != nil { close(newMessages) return } newMessages <- msg } }() for { select { case event := <-subscription.New: if websocket.JSON.Send(ws, &event) != nil { // They disconnected. return nil } case msg, ok := <-newMessages: if !ok { return nil } roomevents.Say(user, msg) } } } return nil }
func (c Product) EditPost(id int64, product models.Product) revel.Result { var tmp models.Product has, _ := c.Engine.Id(id).Get(&tmp) if !has { return c.NotFound("案例不存在") } product.Validate(c.Validation) file, header, err := c.Request.FormFile("image") if err == nil { defer file.Close() if ok := checkImageExt(c.Validation, &file, header, "image"); ok { fileName := uuidFileName(header.Filename) err, ret := qiniuUploadImage(&file, fileName) if err != nil { c.Flash.Error("上传到七牛出错,请检查七牛配置。") return c.Redirect(routes.Product.Edit(id)) } else { if tmp.Image != "" { qiniuDeleteImage(tmp.Image) } product.Image = ret.Key } } } else { product.Image = tmp.Image } if c.Validation.HasErrors() { c.Validation.Keep() c.FlashParams() return c.Redirect(routes.Product.Edit(id)) } // 强制更新允许空值的字段 aff, _ := c.Engine.Id(id).MustCols("site", "repository").Update(&product) if aff > 0 { c.Flash.Success("编辑案例成功") cache.Delete("products") } else { c.Flash.Error("编辑案例失败") return c.Redirect(routes.Product.Edit(id)) } return c.Redirect(routes.Product.Index()) }
func (c Admin) NewCategoryPost(category models.Category) revel.Result { category.Validate(c.Validation) if c.Validation.HasErrors() { c.Validation.Keep() c.FlashParams() return c.Redirect(routes.Admin.NewCategory()) } aff, _ := c.Engine.Insert(&category) if aff > 0 { c.Flash.Success("添加分类成功") cache.Delete("categories") } else { c.Flash.Error("添加分类失败") return c.Redirect(routes.Admin.NewCategory()) } return c.Redirect(routes.Admin.ListCategory()) }
func (c Product) NewPost(product models.Product) revel.Result { product.Validate(c.Validation) file, header, err := c.Request.FormFile("image") if err == nil { defer file.Close() if ok := checkImageExt(c.Validation, &file, header, "image"); ok { fileName := uuidFileName(header.Filename) err, ret := qiniuUploadImage(&file, fileName) if err != nil { c.Flash.Error("上传到七牛出错,请检查七牛配置。") return c.Redirect(routes.Product.New()) } else { product.Image = ret.Key } } } else { err := &revel.ValidationError{ Message: "字段不能为空", Key: "image", } c.Validation.Errors = append(c.Validation.Errors, err) } if c.Validation.HasErrors() { c.Validation.Keep() c.FlashParams() return c.Redirect(routes.Product.New()) } product.User = models.User{Id: c.user().Id} aff, _ := c.Engine.Insert(&product) if aff > 0 { c.Flash.Success("提交案例成功") cache.Delete("products") } else { c.Flash.Error("提交案例失败") } return c.Redirect(routes.Product.Index()) }
func ClearCache(key string) { go cache.Delete(key) }