1 /*
2 * ====================================================================
3 * Licensed to the Apache Software Foundation (ASF) under one
4 * or more contributor license agreements. See the NOTICE file
5 * distributed with this work for additional information
6 * regarding copyright ownership. The ASF licenses this file
7 * to you under the Apache License, Version 2.0 (the
8 * "License"); you may not use this file except in compliance
9 * with the License. You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing,
14 * software distributed under the License is distributed on an
15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 * KIND, either express or implied. See the License for the
17 * specific language governing permissions and limitations
18 * under the License.
19 * ====================================================================
20 *
21 * This software consists of voluntary contributions made by many
22 * individuals on behalf of the Apache Software Foundation. For more
23 * information on the Apache Software Foundation, please see
24 * <http://www.apache.org/>.
25 *
26 */
27
28 package org.apache.hc.core5.http.io.entity;
29
30 import java.io.IOException;
31 import java.io.InputStream;
32 import java.nio.ByteBuffer;
33
34 import org.apache.hc.core5.http.ContentType;
35 import org.apache.hc.core5.util.Args;
36
37 /**
38 * An entity that delivers the contents of a {@link ByteBuffer}.
39 */
40 public class ByteBufferEntity extends AbstractHttpEntity {
41
42 private final ByteBuffer buffer;
43 private final long length;
44
45 public ByteBufferEntity(final ByteBuffer buffer, final ContentType contentType, final String contentEncoding) {
46 super(contentType, contentEncoding);
47 Args.notNull(buffer, "Source byte buffer");
48 this.buffer = buffer;
49 this.length = buffer.remaining();
50 }
51
52 public ByteBufferEntity(final ByteBuffer buffer, final ContentType contentType) {
53 this(buffer, contentType, null);
54 }
55
56 @Override
57 public final boolean isRepeatable() {
58 return false;
59 }
60
61 @Override
62 public final long getContentLength() {
63 return length;
64 }
65
66 @Override
67 public final InputStream getContent() throws IOException, UnsupportedOperationException {
68 return new InputStream() {
69
70 @Override
71 public int read() throws IOException {
72 if (!buffer.hasRemaining()) {
73 return -1;
74 }
75 return buffer.get() & 0xFF;
76 }
77
78 @Override
79 public int read(final byte[] bytes, final int off, final int len) throws IOException {
80 if (!buffer.hasRemaining()) {
81 return -1;
82 }
83 final int chunk = Math.min(len, buffer.remaining());
84 buffer.get(bytes, off, chunk);
85 return chunk;
86 }
87 };
88 }
89
90 @Override
91 public final boolean isStreaming() {
92 return false;
93 }
94
95 @Override
96 public final void close() throws IOException {
97 }
98
99 }