001/*
002 * ====================================================================
003 * Licensed to the Apache Software Foundation (ASF) under one
004 * or more contributor license agreements.  See the NOTICE file
005 * distributed with this work for additional information
006 * regarding copyright ownership.  The ASF licenses this file
007 * to you under the Apache License, Version 2.0 (the
008 * "License"); you may not use this file except in compliance
009 * with the License.  You may obtain a copy of the License at
010 *
011 *   http://www.apache.org/licenses/LICENSE-2.0
012 *
013 * Unless required by applicable law or agreed to in writing,
014 * software distributed under the License is distributed on an
015 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
016 * KIND, either express or implied.  See the License for the
017 * specific language governing permissions and limitations
018 * under the License.
019 * ====================================================================
020 *
021 * This software consists of voluntary contributions made by many
022 * individuals on behalf of the Apache Software Foundation.  For more
023 * information on the Apache Software Foundation, please see
024 * <http://www.apache.org/>.
025 *
026 */
027
028package org.apache.http.nio.entity;
029
030import java.io.IOException;
031import java.io.OutputStream;
032
033import org.apache.http.nio.util.ContentOutputBuffer;
034import org.apache.http.util.Args;
035
036/**
037 * {@link OutputStream} adaptor for {@link ContentOutputBuffer}.
038 *
039 * @since 4.0
040 */
041public class ContentOutputStream extends OutputStream {
042
043    private final ContentOutputBuffer buffer;
044
045    public ContentOutputStream(final ContentOutputBuffer buffer) {
046        super();
047        Args.notNull(buffer, "Output buffer");
048        this.buffer = buffer;
049    }
050
051    @Override
052    public void close() throws IOException {
053        this.buffer.writeCompleted();
054    }
055
056    @Override
057    public void flush() throws IOException {
058    }
059
060    @Override
061    public void write(final byte[] b, final int off, final int len) throws IOException {
062        this.buffer.write(b, off, len);
063    }
064
065    @Override
066    public void write(final byte[] b) throws IOException {
067        if (b == null) {
068            return;
069        }
070        this.buffer.write(b, 0, b.length);
071    }
072
073    @Override
074    public void write(final int b) throws IOException {
075        this.buffer.write(b);
076    }
077
078}