xuhy
2024-09-18 14f62b1e99c93facce4245dbae56b9f43cb0e017
ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TOrderInvoiceController.java
@@ -1,9 +1,37 @@
package com.ruoyi.order.controller;
import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import com.ruoyi.common.core.utils.DateUtils;
import com.ruoyi.common.core.utils.WebUtils;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.PageInfo;
import com.ruoyi.common.log.annotation.Log;
import com.ruoyi.common.log.enums.BusinessType;
import com.ruoyi.order.api.model.TOrderInvoice;
import com.ruoyi.order.api.query.TOrderInvoiceQuery;
import com.ruoyi.order.api.vo.TOrderInvoiceVO;
import com.ruoyi.order.dto.AddOrderInvoice;
import com.ruoyi.order.dto.GetOrderInvoiceList;
import com.ruoyi.order.dto.MyOrderInvoiceInfo;
import com.ruoyi.order.dto.OrderInvoiceList;
import com.ruoyi.order.export.OrderInvoiceExport;
import com.ruoyi.order.service.TOrderInvoiceService;
import io.swagger.annotations.Api;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.BeanUtils;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
/**
 * <p>
@@ -18,5 +46,99 @@
@RequestMapping("/t-order-invoice")
public class TOrderInvoiceController {
   @Resource
   private TOrderInvoiceService orderInvoiceService;
   @ResponseBody
   @PostMapping("/addOrderInvoice")
   @ApiOperation(value = "添加开票申请", tags = {"小程序-充电发票"})
   public AjaxResult addOrderInvoice(@RequestBody AddOrderInvoice addOrderInvoice){
      return orderInvoiceService.addOrderInvoice(addOrderInvoice);
   }
   @PostMapping("/pageList")
   @ApiOperation(value = "查询开票分页列表", tags = {"管理后台-发票管理"})
   public AjaxResult<PageInfo<TOrderInvoiceVO>> pageList(@RequestBody TOrderInvoiceQuery query){
      return AjaxResult.success(orderInvoiceService.pageList(query));
   }
   @GetMapping("/uploadPdf")
   @ApiOperation(value = "上传发票", tags = {"管理后台-发票管理"})
   public AjaxResult<String> uploadPdf(@RequestParam("id") Long id,
                              @RequestParam("invoiceUrl") String invoiceUrl){
      TOrderInvoice orderInvoice = orderInvoiceService.getById(id);
      orderInvoice.setInvoiceUrl(invoiceUrl);
      orderInvoiceService.updateById(orderInvoice);
      return AjaxResult.success();
   }
   @PostMapping("/statusCount")
   @ApiOperation(value = "查询开票状态数量", tags = {"管理后台-发票管理"})
   public AjaxResult<Map<String,Integer>> statusCount(@RequestBody TOrderInvoiceQuery query){
      return AjaxResult.success(orderInvoiceService.statusCount(query));
   }
   /**
    * 发票管理导出
    */
   @ApiOperation(value = "发票管理导出", tags = {"管理后台-发票管理"})
   @Log(title = "发票管理导出", businessType = BusinessType.EXPORT)
   @PutMapping("/export")
   public void export(@RequestBody TOrderInvoiceQuery query)
   {
      List<TOrderInvoiceVO> list = orderInvoiceService.export(query);
      List<OrderInvoiceExport> orderInvoiceExports = new ArrayList<>();
      for (TOrderInvoiceVO orderInvoiceVO : list) {
         OrderInvoiceExport orderInvoiceExport = new OrderInvoiceExport();
         BeanUtils.copyProperties(orderInvoiceVO,orderInvoiceExport);
         if (orderInvoiceVO.getBillingTime()!=null){
            orderInvoiceExport.setBillingTime(DateUtils.localDateTimeToString(orderInvoiceVO.getBillingTime()));
         }
         orderInvoiceExports.add(orderInvoiceExport);
      }
      Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), OrderInvoiceExport.class, orderInvoiceExports);
      HttpServletResponse response = WebUtils.response();
      response.setContentType("application/vnd.ms-excel");
      response.setCharacterEncoding("utf-8");
      ServletOutputStream outputStream = null;
      try {
         String fileName = URLEncoder.encode("发票导出.xls", "utf-8");
         response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
         response.setContentType("application/vnd.ms-excel;charset=UTF-8");
         response.setHeader("Pragma", "no-cache");
         response.setHeader("Cache-Control", "no-cache");
         outputStream = response.getOutputStream();
         workbook.write(outputStream);
      } catch (IOException e) {
         e.printStackTrace();
      } finally {
         try {
            outputStream.close();
         } catch (IOException e) {
            e.printStackTrace();
         }
      }
   }
   @ResponseBody
   @GetMapping("/getMyOrderInvoiceList")
   @ApiOperation(value = "获取开票记录列表", tags = {"小程序-充电发票"})
   public AjaxResult<List<OrderInvoiceList>> getMyOrderInvoiceList(GetOrderInvoiceList query){
      List<OrderInvoiceList> myOrderInvoiceList = orderInvoiceService.getMyOrderInvoiceList(query);
      return AjaxResult.success(myOrderInvoiceList);
   }
   @ResponseBody
   @GetMapping("/getMyOrderInvoiceInfo/{id}")
   @ApiOperation(value = "获取开票记录详情", tags = {"小程序-充电发票"})
   public AjaxResult<MyOrderInvoiceInfo> getMyOrderInvoiceInfo(@PathVariable String id){
      MyOrderInvoiceInfo myOrderInvoiceInfo = orderInvoiceService.getMyOrderInvoiceInfo(id);
      return AjaxResult.success(myOrderInvoiceInfo);
   }
}