parent
968e290a94
commit
35602b9bf8
@ -1,85 +0,0 @@
|
|||||||
package cc.yunxi.controller;
|
|
||||||
|
|
||||||
import io.swagger.annotations.ApiModel;
|
|
||||||
import io.swagger.annotations.ApiOperation;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.springframework.core.io.ClassPathResource;
|
|
||||||
import org.springframework.core.io.Resource;
|
|
||||||
import org.springframework.http.ResponseEntity;
|
|
||||||
import org.springframework.util.StringUtils;
|
|
||||||
import org.springframework.web.bind.annotation.GetMapping;
|
|
||||||
import org.springframework.web.bind.annotation.PathVariable;
|
|
||||||
import org.springframework.web.bind.annotation.RestController;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.nio.charset.StandardCharsets;
|
|
||||||
import java.nio.file.Files;
|
|
||||||
import java.nio.file.Path;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
|
||||||
|
|
||||||
@ApiModel("机柜扫码登录")
|
|
||||||
@Slf4j
|
|
||||||
@RestController("/qrcode")
|
|
||||||
public class QRCodeController {
|
|
||||||
|
|
||||||
private static final Map<String, String> FILE_CONTEXT = new ConcurrentHashMap<>();
|
|
||||||
|
|
||||||
// @ApiOperation("二维码校验文件")
|
|
||||||
// @GetMapping("/qrcode/{filename}")
|
|
||||||
// public void getQRCode(HttpServletResponse response, @PathVariable String filename) throws IOException {
|
|
||||||
// Resource resource = new ClassPathResource("static/" + filename);
|
|
||||||
//
|
|
||||||
// if (resource.exists() || resource.isReadable()) {
|
|
||||||
// // 设置响应头
|
|
||||||
// String contentType = null;
|
|
||||||
// try (InputStream inputStream = resource.getInputStream()) {
|
|
||||||
// contentType = "application/octet-stream";
|
|
||||||
// } catch (IOException e) {
|
|
||||||
// // 如果无法获取输入流,则回退到默认内容类型
|
|
||||||
// contentType = "application/octet-stream";
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// response.setContentType(contentType);
|
|
||||||
// response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + resource.getFilename() + "\"");
|
|
||||||
//
|
|
||||||
// try (InputStream inputStream = resource.getInputStream();
|
|
||||||
// OutputStream outputStream = response.getOutputStream()) {
|
|
||||||
//
|
|
||||||
// byte[] buffer = new byte[4096];
|
|
||||||
// int bytesRead = -1;
|
|
||||||
// while ((bytesRead = inputStream.read(buffer)) != -1) {
|
|
||||||
// outputStream.write(buffer, 0, bytesRead);
|
|
||||||
// }
|
|
||||||
// outputStream.flush();
|
|
||||||
// }
|
|
||||||
// } else {
|
|
||||||
// response.setStatus(HttpServletResponse.SC_NOT_FOUND);
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
@ApiOperation("二维码校验文件")
|
|
||||||
@GetMapping("/qrcode/{filename:.+}")
|
|
||||||
public ResponseEntity<String> getFile(@PathVariable String filename) {
|
|
||||||
if (StringUtils.hasText(filename)) {
|
|
||||||
String context = FILE_CONTEXT.get(filename);
|
|
||||||
if (StringUtils.hasText(context)) return ResponseEntity.ok(context);
|
|
||||||
}
|
|
||||||
Resource resource = new ClassPathResource("static/" + filename);
|
|
||||||
if (resource.exists() || resource.isReadable()) {
|
|
||||||
|
|
||||||
try {
|
|
||||||
Path path = resource.getFile().toPath();
|
|
||||||
String content = new String(Files.readAllBytes(path), StandardCharsets.UTF_8);
|
|
||||||
FILE_CONTEXT.put(filename, content);
|
|
||||||
return ResponseEntity.ok(content);
|
|
||||||
} catch (IOException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
log.error("读取校验文件失败: {}", e.getMessage());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ResponseEntity.notFound().build();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
@ -0,0 +1 @@
|
|||||||
|
175babf4957dd6090b5b36b78cd51c78
|
Loading…
Reference in new issue