io.leopard.web.mvc.DownloadView Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of web4j-view Show documentation
The newest version!
package io.leopard.web.mvc;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.net.URLEncoder;
import java.util.Map;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.view.AbstractUrlBasedView;

public class DownloadView extends ModelAndView {

	public DownloadView(final String filename, final String name) {
		AbstractUrlBasedView view = new AbstractUrlBasedView() {
			@Override
			protected void renderMergedOutputModel(Map model, HttpServletRequest request, HttpServletResponse response) throws Exception {
				String name2 = URLEncoder.encode(name, "UTF-8");
				response.setContentType("application/force-download");
				response.addHeader("Content-Disposition", "attachment;filename=" + name2);

				File file = new File(filename);

				InputStream inputStream = new FileInputStream(file);
				BufferedInputStream bis = new BufferedInputStream(inputStream);
				byte[] bytes = new byte[1024];
				ServletOutputStream out = response.getOutputStream();
				int readLength = 0;
				while ((readLength = bis.read(bytes)) != -1) {
					out.write(bytes, 0, readLength);
				}
				// TODO ahai 释放资源
				inputStream.close();
				bis.close();
				out.flush();
				out.close();
			}
		};
		super.setView(view);
	}

	public DownloadView(final InputStream input, final String name) {
		AbstractUrlBasedView view = new AbstractUrlBasedView() {
			@Override
			protected void renderMergedOutputModel(Map model, HttpServletRequest request, HttpServletResponse response) throws Exception {
				String name2 = URLEncoder.encode(name, "UTF-8");
				response.setContentType("application/force-download");
				response.addHeader("Content-Disposition", "attachment;filename=" + name2);
				byte[] bytes = new byte[1024];
				ServletOutputStream out = response.getOutputStream();
				int readLength = 0;
				while ((readLength = input.read(bytes)) != -1) {
					out.write(bytes, 0, readLength);
				}
				// TODO ahai 释放资源
				input.close();
				out.flush();
				out.close();
			}
		};
		super.setView(view);
	}
}