java实现kml文件下载接口

news2025/2/24 18:32:35

根据业务需求，需提供一个下载kml格式航线文件的HTTP GET接口

示例代码

package com.kyrielx.kmzdemo.controller;

import org.apache.commons.io.FileUtils;
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 java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Objects;

/**
 * @author kyrielx
 * @date 2024/3/19 9:39
 * @desc
 **/
@RestController
@RequestMapping("/kyrielx")
public class DemoController {

    // 文件上传接口，接收一个file参数
    @PostMapping("/uploadKmz")
    public String upload(MultipartFile file) {
        // 保存MultipartFile类型的文件到本地目录
        try {
            // 获取当前项目相对路径地址 target/classes
            String path = Objects.requireNonNull(this.getClass().getClassLoader().getResource("")).getPath();
            System.out.println(path + file.getOriginalFilename());
            file.transferTo(new File( path + file.getOriginalFilename()));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "success";
    }

    // 文件下载接口
    @GetMapping(path = "/downloadKmz/{fileName}")
    public ResponseEntity<byte[]> download(@PathVariable String fileName) {
        // 将本地文件返回给前端
        if(null != fileName){
            try {
                HttpHeaders headers = new HttpHeaders();
                headers.setContentDispositionFormData("attachment", new String(fileName.getBytes(StandardCharsets.UTF_8), "ISO8859-1"));
                headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
                String path = Objects.requireNonNull(this.getClass().getClassLoader().getResource("")).getPath();
                System.out.println(path+fileName);
                File templateFile = new File(path + fileName);
                return new ResponseEntity<>(FileUtils.readFileToByteArray(templateFile), headers, HttpStatus.CREATED);
            }catch (Exception e){
                return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
            }
        }else{
            return new ResponseEntity<>(HttpStatus.BAD_REQUEST);
        }
    }

}