com.ibm.watson.developer_cloud.http.InputStreamRequestBody.java Source code

Introduction

Here is the source code for com.ibm.watson.developer_cloud.http.InputStreamRequestBody.java

Source

/**
 * Copyright 2015 IBM Corp. All Rights Reserved.
 * 
 * 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.ibm.watson.developer_cloud.http;

import java.io.IOException;
import java.io.InputStream;

import okio.BufferedSink;
import okio.Okio;
import okio.Source;

import com.squareup.okhttp.MediaType;
import com.squareup.okhttp.RequestBody;
import com.squareup.okhttp.internal.Util;

/**
 * RequestBody that takes an {@link InputStream}.
 * 
 */
public class InputStreamRequestBody extends RequestBody {

    private InputStream inputStream;
    private MediaType mediaType;

    /**
     * Creates the @link {@link RequestBody} from an @link {@link InputStream}
     * 
     * @param mediaType the media type
     * @param inputStream the input stream
     * @return the request body
     */
    public static RequestBody create(final MediaType mediaType, final InputStream inputStream) {

        return new InputStreamRequestBody(inputStream, mediaType);
    }

    private InputStreamRequestBody(InputStream inputStream, MediaType mediaType) {
        this.inputStream = inputStream;
        this.mediaType = mediaType;
    }

    /*
     * (non-Javadoc)
     * 
     * @see com.squareup.okhttp.RequestBody#contentType()
     */
    @Override
    public MediaType contentType() {
        return mediaType;
    }

    /*
     * (non-Javadoc)
     * 
     * @see com.squareup.okhttp.RequestBody#writeTo(okio.BufferedSink)
     */
    @Override
    public void writeTo(BufferedSink sink) throws IOException {
        Source source = null;
        try {
            source = Okio.source(inputStream);
            sink.writeAll(source);
        } finally {
            Util.closeQuietly(source);
        }
    }
}