package com.cmc.test;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * 使用多執行緒下載防盜鏈圖片
 * 
 * @author chenmc
 * @date 2017年8月30日 下午2:39:23
 */
public class DownloadPicture {

	public static void main(String[] args) throws InterruptedException {
		for (int i = 1; i <= 10; i++) {
			int n = i;
			Thread t = new Thread(){
				@Override
				public synchronized void run() {
					try {
						httpRequest(n);
					} catch (Exception e) {
						e.printStackTrace();
					}
					super.run();
				}
			};
			t.start();
		}
	}

	/**
	 * 發起http請求
	 * 
	 * @author chenmc
	 * @date 2017年8月30日 下午2:36:09
	 * @param n
	 * @throws Exception
	 */
	public static void httpRequest(int n) throws Exception {
		for (int i = (n-1)*5 > 1 ? (n-1)*5 : 1; i < n*5; i++) { 
			long startTime = System.currentTimeMillis();
			String path = "C:/Users/admin/Pictures/桌布/" + String.format("%03d", i) + ".jpg";
			//這裡url寫你想要下載圖片的url
			String uri = String.format("http://bizhi.xxx.com/rluo/%s.jpg", String.format("%03d", i));
			System.out.println(uri);
			URL url = new URL(uri);  
			HttpURLConnection conn = (HttpURLConnection)url.openConnection(); 
			conn.setRequestProperty("referer", "http://bizhi.xxx.com"); //這是破解防盜鏈新增的引數
			conn.setRequestMethod("GET");  
			conn.setConnectTimeout(5 * 1000);  
			InputStream inStream = conn.getInputStream();//通過輸入流獲取圖片資料
			readInputStream(inStream, path);//得到圖片的二進位制資料，並儲存
			long endTime = System.currentTimeMillis();
			System.out.println(path + "--耗時：" + (endTime-startTime)/1000 + "s");
		}
	}
	
	/**
	 * 儲存圖片
	 * 
	 * @author chenmc
	 * @date 2017年8月30日 下午2:35:54
	 * @param inStream
	 * @param path
	 * @throws Exception
	 */
	 public static void readInputStream(InputStream inStream, String path) throws Exception{  
		FileOutputStream fos = new FileOutputStream(new File(path));
		byte[] buffer = new byte[102400];  
		int len = 0;  
		while( (len=inStream.read(buffer)) != -1 ){  
			fos.write(buffer, 0, len);  
		}  
		inStream.close();  
		fos.flush();
		fos.close();
	 }
}