company_controller.go 3.0 KB
package controllers

import (
	"github.com/linmadan/egglib-go/web/beego"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-user/pkg/application/company/command"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-user/pkg/application/company/query"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-user/pkg/application/company/service"
)

type CompanyController struct {
	beego.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)
	companyId, _ := controller.GetInt64(":companyId")
	updateCompanyCommand.CompanyId = companyId
	data, err := companyService.UpdateCompany(updateCompanyCommand)
	controller.Response(data, err)
}

func (controller *CompanyController) GetCompany() {
	companyService := service.NewCompanyService(nil)
	getCompanyQuery := &query.GetCompanyQuery{}
	companyId, _ := controller.GetInt64(":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) ListCompany() {
	companyService := service.NewCompanyService(nil)
	listCompanyQuery := &query.ListCompanyQuery{}
	data, err := companyService.ListCompany(listCompanyQuery)
	controller.Response(data, err)
}

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

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