From e16f5fdfbc1c49f4c519f05b190e96e497253b51 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 13 十一月 2024 17:32:14 +0800
Subject: [PATCH] 1
---
src/main/java/com/moon/server/service/sys/ProxyService.java | 100 +++++++++++++++++++++++++++++++++++++-------------
1 files changed, 74 insertions(+), 26 deletions(-)
diff --git a/src/main/java/com/moon/server/service/sys/ProxyService.java b/src/main/java/com/moon/server/service/sys/ProxyService.java
index 1b03132..a633514 100644
--- a/src/main/java/com/moon/server/service/sys/ProxyService.java
+++ b/src/main/java/com/moon/server/service/sys/ProxyService.java
@@ -47,23 +47,21 @@
/**
* URL浠ｇ悊
*/
- public void proxyUrl(String token, int resId, HttpServletRequest req, HttpServletResponse res) throws Exception {
+ public void proxyUrl(String token, int resId, boolean isRest, HttpServletRequest req, HttpServletResponse res) throws Exception {
// 3.鑾峰彇鐢ㄦ埛
UserEntity ue = getUser(req, res, token);
if (null == ue) {
return;
}
-
- // 9.鑾峰彇璧勬簮瀹炰綋
+ // 9.鑾峰彇璧勬簮瀹炰綋锛宻tatus锛�0-绂佺敤锛�1-鍚敤鍘熷鍦板潃锛�2-鍚敤浠ｇ悊鍦板潃
ResEntity entity = getResEntity(ue, resId);
- if (null == entity || entity.getType() != StaticData.THREE || StringHelper.isNull(entity.getProxy()) || StringHelper.isNull(entity.getUrl())) {
+ if (null == entity || StaticData.I1 > entity.getStatus() || StringHelper.isNull(entity.getProxy()) || StringHelper.isNull(entity.getUrl())) {
WebHelper.writeStr2Page(res, ILLEGAL_RESOURCE);
return;
}
-
insertLog(req, ue, resId);
- String url = getSourceUrl(req, entity, token);
+ String url = getUrl(req, ue, entity, token, isRest);
res.setHeader("token", token);
forward(req, res, entity, url);
}
@@ -101,25 +99,20 @@
if (StringHelper.isEmpty(ip)) {
return WebHelper.writeStr2Page(res, AuthInterceptor.IP_NULL);
}
-
// 5.妫�鏌ラ粦鍚嶅崟
if (!checkBlackList(ip, req)) {
return WebHelper.writeStr2Page(res, AuthInterceptor.BLACK_LIST);
}
-
// 6.admin璺宠繃鏉冮檺妫�娴�
if (StaticData.ADMIN.equals(ue.getUid())) {
return true;
}
-
- // 7.妫�鏌ョ櫧鍚嶅崟
+ // 7.妫�鏌ョ櫧鍚嶅崟鍜孖P涓�鑷存��
if (!checkWhiteList(ip, req)) {
- // 妫�鏌P涓�鑷存��
if (!checkIpSource(ip, token)) {
return WebHelper.writeStr2Page(res, AuthInterceptor.ILLEGAL_TOKEN);
}
}
-
// 8.妫�鏌ョ敤鎴稩D鏄惁绂佺敤
if (sysService.tokenService.isUidDisable(ue)) {
return WebHelper.writeStr2Page(res, AuthInterceptor.USER_LOCK);
@@ -136,11 +129,8 @@
if (blackList == null || blackList.isEmpty()) {
return true;
}
- if (blackList.contains(ip)) {
- return false;
- }
- return true;
+ return !blackList.contains(ip);
}
/**
@@ -156,24 +146,22 @@
}
/**
- * 妫�鏌P涓�鑷存��
+ * 妫�鏌P涓�鑷存�э細鍥哄畾浠ょ墝涓嶆鏌�
*/
private boolean checkIpSource(String ip, String token) {
TokenEntity te = sysService.tokenService.getEntityByToken(token);
- return te.getIp().equals(ip);
+ return StaticData.I1 == te.getType() || te.getIp().equals(ip);
}
/**
* 妫�鏌ヨ祫婧愭潈闄�
*/
private ResEntity getResEntity(UserEntity ue, int resId) {
- String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
- List<ResEntity> rs = permsService.selectRes(uid);
+ List<ResEntity> rs = StaticData.ADMIN.equals(ue.getUid()) ? permsService.selectAllRes() : permsService.selectRes(ue.getUid());
if (null == rs || rs.isEmpty()) {
return null;
}
-
// List<ResEntity> list = rs.stream().filter(resEntity -> resEntity.getId() == resId).collect(Collectors.toList())
for (ResEntity entity : rs) {
if (resId == entity.getId()) {
@@ -234,21 +222,81 @@
}
/**
- * 鑾峰彇鍘熷Url
+ * 鑾峰彇Url
*/
- private String getSourceUrl(HttpServletRequest req, ResEntity entity, String token) {
+ private String getUrl(HttpServletRequest req, UserEntity ue, ResEntity entity, String token, boolean isRest) {
String proxyUrl = entity.getProxy().replace("{token}", token);
int end = req.getRequestURL().indexOf(proxyUrl) + proxyUrl.length();
- String url = entity.getUrl() + req.getRequestURL().substring(end);
- if (null != req.getQueryString()) {
- url = url + (entity.getUrl().contains("?") ? "&" : "?") + req.getQueryString();
+ String url = entity.getUrl().trim() + req.getRequestURL().substring(end);
+ // category锛�0-鍏朵粬锛�1-GisServer锛�2-GeoServer锛�3-鏁扮畝
+ if (StaticData.I2 == entity.getCategory()) {
+ url = getGeoServerUrl(req, ue, entity, url);
+ } else if (StaticData.I3 == entity.getCategory()) {
+ if (null != req.getQueryString()) {
+ url = url + (url.contains("?") ? "&" : "?") + req.getQueryString();
+ }
+ if (isRest) {
+ url = url.replace("/v6/wmts/", "/v6/rest/");
+ }
+ if (!StringHelper.isNull(entity.getArgs())) {
+ url = url + (url.contains("?") ? "&" : "?") + entity.getArgs();
+ }
}
return url;
}
/**
+ * 鑾峰彇GeoServer鍦板潃
+ */
+ private String getGeoServerUrl(HttpServletRequest req, UserEntity ue, ResEntity entity, String url) {
+ if (null == req.getQueryString()) {
+ return url;
+ }
+
+ String str = req.getQueryString();
+ boolean isLower = str.contains(StaticData.SERVICE);
+ String layersKey = isLower ? StaticData.LAYERS : StaticData.LAYERS.toUpperCase();
+ String layers = req.getParameter(layersKey);
+ String request = req.getParameter(isLower ? StaticData.REQUEST : StaticData.REQUEST.toUpperCase());
+ if (!StaticData.GET_CAPABILITIES.equals(request)) {
+ int start = str.indexOf(layersKey);
+ int end = str.indexOf("&", start);
+ layers = filterGeoLayers(ue, layers);
+
+ str = str.replace(str.substring(start, end > -1 ? end : str.length()), layersKey + "=" + layers);
+ } else {
+ List<String> tabs = StaticData.ADMIN.equals(ue.getUid()) ? permsService.selectAllTabs() : permsService.selectTabs(ue.getUid());
+ entity.setTab(null == tabs ? "" : StringHelper.join(tabs, ","));
+ entity.setBak(StaticData.GET_CAPABILITIES);
+ }
+
+ return url + (url.contains("?") ? "&" : "?") + str;
+ }
+
+ /**
+ * 杩囨护GeoServer鍥惧眰
+ */
+ private String filterGeoLayers(UserEntity ue, String layers) {
+ List<String> tabs = StaticData.ADMIN.equals(ue.getUid()) ? permsService.selectAllTabs() : permsService.selectTabs(ue.getUid());
+ if (null == tabs || tabs.isEmpty() || StringHelper.isEmpty(layers)) {
+ return "";
+ }
+
+ StringBuilder sb = new StringBuilder();
+ String[] strs = layers.split(StaticData.COMMA);
+ for (String str : strs) {
+ if (tabs.contains(str)) {
+ sb.append(str).append(",");
+ }
+ }
+ sb.deleteCharAt(sb.length() - 1);
+
+ return sb.toString();
+ }
+
+ /**
* 杞彂璇锋眰
*/
private void forward(HttpServletRequest request, HttpServletResponse response, ResEntity entity, String url) throws Exception {
--
Gitblit v1.9.3