Server：新增代理接口，支持前端任意跨域

Author: TommyLemon

APIJSON-Java-Server/APIJSONBoot/src/main/java/apijson/demo/server/Controller.java

@@ -24,17 +24,33 @@
 
 import java.net.URLDecoder;
 import java.rmi.ServerException;
+import java.util.ArrayList;
+import java.util.Enumeration;
+import java.util.List;
 import java.util.Random;
 import java.util.concurrent.TimeoutException;
 
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpSession;
 
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Service;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.client.RestTemplate;
 
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 
 import apijson.demo.server.model.BaseModel;
@@ -63,6 +79,7 @@
  * <br > 3.调试方便 - 建议使用 APIJSON在线测试工具 或 Postman
  * @author Lemon
  */
+@Service
 @RestController
 @RequestMapping("")
 public class Controller {
@@ -200,15 +217,15 @@ public String openHead(@PathVariable String request, HttpSession session) {
 	public static final String FUNCTION_;
 	public static final String REQUEST_;
 	public static final String ACCESS_;
-	
+
 	public static final String USER_;
 	public static final String PRIVACY_;
 	public static final String VERIFY_; //加下划线后缀是为了避免 Verify 和 verify 都叫VERIFY，分不清
 	static {
 		FUNCTION_ = Function.class.getSimpleName();
 		REQUEST_ = Request.class.getSimpleName();
 		ACCESS_ = Access.class.getSimpleName();
-		
+
 		USER_ = User.class.getSimpleName();
 		PRIVACY_ = Privacy.class.getSimpleName();
 		VERIFY_ = Verify.class.getSimpleName();
@@ -233,8 +250,8 @@ public String openHead(@PathVariable String request, HttpSession session) {
 
 	public static final String TYPE = "type";
 
-	
-	
+
+
 	/**重新加载配置
 	 * @param request
 	 * @return
@@ -261,17 +278,17 @@ public JSONObject reload(@RequestBody String request) {
 		} catch (Exception e) {
 			return DemoParser.extendErrorResult(requestObject, e);
 		}
-		
+
 		JSONResponse response = new JSONResponse(headVerify(Verify.TYPE_RELOAD, phone, verify));
 		response = response.getJSONResponse(VERIFY_);
 		if (JSONResponse.isExist(response) == false) {
 			return DemoParser.extendErrorResult(requestObject, new ConditionErrorException("手机号或验证码错误"));
 		}
 
 		JSONObject result = DemoParser.newSuccessResult();
-		
+
 		boolean reloadAll = StringUtil.isEmpty(type, true) || "ALL".equals(type);
-		
+
 		if (reloadAll || "FUNCTION".equals(type)) {
 			try {
 				result.put(FUNCTION_, DemoFunction.init());
@@ -280,7 +297,7 @@ public JSONObject reload(@RequestBody String request) {
 				result.put(FUNCTION_, DemoParser.newErrorResult(e));
 			}
 		}
-		
+
 		if (reloadAll || "REQUEST".equals(type)) {
 			try {
 				result.put(REQUEST_, StructureUtil.init());
@@ -289,7 +306,7 @@ public JSONObject reload(@RequestBody String request) {
 				result.put(REQUEST_, DemoParser.newErrorResult(e));
 			}
 		}
-		
+
 		if (reloadAll || "ACCESS".equals(type)) {
 			try {
 				result.put(ACCESS_, DemoVerifier.init());
@@ -298,7 +315,7 @@ public JSONObject reload(@RequestBody String request) {
 				result.put(ACCESS_, DemoParser.newErrorResult(e));
 			}
 		}
-		
+
 		return result;
 	}
 
@@ -952,4 +969,59 @@ public JSONObject putBalance(@RequestBody String request, HttpSession session) {
 	}
 
 
+	@Autowired
+	HttpServletRequest request;
+	@Autowired
+	HttpServletResponse response;
+
+	@RequestMapping(value = "/delegate")
+//	@ResponseBody
+	public String delegate(@RequestParam("$_delegate_url") String url, @RequestBody String body, HttpMethod method, HttpSession session){
+		Enumeration<String> names = request.getHeaderNames();
+		HttpHeaders headers = null;
+		String name;
+		if (names != null) {
+			headers = new HttpHeaders();
+			while (names.hasMoreElements()) {
+				name = names.nextElement();
+				headers.add(name, request.getHeader(name));
+			}
+			
+			List<String> cookie = session == null ? null : (List<String>) session.getAttribute("Cookie");
+			if (cookie != null && cookie.isEmpty() == false) {
+				List<String> c = headers.get("Cookie");
+				if (c == null) {
+					c = new ArrayList<>();
+				}
+				c.addAll(cookie);
+				headers.put("Cookie", c);
+			}
+		}
+		try {
+			request.getParameterMap().remove("$_delegate_url");
+		} catch (Exception e) {
+			// TODO: handle exception
+		}
+
+		RestTemplate client = new RestTemplate();
+		//  请勿轻易改变此提交方式，大部分的情况下，提交方式都是表单提交
+		HttpEntity<String> requestEntity = new HttpEntity<>(method == HttpMethod.GET ? JSON.toJSONString(request.getParameterMap()) : body, headers);
+		//  执行HTTP请求
+		ResponseEntity<String> entity = client.exchange(url, method, requestEntity, String.class);
+		
+		HttpHeaders hs = entity.getHeaders();
+		if (session != null && hs != null) {
+			List<String> cookie = hs.get("Set-Cookie");
+			if (cookie != null && cookie.isEmpty() == false) {
+				session.setAttribute("Cookie", cookie);
+			}
+		}
+		return entity.getBody();
+	}
+
+	@GetMapping("v2/api-docs")
+	public JSONObject swaggerAPIDocs() {
+		return new DemoParser().parseResponse(new JSONRequest("Swagger", new JSONRequest())).getJSONObject("Swagger");
+	}
+
 }