yunsu/cloud-model/src/main/java/com/hssx/cloudmodel/controller/CompanyController.java

package com.hssx.cloudmodel.controller;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.hssx.cloudmodel.entity.Company;
import com.hssx.cloudmodel.entity.User;
import com.hssx.cloudmodel.entity.vo.CompanyVO;
import com.hssx.cloudmodel.service.CompanyService;
import com.hssx.cloudmodel.util.HttpRespMsg;
import com.hssx.cloudmodel.util.PageUtil;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;


/**
 * @author 吴涛涛
 * @since 2019-07-26
 */
@Controller
@RequestMapping("/company")
@Slf4j
public class CompanyController {

    @Autowired
    private CompanyService companyService;

    /**
     * 添加/修改公司
     *
     * 参数：companyName 公司名 companyAddress 公司地址
     * companyType公司类型 0-资产方 1-生产方（yLng 经度  xLat 纬度）
     * flag   0-添加，1-修改
     * @return
     */
    @ApiOperation("添加/修改公司")
    @RequestMapping("/add")
    @ResponseBody
    public HttpRespMsg addAndUpdateRole(CompanyVO companyVO, HttpServletRequest request, Integer flag,
                                        HttpServletResponse response) {
        HttpRespMsg msg = new HttpRespMsg();
        msg = companyService.addAndUpdateRole(companyVO, flag);
        return msg;
    }

    /**
     * 删除公司
     *
     * @param company 参数 id 公司id
     * @return
     */
    @ApiOperation("删除公司")
    @RequestMapping("/delete")
    @ResponseBody
    public HttpRespMsg delete(Company company, HttpServletRequest request,
                              HttpServletResponse response) {
        HttpRespMsg msg = new HttpRespMsg();
        msg = companyService.deleteById(company.getId());
        return msg;
    }

    /**
     * 公司列表
     * 参数：pageNum 当前页码，pageSize 每页条数 keyName 关键字查询,companyType 公司类型 0-资产方 1-生产方
     * @return
     */
    @ApiOperation("公司列表")
    @RequestMapping("/list")
    @ResponseBody
    public HttpRespMsg deleteRole(@RequestParam(required = false)String keyName, HttpServletRequest request,
                                  HttpServletResponse response, PageUtil page,@RequestParam(required = false)Integer companyType) {
        HttpRespMsg msg = new HttpRespMsg();
        msg = companyService.pageList(page,keyName,companyType);
        return msg;
    }

    /**
     * 创建模具获取的(资产方)公司列表
     * @return
     */
    @ApiOperation("创建模具获取的公司列表")
    @RequestMapping("/getCompanys")
    @ResponseBody
    public HttpRespMsg getCompanys() {
        HttpRespMsg msg = new HttpRespMsg();
        QueryWrapper<Company> qw = new QueryWrapper<>();
        qw.eq("company_type",0);
        msg.data  = companyService.list(qw);
        return msg;
    }

    /**
     * 创建账号是所获取的公司列表
     * parentId 当前用户的上级id id 当前用户id
     * @return
     */
    @ApiOperation(value = "获取公司列表")
    @RequestMapping("/getCompanyList")
    @ResponseBody
    public HttpRespMsg getCompanyList(User user){
        HttpRespMsg msg = new HttpRespMsg();
        List<Company> list = companyService.getIdAndNamelist(user);
        msg.data = list;
        return msg;
    }

    /**
     * 项目分配生产方公司列表
     * @return
     */
    @ApiOperation(value = "项目分配生产方公司列表")
    @RequestMapping("/addCompanyListToProject")
    @ResponseBody
    public HttpRespMsg addCompanyListToProject(User user){
        HttpRespMsg msg = new HttpRespMsg();
        List<Company> list = companyService.addCompanyListToProject(user);
        msg.data = list;
        return msg;
    }

}