DriverController.java 7.24 KB
package com.ruoyi.driver.controller;

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

import com.ruoyi.common.global.Result;
import com.ruoyi.common.global.ResultCode;
import com.ruoyi.pojo.request.DriverRequestVo;
import com.ruoyi.pojo.request.DriverSignInRequestVo;
import com.ruoyi.pojo.request.FaceRegistrationFeedbackVo;
import com.ruoyi.pojo.request.FaceUpdateReqVo;
import com.ruoyi.pojo.response.DriverResponseVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.driver.domain.Driver;
import com.ruoyi.driver.service.IDriverService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;
import org.springframework.web.multipart.MultipartFile;

/**
 * 驾驶员信息Controller
 *
 * @author 古自健
 * @date 2023-07-04
 */
@RestController
@RequestMapping("/driver/driver")
@Api(tags = "驾驶员管理接口")
public class DriverController extends BaseController {
    @Autowired
    private IDriverService driverService;


    /**
     * 获取驾驶员排班信息
     */
    @GetMapping("/{schedulingDate}/{jobCode}")
    @ApiOperation("获取驾驶员排班信息")
    public AjaxResult getDriverSchedulingInfo(@ApiParam(name = "date", value = "yyyyMMdd", type = "String") @PathVariable("schedulingDate") String schedulingDate, @PathVariable("jobCode") String jobCode) {
        return driverService.getDriverSchedulingInfo(schedulingDate, jobCode);
    }

    /**
     * 获取当前所有有班次信息的工号集合
     */
    @GetMapping("/getDriverSchedulingAll")
    @ApiOperation("获取当前所有有班次信息的工号集合")
    public AjaxResult getDriverSchedulingAll() {
        return driverService.getDriverSchedulingAll();
    }

    /**
     * 查询驾驶员信息列表
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:list')")
    @GetMapping("/list")
    @ApiOperation(value = "查询驾驶员信息列表", notes = "查询驾驶员信息列表")
    public TableDataInfo list(DriverRequestVo driver) {
        startPage();
        List<Driver> list = driverService.selectDriverList(driver);
        return getDataTable(list);
    }

    /**
     * 获取驾驶员信息列表
     *
     * @param driver
     * @return
     */
    @GetMapping("/getDrivers")
    @ApiOperation(value = "获取驾驶员信息列表(全量)", notes = "查询驾驶员信息列表")
    public AjaxResult getDrivers(DriverRequestVo driver) {
        return AjaxResult.success(driverService.getDrivers(driver));
    }

    /**
     * 导出驾驶员信息列表
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:export')")
//    @Log(title = "驾驶员信息", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ApiOperation("导出驾驶员信息列表")
    public void export(HttpServletResponse response, DriverRequestVo driver) {
        List<Driver> list = driverService.selectDriverList(driver);
        ExcelUtil<Driver> util = new ExcelUtil<>(Driver.class);
        util.exportExcel(response, list, "驾驶员信息数据");
    }

    /**
     * 获取驾驶员信息详细信息
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:query')")
    @GetMapping(value = "/{id}")
    @ApiOperation("获取驾驶员信息详细信息")
    public AjaxResult getInfo(@ApiParam(value = "id", required = true) @PathVariable("id") Long id) {
        return success(driverService.selectDriverById(id));
    }

    /**
     * 新增驾驶员信息
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:add')")
    @Log(title = "驾驶员信息", businessType = BusinessType.INSERT)
    @PostMapping
    @ApiOperation("新增驾驶员信息(后台)")
    public AjaxResult add(@ApiParam @RequestBody Driver driver) {
        return toAjax(driverService.insertDriver(driver));
    }


    /**
     * 新增驾驶员信息
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:add')")
//    @Log(title = "驾驶员信息", businessType = BusinessType.INSERT)
    @PostMapping("/driver/sign")
    @ApiOperation("新增驾驶员信息(设备传入,更新人脸数据)")
    public Result<?> addNewDriver(@ApiParam @RequestBody @Validated DriverSignInRequestVo vo) {
        driverService.updateDriverSignInfo(vo);
        return Result.OK();
    }


    /**
     * 修改驾驶员信息
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:edit')")
//    @Log(title = "驾驶员信息", businessType = BusinessType.UPDATE)
    @PutMapping
    @ApiOperation("修改驾驶员信息")
    public AjaxResult edit(@ApiParam("driver") @RequestBody Driver driver) {
        return toAjax(driverService.updateDriver(driver) + 1);
    }


    /**
     * 人脸数据更新接口
     *
     * @param request 请求头
     * @param vo      更新实际数据
     */
    @PostMapping("/update/face")
    @ApiOperation("人脸数据更新接口")
    public Result<?> updateFaceByJob(HttpServletRequest request, @ApiParam @Validated @RequestBody FaceUpdateReqVo vo) {
        try {
            return driverService.updateFaceByJob(request, vo);
        } catch (Exception e) {
            return Result.ERROR(e.getMessage());
        }
    }

    /**
     * 删除驾驶员信息
     */
//    @PreAuthorize("@ss.hasPermi('driver:driver:remove')")
//    @Log(title = "驾驶员信息", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ids}")
    @ApiOperation("删除驾驶员信息")
    public AjaxResult remove(@ApiParam @PathVariable Long[] ids) {
        return toAjax(driverService.deleteDriverByIds(ids));
    }


    /**
     * 校验工号
     *
     * @param driver
     * @return
     */
    @GetMapping("/check")
    @ApiOperation("人事校验,是否为公司员工")
    public AjaxResult checkJobCode(@ApiParam @ModelAttribute Driver driver) {
        return driverService.checkJobCode(driver);
    }


    @PostMapping("/upload")
    @ApiOperation("驾驶员人脸上传接口")
    public AjaxResult uploadImage(MultipartFile file) {
        try {
            return driverService.uploadImage(file);
        } catch (Exception e) {
            return AjaxResult.error(e.getMessage());
        }
    }


    /**
     * 头像下载  根据 工号
     */
    @ApiOperation("头像下载")
    @GetMapping("/download/{jobCode}")
    public void downLoadHeadImage(@PathVariable("jobCode") String jobCode, HttpServletResponse response) {
        driverService.downloadHeadImage(jobCode, response);
    }

    /**
     * 人脸注册状态
     */
    @ApiOperation("注册反馈")
    @PostMapping("/faceSignIn")
    public AjaxResult faceRegistrationFeedback(@RequestBody FaceRegistrationFeedbackVo vo) {
        return driverService.faceRegistrationFeedback(vo.getDeviceId(), vo.getJobCodes());
    }

//    @GetMapping("/questionFeedback")
////    @ApiOperation("人脸识别反馈")
//    public Result<?> questionFeedback() {
//
//        return null;
//    }
}