HttpClient filter中间转发从A tomcat转发至B tomcat

BackFilter.java

主要解决基于HttpGet/HttpPost以及基于HttpPost的附件流转发
import java.io.IOException;
import java.io.InputStream;
import java.util.Collection;
import java.util.Date;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.http.Header;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import com.xinhe99.utils.DateUtil;


public class BackFilter  implements Filter{

	public void destroy() {
		
	}

	public void doFilter(ServletRequest req, ServletResponse res,
			FilterChain chain) throws IOException, ServletException {
		System.out.println("--------------filter--------------------------");
		HttpServletRequest request=(HttpServletRequest) req;
		HttpServletResponse response=(HttpServletResponse) res;
		
		request.setCharacterEncoding(HttpClientBack.encodeUTF);
		Map<String, String> params = new HashMap<String, String>();  
		String method = request.getMethod(); //获取提交类型
		String IP = getRemoteHost(request);  //获取真实Ip
     	String requestURI = request.getRequestURI();//获取请求URI
     	System.out.println(request.getRequestURI());
     	System.out.println(request.getRequestURL());
     	System.out.println(request.getQueryString());
		String scheme = request.getScheme();  //获取协议名称(http或https)
		String realName = "10.168.110.247:8080"; //读取的域名
		String strUr="://";
		String url=scheme + strUr + realName + requestURI;
		Enumeration<String> parameterNames = request.getParameterNames();
		while(parameterNames.hasMoreElements()){
			String key = parameterNames.nextElement();
			String value = request.getParameter(key);
			params.put(key, value);
		}
	
	    byte[]b=new byte[1024];
	    ResponseResult result = null;
//		if("post".equalsIgnoreCase(method)){
//		b = HttpClientBack.post(url, params,IP);
//	}
//	if("get".equalsIgnoreCase(method)){
//		b= HttpClientBack.get(url,IP);
//	}
	    		
		if("post".equalsIgnoreCase(method)){
			// 创建一个通用的多部分解析器
			CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(
					request.getSession().getServletContext());
			// 判断 request 是否有文件上传,即多部分请求
			InputStream in = null;
			if (multipartResolver.isMultipart(request)) {
				System.out.println("wenjian============");
				MultipartHttpServletRequest mreqest = multipartResolver
						.resolveMultipart(request);
				Map<String, MultipartFile> filesMap = mreqest.getFileMap();
				Collection<MultipartFile> fileList = filesMap.values();
				for (MultipartFile mf : fileList) {
					 //获取原始文件名        
		       		  String originalFilename =mf.getOriginalFilename();
		       		 //获取文件流，可以进行处理
		       		  in = mf.getInputStream();
		       		  result = HttpClientBack.postMuti(url, params, IP,originalFilename,in);
		       		 
				}
			}else
			{
			  result = HttpClientBack.post(url, params,IP);
			}
		}
		if("get".equalsIgnoreCase(method)){
			// add query string
			if(null != request.getQueryString()){
				url += "?" + request.getQueryString();
			}
			result= HttpClientBack.get(url,IP);
		}
	    
	    if(null != result){
	    	b = result.getBody();
	    }else{
	    	///do you wangcan beibei xietong chuli 
	    }
		//文件名用日期加时间戳
		String filename=DateUtil.getDate(new Date()).replaceAll("-", "")+String.valueOf(new Date().getTime());
		//设置相应头 
		Header[] headers = result.getHeaders();
		if(null != headers && 0 < headers.length){
			for(Header header : headers){
				response.setHeader(header.getName(), header.getValue());
			}
		}
//		response.setHeader("Content-type", "application/xls;charset=UTF-8");
//		response.setContentType("application/x-msdownload");
//		response.setHeader("Content-Disposition", "attachment;filename="+filename+".xls;charset=UTF-8");
		
		if(b.length !=-1){
			response.getOutputStream().write(b);
		}else{
			response.getOutputStream().write(-1);
		}
       		
	}

	public void init(FilterConfig arg0) throws ServletException {
		
	}

	public String getRemoteHost(javax.servlet.http.HttpServletRequest request){
	    String ip = request.getHeader("x-forwarded-for");
	    if(ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)){
	        ip = request.getHeader("Proxy-Client-IP");
	    }
	    if(ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)){
	        ip = request.getHeader("WL-Proxy-Client-IP");
	    }
	    if(ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)){
	        ip = request.getRemoteAddr();
	    }
	    return ip.equals("0:0:0:0:0:0:0:1")?"127.0.0.1":ip;
	}
	
	public String getUrl() {
		Properties props = new Properties();
	    
	    InputStream in = BackFilter.class.getResourceAsStream("/config/config.properties");

	    try {
			props.load(in);
			in.close();
		} catch (IOException e) {
			e.printStackTrace();
		}
	    // 读取特定属性
	    String key = "target.url";
	    String url = props.getProperty(key);
	    return url;
	}
	
}

　　HttpClientBack.java

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.http.Consts;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.HttpMultipartMode;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.InputStreamBody;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.Args;
import org.apache.http.util.ByteArrayBuffer;
import org.apache.http.util.CharsetUtils;
import org.apache.http.util.EntityUtils;
import org.apache.xmlbeans.impl.common.IOUtil;

public class HttpClientBack {

	public static final String encodeUTF = "UTF-8";

	// 发送post请求
	public static ResponseResult post(String url, Map<String, String> params,
			String ip) {
		CloseableHttpClient httpclient = HttpClientBuilder.create().build();
		byte[] body = new byte[1024];

		// body = invoke(httpclient, post,ip);

		ResponseResult result = null;
		try {
			HttpPost post = postForm(url, params);
			// HttpGet get = new HttpGet(url);
			result = invoke(httpclient, post, ip);
			httpclient.close();
			return result;
		} catch (IOException e) {
			e.printStackTrace();
			return null;
		}

		// try {
		// httpclient.close();
		// } catch (IOException e) {
		// e.printStackTrace();
		// }
		//
		// return body;
	}

	// 发送get请求
	public static ResponseResult get(String url, String ip) {
		CloseableHttpClient httpclient = HttpClientBuilder.create().build();
		// byte[] body = new byte[1024];
		ResponseResult result = null;
		try {
			HttpGet get = new HttpGet(url);
			result = invoke(httpclient, get, ip);
			httpclient.close();
			return result;
		} catch (IOException e) {
			e.printStackTrace();
			return null;
		}

		// return body;
	}

	// 执行
	private static ResponseResult invoke(CloseableHttpClient httpclient,
			HttpUriRequest httpost, String ip) {

		httpost.setHeader("x-forwarded-for_new", ip);
		HttpResponse response = sendRequest(httpclient, httpost);
		return new ResponseResult(paseResponse(response),
				response.getAllHeaders());
	}

	// 解析
	private static byte[] paseResponse(HttpResponse response) {
		HttpEntity entity = response.getEntity();

		byte[] body = null;
		try {

			body = EntityUtils.toByteArray(entity);
		} catch (Exception e) {
			e.printStackTrace();
		}

		return body;
	}

	// 正式发送
	private static HttpResponse sendRequest(CloseableHttpClient httpclient,
			HttpUriRequest httpost) {
		HttpResponse response = null;

		try {
			response = httpclient.execute(httpost);
		} catch (ClientProtocolException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return response;
	}

	// post方式获取并包装参数
	private static HttpPost postForm(String url, Map<String, String> params) {

		HttpPost httpost = new HttpPost(url);
		List<NameValuePair> nvps = new ArrayList<NameValuePair>();

		Set<String> keySet = params.keySet();
		for (String key : keySet) {
			nvps.add(new BasicNameValuePair(key, params.get(key)));
		}

		try {
			httpost.setEntity(new UrlEncodedFormEntity(nvps,
					HttpClientBack.encodeUTF));
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		}
		return httpost;
	}
    /**
     * 处理附件上传
     * @param url  
     * @param params
     * @param ip
     * @param fileName
     * @param targetIn
     * @return
     */
	public static ResponseResult postMuti(String url,
			Map<String, String> params, String ip, String fileName,
			InputStream targetIn) {
		RequestConfig defaultRequestConfig = RequestConfig.custom()
				  .setSocketTimeout(5000)
				  .setConnectTimeout(5000)
				  .setConnectionRequestTimeout(5000).build();
		CloseableHttpClient httpClient = HttpClientBuilder.create().setDefaultRequestConfig(defaultRequestConfig).build();
		
		ResponseResult result = new ResponseResult();
		InputStream localIn = targetIn;
		HttpEntity reqEntity = null;
		InputStream responseIn = null;
		CloseableHttpResponse response = null;
		byte[] datas =null;
		try {
			HttpPost httpPost = postForm(url, params);
			httpPost.setHeader("x-forwarded-for_new", ip);
			InputStreamBody bin = new InputStreamBody(targetIn, fileName);
			StringBody uploadFileName = new StringBody(fileName,
					ContentType.create("text/plain", Consts.UTF_8));
			// 以浏览器兼容模式运行，防止文件名乱码。
			reqEntity = MultipartEntityBuilder.create()
					.setMode(HttpMultipartMode.BROWSER_COMPATIBLE)
					.addPart("file", bin).addPart("filename", uploadFileName)
					.setCharset(CharsetUtils.get("UTF-8")).build();
			httpPost.setEntity(reqEntity);
			System.out.println("发起请求的页面地址 " + httpPost.getRequestLine());
			// 发起请求 并返回请求的响应
			response = httpClient.execute(httpPost);
			responseIn = response.getEntity().getContent();
			datas = toByteArray(responseIn);
			// 打印响应状态
			System.out.println(response.getStatusLine());
			// 获取响应对象
			HttpEntity resEntity = response.getEntity();
			if (resEntity != null) {
				// 打印响应长度
				System.out.println("Response content length: "
						+ resEntity.getContentLength());
				// 打印响应内容
				System.out.println(new String(datas,Charset.forName("UTF-8")));
			}
			// 销毁
			//EntityUtils.consume(resEntity);
			result.setBody(datas);
			result.setHeaders(response.getAllHeaders());
		} catch (Exception ex) {
			ex.printStackTrace();
			return null;
		} finally {
			try {
				if (response != null) {
					response.close();
				}
				if (localIn != null) {
					localIn.close();
				}
				if (responseIn != null) {
					responseIn.close();
				}
				if (httpClient != null) {
					httpClient.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return result;
	}
	private static byte[] toByteArray(InputStream input) throws IOException {
	    ByteArrayOutputStream output = new ByteArrayOutputStream();
	    try{
	    byte[] buffer = new byte[4096];
	    int n = 0;
	    while (-1 != (n = input.read(buffer))) {
	        output.write(buffer, 0, n);
	    }
	    }finally{
	    	if(output!=null){
	    		output.close();
	    	}
	    }
	    return output.toByteArray();
	}
}