From 48d3f1b77adea0609bb2453c7b4a660f2222871f Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期一, 25 八月 2025 08:54:50 +0800
Subject: [PATCH] bug修改

---
 ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/ChargeOrderController.java |   59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 57 insertions(+), 2 deletions(-)

diff --git a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/ChargeOrderController.java b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/ChargeOrderController.java
index 99b3d35..cb17939 100644
--- a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/ChargeOrderController.java
+++ b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/ChargeOrderController.java
@@ -1,19 +1,36 @@
 package com.ruoyi.order.controller;
 
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
 import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.dto.ChargingOrderGroup;
 import com.ruoyi.common.core.dto.ChargingPercentProvinceDto;
+import com.ruoyi.common.core.utils.WebUtils;
 import com.ruoyi.common.core.web.page.PageInfo;
+import com.ruoyi.common.security.service.TokenService;
 import com.ruoyi.order.dto.GetImportOrderDTO;
+import com.ruoyi.order.export.ChargeOrderExport;
 import com.ruoyi.order.model.ChargeOrder;
 import com.ruoyi.order.service.ChargeOrderService;
+import com.ruoyi.other.api.domain.TIntegralRule;
+import com.ruoyi.system.api.domain.SysConfig;
+import com.ruoyi.system.api.domain.SysUser;
+import com.ruoyi.system.api.feignClient.SysUserClient;
+import com.ruoyi.system.api.model.LoginUser;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 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;
 
 @Api(tags = "充电订单")
@@ -22,6 +39,8 @@
 public class ChargeOrderController {
     @Resource
     private ChargeOrderService chargeOrderService;
+    @Resource
+    private SysUserClient sysUserClient;
     @PostMapping(value = "/getBySiteIdAndTime")
     public R<List<ChargingOrderGroup>> getBySiteIdAndTime(@RequestBody ChargingPercentProvinceDto chargingPercentProvinceDto) {
 
@@ -32,16 +51,18 @@
      * 导入充电信息
      */
     @PostMapping("/importExpress")
+    @ApiOperation(value = "导入充电信息", tags = "后台-充电订单-导入充电信息")
     public R importExpress(@RequestParam("file") MultipartFile file) {
        /* JSONObject jsonObject = JSONObject.parseObject(url);
         String url2 = jsonObject.getString("url");*/
         if (file.isEmpty()) {
             return R.fail("请选择要上传的文件");
         }
-
         return chargeOrderService.importExpress(file);
     }
 
+    @Autowired
+    private TokenService tokenService;
     /**
      * 获取导入订单列表
      */
@@ -49,9 +70,43 @@
     @PostMapping("/getImportData")
     @ApiOperation(value = "获取导入订单列表", tags = "后台-充电订单")
     public R<PageInfo<ChargeOrder>> getUserPointPageList(@RequestBody GetImportOrderDTO getImportOrderDTO) {
-
+        LoginUser loginUser = tokenService.getLoginUser();
+        R<SysUser> sysUser = sysUserClient.getSysUser(loginUser.getUserid());
+        SysUser data = sysUser.getData();
+        if (data.getRoleType() == 2){
+            // 只查询当前站点数据
+            getImportOrderDTO.setSiteId(data.getSiteId());
+        }
         PageInfo<ChargeOrder> pageInfo=chargeOrderService.getUserPointPageList(getImportOrderDTO);
 
         return R.ok(pageInfo);
     }
+
+    @ApiOperation(value = "充电订单导入模板下载", tags = "后台-充电订单")
+    @GetMapping("/importChargeOrder")
+    public void importChargeOrder() {
+        List<ChargeOrderExport> chargeOrderExports = new ArrayList<>();
+        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), ChargeOrderExport.class, chargeOrderExports);
+        HttpServletResponse response = WebUtils.response();
+        ServletOutputStream outputStream = null;
+        try {
+            String fileName = URLEncoder.encode("充电订单导入模板.xls", "utf-8");
+            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
+            response.setHeader("content-Type", "application/vnd.ms-excel");
+            response.setHeader("Pragma", "no-cache");
+            response.setHeader("Cache-Control", "no-cache");
+            outputStream = response.getOutputStream();
+            workbook.write(outputStream);
+        } catch (IOException e) {
+            e.printStackTrace();
+            System.out.println("充电订单导入模板下载失败!");
+        } finally {
+            try {
+                outputStream.close();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
 }

--
Gitblit v1.7.1