1

13693261870

2023-02-19 5a6e173298a10b22285b67044fa29ab96e22ef22

 src/main/java/com/lf/server/controller/data/WmtsController.java

@@ -1,8 +1,7 @@
package com.lf.server.controller.data;

import com.lf.server.annotation.SysLog;
import com.lf.server.entity.all.StaticData;
import com.lf.server.helper.GdalHelper;
import com.lf.server.config.PropertiesConfig;
import com.lf.server.helper.StringHelper;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
@@ -10,17 +9,29 @@
import io.swagger.annotations.ApiOperation;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.http.server.ServletServerHttpResponse;
import org.springframework.web.bind.annotation.*;

import javax.servlet.ServletOutputStream;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.util.Date;
import java.util.List;

/**
 * WMTS服务
@@ -30,19 +41,19 @@
@RestController
@RequestMapping("/wmts")
public class WmtsController {
    @Autowired
    PropertiesConfig config;

    private final static Log log = LogFactory.getLog(WmtsController.class);

    @SysLog()
    @ApiOperation(value = "查询WMTS元数据")
    @ApiOperation(value = "获取WMTS元数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "token", value = "令牌", dataType = "String", required = true, defaultValue = "token", paramType = "path")})
    @GetMapping("select/metas/{token}")
    public void selectMetas(String service, String version, String request, @PathVariable(name = "token") String token, HttpServletRequest req, HttpServletResponse res) {
            @ApiImplicitParam(name = "token", value = "令牌", required = true, dataType = "String", defaultValue = "token", paramType = "path")
    })
    @GetMapping("select/{token}/WMTSCapabilities.xml")
    public void selectWmtsCapabilities(@PathVariable(name = "token") String token, HttpServletRequest req, HttpServletResponse res) {
        try {
            if (!StaticData.WMTS.equals(service) || !StaticData.CAPABILITY.equals(request)) {
                return;
            }

            ClassPathResource resource = new ClassPathResource("wmts/web.xml");
            if (!resource.exists()) {
                return;
@@ -53,7 +64,10 @@
            stream.read(data);
            stream.close();

            String url = req.getRequestURL().toString().replace("WMTSCapabilities.xml", "tile");

            String result = new String(data, StandardCharsets.UTF_8);
            result = result.replace("{url}", url);
            byte[] bytes = result.getBytes(StandardCharsets.UTF_8);

            res.setHeader("Content-Type", "application/xml;charset=UTF-8");
@@ -64,4 +78,96 @@
            log.error(ex.getMessage(), ex);
        }
    }

    @ApiOperation(value = "获取WMTS瓦片")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "token", value = "令牌", required = true, dataType = "String", defaultValue = "token", paramType = "path"),
            @ApiImplicitParam(name = "layer", value = "图层类型", required = true, dataType = "String", defaultValue = "img"),
            @ApiImplicitParam(name = "z", value = "层级", required = true, dataType = "Integer"),
            @ApiImplicitParam(name = "x", value = "行号", required = true, dataType = "Integer"),
            @ApiImplicitParam(name = "y", value = "列号", required = true, dataType = "Integer")
    })
    @GetMapping("select/{token}/tile")
    public void selectWmtsTile(@RequestParam("LAYER") String layer, @RequestParam("TILEMATRIX") Integer z, @RequestParam("TILEROW") Integer x,
                               @RequestParam("TILECOL") Integer y, @PathVariable(name = "token") String token,
                               HttpServletRequest req, HttpServletResponse res) {
        try {
            // ServletServerHttpRequest ssRequest = new ServletServerHttpRequest(request)
            if (StringHelper.isEmpty(layer) || null == z || null == x || null == y) {
                return;
            }

            ServletServerHttpRequest ssReq = new ServletServerHttpRequest(req);
            ServletServerHttpResponse ssRes = new ServletServerHttpResponse(res);

            // 检查缓存是否过期
//            if (checkIfNotModify(ssReq, ssRes)) {
//                // 设置缓存头
//                setBrowerCache(ssRes);
//                return;
//            }

            // 设置返回图片类型
            res.setContentType("image/png");
            // 设置缓存参数
            setBrowerCache(ssRes);
            // 通过response对象，获取到输出流
            ServletOutputStream outputStream = res.getOutputStream();
            // 定义输入流，通过输入流读取文件内容
            FileInputStream fileInputStream;

            // y = (1 << z) - y - 1
            y = (int) Math.pow(2, z) - y - 1;
            String path = config.getTilePath() + File.separator + layer + File.separator + z + File.separator + x + File.separator + y + ".png";

            File file = new File(path);
            if (!file.exists() || file.isDirectory()) {
                ClassPathResource resource = new ClassPathResource("wmts/nofound.png");
                fileInputStream = new FileInputStream(resource.getFile());
            } else {
                System.out.println(path);
                fileInputStream = new FileInputStream(file);
            }

            int len = 0;
            byte[] bytes = new byte[1024];
            while ((len = fileInputStream.read(bytes)) != -1) {
                outputStream.write(bytes, 0, len);
                outputStream.flush();
            }

            outputStream.close();
            fileInputStream.close();
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }

    /**
     * 设置浏览器缓存参数
     */
    private void setBrowerCache(ServerHttpResponse res) {
        HttpHeaders headers = res.getHeaders();
        headers.setCacheControl("public, must-revalidate");

        headers.setExpires(System.currentTimeMillis() + 24 * 60 * 60 * 1000);
        headers.setAccessControlMaxAge(Duration.ofHours(24));
        headers.setETag("\"" + new Date().toString() + "\"");
        // must-revalidate
        // response.Cache.SetRevalidation(HttpCacheRevalidation.AllCaches);
    }

    /**
     * 验证缓存参数
     */
    boolean checkIfNotModify(ServerHttpRequest req, ServerHttpResponse res) {
        List<String> etags = req.getHeaders().getIfNoneMatch();
        if (0 == etags.size() || StringHelper.isEmpty(etags.get(0))) {
            return false;
        }

        res.setStatusCode(HttpStatus.NOT_MODIFIED);

        return true;
    }
}