sparrow/services/knowoapi/controllers/rule_chain.go

package controllers

import (
	"github.com/kataras/iris"
	"sparrow/pkg/models"
	"sparrow/services/knowoapi/services"
)

// RuleChainController api
type RuleChainController struct {
	Ctx     iris.Context
	Service services.RuleChainService
	Token   Token
}

// Post ``
// POST /admin/rule_chain
func (a *RuleChainController) Post() {
	ptl := new(models.RuleChain)
	if err := parseBody(a.Ctx, ptl); err != nil {
		badRequest(a.Ctx, err)
		return
	}
	err := a.Service.Create(ptl)
	if err != nil {
		responseError(a.Ctx, ErrDatabase, err.Error())
		return
	}
	done(a.Ctx, ptl)
}

// Delete delete
// DELETE /admin/rule_chain
func (a *RuleChainController) Delete() {
	RuleChain := new(models.RuleChain)
	if err := parseBody(a.Ctx, RuleChain); err != nil {
		badRequest(a.Ctx, err)
		return
	}
	if err := a.Service.Delete(RuleChain); err != nil {
		responseError(a.Ctx, ErrDatabase, err.Error())
		return
	}
	done(a.Ctx, RuleChain)
}

// Put 更新
// PUT /admin/rule_chain
func (a *RuleChainController) Put() {
	RuleChain := new(models.RuleChain)
	if err := parseBody(a.Ctx, RuleChain); err != nil {
		badRequest(a.Ctx, err)
		return
	}
	if err := a.Service.Update(RuleChain); err != nil {
		responseError(a.Ctx, ErrDatabase, err.Error())
		return
	}
	done(a.Ctx, RuleChain)
}

// Get get RuleChains
// GET /admin/rule_chain?pi=&ps=&name=
func (a *RuleChainController) Get() {
	pi, err := a.Ctx.URLParamInt("pi")
	if err != nil {
		badRequest(a.Ctx, err)
		return
	}
	ps, err := a.Ctx.URLParamInt("ps")
	if err != nil {
		badRequest(a.Ctx, err)
		return
	}
	name := a.Ctx.URLParam("name")
	datas, total, err := a.Service.Query(pi, ps, name)
	if err != nil {
		responseError(a.Ctx, ErrDatabase, err.Error())
		return
	}

	done(a.Ctx, map[string]interface{}{
		"list":  datas,
		"total": total,
	})
}