com.wudaosoft.net.httpclient.FileResponseHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wudaosoft-httpclient Show documentation
Wudaosoft Httpclient Tools
There is a newer version: 3.0.0
/**
 *    Copyright 2009-2018 Wudao Software Studio(wudaosoft.com)
 * 
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 * 
 *        http://www.apache.org/licenses/LICENSE-2.0
 * 
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package com.wudaosoft.net.httpclient;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.ResponseHandler;
import org.apache.http.util.Args;

/**
 * @author Changsoul.Wu
 */
public class FileResponseHandler implements ResponseHandler {

	private File file;
	private File dir;

	public FileResponseHandler(final File file) {
		this.file = Args.notNull(file, "file");
	}
	
	public FileResponseHandler(final String dir) {
		this.dir = new File(Args.notBlank(dir, "dir"));
		Args.check(this.dir.isDirectory(), "dir must be a directory");
	}

	@Override
	public File handleResponse(HttpResponse response) throws ClientProtocolException, IOException {
		int status = response.getStatusLine().getStatusCode();

		if (status != 200) {
			throw new ClientProtocolException("Unexpected response status: " + status);
		}

		HttpEntity entity = response.getEntity();

		if (entity == null || !entity.isStreaming()) {
			throw new ClientProtocolException("Response contains no content");
		}
		
		File tempFile = null;
		
		if (this.file != null) {
			tempFile = this.file;
		} else if (this.dir != null) {
			Header contentDisposition = response.getLastHeader("Content-disposition");
			String filename = contentDisposition.getValue().split(";")[1].split("=")[1].replace("\"", "");
			tempFile = new File(this.dir, filename);
		}
		
		Args.notNull(tempFile, "file");
		Args.check(tempFile.canWrite(), "file must be writeable");

		InputStream inputStream = entity.getContent();
		FileOutputStream outputStream = new FileOutputStream(tempFile);

		try {
			
			byte[] buff = new byte[4096];
			int size = -1;
			while ((size = inputStream.read(buff)) != -1) {

				outputStream.write(buff, 0, size);
			}

			outputStream.flush();
			
			return tempFile;
		} finally {
			try {
				outputStream.close();
			}catch (IOException e) {
			}
		}
		
	}

}