com.king.platform.net.http.netty.request.FileRegionHttpBody Maven / Gradle / Ivy

Go to download
// Copyright (C) king.com Ltd 2015
// https://github.com/king/king-http-client
// Author: Magnus Gustafsson
// License: Apache 2.0, https://raw.github.com/king/king-http-client/LICENSE-APACHE

package com.king.platform.net.http.netty.request;

import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.DefaultFileRegion;

import java.io.File;
import java.io.IOException;

public class FileRegionHttpBody implements HttpBody {
	private final File file;
	private final String contentType;
	private final long contentLength;

	public FileRegionHttpBody(File file, String contentType) {
		this.file = file;
		this.contentType = contentType;
		this.contentLength = file.length();
	}


	@Override
	public long getContentLength() {
		return contentLength;
	}

	@Override
	public String getContentType() {
		if (contentType != null) {
			return contentType;
		}
		return "application/binary";
	}

	@Override
	public ChannelFuture writeContent(ChannelHandlerContext ctx) throws IOException {
		Channel channel = ctx.channel();
		return channel.write(new DefaultFileRegion(file, 0, file.length()), channel.newProgressivePromise());
	}

}