java分段流_JAVA，C# 大文件数据流分段输出，多段输出，支持视频图片等（Range，etag操作）...

packagelan.dk.podcastserver.utils.multipart;importlan.dk.podcastserver.utils.MimeTypeUtils;importorg.apache.commons.lang3.StringUtils;importorg.slf4j.Logger;importorg.slf4j.LoggerFactory;importjavax.servlet.ServletOutputStream;importjavax.servlet.http.HttpServletRequest;importjavax.servlet.http.HttpServletResponse;import java.io.*;importjava.nio.file.Files;importjava.nio.file.Path;importjava.nio.file.Paths;importjava.nio.file.attribute.FileTime;importjava.time.LocalDateTime;importjava.time.ZoneId;importjava.time.ZoneOffset;importjava.util.ArrayList;importjava.util.Arrays;importjava.util.List;/*** Created by kevin on 10/02/15.*/

public classMultipartFileSender {protected final Logger logger = LoggerFactory.getLogger(this.getClass());private static final int DEFAULT_BUFFER_SIZE = 20480; //..bytes = 20KB.

private static final long DEFAULT_EXPIRE_TIME = 604800000L; //..ms = 1 week.

private static final String MULTIPART_BOUNDARY = "MULTIPART_BYTERANGES";

Path filepath;

HttpServletRequest request;

HttpServletResponse response;publicMultipartFileSender() {

}public staticMultipartFileSender fromPath(Path path) {return newMultipartFileSender().setFilepath(path);

}public staticMultipartFileSender fromFile(File file) {return newMultipartFileSender().setFilepath(file.toPath());

}public staticMultipartFileSender fromURIString(String uri) {return newMultipartFileSender().setFilepath(Paths.get(uri));

}//** internal setter **// privateMultipartFileSender setFilepath(Path filepath) {this.filepath =filepath;return this;

}publicMultipartFileSender with(HttpServletRequest httpRequest) {

request=httpRequest;return this;

}publicMultipartFileSender with(HttpServletResponse httpResponse) {

response=httpResponse;return this;

}public void serveResource() throwsException {if (response == null || request == null) {return;

}if (!Files.exists(filepath)) {

logger.error("File doesn't exist at URI : {}", filepath.toAbsolutePath().toString());

response.sendError(HttpServletResponse.SC_NOT_FOUND);return;

}

Long length=Files.size(filepath);

String fileName=filepath.getFileName().toString();

FileTime lastModifiedObj=Files.getLastModifiedTime(filepath);if (StringUtils.isEmpty(fileName) || lastModifiedObj == null) {

response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);return;

}long lastModified =LocalDateTime.ofInstant(lastModifiedObj.toInstant(), ZoneId.of(ZoneOffset.systemDefault().getId())).toEpochSecond(ZoneOffset.UTC);

String contentType=MimeTypeUtils.probeContentType(filepath);//Validate request headers for caching ---------------------------------------------------//If-None-Match header should contain "*" or ETag. If so, then return 304.

String ifNoneMatch = request.getHeader("If-None-Match");if (ifNoneMatch != null &&HttpUtils.matches(ifNoneMatch, fileName)) {

response.setHeader("ETag", fileName); //Required in 304.

response.sendError(HttpServletResponse.SC_NOT_MODIFIED);return;

}//If-Modified-Since header should be greater than LastModified. If so, then return 304.//This header is ignored if any If-None-Match header is specified.

long ifModifiedSince = request.getDateHeader("If-Modified-Since");if (ifNoneMatch == null && ifModifiedSince != -1 && ifModifiedSince + 1000 >lastModified) {

response.setHeader("ETag", fileName); //Required in 304.

response.sendError(HttpServletResponse.SC_NOT_MODIFIED);return;

}//Validate request headers for resume ----------------------------------------------------//If-Match header should contain "*" or ETag. If not, then return 412.

String ifMatch = request.getHeader("If-Match");if (ifMatch != null && !HttpUtils.matches(ifMatch, fileName)) {

response.sendError(HttpServletResponse.SC_PRECONDITION_FAILED);return;

}//If-Unmodified-Since header should be greater than LastModified. If not, then return 412.

long ifUnmodifiedSince = request.getDateHeader("If-Unmodified-Since");if (ifUnmodifiedSince != -1 && ifUnmodifiedSince + 1000 <=lastModified) {

response.sendError(HttpServletResponse.SC_PRECONDITION_FAILED);return;

}//Validate and process range -------------------------------------------------------------//Prepare some variables. The full Range represents the complete file.

Range full = new Range(0, length - 1, length);

List ranges = new ArrayList<>();//Validate and process Range and If-Range headers.

String range = request.getHeader("Range");if (range != null) {//Range header should match format "bytes=n-n,n-n,n-n...". If not, then return 416.

if (!range.matches("^bytes=\\d*-\\d*(,\\d*-\\d*)*$")) {

response.setHeader("Content-Range", "bytes */" + length); //Required in 416.

response.sendError(HttpServletResponse.SC_REQUESTED_RANGE_NOT_SATISFIABLE);return;

}

String ifRange= request.getHeader("If-Range");if (ifRange != null && !ifRange.equals(fileName)) {try{long ifRangeTime = request.getDateHeader("If-Range"); //Throws IAE if invalid.

if (ifRangeTime != -1) {

ranges.add(full);

}

}catch(IllegalArgumentException ignore) {

ranges.add(full);

}

}//If any valid If-Range header, then process each part of byte range.

if(ranges.isEmpty()) {for (String part : range.substring(6).split(",")) {//Assuming a file with length of 100, the following examples returns bytes at://50-80 (50 to 80), 40- (40 to length=100), -20 (length-20=80 to length=100).

long start = Range.sublong(part, 0, part.indexOf("-"));long end = Range.sublong(part, part.indexOf("-") + 1, part.length());if (start == -1) {

start= length -end;

end= length - 1;

}else if (end == -1 || end > length - 1) {

end= length - 1;

}//Check if Range is syntactically valid. If not, then return 416.

if (start >end) {

response.setHeader("Content-Range", "bytes */" + length); //Required in 416.

response.sendError(HttpServletResponse.SC_REQUESTED_RANGE_NOT_SATISFIABLE);return;

}//Add range.

ranges.add(newRange(start, end, length));

}

}

}//Prepare and initialize response --------------------------------------------------------//Get content type by file name and set content disposition.

String disposition = "inline";//If content type is unknown, then set the default value.//For all content types, see:http://www.w3schools.com/media/media_mimeref.asp

//To add new content types, add new mime-mapping entry in web.xml.

if (contentType == null) {

contentType= "application/octet-stream";

}else if (!contentType.startsWith("image")) {//Else, expect for images, determine content disposition. If content type is supported by//the browser, then set to inline, else attachment which will pop a 'save as' dialogue.

String accept = request.getHeader("Accept");

disposition= accept != null && HttpUtils.accepts(accept, contentType) ? "inline" : "attachment";

}

logger.debug("Content-Type : {}", contentType);//Initialize response.

response.reset();

response.setBufferSize(DEFAULT_BUFFER_SIZE);

response.setHeader("Content-Type", contentType);

response.setHeader("Content-Disposition", disposition + ";filename=\"" + fileName + "\"");

logger.debug("Content-Disposition : {}", disposition);

response.setHeader("Accept-Ranges", "bytes");

response.setHeader("ETag", fileName);

response.setDateHeader("Last-Modified", lastModified);

response.setDateHeader("Expires", System.currentTimeMillis() +DEFAULT_EXPIRE_TIME);//Send requested file (part(s)) to client ------------------------------------------------//Prepare streams.

try (InputStream input = newBufferedInputStream(Files.newInputStream(filepath));

OutputStream output=response.getOutputStream()) {if (ranges.isEmpty() || ranges.get(0) ==full) {//Return full file.

logger.info("Return full file");

response.setContentType(contentType);

response.setHeader("Content-Range", "bytes " + full.start + "-" + full.end + "/" +full.total);

response.setHeader("Content-Length", String.valueOf(full.length));

Range.copy(input, output, length, full.start, full.length);

}else if (ranges.size() == 1) {//Return single part of file.

Range r = ranges.get(0);

logger.info("Return 1 part of file : from ({}) to ({})", r.start, r.end);

response.setContentType(contentType);

response.setHeader("Content-Range", "bytes " + r.start + "-" + r.end + "/" +r.total);

response.setHeader("Content-Length", String.valueOf(r.length));

response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);//206.//Copy single part range.

Range.copy(input, output, length, r.start, r.length);

}else{//Return multiple parts of file.

response.setContentType("multipart/byteranges; boundary=" +MULTIPART_BOUNDARY);

response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);//206.//Cast back to ServletOutputStream to get the easy println methods.

ServletOutputStream sos =(ServletOutputStream) output;//Copy multi part range.

for(Range r : ranges) {

logger.info("Return multi part of file : from ({}) to ({})", r.start, r.end);//Add multipart boundary and header fields for every range.

sos.println();

sos.println("--" +MULTIPART_BOUNDARY);

sos.println("Content-Type: " +contentType);

sos.println("Content-Range: bytes " + r.start + "-" + r.end + "/" +r.total);//Copy single part range of multi part range.

Range.copy(input, output, length, r.start, r.length);

}//End with multipart boundary.

sos.println();

sos.println("--" + MULTIPART_BOUNDARY + "--");

}

}

}private static classRange {longstart;longend;longlength;longtotal;/*** Construct a byte range.

*@paramstart Start of the byte range.

*@paramend End of the byte range.

*@paramtotal Total length of the byte source.*/

public Range(long start, long end, longtotal) {this.start =start;this.end =end;this.length = end - start + 1;this.total =total;

}public static long sublong(String value, int beginIndex, intendIndex) {

String substring=value.substring(beginIndex, endIndex);return (substring.length() > 0) ? Long.parseLong(substring) : -1;

}private static void copy(InputStream input, OutputStream output, long inputSize, long start, long length) throwsIOException {byte[] buffer = new byte[DEFAULT_BUFFER_SIZE];intread;if (inputSize ==length) {//Write full range.

while ((read = input.read(buffer)) > 0) {

output.write(buffer,0, read);

output.flush();

}

}else{

input.skip(start);long toRead =length;while ((read = input.read(buffer)) > 0) {if ((toRead -= read) > 0) {

output.write(buffer,0, read);

output.flush();

}else{

output.write(buffer,0, (int) toRead +read);

output.flush();break;

}

}

}

}

}private static classHttpUtils {/*** Returns true if the given accept header accepts the given value.

*@paramacceptHeader The accept header.

*@paramtoAccept The value to be accepted.

*@returnTrue if the given accept header accepts the given value.*/

public static booleanaccepts(String acceptHeader, String toAccept) {

String[] acceptValues= acceptHeader.split("\\s*(,|;)\\s*");

Arrays.sort(acceptValues);return Arrays.binarySearch(acceptValues, toAccept) > -1

|| Arrays.binarySearch(acceptValues, toAccept.replaceAll("/.*$", "/*")) > -1

|| Arrays.binarySearch(acceptValues, "*/*") > -1;

}/*** Returns true if the given match header matches the given value.

*@parammatchHeader The match header.

*@paramtoMatch The value to be matched.

*@returnTrue if the given match header matches the given value.*/

public static booleanmatches(String matchHeader, String toMatch) {

String[] matchValues= matchHeader.split("\\s*,\\s*");

Arrays.sort(matchValues);return Arrays.binarySearch(matchValues, toMatch) > -1

|| Arrays.binarySearch(matchValues, "*") > -1;

}

}

}