package com.terra.proxy.controller;
|
|
import cn.hutool.core.bean.BeanUtil;
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.fastjson.JSON;
|
import com.terra.proxy.bean.*;
|
import com.terra.proxy.config.SysTemPropertyConfig;
|
import com.terra.proxy.intercepter.ServiceExcelListener;
|
import com.terra.proxy.mapper.LogMapper;
|
import com.terra.proxy.properties.TerraProperties;
|
import com.terra.proxy.service.Impl.ServerRegisterServiceImpl;
|
import com.terra.proxy.util.JedisUtils;
|
import com.terra.proxy.util.JwtUtils;
|
import com.terra.proxy.util.Result;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import redis.clients.jedis.Jedis;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URI;
|
import java.net.URISyntaxException;
|
import java.net.URLEncoder;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.Objects;
|
|
@RestController
|
@RequestMapping("/excel")
|
public class ExcelController {
|
@Autowired
|
private SysTemPropertyConfig sys;
|
|
@Autowired
|
private ServerRegisterServiceImpl serverRegisterService;
|
|
@Autowired
|
private TerraProperties properties;
|
|
@Autowired
|
private JwtUtils JwtUtils;
|
|
@Autowired
|
private LogMapper logMapper;
|
|
private static String fileName = "服务注册模板Excel.xlsx";
|
|
private static String fileResultName = "服务注册结果.xlsx";
|
@Value("${sys.jwt.expire}")
|
private Long expireSeconds;
|
|
/*
|
* 获取注册服务的Excel模板
|
*
|
* */
|
@GetMapping("getExcelModle")
|
public ResponseEntity getExcelModle() {
|
String uploadPath = sys.getUploadPath();
|
File file = new File(uploadPath);
|
if (!file.exists()) {
|
file.mkdir();
|
}
|
ArrayList<ServerExcelApplyEntity> userEntities = new ArrayList<>();
|
userEntities.add(new ServerExcelApplyEntity());
|
EasyExcel.write(uploadPath + File.separator + fileName, ServerExcelApplyEntity.class).sheet("服务注册表").doWrite(userEntities);
|
return ResponseEntity.ok().body(null);
|
}
|
|
/*
|
* 获取注册服务的Excel文件
|
*
|
* */
|
@GetMapping("getExcel")
|
public ResponseEntity getExcel(HttpServletResponse response) throws IOException {
|
String uploadPath = sys.getUploadPath();
|
FileInputStream fileInputStream = null;
|
ServletOutputStream outputStream = response.getOutputStream();
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "utf-8"));
|
File file = new File(uploadPath + File.separator + fileName);
|
if (file.exists() && file.canRead()) {
|
fileInputStream = new FileInputStream(file);
|
byte[] bytes = new byte[1024];
|
int length;
|
while ((length = fileInputStream.read(bytes)) != -1) {
|
outputStream.write(bytes, 0, length);
|
}
|
}
|
if (null != fileInputStream) {
|
fileInputStream.close();
|
}
|
if (null != outputStream) {
|
outputStream.flush();
|
outputStream.close();
|
}
|
return ResponseEntity.ok().body(null);
|
}
|
|
/*
|
* 导入注册服务的Excel文件
|
*
|
* */
|
@PostMapping("importExcel")
|
public String importExcel(@RequestPart("file") MultipartFile file, HttpServletRequest req) {
|
InputStream inputStream;
|
List<ServerExcelApplyEntity> serverExcelEntityList;
|
List<ServerExcelResultEntity> excelEntities = new ArrayList<>();
|
Jedis jedis = JedisUtils.getJedis();
|
try {
|
inputStream = file.getInputStream();
|
serverExcelEntityList = EasyExcel.read(inputStream, ServerExcelApplyEntity.class, new ServiceExcelListener()).doReadAllSync();
|
//注册服务 申请token
|
serverExcelEntityList.stream().forEach(ServerExcelApplyEntity -> excelEntities.add(addArcGisServer(req, ServerExcelApplyEntity, jedis)));
|
returnExcel(excelEntities);
|
return JSON.toJSONString(Result.ok().put("msg", "").put("data", ""));
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
if (null != jedis) {
|
JedisUtils.close(jedis);
|
}
|
}
|
return "";
|
}
|
|
/*
|
* 获取注册服务结果的Excel文件
|
*
|
* */
|
@GetMapping("getExcelResult")
|
public ResponseEntity getExcelResult(HttpServletResponse response) throws IOException {
|
String uploadPath = sys.getUploadPath();
|
FileInputStream fileInputStream = null;
|
ServletOutputStream outputStream = response.getOutputStream();
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileResultName, "utf-8"));
|
File file = new File(uploadPath + File.separator + fileResultName);
|
if (file.exists() && file.canRead()) {
|
fileInputStream = new FileInputStream(file);
|
byte[] bytes = new byte[1024];
|
int length;
|
while ((length = fileInputStream.read(bytes)) != -1) {
|
outputStream.write(bytes, 0, length);
|
}
|
}
|
if (null != fileInputStream) {
|
fileInputStream.close();
|
}
|
if (null != outputStream) {
|
outputStream.flush();
|
outputStream.close();
|
}
|
return ResponseEntity.ok().body(null);
|
}
|
|
|
private void returnExcel(List serverExcelEntityList) throws IOException {
|
File file = new File(sys.getUploadPath() + File.separator + fileResultName);
|
FileOutputStream fileOutputStream = new FileOutputStream(file);
|
EasyExcel.write(fileOutputStream, ServerExcelResultEntity.class).sheet("服务注册结果").doWrite(serverExcelEntityList);
|
fileOutputStream.close();
|
}
|
|
private String genToken(HttpServletRequest req, ServerExcelApplyEntity serverExcelEntity, Jedis jedis) {
|
String identyinfo = serverExcelEntity.getApplySys();
|
String resourceid = serverExcelEntity.getZyId().toString();
|
Long expiration = serverExcelEntity.getTime();
|
try {
|
CustomerToken cutToken = new CustomerToken();
|
cutToken.setResourceId(resourceid);
|
cutToken.setUserid("0");
|
cutToken.setAppUrl("");
|
cutToken.setAppId("");
|
cutToken.setClientIp(req.getRemoteHost());
|
cutToken.setRemark(identyinfo == null ? "" : identyinfo);
|
String token = expiration != null ? JwtUtils.generateToken(cutToken.toString(), expiration) : JwtUtils.generateToken(cutToken.toString(), expireSeconds);
|
TokenRecord record = new TokenRecord();
|
record.setApplytime(new Date());
|
record.setExpiration(expiration != null ? expiration.intValue() : expireSeconds.intValue());
|
record.setIdentyinfo(identyinfo);
|
record.setTokeninfo(cutToken.toString());
|
if (!StringUtils.isEmpty(resourceid)) {
|
record.setResourceid(Integer.valueOf(resourceid));
|
}
|
record.setToken(token);
|
record.setTokenapplyer(req.getRemoteHost());
|
logMapper.saveTokenRecord(record);
|
expiration = expiration != null ? expiration : expireSeconds;
|
jedis.setex("zytoken:" + token, expiration.intValue(), expiration.toString());
|
return token;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return null;
|
}
|
|
private ServerExcelResultEntity addArcGisServer(HttpServletRequest req, ServerExcelApplyEntity serverExcelApplyEntity, Jedis jedis) {
|
if (StringUtils.isEmpty(serverExcelApplyEntity.getPubliceIs())) {
|
serverExcelApplyEntity.setPubliceIs("0");
|
}
|
Integer isPublic = Integer.valueOf(serverExcelApplyEntity.getPubliceIs());
|
ServerRgBean serverRgBean = new ServerRgBean();
|
serverRgBean.setEnable(1);
|
serverRgBean.setIspublic(isPublic);
|
serverRgBean.setFromsys("HYJGPT");
|
serverRgBean.setResourceid(serverExcelApplyEntity.getZyId());
|
serverRgBean.setUrlid(null);
|
serverRgBean.setContextpath(null);
|
serverRgBean.setSuffix(null);
|
serverRgBean.setServername(serverExcelApplyEntity.getServerName());
|
serverRgBean.setServerurl(serverExcelApplyEntity.getServerUrl());
|
String servername = judgeServerType(serverExcelApplyEntity.getServerUrl());
|
serverRegisterService.addServer(serverRgBean);
|
ServerExcelResultEntity serverExcelEntity = getServerExcelResultEntity(serverExcelApplyEntity);
|
String proxyurl = new StringBuilder(properties.getProxy().getHost()).append("/arcgis/rest/services/").append(serverExcelApplyEntity.getZyId()).append("/").append(servername).toString();
|
serverExcelEntity.setProxyurl(proxyurl);
|
String useToken = serverExcelApplyEntity.getUseToken();
|
String autoToken = serverExcelApplyEntity.getAutoToken();
|
if (Objects.equals("0", useToken)) {
|
return serverExcelEntity;
|
}
|
//生成token genToken
|
String token = genToken(req, serverExcelApplyEntity, jedis);
|
serverExcelEntity.setToken(token);
|
//加入redis中完成自动续约
|
if (Objects.equals(autoToken, "1")) {
|
jedis.sadd("autoToken", token);
|
}
|
return serverExcelEntity;
|
}
|
|
private ServerExcelResultEntity getServerExcelResultEntity(ServerExcelApplyEntity serverExcelApplyEntity) {
|
ServerExcelResultEntity resultEntity = new ServerExcelResultEntity();
|
BeanUtil.copyProperties(serverExcelApplyEntity, resultEntity);
|
return resultEntity;
|
}
|
|
public static String judgeServerType(String serverUrl) {
|
String type = null;
|
try {
|
URI uri = new URI(serverUrl);
|
String path = uri.getPath();
|
String[] t = path.split("/");
|
if (isdigit(t[t.length - 1])) {
|
type = t[t.length - 2];
|
} else {
|
type = t[t.length - 1];
|
}
|
} catch (URISyntaxException e) {
|
e.printStackTrace();
|
}
|
return type;
|
}
|
|
public static boolean isdigit(String str) {
|
return str.matches("[0-9]+");
|
}
|
|
|
}
|
