审查视图

pkg/port/beego/controllers/evaluation_rule_controller.go 3.3 KB
1 2 3 4 5 6 7 8 9 10 11 12 13 14
package controllers

import (
	"github.com/linmadan/egglib-go/core/application"
	"github.com/linmadan/egglib-go/web/beego"
	service "gitlab.fjmaimaimai.com/allied-creation/performance/pkg/application/evaluation_rule"
	"gitlab.fjmaimaimai.com/allied-creation/performance/pkg/application/evaluation_rule/command"
	"gitlab.fjmaimaimai.com/allied-creation/performance/pkg/port/beego/middlewares"
)

type RuleController struct {
	beego.BaseController
}
郑周 authored
15
func (controller *RuleController) CreateRule() {
16 17 18 19 20
	ruService := service.NewEvaluationRuleService()
	in := &command.CreateRuleCommand{}
	if err := controller.Unmarshal(in); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {
郑周 authored
21 22 23
		ua := middlewares.GetUser(controller.Ctx)
		in.CompanyId = ua.CompanyId
		in.CreatorId = ua.UserId
24 25 26 27 28 29 30 31 32 33
		controller.Response(ruService.Create(in))
	}
}

func (controller *RuleController) UpdateRule() {
	ruService := service.NewEvaluationRuleService()
	in := &command.UpdateRuleCommand{}
	if err := controller.Unmarshal(in); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {
郑周 authored
34 35
		ua := middlewares.GetUser(controller.Ctx)
		in.CompanyId = ua.CompanyId
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
		controller.Response(ruService.Update(in))
	}
}

func (controller *RuleController) GetRule() {
	ruService := service.NewEvaluationRuleService()
	in := &command.GetRuleCommand{}
	if id, err := controller.GetInt64(":Id"); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {
		in.Id = id
		controller.Response(ruService.Get(in))
	}
}

func (controller *RuleController) RemoveRule() {
	ruService := service.NewEvaluationRuleService()
	in := &command.DeleteRuleCommand{}
	if err := controller.Unmarshal(in); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {
		controller.Response(ruService.Remove(in))
	}
}
郑周 authored
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
//func (controller *RuleController) ListRule() {
//	ruService := service.NewEvaluationRuleService()
//	in := &command.QueryRuleCommand{}
//	in.Type = -1
//	if err := controller.Unmarshal(in); err != nil {
//		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
//	} else {
//		if len(in.NameOrRemark) > 0 {
//			in.NameOrRemark = "%" + in.NameOrRemark + "%"
//		}
//		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
//		controller.Response(ruService.List(in))
//	}
//}

func (controller *RuleController) ListRuleRelCreator() {
77 78
	ruService := service.NewEvaluationRuleService()
	in := &command.QueryRuleCommand{}
郑周 authored
79
	in.Type = -1
80 81 82
	if err := controller.Unmarshal(in); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {
郑周 authored
83 84 85
		if len(in.NameOrRemark) > 0 {
			in.NameOrRemark = "%" + in.NameOrRemark + "%"
		}
郑周 authored
86 87
		ua := middlewares.GetUser(controller.Ctx)
		in.CompanyId = ua.CompanyId
88 89 90
		controller.Response(ruService.ListRelCreator(in))
	}
}
郑周 authored
91 92 93 94 95 96 97 98 99 100 101 102 103

func (controller *RuleController) ListCreator() {
	ruService := service.NewEvaluationRuleService()
	in := &command.QueryCreatorCommand{}
	if err := controller.Unmarshal(in); err != nil {
		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
	} else {

		ua := middlewares.GetUser(controller.Ctx)
		in.CompanyId = ua.CompanyId
		controller.Response(ruService.ListCreator(in))
	}
}