package com.xunlei;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;

/**
 * @author Hank
 * @date 2015-4-22 下午4:19:31
 */
public class HttpConnection {
	public HttpURLConnection conn;

	/**
	 * 创建连接
	 */
	public HttpURLConnection createConnection(String strUrl, String requestMethod, int timeout) {
		try {
			URL url = new URL(strUrl);
			conn = (HttpURLConnection) url.openConnection();
			conn.setRequestMethod(requestMethod);
			conn.setConnectTimeout(timeout);
			conn.setReadTimeout(timeout);
		} catch (MalformedURLException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return conn;
	}

	/**
	 * 从连接中获得byte数组
	 */
	public byte[] getByte() {
		InputStream inStream = null;
		byte[] b = null;
		try {
			inStream = conn.getInputStream();// 通过输入流获取数据
			b = readInputStream(inStream);// 得到二进制数据
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (inStream != null) {
				try {
					inStream.close();
				} catch (IOException e) {
				}
			}
		}
		return b;
	}

	private byte[] readInputStream(InputStream inStream) {
		byte[] b = null;
		ByteArrayOutputStream outStream = new ByteArrayOutputStream();
		try {
			byte[] buffer = new byte[1024];
			int len = 0;
			while ((len = inStream.read(buffer)) != -1) {
				outStream.write(buffer, 0, len);
			}
			b = outStream.toByteArray();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (outStream != null)
				try {
					outStream.close();
				} catch (IOException e) {
				}
		}
		return b;
	}

	/**
	 * 读取文件
	 */
	public byte[] inPut(String filePath) {
		File file = new File(filePath);
		if (file.canRead()) {
			FileInputStream fis = null;
			try {
				fis = new FileInputStream(file);
				return this.readInputStream(fis);
			} catch (Exception e) {
				e.printStackTrace();
			} finally {
				if (fis != null)
					try {
						fis.close();
					} catch (IOException e) {
					}
			}
		}
		return null;
	}

	/**
	 * 输出文件
	 */
	public void outPut(byte[] bs, String filePath) {
		FileOutputStream fos = null;
		try {
			fos = new FileOutputStream(new File(filePath));
			fos.write(bs);
			fos.flush();
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (fos != null)
				try {
					fos.close();
				} catch (IOException e) {
				}
		}
	}
}