com.king.platform.net.http.netty.request.FileRegionHttpBody.java Source code

Introduction

Here is the source code for com.king.platform.net.http.netty.request.FileRegionHttpBody.java

Source

// 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());
    }

}