无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
rain
2024-07-31 4e8892f75a0d9b76e147cccdb1f188c272aed4c5 
 src/main/java/com/dji/sample/patches/controller/PatchesController.java


@@ -10,19 +10,21 @@


import com.dji.sample.patches.model.entity.LotInfo;


import com.dji.sample.patches.service.GetPatchesService;


import com.dji.sample.patches.service.ShpToDataSourceService;


import com.dji.sample.patches.utils.DistrictCodeUtils;


import com.dji.sample.patches.utils.GeoToolsUtil;


import com.dji.sample.patches.utils.PointPO;


import com.dji.sample.patches.utils.TimerUtil;


import com.dji.sample.patches.utils.*;


import com.dji.sample.territory.service.impl.TbDkjbxxServiceImpl;


import com.dji.sample.wayline.model.entity.WaylineFileEntity;


import com.dji.sample.wayline.service.IWaylineFileService;


import com.dji.sample.wayline.service.IWaylineJobService;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.http.HttpHeaders;


import org.springframework.http.HttpStatus;


import org.springframework.http.MediaType;


import org.springframework.http.ResponseEntity;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.multipart.MultipartFile;




import javax.servlet.http.HttpServletRequest;


import java.io.ByteArrayInputStream;


import java.io.IOException;


import java.net.URL;


import java.sql.SQLException;


@@ -269,4 +271,24 @@


        }


        return ResponseResult.success(infoMap);


    }




    @GetMapping("/getExcel")


    public ResponseEntity<byte[]> exportExcel(@RequestParam String workspaceId) {


        List<LotInfo> lotInfos = getPatchesService.listLotinfo(workspaceId);


        try {


            ByteArrayInputStream excelStream = ExcelUtil.exportLotInfoToExcel(lotInfos);


            byte[] excelData = excelStream.readAllBytes();


            excelStream.close();




            HttpHeaders headers = new HttpHeaders();


            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=lot_info_data.xlsx");


            headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_OCTET_STREAM_VALUE);


            headers.setContentLength(excelData.length);




            return new ResponseEntity<>(excelData, headers, HttpStatus.OK);


        } catch (IOException e) {


            e.printStackTrace();


            return new ResponseEntity<>(("Failed to generate Excel file.").getBytes(), HttpStatus.INTERNAL_SERVER_ERROR);


        }


    }


}