func (c *MockRuleController) Get() { var rule models.MockRule var contents []*models.MockContent id, _ := strconv.Atoi(c.Ctx.Input.Param(":id")) o := orm.NewOrm() rule.Id = id err := o.Read(&rule) if err == orm.ErrNoRows || err == orm.ErrMissPK { c.ShowError("Can't find data.") return } rule.UnmarshalHeaderAndParams() _, err = o.QueryTable("mock_content").Filter("ruleId", id).All(&contents) rule.Contents = contents c.ShowSuccess(rule) }
func (c *MockRuleController) Toggle() { var rule models.MockRule id, _ := strconv.Atoi(c.Ctx.Input.Param(":id")) rule.Id = id o := orm.NewOrm() err := o.Read(&rule) if err == orm.ErrNoRows || err == orm.ErrMissPK { c.ShowError("Can't find data.") return } rule.Enable = !rule.Enable _, err = o.Update(&rule) if err == orm.ErrNoRows || err == orm.ErrMissPK { c.ShowError(err.Error()) return } refreshRuleCache() c.ShowSuccess(nil) }