import { Body, Controller, Del, Get, Inject, Param, Post, Query } from '@midwayjs/core'; import { ApiResponse, ApiTags, ApiQuery } from '@midwayjs/swagger'; import { Validate } from '@midwayjs/validate'; import { BaseController } from '../../frame/BaseController'; import { omit, pick } from 'lodash'; import { ErrorCode, ServiceError } from '../../error/service.error'; import { CompanyService } from '../../service/users/company.service'; import { CVO_company, FVO_company, QVO_company, UVAO_company } from '../../interface/users/company.interface'; import { Context } from '@midwayjs/koa'; import { ServiceUtilService } from '../../service/serviceUtil.service'; const namePrefix = '企业'; @ApiTags(['企业']) @Controller('/company', { tagName: namePrefix }) export class CompanyController implements BaseController { @Inject() service: CompanyService; @Inject() ctx: Context; @Inject() serviceUtil: ServiceUtilService; @Get('/') @ApiTags('列表查询') @ApiQuery({ name: 'query' }) @ApiResponse({ type: QVO_company }) async index(@Query() query: object) { const qobj = omit(query, ['skip', 'limit']); const others = pick(query, ['skip', 'limit']); const result = await this.service.query(qobj, others); return result; } @Get('/:id') @ApiTags('单查询') @ApiResponse({ type: FVO_company }) async fetch(@Param('id') id: number) { const data = await this.service.fetch({ id }); const result = new FVO_company(data); return result; } @Post('/', { routerName: `创建${namePrefix}` }) @ApiTags('创建数据') @Validate() @ApiResponse({ type: CVO_company }) async create(@Body() data: object) { const dbData = await this.service.create(data); await this.serviceUtil.fillIdentity(data, 'Company'); const result = new CVO_company(dbData); return result; } @Post('/:id', { routerName: `修改${namePrefix}` }) @ApiTags('修改数据') @Validate() @ApiResponse({ type: UVAO_company }) async update(@Param('id') id: number, @Body() data: object) { if (!id) throw new ServiceError(ErrorCode.ID_NOT_FOUND); const result = await this.service.update({ id }, data); await this.serviceUtil.fillIdentity(data, 'Company'); return result; } @Del('/:id', { routerName: `删除${namePrefix}` }) @ApiTags('删除数据') @Validate() async delete(@Param('id') id: number) { if (!id) throw new ServiceError(ErrorCode.ID_NOT_FOUND); const result = await this.service.delete({ id }); return result; } @Get('/list') async list(@Query() query: object) { const qobj = omit(query, ['skip', 'limit']); const others = pick(query, ['skip', 'limit']); const { data, total } = await this.service.query(qobj, others); for (let i of data) { i = await this.serviceUtil.fillOnwer(i); } return { data, total }; } @Get('/contact') @ApiTags('列表查询') @ApiQuery({ name: 'query' }) @ApiResponse({ type: QVO_company }) async contact(@Query() query: object) { const qobj = omit(query, ['skip', 'limit']); const others = pick(query, ['skip', 'limit']); const { data, total } = await this.service.query(qobj, others); const list = await this.service.fillMatch(data); return { data: list, total }; } @Get('/detail/:id') @ApiResponse({ type: FVO_company }) async detail(@Param('id') id: string) { let data = await this.service.fetch({ id }); data = await this.serviceUtil.fillOnwer(data); data = await this.serviceUtil.fillCollection(data, 'company'); return data; } }