company_controller.go 2.9 KB
package backgroud_client

import (
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-gateway/pkg/application/background/company/command"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-gateway/pkg/application/background/company/query"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-gateway/pkg/application/background/company/service"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-gateway/pkg/port/beego/controllers"
)

type CompanyController struct {
	controllers.BaseController
}

// func (controller *CompanyController) CreateCompany() {
// 	companyService := service.NewCompanyService(nil)
// 	createCompanyCommand := &command.CreateCompanyCommand{}
// 	controller.Unmarshal(createCompanyCommand)
// 	data, err := companyService.CreateCompany(createCompanyCommand)
// 	controller.Response(data, err)
// }

func (controller *CompanyController) UpdateCompany() {
	companyService := service.NewCompanyService(nil)
	updateCompanyCommand := &command.UpdateCompanyCommand{}
	controller.Unmarshal(updateCompanyCommand)
	updateCompanyCommand.Operator = controller.GetOperator()
	data, err := companyService.UpdateCompany(updateCompanyCommand)
	controller.Response(data, err)
}

func (controller *CompanyController) GetCompany() {
	companyService := service.NewCompanyService(nil)
	getCompanyQuery := &query.GetCompanyQuery{}
	companyId, _ := controller.GetInt(":companyId")
	getCompanyQuery.CompanyId = companyId
	data, err := companyService.GetCompany(getCompanyQuery)
	controller.Response(data, err)
}

// func (controller *CompanyController) RemoveCompany() {
// 	companyService := service.NewCompanyService(nil)
// 	removeCompanyCommand := &command.RemoveCompanyCommand{}
// 	controller.Unmarshal(removeCompanyCommand)
// 	companyId, _ := controller.GetInt64(":companyId")
// 	removeCompanyCommand.CompanyId = companyId
// 	data, err := companyService.RemoveCompany(removeCompanyCommand)
// 	controller.Response(data, err)
// }

func (controller *CompanyController) SearchCompany() {
	companyService := service.NewCompanyService(nil)
	listCompanyQuery := &query.ListCompanyQuery{}
	controller.Unmarshal(listCompanyQuery)
	cnt, data, err := companyService.ListCompany(listCompanyQuery)
	controller.ReturnPageListData(cnt, data, err, listCompanyQuery.PageNumber)
}

func (controller *CompanyController) AuditCompany() {
	companyService := service.NewCompanyService(nil)
	cmd := &command.AuditCompanyCommand{}
	controller.Unmarshal(cmd)
	companyId, _ := controller.GetInt64(":companyId")
	cmd.CompanyId = companyId
	data, err := companyService.AuditCompany(cmd)
	controller.Response(data, err)
}

func (controller *CompanyController) EnableCompany() {
	companyService := service.NewCompanyService(nil)
	cmd := &command.EnableCompanyCommand{}
	controller.Unmarshal(cmd)
	companyId, _ := controller.GetInt64(":companyId")
	cmd.CompanyId = companyId
	data, err := companyService.EnableCompany(cmd)
	controller.Response(data, err)
}