扩展转发

magic-boot/src/main/java/org/ssssssss/magicboot/extension/ResponseFunctionExtension.java

@@ -0,0 +1,39 @@
+package org.ssssssss.magicboot.extension;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.context.request.RequestAttributes;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+import org.ssssssss.magicapi.context.RequestContext;
+import org.ssssssss.magicapi.modules.ResponseModule;
+import org.ssssssss.script.annotation.Comment;
+import org.ssssssss.script.functions.ExtensionMethod;
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+
+@Component
+public class ResponseFunctionExtension implements ExtensionMethod {
+
+    @Override
+	public Class<?> support() {
+		return ResponseModule.class;
+	}
+
+	private HttpServletResponse getResponse() {
+		RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
+		if (requestAttributes instanceof ServletRequestAttributes) {
+			return ((ServletRequestAttributes) requestAttributes).getResponse();
+		}
+		return null;
+	}
+
+    @Comment("转发")
+	public ResponseModule.NullValue dispatcher(ResponseModule responseModule, String url) throws ServletException, IOException {
+		HttpServletRequest request = RequestContext.getHttpServletRequest();
+		request.getRequestDispatcher(url).forward(request, getResponse());
+		return responseModule.end();
+	}
+
+}