无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
zrj
2024-08-20 8b7b86f12fab51fe21b5575f4ffd524fbeaf412b 
 src/main/java/com/dji/sample/patches/controller/PatchesController.java


@@ -6,23 +6,26 @@


import com.dji.sample.common.util.MinioUrlUtils;


import com.dji.sample.media.model.MediaFileEntity;


import com.dji.sample.media.service.impl.FileServiceImpl;


import com.dji.sample.patches.constant.FileSufConstant;


import com.dji.sample.patches.model.Param.PatchesParam;


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;


@@ -70,7 +73,7 @@


                                                        @RequestParam(name = "xmc", required = false) String xmc,


                                                        @RequestParam(name = "investigate", required = false) Integer investigate,


                                                        @RequestParam(name = "dkmj", required = false) Double dkmj,


                                                        @RequestParam(name = "isPlan",required = false) Integer isPlan


                                                        @RequestParam(name = "isPush",required = false) Integer isPush


                                                        ) {


        fileService.getNoaddFile();


        //调用service分页查询


@@ -84,7 +87,7 @@


                .xmc(xmc)


                .investigate(investigate)


                .dkmj(dkmj)


                .isPlan(isPlan)


                .isPush(isPush)


                .build();


        PaginationData<LotInfo> data = getPatchesService.limitGet(param);


        return ResponseResult.success(data);


@@ -176,7 +179,7 @@


//            timerUtil.mytask5();


//            timerUtil.mytask6();


//            timerUtil.mytask7();




            timerUtil.GtdbOperation();


        } catch (Exception e) {


            throw new RuntimeException("db存储发送出现异常" + e);


        }


@@ -253,6 +256,10 @@


                                 @RequestParam double airportLat,


                                 @RequestParam double airportLon,


                                 HttpServletRequest request) throws Exception {


        // 检查格式是否正确


        if(!checkFileType(file.getOriginalFilename())){


            throw new RuntimeException("文件格式异常");


        }


        CustomClaim customClaim = (CustomClaim) request.getAttribute(TOKEN_CLAIM);


        String creator = customClaim.getUsername();


        MultipartFile multipartFile = shpToDataSourceService.insertGeo(file, workspaceId, waylineName, airportLat, airportLon, creator);


@@ -269,4 +276,41 @@


        }


        return ResponseResult.success(infoMap);


    }




    /**


     * 检查文件格式是否正确


     * @param originalFilename


     * @return


     */


    private boolean checkFileType(String originalFilename) {


        if (originalFilename.endsWith(FileSufConstant.FILE_SUFFIX_ZIP) ||


                originalFilename.endsWith(FileSufConstant.FILE_SUFFIX_KML) ||


                originalFilename.endsWith(FileSufConstant.FILE_SUFFIX_KMZ)


        ){


            return true;


        }


        // 返回


        return false;


    }




    @GetMapping("/getExcel")


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


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


        try {


            List<LotInfo> lotInfos = getPatchesService.getLotInfosByIDs(ids);


            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<>(("生成excel失败.").getBytes(), HttpStatus.INTERNAL_SERVER_ERROR);


        }


    }


}