ProxyServlet代理

来源:互联网 发布:tinycore linux有啥用 编辑:程序博客网 时间:2024/06/05 10:24
import java.io.IOException;import java.io.InputStream;import java.io.OutputStream;import java.io.UnsupportedEncodingException;import java.net.HttpURLConnection;import java.net.MalformedURLException;import java.net.URL;import java.net.URLEncoder;import java.util.Iterator;import java.util.Map;import java.util.Map.Entry;import java.util.regex.Pattern;import javax.servlet.ServletException;import javax.servlet.http.HttpServlet;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import org.apache.commons.lang.StringUtils;public class ProxyServlet extends HttpServlet {/** *  */private static final long serialVersionUID = 1L;private String url;/** * 对servlet进行请求处理,并将结果在指定输出流中输出 *  * @param os * @param servletName * @param parm * @throws IOException * @throws MalformedURLException */private void process(HttpServletRequest req, HttpServletResponse resp,String[] target) throws MalformedURLException, IOException {// 取得连接HttpURLConnection huc = (HttpURLConnection) new URL(url + target[0]).openConnection();// 设置连接属性huc.setDoOutput(true);huc.setRequestMethod("POST");huc.setUseCaches(false);huc.setInstanceFollowRedirects(true);huc.setRequestProperty("Content-Type","application/x-www-form-urlencoded");huc.connect();// 往目标servlet中提供参数OutputStream targetOS = null;try {targetOS = huc.getOutputStream();targetOS.write(target[1].getBytes());        targetOS.flush();        targetOS.close();} catch (IOException e) {e.printStackTrace();}// 取得页面输出,并设置页面编码及缓存设置int r;String cb = req.getParameter("callback");resp.setContentType(huc.getContentType());resp.setHeader("Cache-Control", huc.getHeaderField("Cache-Control"));resp.setHeader("Pragma", huc.getHeaderField("Pragma"));resp.setHeader("Expires", huc.getHeaderField("Expires"));resp.setStatus(huc.getResponseCode());// 将目标servlet的输入流直接往页面输出try {OutputStream os = resp.getOutputStream();        InputStream targetIS = huc.getInputStream();if(cb!=null){os.write((cb + "(").getBytes());}while ((r = targetIS.read()) != -1) {os.write(r);}if(cb!=null){os.write((")").getBytes());}targetIS.close();os.flush();os.close();huc.disconnect();} catch (IOException e) {// TODO: handle exceptionresp.sendError(huc.getResponseCode());huc.disconnect();}}/** * 将参数中的目标分离成由目标servlet名称和参数组成的数组 *  * @param queryString * @return * @throws UnsupportedEncodingException */private String[] parse(Map map) throws UnsupportedEncodingException {String[] arr = { "", "" };Pattern pattern = Pattern.compile("^_[0-9]+$");Iterator iter = map.entrySet().iterator();while (iter.hasNext()) {Map.Entry me = (Entry) iter.next();String[] varr = (String[]) me.getValue();if (!(StringUtils.equals("callback", me.getKey().toString()) )) {if( !((pattern.matcher(me.getKey().toString())).find())){if ("servletName".equals(me.getKey())) {// 取出servlet名称arr[0] = varr[0];} else {// 重新组装参数字符串for (int i = 0; i < varr.length; i++) {// 参数需要进行转码,实现字符集的统一arr[1] += "&" + me.getKey() + "="+ URLEncoder.encode(varr[i], "utf-8");}}}}}arr[1] = arr[1].replaceAll("^&", "");return arr;}@Overridepublic void init() throws ServletException {// 设置目标服务器地址url = this.getInitParameter("url");if (!url.endsWith("/"))url = url + "/";}@Overrideprotected void doGet(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {this.doPost(req, resp);}@Overrideprotected void doPost(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {String[] target = parse(req.getParameterMap());process(req, resp, target);}}

原创粉丝点击