| <!DOCTYPE HTML> |
| <html lang="en"> |
| <head> |
| <!-- Generated by javadoc (17) --> |
| <title>Source code</title> |
| <meta name="viewport" content="width=device-width, initial-scale=1"> |
| <meta name="description" content="source: package: org.apache.hadoop.hbase.nio, class: MultiByteBuff"> |
| <meta name="generator" content="javadoc/SourceToHTMLConverter"> |
| <link rel="stylesheet" type="text/css" href="../../../../../../stylesheet.css" title="Style"> |
| </head> |
| <body class="source-page"> |
| <main role="main"> |
| <div class="source-container"> |
| <pre><span class="source-line-no">001</span><span id="line-1">/*</span> |
| <span class="source-line-no">002</span><span id="line-2"> * Licensed to the Apache Software Foundation (ASF) under one</span> |
| <span class="source-line-no">003</span><span id="line-3"> * or more contributor license agreements. See the NOTICE file</span> |
| <span class="source-line-no">004</span><span id="line-4"> * distributed with this work for additional information</span> |
| <span class="source-line-no">005</span><span id="line-5"> * regarding copyright ownership. The ASF licenses this file</span> |
| <span class="source-line-no">006</span><span id="line-6"> * to you under the Apache License, Version 2.0 (the</span> |
| <span class="source-line-no">007</span><span id="line-7"> * "License"); you may not use this file except in compliance</span> |
| <span class="source-line-no">008</span><span id="line-8"> * with the License. You may obtain a copy of the License at</span> |
| <span class="source-line-no">009</span><span id="line-9"> *</span> |
| <span class="source-line-no">010</span><span id="line-10"> * http://www.apache.org/licenses/LICENSE-2.0</span> |
| <span class="source-line-no">011</span><span id="line-11"> *</span> |
| <span class="source-line-no">012</span><span id="line-12"> * Unless required by applicable law or agreed to in writing, software</span> |
| <span class="source-line-no">013</span><span id="line-13"> * distributed under the License is distributed on an "AS IS" BASIS,</span> |
| <span class="source-line-no">014</span><span id="line-14"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span> |
| <span class="source-line-no">015</span><span id="line-15"> * See the License for the specific language governing permissions and</span> |
| <span class="source-line-no">016</span><span id="line-16"> * limitations under the License.</span> |
| <span class="source-line-no">017</span><span id="line-17"> */</span> |
| <span class="source-line-no">018</span><span id="line-18">package org.apache.hadoop.hbase.nio;</span> |
| <span class="source-line-no">019</span><span id="line-19"></span> |
| <span class="source-line-no">020</span><span id="line-20">import static org.apache.hadoop.hbase.io.ByteBuffAllocator.NONE;</span> |
| <span class="source-line-no">021</span><span id="line-21"></span> |
| <span class="source-line-no">022</span><span id="line-22">import java.io.IOException;</span> |
| <span class="source-line-no">023</span><span id="line-23">import java.nio.BufferOverflowException;</span> |
| <span class="source-line-no">024</span><span id="line-24">import java.nio.BufferUnderflowException;</span> |
| <span class="source-line-no">025</span><span id="line-25">import java.nio.ByteBuffer;</span> |
| <span class="source-line-no">026</span><span id="line-26">import java.nio.InvalidMarkException;</span> |
| <span class="source-line-no">027</span><span id="line-27">import java.nio.channels.FileChannel;</span> |
| <span class="source-line-no">028</span><span id="line-28">import java.nio.channels.ReadableByteChannel;</span> |
| <span class="source-line-no">029</span><span id="line-29">import java.util.Iterator;</span> |
| <span class="source-line-no">030</span><span id="line-30">import java.util.NoSuchElementException;</span> |
| <span class="source-line-no">031</span><span id="line-31">import org.apache.hadoop.hbase.io.ByteBuffAllocator.Recycler;</span> |
| <span class="source-line-no">032</span><span id="line-32">import org.apache.hadoop.hbase.util.ByteBufferUtils;</span> |
| <span class="source-line-no">033</span><span id="line-33">import org.apache.hadoop.hbase.util.Bytes;</span> |
| <span class="source-line-no">034</span><span id="line-34">import org.apache.hadoop.hbase.util.ObjectIntPair;</span> |
| <span class="source-line-no">035</span><span id="line-35">import org.apache.yetus.audience.InterfaceAudience;</span> |
| <span class="source-line-no">036</span><span id="line-36"></span> |
| <span class="source-line-no">037</span><span id="line-37">/**</span> |
| <span class="source-line-no">038</span><span id="line-38"> * Provides a unified view of all the underlying ByteBuffers and will look as if a bigger sequential</span> |
| <span class="source-line-no">039</span><span id="line-39"> * buffer. This class provides similar APIs as in {@link ByteBuffer} to put/get int, short, long etc</span> |
| <span class="source-line-no">040</span><span id="line-40"> * and doing operations like mark, reset, slice etc. This has to be used when data is split across</span> |
| <span class="source-line-no">041</span><span id="line-41"> * multiple byte buffers and we don't want copy them to single buffer for reading from it.</span> |
| <span class="source-line-no">042</span><span id="line-42"> */</span> |
| <span class="source-line-no">043</span><span id="line-43">@InterfaceAudience.Private</span> |
| <span class="source-line-no">044</span><span id="line-44">public class MultiByteBuff extends ByteBuff {</span> |
| <span class="source-line-no">045</span><span id="line-45"></span> |
| <span class="source-line-no">046</span><span id="line-46"> private final ByteBuffer[] items;</span> |
| <span class="source-line-no">047</span><span id="line-47"> // Pointer to the current item in the MBB</span> |
| <span class="source-line-no">048</span><span id="line-48"> private ByteBuffer curItem = null;</span> |
| <span class="source-line-no">049</span><span id="line-49"> // Index of the current item in the MBB</span> |
| <span class="source-line-no">050</span><span id="line-50"> private int curItemIndex = 0;</span> |
| <span class="source-line-no">051</span><span id="line-51"></span> |
| <span class="source-line-no">052</span><span id="line-52"> private int limit = 0;</span> |
| <span class="source-line-no">053</span><span id="line-53"> private int limitedItemIndex;</span> |
| <span class="source-line-no">054</span><span id="line-54"> private int markedItemIndex = -1;</span> |
| <span class="source-line-no">055</span><span id="line-55"> private final int[] itemBeginPos;</span> |
| <span class="source-line-no">056</span><span id="line-56"></span> |
| <span class="source-line-no">057</span><span id="line-57"> private Iterator<ByteBuffer> buffsIterator = new Iterator<ByteBuffer>() {</span> |
| <span class="source-line-no">058</span><span id="line-58"> @Override</span> |
| <span class="source-line-no">059</span><span id="line-59"> public boolean hasNext() {</span> |
| <span class="source-line-no">060</span><span id="line-60"> return curItemIndex < limitedItemIndex</span> |
| <span class="source-line-no">061</span><span id="line-61"> || (curItemIndex == limitedItemIndex && items[curItemIndex].hasRemaining());</span> |
| <span class="source-line-no">062</span><span id="line-62"> }</span> |
| <span class="source-line-no">063</span><span id="line-63"></span> |
| <span class="source-line-no">064</span><span id="line-64"> @Override</span> |
| <span class="source-line-no">065</span><span id="line-65"> public ByteBuffer next() {</span> |
| <span class="source-line-no">066</span><span id="line-66"> if (curItemIndex >= items.length) {</span> |
| <span class="source-line-no">067</span><span id="line-67"> throw new NoSuchElementException("items overflow");</span> |
| <span class="source-line-no">068</span><span id="line-68"> }</span> |
| <span class="source-line-no">069</span><span id="line-69"> curItem = items[curItemIndex++];</span> |
| <span class="source-line-no">070</span><span id="line-70"> return curItem;</span> |
| <span class="source-line-no">071</span><span id="line-71"> }</span> |
| <span class="source-line-no">072</span><span id="line-72"> };</span> |
| <span class="source-line-no">073</span><span id="line-73"></span> |
| <span class="source-line-no">074</span><span id="line-74"> public MultiByteBuff(ByteBuffer... items) {</span> |
| <span class="source-line-no">075</span><span id="line-75"> this(NONE, items);</span> |
| <span class="source-line-no">076</span><span id="line-76"> }</span> |
| <span class="source-line-no">077</span><span id="line-77"></span> |
| <span class="source-line-no">078</span><span id="line-78"> public MultiByteBuff(Recycler recycler, ByteBuffer... items) {</span> |
| <span class="source-line-no">079</span><span id="line-79"> this(new RefCnt(recycler), items);</span> |
| <span class="source-line-no">080</span><span id="line-80"> }</span> |
| <span class="source-line-no">081</span><span id="line-81"></span> |
| <span class="source-line-no">082</span><span id="line-82"> MultiByteBuff(RefCnt refCnt, ByteBuffer... items) {</span> |
| <span class="source-line-no">083</span><span id="line-83"> this.refCnt = refCnt;</span> |
| <span class="source-line-no">084</span><span id="line-84"> assert items != null;</span> |
| <span class="source-line-no">085</span><span id="line-85"> assert items.length > 0;</span> |
| <span class="source-line-no">086</span><span id="line-86"> this.items = items;</span> |
| <span class="source-line-no">087</span><span id="line-87"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">088</span><span id="line-88"> // See below optimization in getInt(int) where we check whether the given index land in current</span> |
| <span class="source-line-no">089</span><span id="line-89"> // item. For this we need to check whether the passed index is less than the next item begin</span> |
| <span class="source-line-no">090</span><span id="line-90"> // offset. To handle this effectively for the last item buffer, we add an extra item into this</span> |
| <span class="source-line-no">091</span><span id="line-91"> // array.</span> |
| <span class="source-line-no">092</span><span id="line-92"> itemBeginPos = new int[items.length + 1];</span> |
| <span class="source-line-no">093</span><span id="line-93"> int offset = 0;</span> |
| <span class="source-line-no">094</span><span id="line-94"> for (int i = 0; i < items.length; i++) {</span> |
| <span class="source-line-no">095</span><span id="line-95"> ByteBuffer item = items[i];</span> |
| <span class="source-line-no">096</span><span id="line-96"> item.rewind();</span> |
| <span class="source-line-no">097</span><span id="line-97"> itemBeginPos[i] = offset;</span> |
| <span class="source-line-no">098</span><span id="line-98"> int l = item.limit() - item.position();</span> |
| <span class="source-line-no">099</span><span id="line-99"> offset += l;</span> |
| <span class="source-line-no">100</span><span id="line-100"> }</span> |
| <span class="source-line-no">101</span><span id="line-101"> this.limit = offset;</span> |
| <span class="source-line-no">102</span><span id="line-102"> this.itemBeginPos[items.length] = offset + 1;</span> |
| <span class="source-line-no">103</span><span id="line-103"> this.limitedItemIndex = this.items.length - 1;</span> |
| <span class="source-line-no">104</span><span id="line-104"> }</span> |
| <span class="source-line-no">105</span><span id="line-105"></span> |
| <span class="source-line-no">106</span><span id="line-106"> private MultiByteBuff(RefCnt refCnt, ByteBuffer[] items, int[] itemBeginPos, int limit,</span> |
| <span class="source-line-no">107</span><span id="line-107"> int limitedIndex, int curItemIndex, int markedIndex) {</span> |
| <span class="source-line-no">108</span><span id="line-108"> this.refCnt = refCnt;</span> |
| <span class="source-line-no">109</span><span id="line-109"> this.items = items;</span> |
| <span class="source-line-no">110</span><span id="line-110"> this.curItemIndex = curItemIndex;</span> |
| <span class="source-line-no">111</span><span id="line-111"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">112</span><span id="line-112"> this.itemBeginPos = itemBeginPos;</span> |
| <span class="source-line-no">113</span><span id="line-113"> this.limit = limit;</span> |
| <span class="source-line-no">114</span><span id="line-114"> this.limitedItemIndex = limitedIndex;</span> |
| <span class="source-line-no">115</span><span id="line-115"> this.markedItemIndex = markedIndex;</span> |
| <span class="source-line-no">116</span><span id="line-116"> }</span> |
| <span class="source-line-no">117</span><span id="line-117"></span> |
| <span class="source-line-no">118</span><span id="line-118"> /**</span> |
| <span class="source-line-no">119</span><span id="line-119"> * @throws UnsupportedOperationException MBB does not support array based operations</span> |
| <span class="source-line-no">120</span><span id="line-120"> */</span> |
| <span class="source-line-no">121</span><span id="line-121"> @Override</span> |
| <span class="source-line-no">122</span><span id="line-122"> public byte[] array() {</span> |
| <span class="source-line-no">123</span><span id="line-123"> throw new UnsupportedOperationException();</span> |
| <span class="source-line-no">124</span><span id="line-124"> }</span> |
| <span class="source-line-no">125</span><span id="line-125"></span> |
| <span class="source-line-no">126</span><span id="line-126"> /**</span> |
| <span class="source-line-no">127</span><span id="line-127"> * @throws UnsupportedOperationException MBB does not support array based operations</span> |
| <span class="source-line-no">128</span><span id="line-128"> */</span> |
| <span class="source-line-no">129</span><span id="line-129"> @Override</span> |
| <span class="source-line-no">130</span><span id="line-130"> public int arrayOffset() {</span> |
| <span class="source-line-no">131</span><span id="line-131"> throw new UnsupportedOperationException();</span> |
| <span class="source-line-no">132</span><span id="line-132"> }</span> |
| <span class="source-line-no">133</span><span id="line-133"></span> |
| <span class="source-line-no">134</span><span id="line-134"> /** Returns false. MBB does not support array based operations */</span> |
| <span class="source-line-no">135</span><span id="line-135"> @Override</span> |
| <span class="source-line-no">136</span><span id="line-136"> public boolean hasArray() {</span> |
| <span class="source-line-no">137</span><span id="line-137"> return false;</span> |
| <span class="source-line-no">138</span><span id="line-138"> }</span> |
| <span class="source-line-no">139</span><span id="line-139"></span> |
| <span class="source-line-no">140</span><span id="line-140"> /** Returns the total capacity of this MultiByteBuffer. */</span> |
| <span class="source-line-no">141</span><span id="line-141"> @Override</span> |
| <span class="source-line-no">142</span><span id="line-142"> public int capacity() {</span> |
| <span class="source-line-no">143</span><span id="line-143"> checkRefCount();</span> |
| <span class="source-line-no">144</span><span id="line-144"> int c = 0;</span> |
| <span class="source-line-no">145</span><span id="line-145"> for (ByteBuffer item : this.items) {</span> |
| <span class="source-line-no">146</span><span id="line-146"> c += item.capacity();</span> |
| <span class="source-line-no">147</span><span id="line-147"> }</span> |
| <span class="source-line-no">148</span><span id="line-148"> return c;</span> |
| <span class="source-line-no">149</span><span id="line-149"> }</span> |
| <span class="source-line-no">150</span><span id="line-150"></span> |
| <span class="source-line-no">151</span><span id="line-151"> /**</span> |
| <span class="source-line-no">152</span><span id="line-152"> * Fetches the byte at the given index. Does not change position of the underlying ByteBuffers</span> |
| <span class="source-line-no">153</span><span id="line-153"> * @return the byte at the given index</span> |
| <span class="source-line-no">154</span><span id="line-154"> */</span> |
| <span class="source-line-no">155</span><span id="line-155"> @Override</span> |
| <span class="source-line-no">156</span><span id="line-156"> public byte get(int index) {</span> |
| <span class="source-line-no">157</span><span id="line-157"> checkRefCount();</span> |
| <span class="source-line-no">158</span><span id="line-158"> int itemIndex = getItemIndex(index);</span> |
| <span class="source-line-no">159</span><span id="line-159"> return ByteBufferUtils.toByte(this.items[itemIndex], index - this.itemBeginPos[itemIndex]);</span> |
| <span class="source-line-no">160</span><span id="line-160"> }</span> |
| <span class="source-line-no">161</span><span id="line-161"></span> |
| <span class="source-line-no">162</span><span id="line-162"> @Override</span> |
| <span class="source-line-no">163</span><span id="line-163"> public byte getByteAfterPosition(int offset) {</span> |
| <span class="source-line-no">164</span><span id="line-164"> checkRefCount();</span> |
| <span class="source-line-no">165</span><span id="line-165"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">166</span><span id="line-166"> int index = offset + this.position();</span> |
| <span class="source-line-no">167</span><span id="line-167"> int itemIndex = getItemIndexFromCurItemIndex(index);</span> |
| <span class="source-line-no">168</span><span id="line-168"> return ByteBufferUtils.toByte(this.items[itemIndex], index - this.itemBeginPos[itemIndex]);</span> |
| <span class="source-line-no">169</span><span id="line-169"> }</span> |
| <span class="source-line-no">170</span><span id="line-170"></span> |
| <span class="source-line-no">171</span><span id="line-171"> /*</span> |
| <span class="source-line-no">172</span><span id="line-172"> * Returns in which sub ByteBuffer, the given element index will be available.</span> |
| <span class="source-line-no">173</span><span id="line-173"> */</span> |
| <span class="source-line-no">174</span><span id="line-174"> private int getItemIndex(int elemIndex) {</span> |
| <span class="source-line-no">175</span><span id="line-175"> if (elemIndex < 0) {</span> |
| <span class="source-line-no">176</span><span id="line-176"> throw new IndexOutOfBoundsException();</span> |
| <span class="source-line-no">177</span><span id="line-177"> }</span> |
| <span class="source-line-no">178</span><span id="line-178"> int index = 1;</span> |
| <span class="source-line-no">179</span><span id="line-179"> while (elemIndex >= this.itemBeginPos[index]) {</span> |
| <span class="source-line-no">180</span><span id="line-180"> index++;</span> |
| <span class="source-line-no">181</span><span id="line-181"> if (index == this.itemBeginPos.length) {</span> |
| <span class="source-line-no">182</span><span id="line-182"> throw new IndexOutOfBoundsException();</span> |
| <span class="source-line-no">183</span><span id="line-183"> }</span> |
| <span class="source-line-no">184</span><span id="line-184"> }</span> |
| <span class="source-line-no">185</span><span id="line-185"> return index - 1;</span> |
| <span class="source-line-no">186</span><span id="line-186"> }</span> |
| <span class="source-line-no">187</span><span id="line-187"></span> |
| <span class="source-line-no">188</span><span id="line-188"> /*</span> |
| <span class="source-line-no">189</span><span id="line-189"> * Returns in which sub ByteBuffer, the given element index will be available. In this case we are</span> |
| <span class="source-line-no">190</span><span id="line-190"> * sure that the item will be after MBB's current position</span> |
| <span class="source-line-no">191</span><span id="line-191"> */</span> |
| <span class="source-line-no">192</span><span id="line-192"> private int getItemIndexFromCurItemIndex(int elemIndex) {</span> |
| <span class="source-line-no">193</span><span id="line-193"> int index = this.curItemIndex;</span> |
| <span class="source-line-no">194</span><span id="line-194"> while (elemIndex >= this.itemBeginPos[index]) {</span> |
| <span class="source-line-no">195</span><span id="line-195"> index++;</span> |
| <span class="source-line-no">196</span><span id="line-196"> if (index == this.itemBeginPos.length) {</span> |
| <span class="source-line-no">197</span><span id="line-197"> throw new IndexOutOfBoundsException();</span> |
| <span class="source-line-no">198</span><span id="line-198"> }</span> |
| <span class="source-line-no">199</span><span id="line-199"> }</span> |
| <span class="source-line-no">200</span><span id="line-200"> return index - 1;</span> |
| <span class="source-line-no">201</span><span id="line-201"> }</span> |
| <span class="source-line-no">202</span><span id="line-202"></span> |
| <span class="source-line-no">203</span><span id="line-203"> /**</span> |
| <span class="source-line-no">204</span><span id="line-204"> * Fetches the int at the given index. Does not change position of the underlying ByteBuffers</span> |
| <span class="source-line-no">205</span><span id="line-205"> * @return the int value at the given index</span> |
| <span class="source-line-no">206</span><span id="line-206"> */</span> |
| <span class="source-line-no">207</span><span id="line-207"> @Override</span> |
| <span class="source-line-no">208</span><span id="line-208"> public int getInt(int index) {</span> |
| <span class="source-line-no">209</span><span id="line-209"> checkRefCount();</span> |
| <span class="source-line-no">210</span><span id="line-210"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">211</span><span id="line-211"> int itemIndex;</span> |
| <span class="source-line-no">212</span><span id="line-212"> if (</span> |
| <span class="source-line-no">213</span><span id="line-213"> this.itemBeginPos[this.curItemIndex] <= index</span> |
| <span class="source-line-no">214</span><span id="line-214"> && this.itemBeginPos[this.curItemIndex + 1] > index</span> |
| <span class="source-line-no">215</span><span id="line-215"> ) {</span> |
| <span class="source-line-no">216</span><span id="line-216"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">217</span><span id="line-217"> } else {</span> |
| <span class="source-line-no">218</span><span id="line-218"> itemIndex = getItemIndex(index);</span> |
| <span class="source-line-no">219</span><span id="line-219"> }</span> |
| <span class="source-line-no">220</span><span id="line-220"> return getInt(index, itemIndex);</span> |
| <span class="source-line-no">221</span><span id="line-221"> }</span> |
| <span class="source-line-no">222</span><span id="line-222"></span> |
| <span class="source-line-no">223</span><span id="line-223"> @Override</span> |
| <span class="source-line-no">224</span><span id="line-224"> public int getIntAfterPosition(int offset) {</span> |
| <span class="source-line-no">225</span><span id="line-225"> checkRefCount();</span> |
| <span class="source-line-no">226</span><span id="line-226"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">227</span><span id="line-227"> int index = offset + this.position();</span> |
| <span class="source-line-no">228</span><span id="line-228"> int itemIndex;</span> |
| <span class="source-line-no">229</span><span id="line-229"> if (this.itemBeginPos[this.curItemIndex + 1] > index) {</span> |
| <span class="source-line-no">230</span><span id="line-230"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">231</span><span id="line-231"> } else {</span> |
| <span class="source-line-no">232</span><span id="line-232"> itemIndex = getItemIndexFromCurItemIndex(index);</span> |
| <span class="source-line-no">233</span><span id="line-233"> }</span> |
| <span class="source-line-no">234</span><span id="line-234"> return getInt(index, itemIndex);</span> |
| <span class="source-line-no">235</span><span id="line-235"> }</span> |
| <span class="source-line-no">236</span><span id="line-236"></span> |
| <span class="source-line-no">237</span><span id="line-237"> /**</span> |
| <span class="source-line-no">238</span><span id="line-238"> * Fetches the short at the given index. Does not change position of the underlying ByteBuffers</span> |
| <span class="source-line-no">239</span><span id="line-239"> * @return the short value at the given index</span> |
| <span class="source-line-no">240</span><span id="line-240"> */</span> |
| <span class="source-line-no">241</span><span id="line-241"> @Override</span> |
| <span class="source-line-no">242</span><span id="line-242"> public short getShort(int index) {</span> |
| <span class="source-line-no">243</span><span id="line-243"> checkRefCount();</span> |
| <span class="source-line-no">244</span><span id="line-244"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">245</span><span id="line-245"> int itemIndex;</span> |
| <span class="source-line-no">246</span><span id="line-246"> if (</span> |
| <span class="source-line-no">247</span><span id="line-247"> this.itemBeginPos[this.curItemIndex] <= index</span> |
| <span class="source-line-no">248</span><span id="line-248"> && this.itemBeginPos[this.curItemIndex + 1] > index</span> |
| <span class="source-line-no">249</span><span id="line-249"> ) {</span> |
| <span class="source-line-no">250</span><span id="line-250"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">251</span><span id="line-251"> } else {</span> |
| <span class="source-line-no">252</span><span id="line-252"> itemIndex = getItemIndex(index);</span> |
| <span class="source-line-no">253</span><span id="line-253"> }</span> |
| <span class="source-line-no">254</span><span id="line-254"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">255</span><span id="line-255"> int offsetInItem = index - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">256</span><span id="line-256"> if (item.limit() - offsetInItem >= Bytes.SIZEOF_SHORT) {</span> |
| <span class="source-line-no">257</span><span id="line-257"> return ByteBufferUtils.toShort(item, offsetInItem);</span> |
| <span class="source-line-no">258</span><span id="line-258"> }</span> |
| <span class="source-line-no">259</span><span id="line-259"> if (items.length - 1 == itemIndex) {</span> |
| <span class="source-line-no">260</span><span id="line-260"> // means cur item is the last one and we wont be able to read a int. Throw exception</span> |
| <span class="source-line-no">261</span><span id="line-261"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">262</span><span id="line-262"> }</span> |
| <span class="source-line-no">263</span><span id="line-263"> ByteBuffer nextItem = items[itemIndex + 1];</span> |
| <span class="source-line-no">264</span><span id="line-264"> // Get available one byte from this item and remaining one from next</span> |
| <span class="source-line-no">265</span><span id="line-265"> short n = 0;</span> |
| <span class="source-line-no">266</span><span id="line-266"> n = (short) (n ^ (ByteBufferUtils.toByte(item, offsetInItem) & 0xFF));</span> |
| <span class="source-line-no">267</span><span id="line-267"> n = (short) (n << 8);</span> |
| <span class="source-line-no">268</span><span id="line-268"> n = (short) (n ^ (ByteBufferUtils.toByte(nextItem, 0) & 0xFF));</span> |
| <span class="source-line-no">269</span><span id="line-269"> return n;</span> |
| <span class="source-line-no">270</span><span id="line-270"> }</span> |
| <span class="source-line-no">271</span><span id="line-271"></span> |
| <span class="source-line-no">272</span><span id="line-272"> @Override</span> |
| <span class="source-line-no">273</span><span id="line-273"> public short getShortAfterPosition(int offset) {</span> |
| <span class="source-line-no">274</span><span id="line-274"> checkRefCount();</span> |
| <span class="source-line-no">275</span><span id="line-275"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">276</span><span id="line-276"> int index = offset + this.position();</span> |
| <span class="source-line-no">277</span><span id="line-277"> int itemIndex;</span> |
| <span class="source-line-no">278</span><span id="line-278"> if (this.itemBeginPos[this.curItemIndex + 1] > index) {</span> |
| <span class="source-line-no">279</span><span id="line-279"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">280</span><span id="line-280"> } else {</span> |
| <span class="source-line-no">281</span><span id="line-281"> itemIndex = getItemIndexFromCurItemIndex(index);</span> |
| <span class="source-line-no">282</span><span id="line-282"> }</span> |
| <span class="source-line-no">283</span><span id="line-283"> return getShort(index, itemIndex);</span> |
| <span class="source-line-no">284</span><span id="line-284"> }</span> |
| <span class="source-line-no">285</span><span id="line-285"></span> |
| <span class="source-line-no">286</span><span id="line-286"> private int getInt(int index, int itemIndex) {</span> |
| <span class="source-line-no">287</span><span id="line-287"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">288</span><span id="line-288"> int offsetInItem = index - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">289</span><span id="line-289"> int remainingLen = item.limit() - offsetInItem;</span> |
| <span class="source-line-no">290</span><span id="line-290"> if (remainingLen >= Bytes.SIZEOF_INT) {</span> |
| <span class="source-line-no">291</span><span id="line-291"> return ByteBufferUtils.toInt(item, offsetInItem);</span> |
| <span class="source-line-no">292</span><span id="line-292"> }</span> |
| <span class="source-line-no">293</span><span id="line-293"> if (items.length - 1 == itemIndex) {</span> |
| <span class="source-line-no">294</span><span id="line-294"> // means cur item is the last one and we wont be able to read a int. Throw exception</span> |
| <span class="source-line-no">295</span><span id="line-295"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">296</span><span id="line-296"> }</span> |
| <span class="source-line-no">297</span><span id="line-297"> int l = 0;</span> |
| <span class="source-line-no">298</span><span id="line-298"> for (int i = 0; i < Bytes.SIZEOF_INT; i++) {</span> |
| <span class="source-line-no">299</span><span id="line-299"> l <<= 8;</span> |
| <span class="source-line-no">300</span><span id="line-300"> l ^= get(index + i) & 0xFF;</span> |
| <span class="source-line-no">301</span><span id="line-301"> }</span> |
| <span class="source-line-no">302</span><span id="line-302"> return l;</span> |
| <span class="source-line-no">303</span><span id="line-303"> }</span> |
| <span class="source-line-no">304</span><span id="line-304"></span> |
| <span class="source-line-no">305</span><span id="line-305"> private short getShort(int index, int itemIndex) {</span> |
| <span class="source-line-no">306</span><span id="line-306"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">307</span><span id="line-307"> int offsetInItem = index - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">308</span><span id="line-308"> int remainingLen = item.limit() - offsetInItem;</span> |
| <span class="source-line-no">309</span><span id="line-309"> if (remainingLen >= Bytes.SIZEOF_SHORT) {</span> |
| <span class="source-line-no">310</span><span id="line-310"> return ByteBufferUtils.toShort(item, offsetInItem);</span> |
| <span class="source-line-no">311</span><span id="line-311"> }</span> |
| <span class="source-line-no">312</span><span id="line-312"> if (items.length - 1 == itemIndex) {</span> |
| <span class="source-line-no">313</span><span id="line-313"> // means cur item is the last one and we wont be able to read a short. Throw exception</span> |
| <span class="source-line-no">314</span><span id="line-314"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">315</span><span id="line-315"> }</span> |
| <span class="source-line-no">316</span><span id="line-316"> ByteBuffer nextItem = items[itemIndex + 1];</span> |
| <span class="source-line-no">317</span><span id="line-317"> // Get available bytes from this item and remaining from next</span> |
| <span class="source-line-no">318</span><span id="line-318"> short l = 0;</span> |
| <span class="source-line-no">319</span><span id="line-319"> for (int i = offsetInItem; i < item.capacity(); i++) {</span> |
| <span class="source-line-no">320</span><span id="line-320"> l = (short) (l << 8);</span> |
| <span class="source-line-no">321</span><span id="line-321"> l = (short) (l ^ (ByteBufferUtils.toByte(item, i) & 0xFF));</span> |
| <span class="source-line-no">322</span><span id="line-322"> }</span> |
| <span class="source-line-no">323</span><span id="line-323"> for (int i = 0; i < Bytes.SIZEOF_SHORT - remainingLen; i++) {</span> |
| <span class="source-line-no">324</span><span id="line-324"> l = (short) (l << 8);</span> |
| <span class="source-line-no">325</span><span id="line-325"> l = (short) (l ^ (ByteBufferUtils.toByte(nextItem, i) & 0xFF));</span> |
| <span class="source-line-no">326</span><span id="line-326"> }</span> |
| <span class="source-line-no">327</span><span id="line-327"> return l;</span> |
| <span class="source-line-no">328</span><span id="line-328"> }</span> |
| <span class="source-line-no">329</span><span id="line-329"></span> |
| <span class="source-line-no">330</span><span id="line-330"> private long getLong(int index, int itemIndex) {</span> |
| <span class="source-line-no">331</span><span id="line-331"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">332</span><span id="line-332"> int offsetInItem = index - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">333</span><span id="line-333"> int remainingLen = item.limit() - offsetInItem;</span> |
| <span class="source-line-no">334</span><span id="line-334"> if (remainingLen >= Bytes.SIZEOF_LONG) {</span> |
| <span class="source-line-no">335</span><span id="line-335"> return ByteBufferUtils.toLong(item, offsetInItem);</span> |
| <span class="source-line-no">336</span><span id="line-336"> }</span> |
| <span class="source-line-no">337</span><span id="line-337"> if (items.length - 1 == itemIndex) {</span> |
| <span class="source-line-no">338</span><span id="line-338"> // means cur item is the last one and we wont be able to read a long. Throw exception</span> |
| <span class="source-line-no">339</span><span id="line-339"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">340</span><span id="line-340"> }</span> |
| <span class="source-line-no">341</span><span id="line-341"> long l = 0;</span> |
| <span class="source-line-no">342</span><span id="line-342"> for (int i = 0; i < Bytes.SIZEOF_LONG; i++) {</span> |
| <span class="source-line-no">343</span><span id="line-343"> l <<= 8;</span> |
| <span class="source-line-no">344</span><span id="line-344"> l ^= get(index + i) & 0xFF;</span> |
| <span class="source-line-no">345</span><span id="line-345"> }</span> |
| <span class="source-line-no">346</span><span id="line-346"> return l;</span> |
| <span class="source-line-no">347</span><span id="line-347"> }</span> |
| <span class="source-line-no">348</span><span id="line-348"></span> |
| <span class="source-line-no">349</span><span id="line-349"> /**</span> |
| <span class="source-line-no">350</span><span id="line-350"> * Fetches the long at the given index. Does not change position of the underlying ByteBuffers</span> |
| <span class="source-line-no">351</span><span id="line-351"> * @return the long value at the given index</span> |
| <span class="source-line-no">352</span><span id="line-352"> */</span> |
| <span class="source-line-no">353</span><span id="line-353"> @Override</span> |
| <span class="source-line-no">354</span><span id="line-354"> public long getLong(int index) {</span> |
| <span class="source-line-no">355</span><span id="line-355"> checkRefCount();</span> |
| <span class="source-line-no">356</span><span id="line-356"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">357</span><span id="line-357"> int itemIndex;</span> |
| <span class="source-line-no">358</span><span id="line-358"> if (</span> |
| <span class="source-line-no">359</span><span id="line-359"> this.itemBeginPos[this.curItemIndex] <= index</span> |
| <span class="source-line-no">360</span><span id="line-360"> && this.itemBeginPos[this.curItemIndex + 1] > index</span> |
| <span class="source-line-no">361</span><span id="line-361"> ) {</span> |
| <span class="source-line-no">362</span><span id="line-362"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">363</span><span id="line-363"> } else {</span> |
| <span class="source-line-no">364</span><span id="line-364"> itemIndex = getItemIndex(index);</span> |
| <span class="source-line-no">365</span><span id="line-365"> }</span> |
| <span class="source-line-no">366</span><span id="line-366"> return getLong(index, itemIndex);</span> |
| <span class="source-line-no">367</span><span id="line-367"> }</span> |
| <span class="source-line-no">368</span><span id="line-368"></span> |
| <span class="source-line-no">369</span><span id="line-369"> @Override</span> |
| <span class="source-line-no">370</span><span id="line-370"> public long getLongAfterPosition(int offset) {</span> |
| <span class="source-line-no">371</span><span id="line-371"> checkRefCount();</span> |
| <span class="source-line-no">372</span><span id="line-372"> // Mostly the index specified will land within this current item. Short circuit for that</span> |
| <span class="source-line-no">373</span><span id="line-373"> int index = offset + this.position();</span> |
| <span class="source-line-no">374</span><span id="line-374"> int itemIndex;</span> |
| <span class="source-line-no">375</span><span id="line-375"> if (this.itemBeginPos[this.curItemIndex + 1] > index) {</span> |
| <span class="source-line-no">376</span><span id="line-376"> itemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">377</span><span id="line-377"> } else {</span> |
| <span class="source-line-no">378</span><span id="line-378"> itemIndex = getItemIndexFromCurItemIndex(index);</span> |
| <span class="source-line-no">379</span><span id="line-379"> }</span> |
| <span class="source-line-no">380</span><span id="line-380"> return getLong(index, itemIndex);</span> |
| <span class="source-line-no">381</span><span id="line-381"> }</span> |
| <span class="source-line-no">382</span><span id="line-382"></span> |
| <span class="source-line-no">383</span><span id="line-383"> /** Returns this MBB's current position */</span> |
| <span class="source-line-no">384</span><span id="line-384"> @Override</span> |
| <span class="source-line-no">385</span><span id="line-385"> public int position() {</span> |
| <span class="source-line-no">386</span><span id="line-386"> checkRefCount();</span> |
| <span class="source-line-no">387</span><span id="line-387"> return itemBeginPos[this.curItemIndex] + this.curItem.position();</span> |
| <span class="source-line-no">388</span><span id="line-388"> }</span> |
| <span class="source-line-no">389</span><span id="line-389"></span> |
| <span class="source-line-no">390</span><span id="line-390"> /**</span> |
| <span class="source-line-no">391</span><span id="line-391"> * Sets this MBB's position to the given value.</span> |
| <span class="source-line-no">392</span><span id="line-392"> * @return this object</span> |
| <span class="source-line-no">393</span><span id="line-393"> */</span> |
| <span class="source-line-no">394</span><span id="line-394"> @Override</span> |
| <span class="source-line-no">395</span><span id="line-395"> public MultiByteBuff position(int position) {</span> |
| <span class="source-line-no">396</span><span id="line-396"> checkRefCount();</span> |
| <span class="source-line-no">397</span><span id="line-397"> // Short circuit for positioning within the cur item. Mostly that is the case.</span> |
| <span class="source-line-no">398</span><span id="line-398"> if (</span> |
| <span class="source-line-no">399</span><span id="line-399"> this.itemBeginPos[this.curItemIndex] <= position</span> |
| <span class="source-line-no">400</span><span id="line-400"> && this.itemBeginPos[this.curItemIndex + 1] > position</span> |
| <span class="source-line-no">401</span><span id="line-401"> ) {</span> |
| <span class="source-line-no">402</span><span id="line-402"> this.curItem.position(position - this.itemBeginPos[this.curItemIndex]);</span> |
| <span class="source-line-no">403</span><span id="line-403"> return this;</span> |
| <span class="source-line-no">404</span><span id="line-404"> }</span> |
| <span class="source-line-no">405</span><span id="line-405"> int itemIndex = getItemIndex(position);</span> |
| <span class="source-line-no">406</span><span id="line-406"> // All items from 0 - curItem-1 set position at end.</span> |
| <span class="source-line-no">407</span><span id="line-407"> for (int i = 0; i < itemIndex; i++) {</span> |
| <span class="source-line-no">408</span><span id="line-408"> this.items[i].position(this.items[i].limit());</span> |
| <span class="source-line-no">409</span><span id="line-409"> }</span> |
| <span class="source-line-no">410</span><span id="line-410"> // All items after curItem set position at begin</span> |
| <span class="source-line-no">411</span><span id="line-411"> for (int i = itemIndex + 1; i < this.items.length; i++) {</span> |
| <span class="source-line-no">412</span><span id="line-412"> this.items[i].position(0);</span> |
| <span class="source-line-no">413</span><span id="line-413"> }</span> |
| <span class="source-line-no">414</span><span id="line-414"> this.curItem = this.items[itemIndex];</span> |
| <span class="source-line-no">415</span><span id="line-415"> this.curItem.position(position - this.itemBeginPos[itemIndex]);</span> |
| <span class="source-line-no">416</span><span id="line-416"> this.curItemIndex = itemIndex;</span> |
| <span class="source-line-no">417</span><span id="line-417"> return this;</span> |
| <span class="source-line-no">418</span><span id="line-418"> }</span> |
| <span class="source-line-no">419</span><span id="line-419"></span> |
| <span class="source-line-no">420</span><span id="line-420"> /**</span> |
| <span class="source-line-no">421</span><span id="line-421"> * Rewinds this MBB and the position is set to 0</span> |
| <span class="source-line-no">422</span><span id="line-422"> * @return this object</span> |
| <span class="source-line-no">423</span><span id="line-423"> */</span> |
| <span class="source-line-no">424</span><span id="line-424"> @Override</span> |
| <span class="source-line-no">425</span><span id="line-425"> public MultiByteBuff rewind() {</span> |
| <span class="source-line-no">426</span><span id="line-426"> checkRefCount();</span> |
| <span class="source-line-no">427</span><span id="line-427"> for (int i = 0; i < this.items.length; i++) {</span> |
| <span class="source-line-no">428</span><span id="line-428"> this.items[i].rewind();</span> |
| <span class="source-line-no">429</span><span id="line-429"> }</span> |
| <span class="source-line-no">430</span><span id="line-430"> this.curItemIndex = 0;</span> |
| <span class="source-line-no">431</span><span id="line-431"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">432</span><span id="line-432"> this.markedItemIndex = -1;</span> |
| <span class="source-line-no">433</span><span id="line-433"> return this;</span> |
| <span class="source-line-no">434</span><span id="line-434"> }</span> |
| <span class="source-line-no">435</span><span id="line-435"></span> |
| <span class="source-line-no">436</span><span id="line-436"> /**</span> |
| <span class="source-line-no">437</span><span id="line-437"> * Marks the current position of the MBB</span> |
| <span class="source-line-no">438</span><span id="line-438"> * @return this object</span> |
| <span class="source-line-no">439</span><span id="line-439"> */</span> |
| <span class="source-line-no">440</span><span id="line-440"> @Override</span> |
| <span class="source-line-no">441</span><span id="line-441"> public MultiByteBuff mark() {</span> |
| <span class="source-line-no">442</span><span id="line-442"> checkRefCount();</span> |
| <span class="source-line-no">443</span><span id="line-443"> this.markedItemIndex = this.curItemIndex;</span> |
| <span class="source-line-no">444</span><span id="line-444"> this.curItem.mark();</span> |
| <span class="source-line-no">445</span><span id="line-445"> return this;</span> |
| <span class="source-line-no">446</span><span id="line-446"> }</span> |
| <span class="source-line-no">447</span><span id="line-447"></span> |
| <span class="source-line-no">448</span><span id="line-448"> /**</span> |
| <span class="source-line-no">449</span><span id="line-449"> * Similar to {@link ByteBuffer}.reset(), ensures that this MBB is reset back to last marked</span> |
| <span class="source-line-no">450</span><span id="line-450"> * position.</span> |
| <span class="source-line-no">451</span><span id="line-451"> * @return This MBB</span> |
| <span class="source-line-no">452</span><span id="line-452"> */</span> |
| <span class="source-line-no">453</span><span id="line-453"> @Override</span> |
| <span class="source-line-no">454</span><span id="line-454"> public MultiByteBuff reset() {</span> |
| <span class="source-line-no">455</span><span id="line-455"> checkRefCount();</span> |
| <span class="source-line-no">456</span><span id="line-456"> // when the buffer is moved to the next one.. the reset should happen on the previous marked</span> |
| <span class="source-line-no">457</span><span id="line-457"> // item and the new one should be taken as the base</span> |
| <span class="source-line-no">458</span><span id="line-458"> if (this.markedItemIndex < 0) throw new InvalidMarkException();</span> |
| <span class="source-line-no">459</span><span id="line-459"> ByteBuffer markedItem = this.items[this.markedItemIndex];</span> |
| <span class="source-line-no">460</span><span id="line-460"> markedItem.reset();</span> |
| <span class="source-line-no">461</span><span id="line-461"> this.curItem = markedItem;</span> |
| <span class="source-line-no">462</span><span id="line-462"> // All items after the marked position upto the current item should be reset to 0</span> |
| <span class="source-line-no">463</span><span id="line-463"> for (int i = this.curItemIndex; i > this.markedItemIndex; i--) {</span> |
| <span class="source-line-no">464</span><span id="line-464"> this.items[i].position(0);</span> |
| <span class="source-line-no">465</span><span id="line-465"> }</span> |
| <span class="source-line-no">466</span><span id="line-466"> this.curItemIndex = this.markedItemIndex;</span> |
| <span class="source-line-no">467</span><span id="line-467"> return this;</span> |
| <span class="source-line-no">468</span><span id="line-468"> }</span> |
| <span class="source-line-no">469</span><span id="line-469"></span> |
| <span class="source-line-no">470</span><span id="line-470"> /**</span> |
| <span class="source-line-no">471</span><span id="line-471"> * Returns the number of elements between the current position and the limit.</span> |
| <span class="source-line-no">472</span><span id="line-472"> * @return the remaining elements in this MBB</span> |
| <span class="source-line-no">473</span><span id="line-473"> */</span> |
| <span class="source-line-no">474</span><span id="line-474"> @Override</span> |
| <span class="source-line-no">475</span><span id="line-475"> public int remaining() {</span> |
| <span class="source-line-no">476</span><span id="line-476"> checkRefCount();</span> |
| <span class="source-line-no">477</span><span id="line-477"> int remain = 0;</span> |
| <span class="source-line-no">478</span><span id="line-478"> for (int i = curItemIndex; i < items.length; i++) {</span> |
| <span class="source-line-no">479</span><span id="line-479"> remain += items[i].remaining();</span> |
| <span class="source-line-no">480</span><span id="line-480"> }</span> |
| <span class="source-line-no">481</span><span id="line-481"> return remain;</span> |
| <span class="source-line-no">482</span><span id="line-482"> }</span> |
| <span class="source-line-no">483</span><span id="line-483"></span> |
| <span class="source-line-no">484</span><span id="line-484"> /**</span> |
| <span class="source-line-no">485</span><span id="line-485"> * Returns true if there are elements between the current position and the limt</span> |
| <span class="source-line-no">486</span><span id="line-486"> * @return true if there are elements, false otherwise</span> |
| <span class="source-line-no">487</span><span id="line-487"> */</span> |
| <span class="source-line-no">488</span><span id="line-488"> @Override</span> |
| <span class="source-line-no">489</span><span id="line-489"> public final boolean hasRemaining() {</span> |
| <span class="source-line-no">490</span><span id="line-490"> checkRefCount();</span> |
| <span class="source-line-no">491</span><span id="line-491"> return this.curItem.hasRemaining() || (this.curItemIndex < this.limitedItemIndex</span> |
| <span class="source-line-no">492</span><span id="line-492"> && this.items[this.curItemIndex + 1].hasRemaining());</span> |
| <span class="source-line-no">493</span><span id="line-493"> }</span> |
| <span class="source-line-no">494</span><span id="line-494"></span> |
| <span class="source-line-no">495</span><span id="line-495"> /**</span> |
| <span class="source-line-no">496</span><span id="line-496"> * A relative method that returns byte at the current position. Increments the current position by</span> |
| <span class="source-line-no">497</span><span id="line-497"> * the size of a byte.</span> |
| <span class="source-line-no">498</span><span id="line-498"> * @return the byte at the current position</span> |
| <span class="source-line-no">499</span><span id="line-499"> */</span> |
| <span class="source-line-no">500</span><span id="line-500"> @Override</span> |
| <span class="source-line-no">501</span><span id="line-501"> public byte get() {</span> |
| <span class="source-line-no">502</span><span id="line-502"> checkRefCount();</span> |
| <span class="source-line-no">503</span><span id="line-503"> if (this.curItem.remaining() == 0) {</span> |
| <span class="source-line-no">504</span><span id="line-504"> if (items.length - 1 == this.curItemIndex) {</span> |
| <span class="source-line-no">505</span><span id="line-505"> // means cur item is the last one and we wont be able to read a long. Throw exception</span> |
| <span class="source-line-no">506</span><span id="line-506"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">507</span><span id="line-507"> }</span> |
| <span class="source-line-no">508</span><span id="line-508"> this.curItemIndex++;</span> |
| <span class="source-line-no">509</span><span id="line-509"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">510</span><span id="line-510"> }</span> |
| <span class="source-line-no">511</span><span id="line-511"> return this.curItem.get();</span> |
| <span class="source-line-no">512</span><span id="line-512"> }</span> |
| <span class="source-line-no">513</span><span id="line-513"></span> |
| <span class="source-line-no">514</span><span id="line-514"> /**</span> |
| <span class="source-line-no">515</span><span id="line-515"> * Returns the short value at the current position. Also advances the position by the size of</span> |
| <span class="source-line-no">516</span><span id="line-516"> * short</span> |
| <span class="source-line-no">517</span><span id="line-517"> * @return the short value at the current position</span> |
| <span class="source-line-no">518</span><span id="line-518"> */</span> |
| <span class="source-line-no">519</span><span id="line-519"> @Override</span> |
| <span class="source-line-no">520</span><span id="line-520"> public short getShort() {</span> |
| <span class="source-line-no">521</span><span id="line-521"> checkRefCount();</span> |
| <span class="source-line-no">522</span><span id="line-522"> int remaining = this.curItem.remaining();</span> |
| <span class="source-line-no">523</span><span id="line-523"> if (remaining >= Bytes.SIZEOF_SHORT) {</span> |
| <span class="source-line-no">524</span><span id="line-524"> return this.curItem.getShort();</span> |
| <span class="source-line-no">525</span><span id="line-525"> }</span> |
| <span class="source-line-no">526</span><span id="line-526"> short n = 0;</span> |
| <span class="source-line-no">527</span><span id="line-527"> n = (short) (n ^ (get() & 0xFF));</span> |
| <span class="source-line-no">528</span><span id="line-528"> n = (short) (n << 8);</span> |
| <span class="source-line-no">529</span><span id="line-529"> n = (short) (n ^ (get() & 0xFF));</span> |
| <span class="source-line-no">530</span><span id="line-530"> return n;</span> |
| <span class="source-line-no">531</span><span id="line-531"> }</span> |
| <span class="source-line-no">532</span><span id="line-532"></span> |
| <span class="source-line-no">533</span><span id="line-533"> /**</span> |
| <span class="source-line-no">534</span><span id="line-534"> * Returns the int value at the current position. Also advances the position by the size of int</span> |
| <span class="source-line-no">535</span><span id="line-535"> * @return the int value at the current position</span> |
| <span class="source-line-no">536</span><span id="line-536"> */</span> |
| <span class="source-line-no">537</span><span id="line-537"> @Override</span> |
| <span class="source-line-no">538</span><span id="line-538"> public int getInt() {</span> |
| <span class="source-line-no">539</span><span id="line-539"> checkRefCount();</span> |
| <span class="source-line-no">540</span><span id="line-540"> int remaining = this.curItem.remaining();</span> |
| <span class="source-line-no">541</span><span id="line-541"> if (remaining >= Bytes.SIZEOF_INT) {</span> |
| <span class="source-line-no">542</span><span id="line-542"> return this.curItem.getInt();</span> |
| <span class="source-line-no">543</span><span id="line-543"> }</span> |
| <span class="source-line-no">544</span><span id="line-544"> int n = 0;</span> |
| <span class="source-line-no">545</span><span id="line-545"> for (int i = 0; i < Bytes.SIZEOF_INT; i++) {</span> |
| <span class="source-line-no">546</span><span id="line-546"> n <<= 8;</span> |
| <span class="source-line-no">547</span><span id="line-547"> n ^= get() & 0xFF;</span> |
| <span class="source-line-no">548</span><span id="line-548"> }</span> |
| <span class="source-line-no">549</span><span id="line-549"> return n;</span> |
| <span class="source-line-no">550</span><span id="line-550"> }</span> |
| <span class="source-line-no">551</span><span id="line-551"></span> |
| <span class="source-line-no">552</span><span id="line-552"> /**</span> |
| <span class="source-line-no">553</span><span id="line-553"> * Returns the long value at the current position. Also advances the position by the size of long</span> |
| <span class="source-line-no">554</span><span id="line-554"> * @return the long value at the current position</span> |
| <span class="source-line-no">555</span><span id="line-555"> */</span> |
| <span class="source-line-no">556</span><span id="line-556"> @Override</span> |
| <span class="source-line-no">557</span><span id="line-557"> public long getLong() {</span> |
| <span class="source-line-no">558</span><span id="line-558"> checkRefCount();</span> |
| <span class="source-line-no">559</span><span id="line-559"> int remaining = this.curItem.remaining();</span> |
| <span class="source-line-no">560</span><span id="line-560"> if (remaining >= Bytes.SIZEOF_LONG) {</span> |
| <span class="source-line-no">561</span><span id="line-561"> return this.curItem.getLong();</span> |
| <span class="source-line-no">562</span><span id="line-562"> }</span> |
| <span class="source-line-no">563</span><span id="line-563"> long l = 0;</span> |
| <span class="source-line-no">564</span><span id="line-564"> for (int i = 0; i < Bytes.SIZEOF_LONG; i++) {</span> |
| <span class="source-line-no">565</span><span id="line-565"> l <<= 8;</span> |
| <span class="source-line-no">566</span><span id="line-566"> l ^= get() & 0xFF;</span> |
| <span class="source-line-no">567</span><span id="line-567"> }</span> |
| <span class="source-line-no">568</span><span id="line-568"> return l;</span> |
| <span class="source-line-no">569</span><span id="line-569"> }</span> |
| <span class="source-line-no">570</span><span id="line-570"></span> |
| <span class="source-line-no">571</span><span id="line-571"> /**</span> |
| <span class="source-line-no">572</span><span id="line-572"> * Copies the content from this MBB's current position to the byte array and fills it. Also</span> |
| <span class="source-line-no">573</span><span id="line-573"> * advances the position of the MBB by the length of the byte[].</span> |
| <span class="source-line-no">574</span><span id="line-574"> */</span> |
| <span class="source-line-no">575</span><span id="line-575"> @Override</span> |
| <span class="source-line-no">576</span><span id="line-576"> public void get(byte[] dst) {</span> |
| <span class="source-line-no">577</span><span id="line-577"> get(dst, 0, dst.length);</span> |
| <span class="source-line-no">578</span><span id="line-578"> }</span> |
| <span class="source-line-no">579</span><span id="line-579"></span> |
| <span class="source-line-no">580</span><span id="line-580"> /**</span> |
| <span class="source-line-no">581</span><span id="line-581"> * Copies the specified number of bytes from this MBB's current position to the byte[]'s offset.</span> |
| <span class="source-line-no">582</span><span id="line-582"> * Also advances the position of the MBB by the given length.</span> |
| <span class="source-line-no">583</span><span id="line-583"> */</span> |
| <span class="source-line-no">584</span><span id="line-584"> @Override</span> |
| <span class="source-line-no">585</span><span id="line-585"> public void get(byte[] dst, int offset, int length) {</span> |
| <span class="source-line-no">586</span><span id="line-586"> checkRefCount();</span> |
| <span class="source-line-no">587</span><span id="line-587"> while (length > 0) {</span> |
| <span class="source-line-no">588</span><span id="line-588"> int toRead = Math.min(length, this.curItem.remaining());</span> |
| <span class="source-line-no">589</span><span id="line-589"> ByteBufferUtils.copyFromBufferToArray(dst, this.curItem, this.curItem.position(), offset,</span> |
| <span class="source-line-no">590</span><span id="line-590"> toRead);</span> |
| <span class="source-line-no">591</span><span id="line-591"> this.curItem.position(this.curItem.position() + toRead);</span> |
| <span class="source-line-no">592</span><span id="line-592"> length -= toRead;</span> |
| <span class="source-line-no">593</span><span id="line-593"> if (length == 0) break;</span> |
| <span class="source-line-no">594</span><span id="line-594"> this.curItemIndex++;</span> |
| <span class="source-line-no">595</span><span id="line-595"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">596</span><span id="line-596"> offset += toRead;</span> |
| <span class="source-line-no">597</span><span id="line-597"> }</span> |
| <span class="source-line-no">598</span><span id="line-598"> }</span> |
| <span class="source-line-no">599</span><span id="line-599"></span> |
| <span class="source-line-no">600</span><span id="line-600"> @Override</span> |
| <span class="source-line-no">601</span><span id="line-601"> public void get(int sourceOffset, byte[] dst, int offset, int length) {</span> |
| <span class="source-line-no">602</span><span id="line-602"> checkRefCount();</span> |
| <span class="source-line-no">603</span><span id="line-603"> int itemIndex = getItemIndex(sourceOffset);</span> |
| <span class="source-line-no">604</span><span id="line-604"> ByteBuffer item = this.items[itemIndex];</span> |
| <span class="source-line-no">605</span><span id="line-605"> sourceOffset = sourceOffset - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">606</span><span id="line-606"> while (length > 0) {</span> |
| <span class="source-line-no">607</span><span id="line-607"> int toRead = Math.min((item.limit() - sourceOffset), length);</span> |
| <span class="source-line-no">608</span><span id="line-608"> ByteBufferUtils.copyFromBufferToArray(dst, item, sourceOffset, offset, toRead);</span> |
| <span class="source-line-no">609</span><span id="line-609"> length -= toRead;</span> |
| <span class="source-line-no">610</span><span id="line-610"> if (length == 0) break;</span> |
| <span class="source-line-no">611</span><span id="line-611"> itemIndex++;</span> |
| <span class="source-line-no">612</span><span id="line-612"> item = this.items[itemIndex];</span> |
| <span class="source-line-no">613</span><span id="line-613"> offset += toRead;</span> |
| <span class="source-line-no">614</span><span id="line-614"> sourceOffset = 0;</span> |
| <span class="source-line-no">615</span><span id="line-615"> }</span> |
| <span class="source-line-no">616</span><span id="line-616"> }</span> |
| <span class="source-line-no">617</span><span id="line-617"></span> |
| <span class="source-line-no">618</span><span id="line-618"> /**</span> |
| <span class="source-line-no">619</span><span id="line-619"> * Marks the limit of this MBB.</span> |
| <span class="source-line-no">620</span><span id="line-620"> * @return This MBB</span> |
| <span class="source-line-no">621</span><span id="line-621"> */</span> |
| <span class="source-line-no">622</span><span id="line-622"> @Override</span> |
| <span class="source-line-no">623</span><span id="line-623"> public MultiByteBuff limit(int limit) {</span> |
| <span class="source-line-no">624</span><span id="line-624"> checkRefCount();</span> |
| <span class="source-line-no">625</span><span id="line-625"> this.limit = limit;</span> |
| <span class="source-line-no">626</span><span id="line-626"> // Normally the limit will try to limit within the last BB item</span> |
| <span class="source-line-no">627</span><span id="line-627"> int limitedIndexBegin = this.itemBeginPos[this.limitedItemIndex];</span> |
| <span class="source-line-no">628</span><span id="line-628"> if (limit >= limitedIndexBegin && limit < this.itemBeginPos[this.limitedItemIndex + 1]) {</span> |
| <span class="source-line-no">629</span><span id="line-629"> this.items[this.limitedItemIndex].limit(limit - limitedIndexBegin);</span> |
| <span class="source-line-no">630</span><span id="line-630"> return this;</span> |
| <span class="source-line-no">631</span><span id="line-631"> }</span> |
| <span class="source-line-no">632</span><span id="line-632"> int itemIndex = getItemIndex(limit);</span> |
| <span class="source-line-no">633</span><span id="line-633"> int beginOffset = this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">634</span><span id="line-634"> int offsetInItem = limit - beginOffset;</span> |
| <span class="source-line-no">635</span><span id="line-635"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">636</span><span id="line-636"> item.limit(offsetInItem);</span> |
| <span class="source-line-no">637</span><span id="line-637"> for (int i = this.limitedItemIndex; i < itemIndex; i++) {</span> |
| <span class="source-line-no">638</span><span id="line-638"> this.items[i].limit(this.items[i].capacity());</span> |
| <span class="source-line-no">639</span><span id="line-639"> }</span> |
| <span class="source-line-no">640</span><span id="line-640"> this.limitedItemIndex = itemIndex;</span> |
| <span class="source-line-no">641</span><span id="line-641"> for (int i = itemIndex + 1; i < this.items.length; i++) {</span> |
| <span class="source-line-no">642</span><span id="line-642"> this.items[i].limit(this.items[i].position());</span> |
| <span class="source-line-no">643</span><span id="line-643"> }</span> |
| <span class="source-line-no">644</span><span id="line-644"> return this;</span> |
| <span class="source-line-no">645</span><span id="line-645"> }</span> |
| <span class="source-line-no">646</span><span id="line-646"></span> |
| <span class="source-line-no">647</span><span id="line-647"> /**</span> |
| <span class="source-line-no">648</span><span id="line-648"> * Returns the limit of this MBB</span> |
| <span class="source-line-no">649</span><span id="line-649"> * @return limit of the MBB</span> |
| <span class="source-line-no">650</span><span id="line-650"> */</span> |
| <span class="source-line-no">651</span><span id="line-651"> @Override</span> |
| <span class="source-line-no">652</span><span id="line-652"> public int limit() {</span> |
| <span class="source-line-no">653</span><span id="line-653"> return this.limit;</span> |
| <span class="source-line-no">654</span><span id="line-654"> }</span> |
| <span class="source-line-no">655</span><span id="line-655"></span> |
| <span class="source-line-no">656</span><span id="line-656"> /**</span> |
| <span class="source-line-no">657</span><span id="line-657"> * Returns an MBB which is a sliced version of this MBB. The position, limit and mark of the new</span> |
| <span class="source-line-no">658</span><span id="line-658"> * MBB will be independent than that of the original MBB. The content of the new MBB will start at</span> |
| <span class="source-line-no">659</span><span id="line-659"> * this MBB's current position</span> |
| <span class="source-line-no">660</span><span id="line-660"> * @return a sliced MBB</span> |
| <span class="source-line-no">661</span><span id="line-661"> */</span> |
| <span class="source-line-no">662</span><span id="line-662"> @Override</span> |
| <span class="source-line-no">663</span><span id="line-663"> public MultiByteBuff slice() {</span> |
| <span class="source-line-no">664</span><span id="line-664"> checkRefCount();</span> |
| <span class="source-line-no">665</span><span id="line-665"> ByteBuffer[] copy = new ByteBuffer[this.limitedItemIndex - this.curItemIndex + 1];</span> |
| <span class="source-line-no">666</span><span id="line-666"> for (int i = curItemIndex, j = 0; i <= this.limitedItemIndex; i++, j++) {</span> |
| <span class="source-line-no">667</span><span id="line-667"> copy[j] = this.items[i].slice();</span> |
| <span class="source-line-no">668</span><span id="line-668"> }</span> |
| <span class="source-line-no">669</span><span id="line-669"> return new MultiByteBuff(refCnt, copy);</span> |
| <span class="source-line-no">670</span><span id="line-670"> }</span> |
| <span class="source-line-no">671</span><span id="line-671"></span> |
| <span class="source-line-no">672</span><span id="line-672"> /**</span> |
| <span class="source-line-no">673</span><span id="line-673"> * Returns an MBB which is a duplicate version of this MBB. The position, limit and mark of the</span> |
| <span class="source-line-no">674</span><span id="line-674"> * new MBB will be independent than that of the original MBB. The content of the new MBB will</span> |
| <span class="source-line-no">675</span><span id="line-675"> * start at this MBB's current position The position, limit and mark of the new MBB would be</span> |
| <span class="source-line-no">676</span><span id="line-676"> * identical to this MBB in terms of values.</span> |
| <span class="source-line-no">677</span><span id="line-677"> * @return a duplicated MBB</span> |
| <span class="source-line-no">678</span><span id="line-678"> */</span> |
| <span class="source-line-no">679</span><span id="line-679"> @Override</span> |
| <span class="source-line-no">680</span><span id="line-680"> public MultiByteBuff duplicate() {</span> |
| <span class="source-line-no">681</span><span id="line-681"> checkRefCount();</span> |
| <span class="source-line-no">682</span><span id="line-682"> ByteBuffer[] itemsCopy = new ByteBuffer[this.items.length];</span> |
| <span class="source-line-no">683</span><span id="line-683"> for (int i = 0; i < this.items.length; i++) {</span> |
| <span class="source-line-no">684</span><span id="line-684"> itemsCopy[i] = items[i].duplicate();</span> |
| <span class="source-line-no">685</span><span id="line-685"> }</span> |
| <span class="source-line-no">686</span><span id="line-686"> return new MultiByteBuff(refCnt, itemsCopy, this.itemBeginPos, this.limit,</span> |
| <span class="source-line-no">687</span><span id="line-687"> this.limitedItemIndex, this.curItemIndex, this.markedItemIndex);</span> |
| <span class="source-line-no">688</span><span id="line-688"> }</span> |
| <span class="source-line-no">689</span><span id="line-689"></span> |
| <span class="source-line-no">690</span><span id="line-690"> /**</span> |
| <span class="source-line-no">691</span><span id="line-691"> * Writes a byte to this MBB at the current position and increments the position</span> |
| <span class="source-line-no">692</span><span id="line-692"> * @return this object</span> |
| <span class="source-line-no">693</span><span id="line-693"> */</span> |
| <span class="source-line-no">694</span><span id="line-694"> @Override</span> |
| <span class="source-line-no">695</span><span id="line-695"> public MultiByteBuff put(byte b) {</span> |
| <span class="source-line-no">696</span><span id="line-696"> checkRefCount();</span> |
| <span class="source-line-no">697</span><span id="line-697"> if (this.curItem.remaining() == 0) {</span> |
| <span class="source-line-no">698</span><span id="line-698"> if (this.curItemIndex == this.items.length - 1) {</span> |
| <span class="source-line-no">699</span><span id="line-699"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">700</span><span id="line-700"> }</span> |
| <span class="source-line-no">701</span><span id="line-701"> this.curItemIndex++;</span> |
| <span class="source-line-no">702</span><span id="line-702"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">703</span><span id="line-703"> }</span> |
| <span class="source-line-no">704</span><span id="line-704"> this.curItem.put(b);</span> |
| <span class="source-line-no">705</span><span id="line-705"> return this;</span> |
| <span class="source-line-no">706</span><span id="line-706"> }</span> |
| <span class="source-line-no">707</span><span id="line-707"></span> |
| <span class="source-line-no">708</span><span id="line-708"> /**</span> |
| <span class="source-line-no">709</span><span id="line-709"> * Writes a byte to this MBB at the given index and won't affect the position of any of the</span> |
| <span class="source-line-no">710</span><span id="line-710"> * buffers.</span> |
| <span class="source-line-no">711</span><span id="line-711"> * @return this object</span> |
| <span class="source-line-no">712</span><span id="line-712"> * @throws IndexOutOfBoundsException If <tt>index</tt> is negative or not smaller than the</span> |
| <span class="source-line-no">713</span><span id="line-713"> * {@link MultiByteBuff#limit}</span> |
| <span class="source-line-no">714</span><span id="line-714"> */</span> |
| <span class="source-line-no">715</span><span id="line-715"> @Override</span> |
| <span class="source-line-no">716</span><span id="line-716"> public MultiByteBuff put(int index, byte b) {</span> |
| <span class="source-line-no">717</span><span id="line-717"> checkRefCount();</span> |
| <span class="source-line-no">718</span><span id="line-718"> int itemIndex = getItemIndex(index);</span> |
| <span class="source-line-no">719</span><span id="line-719"> ByteBuffer item = items[itemIndex];</span> |
| <span class="source-line-no">720</span><span id="line-720"> item.put(index - itemBeginPos[itemIndex], b);</span> |
| <span class="source-line-no">721</span><span id="line-721"> return this;</span> |
| <span class="source-line-no">722</span><span id="line-722"> }</span> |
| <span class="source-line-no">723</span><span id="line-723"></span> |
| <span class="source-line-no">724</span><span id="line-724"> /**</span> |
| <span class="source-line-no">725</span><span id="line-725"> * Copies from a src BB to this MBB. This will be absolute positional copying and won't affect the</span> |
| <span class="source-line-no">726</span><span id="line-726"> * position of any of the buffers.</span> |
| <span class="source-line-no">727</span><span id="line-727"> * @param destOffset the position in this MBB to which the copy should happen</span> |
| <span class="source-line-no">728</span><span id="line-728"> * @param src the src MBB</span> |
| <span class="source-line-no">729</span><span id="line-729"> * @param srcOffset the offset in the src MBB from where the elements should be read</span> |
| <span class="source-line-no">730</span><span id="line-730"> * @param length the length upto which the copy should happen</span> |
| <span class="source-line-no">731</span><span id="line-731"> * @throws BufferUnderflowException If there are fewer than length bytes remaining in src</span> |
| <span class="source-line-no">732</span><span id="line-732"> * ByteBuff.</span> |
| <span class="source-line-no">733</span><span id="line-733"> * @throws BufferOverflowException If there is insufficient available space in this MBB for</span> |
| <span class="source-line-no">734</span><span id="line-734"> * length bytes.</span> |
| <span class="source-line-no">735</span><span id="line-735"> */</span> |
| <span class="source-line-no">736</span><span id="line-736"> @Override</span> |
| <span class="source-line-no">737</span><span id="line-737"> public MultiByteBuff put(int destOffset, ByteBuff src, int srcOffset, int length) {</span> |
| <span class="source-line-no">738</span><span id="line-738"> checkRefCount();</span> |
| <span class="source-line-no">739</span><span id="line-739"> int destItemIndex = getItemIndex(destOffset);</span> |
| <span class="source-line-no">740</span><span id="line-740"> int srcItemIndex = getItemIndexForByteBuff(src, srcOffset, length);</span> |
| <span class="source-line-no">741</span><span id="line-741"></span> |
| <span class="source-line-no">742</span><span id="line-742"> ByteBuffer destItem = this.items[destItemIndex];</span> |
| <span class="source-line-no">743</span><span id="line-743"> destOffset = this.getRelativeOffset(destOffset, destItemIndex);</span> |
| <span class="source-line-no">744</span><span id="line-744"></span> |
| <span class="source-line-no">745</span><span id="line-745"> ByteBuffer srcItem = getItemByteBuffer(src, srcItemIndex);</span> |
| <span class="source-line-no">746</span><span id="line-746"> srcOffset = getRelativeOffsetForByteBuff(src, srcOffset, srcItemIndex);</span> |
| <span class="source-line-no">747</span><span id="line-747"></span> |
| <span class="source-line-no">748</span><span id="line-748"> while (length > 0) {</span> |
| <span class="source-line-no">749</span><span id="line-749"> int toWrite = destItem.limit() - destOffset;</span> |
| <span class="source-line-no">750</span><span id="line-750"> if (toWrite <= 0) {</span> |
| <span class="source-line-no">751</span><span id="line-751"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">752</span><span id="line-752"> }</span> |
| <span class="source-line-no">753</span><span id="line-753"> int toRead = srcItem.limit() - srcOffset;</span> |
| <span class="source-line-no">754</span><span id="line-754"> if (toRead <= 0) {</span> |
| <span class="source-line-no">755</span><span id="line-755"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">756</span><span id="line-756"> }</span> |
| <span class="source-line-no">757</span><span id="line-757"> int toMove = Math.min(length, Math.min(toRead, toWrite));</span> |
| <span class="source-line-no">758</span><span id="line-758"> ByteBufferUtils.copyFromBufferToBuffer(srcItem, destItem, srcOffset, destOffset, toMove);</span> |
| <span class="source-line-no">759</span><span id="line-759"> length -= toMove;</span> |
| <span class="source-line-no">760</span><span id="line-760"> if (length == 0) {</span> |
| <span class="source-line-no">761</span><span id="line-761"> break;</span> |
| <span class="source-line-no">762</span><span id="line-762"> }</span> |
| <span class="source-line-no">763</span><span id="line-763"> if (toRead < toWrite) {</span> |
| <span class="source-line-no">764</span><span id="line-764"> if (++srcItemIndex >= getItemByteBufferCount(src)) {</span> |
| <span class="source-line-no">765</span><span id="line-765"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">766</span><span id="line-766"> }</span> |
| <span class="source-line-no">767</span><span id="line-767"> srcItem = getItemByteBuffer(src, srcItemIndex);</span> |
| <span class="source-line-no">768</span><span id="line-768"> srcOffset = 0;</span> |
| <span class="source-line-no">769</span><span id="line-769"> destOffset += toMove;</span> |
| <span class="source-line-no">770</span><span id="line-770"> } else if (toRead > toWrite) {</span> |
| <span class="source-line-no">771</span><span id="line-771"> if (++destItemIndex >= this.items.length) {</span> |
| <span class="source-line-no">772</span><span id="line-772"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">773</span><span id="line-773"> }</span> |
| <span class="source-line-no">774</span><span id="line-774"> destItem = this.items[destItemIndex];</span> |
| <span class="source-line-no">775</span><span id="line-775"> destOffset = 0;</span> |
| <span class="source-line-no">776</span><span id="line-776"> srcOffset += toMove;</span> |
| <span class="source-line-no">777</span><span id="line-777"> } else {</span> |
| <span class="source-line-no">778</span><span id="line-778"> // toRead = toWrite case</span> |
| <span class="source-line-no">779</span><span id="line-779"> if (++srcItemIndex >= getItemByteBufferCount(src)) {</span> |
| <span class="source-line-no">780</span><span id="line-780"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">781</span><span id="line-781"> }</span> |
| <span class="source-line-no">782</span><span id="line-782"> srcItem = getItemByteBuffer(src, srcItemIndex);</span> |
| <span class="source-line-no">783</span><span id="line-783"> srcOffset = 0;</span> |
| <span class="source-line-no">784</span><span id="line-784"> if (++destItemIndex >= this.items.length) {</span> |
| <span class="source-line-no">785</span><span id="line-785"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">786</span><span id="line-786"> }</span> |
| <span class="source-line-no">787</span><span id="line-787"> destItem = this.items[destItemIndex];</span> |
| <span class="source-line-no">788</span><span id="line-788"> destOffset = 0;</span> |
| <span class="source-line-no">789</span><span id="line-789"> }</span> |
| <span class="source-line-no">790</span><span id="line-790"> }</span> |
| <span class="source-line-no">791</span><span id="line-791"> return this;</span> |
| <span class="source-line-no">792</span><span id="line-792"> }</span> |
| <span class="source-line-no">793</span><span id="line-793"></span> |
| <span class="source-line-no">794</span><span id="line-794"> private static ByteBuffer getItemByteBuffer(ByteBuff buf, int byteBufferIndex) {</span> |
| <span class="source-line-no">795</span><span id="line-795"> if (buf instanceof SingleByteBuff) {</span> |
| <span class="source-line-no">796</span><span id="line-796"> if (byteBufferIndex != 0) {</span> |
| <span class="source-line-no">797</span><span id="line-797"> throw new IndexOutOfBoundsException(</span> |
| <span class="source-line-no">798</span><span id="line-798"> "index:[" + byteBufferIndex + "],but only index 0 is valid.");</span> |
| <span class="source-line-no">799</span><span id="line-799"> }</span> |
| <span class="source-line-no">800</span><span id="line-800"> return buf.nioByteBuffers()[0];</span> |
| <span class="source-line-no">801</span><span id="line-801"> }</span> |
| <span class="source-line-no">802</span><span id="line-802"> MultiByteBuff multiByteBuff = (MultiByteBuff) buf;</span> |
| <span class="source-line-no">803</span><span id="line-803"> if (byteBufferIndex < 0 || byteBufferIndex >= multiByteBuff.items.length) {</span> |
| <span class="source-line-no">804</span><span id="line-804"> throw new IndexOutOfBoundsException("index:[" + byteBufferIndex + "],but only index [0-"</span> |
| <span class="source-line-no">805</span><span id="line-805"> + multiByteBuff.items.length + ") is valid.");</span> |
| <span class="source-line-no">806</span><span id="line-806"> }</span> |
| <span class="source-line-no">807</span><span id="line-807"> return multiByteBuff.items[byteBufferIndex];</span> |
| <span class="source-line-no">808</span><span id="line-808"> }</span> |
| <span class="source-line-no">809</span><span id="line-809"></span> |
| <span class="source-line-no">810</span><span id="line-810"> private static int getItemIndexForByteBuff(ByteBuff byteBuff, int offset, int length) {</span> |
| <span class="source-line-no">811</span><span id="line-811"> if (byteBuff instanceof SingleByteBuff) {</span> |
| <span class="source-line-no">812</span><span id="line-812"> ByteBuffer byteBuffer = byteBuff.nioByteBuffers()[0];</span> |
| <span class="source-line-no">813</span><span id="line-813"> if (offset + length > byteBuffer.limit()) {</span> |
| <span class="source-line-no">814</span><span id="line-814"> throw new BufferUnderflowException();</span> |
| <span class="source-line-no">815</span><span id="line-815"> }</span> |
| <span class="source-line-no">816</span><span id="line-816"> return 0;</span> |
| <span class="source-line-no">817</span><span id="line-817"> }</span> |
| <span class="source-line-no">818</span><span id="line-818"> MultiByteBuff multiByteBuff = (MultiByteBuff) byteBuff;</span> |
| <span class="source-line-no">819</span><span id="line-819"> return multiByteBuff.getItemIndex(offset);</span> |
| <span class="source-line-no">820</span><span id="line-820"> }</span> |
| <span class="source-line-no">821</span><span id="line-821"></span> |
| <span class="source-line-no">822</span><span id="line-822"> private static int getRelativeOffsetForByteBuff(ByteBuff byteBuff, int globalOffset,</span> |
| <span class="source-line-no">823</span><span id="line-823"> int itemIndex) {</span> |
| <span class="source-line-no">824</span><span id="line-824"> if (byteBuff instanceof SingleByteBuff) {</span> |
| <span class="source-line-no">825</span><span id="line-825"> if (itemIndex != 0) {</span> |
| <span class="source-line-no">826</span><span id="line-826"> throw new IndexOutOfBoundsException("index:[" + itemIndex + "],but only index 0 is valid.");</span> |
| <span class="source-line-no">827</span><span id="line-827"> }</span> |
| <span class="source-line-no">828</span><span id="line-828"> return globalOffset;</span> |
| <span class="source-line-no">829</span><span id="line-829"> }</span> |
| <span class="source-line-no">830</span><span id="line-830"> return ((MultiByteBuff) byteBuff).getRelativeOffset(globalOffset, itemIndex);</span> |
| <span class="source-line-no">831</span><span id="line-831"> }</span> |
| <span class="source-line-no">832</span><span id="line-832"></span> |
| <span class="source-line-no">833</span><span id="line-833"> private int getRelativeOffset(int globalOffset, int itemIndex) {</span> |
| <span class="source-line-no">834</span><span id="line-834"> if (itemIndex < 0 || itemIndex >= this.items.length) {</span> |
| <span class="source-line-no">835</span><span id="line-835"> throw new IndexOutOfBoundsException(</span> |
| <span class="source-line-no">836</span><span id="line-836"> "index:[" + itemIndex + "],but only index [0-" + this.items.length + ") is valid.");</span> |
| <span class="source-line-no">837</span><span id="line-837"> }</span> |
| <span class="source-line-no">838</span><span id="line-838"> return globalOffset - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">839</span><span id="line-839"> }</span> |
| <span class="source-line-no">840</span><span id="line-840"></span> |
| <span class="source-line-no">841</span><span id="line-841"> private static int getItemByteBufferCount(ByteBuff buf) {</span> |
| <span class="source-line-no">842</span><span id="line-842"> return (buf instanceof SingleByteBuff) ? 1 : ((MultiByteBuff) buf).items.length;</span> |
| <span class="source-line-no">843</span><span id="line-843"> }</span> |
| <span class="source-line-no">844</span><span id="line-844"></span> |
| <span class="source-line-no">845</span><span id="line-845"> /**</span> |
| <span class="source-line-no">846</span><span id="line-846"> * Writes an int to this MBB at its current position. Also advances the position by size of int</span> |
| <span class="source-line-no">847</span><span id="line-847"> * @param val Int value to write</span> |
| <span class="source-line-no">848</span><span id="line-848"> * @return this object</span> |
| <span class="source-line-no">849</span><span id="line-849"> */</span> |
| <span class="source-line-no">850</span><span id="line-850"> @Override</span> |
| <span class="source-line-no">851</span><span id="line-851"> public MultiByteBuff putInt(int val) {</span> |
| <span class="source-line-no">852</span><span id="line-852"> checkRefCount();</span> |
| <span class="source-line-no">853</span><span id="line-853"> if (this.curItem.remaining() >= Bytes.SIZEOF_INT) {</span> |
| <span class="source-line-no">854</span><span id="line-854"> this.curItem.putInt(val);</span> |
| <span class="source-line-no">855</span><span id="line-855"> return this;</span> |
| <span class="source-line-no">856</span><span id="line-856"> }</span> |
| <span class="source-line-no">857</span><span id="line-857"> if (this.curItemIndex == this.items.length - 1) {</span> |
| <span class="source-line-no">858</span><span id="line-858"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">859</span><span id="line-859"> }</span> |
| <span class="source-line-no">860</span><span id="line-860"> // During read, we will read as byte by byte for this case. So just write in Big endian</span> |
| <span class="source-line-no">861</span><span id="line-861"> put(int3(val));</span> |
| <span class="source-line-no">862</span><span id="line-862"> put(int2(val));</span> |
| <span class="source-line-no">863</span><span id="line-863"> put(int1(val));</span> |
| <span class="source-line-no">864</span><span id="line-864"> put(int0(val));</span> |
| <span class="source-line-no">865</span><span id="line-865"> return this;</span> |
| <span class="source-line-no">866</span><span id="line-866"> }</span> |
| <span class="source-line-no">867</span><span id="line-867"></span> |
| <span class="source-line-no">868</span><span id="line-868"> private static byte int3(int x) {</span> |
| <span class="source-line-no">869</span><span id="line-869"> return (byte) (x >> 24);</span> |
| <span class="source-line-no">870</span><span id="line-870"> }</span> |
| <span class="source-line-no">871</span><span id="line-871"></span> |
| <span class="source-line-no">872</span><span id="line-872"> private static byte int2(int x) {</span> |
| <span class="source-line-no">873</span><span id="line-873"> return (byte) (x >> 16);</span> |
| <span class="source-line-no">874</span><span id="line-874"> }</span> |
| <span class="source-line-no">875</span><span id="line-875"></span> |
| <span class="source-line-no">876</span><span id="line-876"> private static byte int1(int x) {</span> |
| <span class="source-line-no">877</span><span id="line-877"> return (byte) (x >> 8);</span> |
| <span class="source-line-no">878</span><span id="line-878"> }</span> |
| <span class="source-line-no">879</span><span id="line-879"></span> |
| <span class="source-line-no">880</span><span id="line-880"> private static byte int0(int x) {</span> |
| <span class="source-line-no">881</span><span id="line-881"> return (byte) x;</span> |
| <span class="source-line-no">882</span><span id="line-882"> }</span> |
| <span class="source-line-no">883</span><span id="line-883"></span> |
| <span class="source-line-no">884</span><span id="line-884"> /** Copies from the given byte[] to this MBB */</span> |
| <span class="source-line-no">885</span><span id="line-885"> @Override</span> |
| <span class="source-line-no">886</span><span id="line-886"> public final MultiByteBuff put(byte[] src) {</span> |
| <span class="source-line-no">887</span><span id="line-887"> return put(src, 0, src.length);</span> |
| <span class="source-line-no">888</span><span id="line-888"> }</span> |
| <span class="source-line-no">889</span><span id="line-889"></span> |
| <span class="source-line-no">890</span><span id="line-890"> /** Copies from the given byte[] to this MBB. */</span> |
| <span class="source-line-no">891</span><span id="line-891"> @Override</span> |
| <span class="source-line-no">892</span><span id="line-892"> public MultiByteBuff put(byte[] src, int offset, int length) {</span> |
| <span class="source-line-no">893</span><span id="line-893"> checkRefCount();</span> |
| <span class="source-line-no">894</span><span id="line-894"> if (this.curItem.remaining() >= length) {</span> |
| <span class="source-line-no">895</span><span id="line-895"> ByteBufferUtils.copyFromArrayToBuffer(this.curItem, src, offset, length);</span> |
| <span class="source-line-no">896</span><span id="line-896"> return this;</span> |
| <span class="source-line-no">897</span><span id="line-897"> }</span> |
| <span class="source-line-no">898</span><span id="line-898"> int end = offset + length;</span> |
| <span class="source-line-no">899</span><span id="line-899"> for (int i = offset; i < end; i++) {</span> |
| <span class="source-line-no">900</span><span id="line-900"> this.put(src[i]);</span> |
| <span class="source-line-no">901</span><span id="line-901"> }</span> |
| <span class="source-line-no">902</span><span id="line-902"> return this;</span> |
| <span class="source-line-no">903</span><span id="line-903"> }</span> |
| <span class="source-line-no">904</span><span id="line-904"></span> |
| <span class="source-line-no">905</span><span id="line-905"> /**</span> |
| <span class="source-line-no">906</span><span id="line-906"> * Writes a long to this MBB at its current position. Also advances the position by size of long</span> |
| <span class="source-line-no">907</span><span id="line-907"> * @param val Long value to write</span> |
| <span class="source-line-no">908</span><span id="line-908"> * @return this object</span> |
| <span class="source-line-no">909</span><span id="line-909"> */</span> |
| <span class="source-line-no">910</span><span id="line-910"> @Override</span> |
| <span class="source-line-no">911</span><span id="line-911"> public MultiByteBuff putLong(long val) {</span> |
| <span class="source-line-no">912</span><span id="line-912"> checkRefCount();</span> |
| <span class="source-line-no">913</span><span id="line-913"> if (this.curItem.remaining() >= Bytes.SIZEOF_LONG) {</span> |
| <span class="source-line-no">914</span><span id="line-914"> this.curItem.putLong(val);</span> |
| <span class="source-line-no">915</span><span id="line-915"> return this;</span> |
| <span class="source-line-no">916</span><span id="line-916"> }</span> |
| <span class="source-line-no">917</span><span id="line-917"> if (this.curItemIndex == this.items.length - 1) {</span> |
| <span class="source-line-no">918</span><span id="line-918"> throw new BufferOverflowException();</span> |
| <span class="source-line-no">919</span><span id="line-919"> }</span> |
| <span class="source-line-no">920</span><span id="line-920"> // During read, we will read as byte by byte for this case. So just write in Big endian</span> |
| <span class="source-line-no">921</span><span id="line-921"> put(long7(val));</span> |
| <span class="source-line-no">922</span><span id="line-922"> put(long6(val));</span> |
| <span class="source-line-no">923</span><span id="line-923"> put(long5(val));</span> |
| <span class="source-line-no">924</span><span id="line-924"> put(long4(val));</span> |
| <span class="source-line-no">925</span><span id="line-925"> put(long3(val));</span> |
| <span class="source-line-no">926</span><span id="line-926"> put(long2(val));</span> |
| <span class="source-line-no">927</span><span id="line-927"> put(long1(val));</span> |
| <span class="source-line-no">928</span><span id="line-928"> put(long0(val));</span> |
| <span class="source-line-no">929</span><span id="line-929"> return this;</span> |
| <span class="source-line-no">930</span><span id="line-930"> }</span> |
| <span class="source-line-no">931</span><span id="line-931"></span> |
| <span class="source-line-no">932</span><span id="line-932"> private static byte long7(long x) {</span> |
| <span class="source-line-no">933</span><span id="line-933"> return (byte) (x >> 56);</span> |
| <span class="source-line-no">934</span><span id="line-934"> }</span> |
| <span class="source-line-no">935</span><span id="line-935"></span> |
| <span class="source-line-no">936</span><span id="line-936"> private static byte long6(long x) {</span> |
| <span class="source-line-no">937</span><span id="line-937"> return (byte) (x >> 48);</span> |
| <span class="source-line-no">938</span><span id="line-938"> }</span> |
| <span class="source-line-no">939</span><span id="line-939"></span> |
| <span class="source-line-no">940</span><span id="line-940"> private static byte long5(long x) {</span> |
| <span class="source-line-no">941</span><span id="line-941"> return (byte) (x >> 40);</span> |
| <span class="source-line-no">942</span><span id="line-942"> }</span> |
| <span class="source-line-no">943</span><span id="line-943"></span> |
| <span class="source-line-no">944</span><span id="line-944"> private static byte long4(long x) {</span> |
| <span class="source-line-no">945</span><span id="line-945"> return (byte) (x >> 32);</span> |
| <span class="source-line-no">946</span><span id="line-946"> }</span> |
| <span class="source-line-no">947</span><span id="line-947"></span> |
| <span class="source-line-no">948</span><span id="line-948"> private static byte long3(long x) {</span> |
| <span class="source-line-no">949</span><span id="line-949"> return (byte) (x >> 24);</span> |
| <span class="source-line-no">950</span><span id="line-950"> }</span> |
| <span class="source-line-no">951</span><span id="line-951"></span> |
| <span class="source-line-no">952</span><span id="line-952"> private static byte long2(long x) {</span> |
| <span class="source-line-no">953</span><span id="line-953"> return (byte) (x >> 16);</span> |
| <span class="source-line-no">954</span><span id="line-954"> }</span> |
| <span class="source-line-no">955</span><span id="line-955"></span> |
| <span class="source-line-no">956</span><span id="line-956"> private static byte long1(long x) {</span> |
| <span class="source-line-no">957</span><span id="line-957"> return (byte) (x >> 8);</span> |
| <span class="source-line-no">958</span><span id="line-958"> }</span> |
| <span class="source-line-no">959</span><span id="line-959"></span> |
| <span class="source-line-no">960</span><span id="line-960"> private static byte long0(long x) {</span> |
| <span class="source-line-no">961</span><span id="line-961"> return (byte) x;</span> |
| <span class="source-line-no">962</span><span id="line-962"> }</span> |
| <span class="source-line-no">963</span><span id="line-963"></span> |
| <span class="source-line-no">964</span><span id="line-964"> /**</span> |
| <span class="source-line-no">965</span><span id="line-965"> * Jumps the current position of this MBB by specified length.</span> |
| <span class="source-line-no">966</span><span id="line-966"> */</span> |
| <span class="source-line-no">967</span><span id="line-967"> @Override</span> |
| <span class="source-line-no">968</span><span id="line-968"> public MultiByteBuff skip(int length) {</span> |
| <span class="source-line-no">969</span><span id="line-969"> checkRefCount();</span> |
| <span class="source-line-no">970</span><span id="line-970"> // Get available bytes from this item and remaining from next</span> |
| <span class="source-line-no">971</span><span id="line-971"> int jump = 0;</span> |
| <span class="source-line-no">972</span><span id="line-972"> while (true) {</span> |
| <span class="source-line-no">973</span><span id="line-973"> jump = this.curItem.remaining();</span> |
| <span class="source-line-no">974</span><span id="line-974"> if (jump >= length) {</span> |
| <span class="source-line-no">975</span><span id="line-975"> this.curItem.position(this.curItem.position() + length);</span> |
| <span class="source-line-no">976</span><span id="line-976"> break;</span> |
| <span class="source-line-no">977</span><span id="line-977"> }</span> |
| <span class="source-line-no">978</span><span id="line-978"> this.curItem.position(this.curItem.position() + jump);</span> |
| <span class="source-line-no">979</span><span id="line-979"> length -= jump;</span> |
| <span class="source-line-no">980</span><span id="line-980"> this.curItemIndex++;</span> |
| <span class="source-line-no">981</span><span id="line-981"> this.curItem = this.items[this.curItemIndex];</span> |
| <span class="source-line-no">982</span><span id="line-982"> }</span> |
| <span class="source-line-no">983</span><span id="line-983"> return this;</span> |
| <span class="source-line-no">984</span><span id="line-984"> }</span> |
| <span class="source-line-no">985</span><span id="line-985"></span> |
| <span class="source-line-no">986</span><span id="line-986"> /**</span> |
| <span class="source-line-no">987</span><span id="line-987"> * Jumps back the current position of this MBB by specified length.</span> |
| <span class="source-line-no">988</span><span id="line-988"> */</span> |
| <span class="source-line-no">989</span><span id="line-989"> @Override</span> |
| <span class="source-line-no">990</span><span id="line-990"> public MultiByteBuff moveBack(int length) {</span> |
| <span class="source-line-no">991</span><span id="line-991"> checkRefCount();</span> |
| <span class="source-line-no">992</span><span id="line-992"> while (length != 0) {</span> |
| <span class="source-line-no">993</span><span id="line-993"> if (length > curItem.position()) {</span> |
| <span class="source-line-no">994</span><span id="line-994"> length -= curItem.position();</span> |
| <span class="source-line-no">995</span><span id="line-995"> this.curItem.position(0);</span> |
| <span class="source-line-no">996</span><span id="line-996"> this.curItemIndex--;</span> |
| <span class="source-line-no">997</span><span id="line-997"> this.curItem = this.items[curItemIndex];</span> |
| <span class="source-line-no">998</span><span id="line-998"> } else {</span> |
| <span class="source-line-no">999</span><span id="line-999"> this.curItem.position(curItem.position() - length);</span> |
| <span class="source-line-no">1000</span><span id="line-1000"> break;</span> |
| <span class="source-line-no">1001</span><span id="line-1001"> }</span> |
| <span class="source-line-no">1002</span><span id="line-1002"> }</span> |
| <span class="source-line-no">1003</span><span id="line-1003"> return this;</span> |
| <span class="source-line-no">1004</span><span id="line-1004"> }</span> |
| <span class="source-line-no">1005</span><span id="line-1005"></span> |
| <span class="source-line-no">1006</span><span id="line-1006"> /**</span> |
| <span class="source-line-no">1007</span><span id="line-1007"> * Returns bytes from current position till length specified, as a single ByteBuffer. When all</span> |
| <span class="source-line-no">1008</span><span id="line-1008"> * these bytes happen to be in a single ByteBuffer, which this object wraps, that ByteBuffer item</span> |
| <span class="source-line-no">1009</span><span id="line-1009"> * as such will be returned. So users are warned not to change the position or limit of this</span> |
| <span class="source-line-no">1010</span><span id="line-1010"> * returned ByteBuffer. The position of the returned byte buffer is at the begin of the required</span> |
| <span class="source-line-no">1011</span><span id="line-1011"> * bytes. When the required bytes happen to span across multiple ByteBuffers, this API will copy</span> |
| <span class="source-line-no">1012</span><span id="line-1012"> * the bytes to a newly created ByteBuffer of required size and return that.</span> |
| <span class="source-line-no">1013</span><span id="line-1013"> * @param length number of bytes required.</span> |
| <span class="source-line-no">1014</span><span id="line-1014"> * @return bytes from current position till length specified, as a single ByteButter.</span> |
| <span class="source-line-no">1015</span><span id="line-1015"> */</span> |
| <span class="source-line-no">1016</span><span id="line-1016"> @Override</span> |
| <span class="source-line-no">1017</span><span id="line-1017"> public ByteBuffer asSubByteBuffer(int length) {</span> |
| <span class="source-line-no">1018</span><span id="line-1018"> checkRefCount();</span> |
| <span class="source-line-no">1019</span><span id="line-1019"> if (this.curItem.remaining() >= length) {</span> |
| <span class="source-line-no">1020</span><span id="line-1020"> return this.curItem;</span> |
| <span class="source-line-no">1021</span><span id="line-1021"> }</span> |
| <span class="source-line-no">1022</span><span id="line-1022"> int offset = 0;</span> |
| <span class="source-line-no">1023</span><span id="line-1023"> byte[] dupB = new byte[length];</span> |
| <span class="source-line-no">1024</span><span id="line-1024"> int locCurItemIndex = curItemIndex;</span> |
| <span class="source-line-no">1025</span><span id="line-1025"> ByteBuffer locCurItem = curItem;</span> |
| <span class="source-line-no">1026</span><span id="line-1026"> while (length > 0) {</span> |
| <span class="source-line-no">1027</span><span id="line-1027"> int toRead = Math.min(length, locCurItem.remaining());</span> |
| <span class="source-line-no">1028</span><span id="line-1028"> ByteBufferUtils.copyFromBufferToArray(dupB, locCurItem, locCurItem.position(), offset,</span> |
| <span class="source-line-no">1029</span><span id="line-1029"> toRead);</span> |
| <span class="source-line-no">1030</span><span id="line-1030"> length -= toRead;</span> |
| <span class="source-line-no">1031</span><span id="line-1031"> if (length == 0) break;</span> |
| <span class="source-line-no">1032</span><span id="line-1032"> locCurItemIndex++;</span> |
| <span class="source-line-no">1033</span><span id="line-1033"> locCurItem = this.items[locCurItemIndex];</span> |
| <span class="source-line-no">1034</span><span id="line-1034"> offset += toRead;</span> |
| <span class="source-line-no">1035</span><span id="line-1035"> }</span> |
| <span class="source-line-no">1036</span><span id="line-1036"> return ByteBuffer.wrap(dupB);</span> |
| <span class="source-line-no">1037</span><span id="line-1037"> }</span> |
| <span class="source-line-no">1038</span><span id="line-1038"></span> |
| <span class="source-line-no">1039</span><span id="line-1039"> /**</span> |
| <span class="source-line-no">1040</span><span id="line-1040"> * Returns bytes from given offset till length specified, as a single ByteBuffer. When all these</span> |
| <span class="source-line-no">1041</span><span id="line-1041"> * bytes happen to be in a single ByteBuffer, which this object wraps, that ByteBuffer item as</span> |
| <span class="source-line-no">1042</span><span id="line-1042"> * such will be returned (with offset in this ByteBuffer where the bytes starts). So users are</span> |
| <span class="source-line-no">1043</span><span id="line-1043"> * warned not to change the position or limit of this returned ByteBuffer. When the required bytes</span> |
| <span class="source-line-no">1044</span><span id="line-1044"> * happen to span across multiple ByteBuffers, this API will copy the bytes to a newly created</span> |
| <span class="source-line-no">1045</span><span id="line-1045"> * ByteBuffer of required size and return that.</span> |
| <span class="source-line-no">1046</span><span id="line-1046"> * @param offset the offset in this MBB from where the subBuffer should be created</span> |
| <span class="source-line-no">1047</span><span id="line-1047"> * @param length the length of the subBuffer</span> |
| <span class="source-line-no">1048</span><span id="line-1048"> * @param pair a pair that will have the bytes from the current position till length specified,</span> |
| <span class="source-line-no">1049</span><span id="line-1049"> * as a single ByteBuffer and offset in that Buffer where the bytes starts. The</span> |
| <span class="source-line-no">1050</span><span id="line-1050"> * method would set the values on the pair that is passed in by the caller</span> |
| <span class="source-line-no">1051</span><span id="line-1051"> */</span> |
| <span class="source-line-no">1052</span><span id="line-1052"> @Override</span> |
| <span class="source-line-no">1053</span><span id="line-1053"> public void asSubByteBuffer(int offset, int length, ObjectIntPair<ByteBuffer> pair) {</span> |
| <span class="source-line-no">1054</span><span id="line-1054"> checkRefCount();</span> |
| <span class="source-line-no">1055</span><span id="line-1055"> if (this.itemBeginPos[this.curItemIndex] <= offset) {</span> |
| <span class="source-line-no">1056</span><span id="line-1056"> int relOffsetInCurItem = offset - this.itemBeginPos[this.curItemIndex];</span> |
| <span class="source-line-no">1057</span><span id="line-1057"> if (this.curItem.limit() - relOffsetInCurItem >= length) {</span> |
| <span class="source-line-no">1058</span><span id="line-1058"> pair.setFirst(this.curItem);</span> |
| <span class="source-line-no">1059</span><span id="line-1059"> pair.setSecond(relOffsetInCurItem);</span> |
| <span class="source-line-no">1060</span><span id="line-1060"> return;</span> |
| <span class="source-line-no">1061</span><span id="line-1061"> }</span> |
| <span class="source-line-no">1062</span><span id="line-1062"> }</span> |
| <span class="source-line-no">1063</span><span id="line-1063"> int itemIndex = getItemIndex(offset);</span> |
| <span class="source-line-no">1064</span><span id="line-1064"> ByteBuffer item = this.items[itemIndex];</span> |
| <span class="source-line-no">1065</span><span id="line-1065"> offset = offset - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">1066</span><span id="line-1066"> if (item.limit() - offset >= length) {</span> |
| <span class="source-line-no">1067</span><span id="line-1067"> pair.setFirst(item);</span> |
| <span class="source-line-no">1068</span><span id="line-1068"> pair.setSecond(offset);</span> |
| <span class="source-line-no">1069</span><span id="line-1069"> return;</span> |
| <span class="source-line-no">1070</span><span id="line-1070"> }</span> |
| <span class="source-line-no">1071</span><span id="line-1071"> byte[] dst = new byte[length];</span> |
| <span class="source-line-no">1072</span><span id="line-1072"> int destOffset = 0;</span> |
| <span class="source-line-no">1073</span><span id="line-1073"> while (length > 0) {</span> |
| <span class="source-line-no">1074</span><span id="line-1074"> int toRead = Math.min(length, item.limit() - offset);</span> |
| <span class="source-line-no">1075</span><span id="line-1075"> ByteBufferUtils.copyFromBufferToArray(dst, item, offset, destOffset, toRead);</span> |
| <span class="source-line-no">1076</span><span id="line-1076"> length -= toRead;</span> |
| <span class="source-line-no">1077</span><span id="line-1077"> if (length == 0) break;</span> |
| <span class="source-line-no">1078</span><span id="line-1078"> itemIndex++;</span> |
| <span class="source-line-no">1079</span><span id="line-1079"> item = this.items[itemIndex];</span> |
| <span class="source-line-no">1080</span><span id="line-1080"> destOffset += toRead;</span> |
| <span class="source-line-no">1081</span><span id="line-1081"> offset = 0;</span> |
| <span class="source-line-no">1082</span><span id="line-1082"> }</span> |
| <span class="source-line-no">1083</span><span id="line-1083"> pair.setFirst(ByteBuffer.wrap(dst));</span> |
| <span class="source-line-no">1084</span><span id="line-1084"> pair.setSecond(0);</span> |
| <span class="source-line-no">1085</span><span id="line-1085"> }</span> |
| <span class="source-line-no">1086</span><span id="line-1086"></span> |
| <span class="source-line-no">1087</span><span id="line-1087"> /**</span> |
| <span class="source-line-no">1088</span><span id="line-1088"> * Copies the content from an this MBB to a ByteBuffer</span> |
| <span class="source-line-no">1089</span><span id="line-1089"> * @param out the ByteBuffer to which the copy has to happen, its position will be</span> |
| <span class="source-line-no">1090</span><span id="line-1090"> * advanced.</span> |
| <span class="source-line-no">1091</span><span id="line-1091"> * @param sourceOffset the offset in the MBB from which the elements has to be copied</span> |
| <span class="source-line-no">1092</span><span id="line-1092"> * @param length the length in the MBB upto which the elements has to be copied</span> |
| <span class="source-line-no">1093</span><span id="line-1093"> */</span> |
| <span class="source-line-no">1094</span><span id="line-1094"> @Override</span> |
| <span class="source-line-no">1095</span><span id="line-1095"> public void get(ByteBuffer out, int sourceOffset, int length) {</span> |
| <span class="source-line-no">1096</span><span id="line-1096"> checkRefCount();</span> |
| <span class="source-line-no">1097</span><span id="line-1097"> int itemIndex = getItemIndex(sourceOffset);</span> |
| <span class="source-line-no">1098</span><span id="line-1098"> ByteBuffer in = this.items[itemIndex];</span> |
| <span class="source-line-no">1099</span><span id="line-1099"> sourceOffset = sourceOffset - this.itemBeginPos[itemIndex];</span> |
| <span class="source-line-no">1100</span><span id="line-1100"> while (length > 0) {</span> |
| <span class="source-line-no">1101</span><span id="line-1101"> int toRead = Math.min(in.limit() - sourceOffset, length);</span> |
| <span class="source-line-no">1102</span><span id="line-1102"> ByteBufferUtils.copyFromBufferToBuffer(in, out, sourceOffset, toRead);</span> |
| <span class="source-line-no">1103</span><span id="line-1103"> length -= toRead;</span> |
| <span class="source-line-no">1104</span><span id="line-1104"> if (length == 0) {</span> |
| <span class="source-line-no">1105</span><span id="line-1105"> break;</span> |
| <span class="source-line-no">1106</span><span id="line-1106"> }</span> |
| <span class="source-line-no">1107</span><span id="line-1107"> itemIndex++;</span> |
| <span class="source-line-no">1108</span><span id="line-1108"> in = this.items[itemIndex];</span> |
| <span class="source-line-no">1109</span><span id="line-1109"> sourceOffset = 0;</span> |
| <span class="source-line-no">1110</span><span id="line-1110"> }</span> |
| <span class="source-line-no">1111</span><span id="line-1111"> }</span> |
| <span class="source-line-no">1112</span><span id="line-1112"></span> |
| <span class="source-line-no">1113</span><span id="line-1113"> /**</span> |
| <span class="source-line-no">1114</span><span id="line-1114"> * Copy the content from this MBB to a byte[] based on the given offset and length the position</span> |
| <span class="source-line-no">1115</span><span id="line-1115"> * from where the copy should start the length upto which the copy has to be done</span> |
| <span class="source-line-no">1116</span><span id="line-1116"> * @return byte[] with the copied contents from this MBB.</span> |
| <span class="source-line-no">1117</span><span id="line-1117"> */</span> |
| <span class="source-line-no">1118</span><span id="line-1118"> @Override</span> |
| <span class="source-line-no">1119</span><span id="line-1119"> public byte[] toBytes(int offset, int length) {</span> |
| <span class="source-line-no">1120</span><span id="line-1120"> checkRefCount();</span> |
| <span class="source-line-no">1121</span><span id="line-1121"> byte[] output = new byte[length];</span> |
| <span class="source-line-no">1122</span><span id="line-1122"> this.get(offset, output, 0, length);</span> |
| <span class="source-line-no">1123</span><span id="line-1123"> return output;</span> |
| <span class="source-line-no">1124</span><span id="line-1124"> }</span> |
| <span class="source-line-no">1125</span><span id="line-1125"></span> |
| <span class="source-line-no">1126</span><span id="line-1126"> private int internalRead(ReadableByteChannel channel, long offset, ChannelReader reader)</span> |
| <span class="source-line-no">1127</span><span id="line-1127"> throws IOException {</span> |
| <span class="source-line-no">1128</span><span id="line-1128"> checkRefCount();</span> |
| <span class="source-line-no">1129</span><span id="line-1129"> int total = 0;</span> |
| <span class="source-line-no">1130</span><span id="line-1130"> while (buffsIterator.hasNext()) {</span> |
| <span class="source-line-no">1131</span><span id="line-1131"> ByteBuffer buffer = buffsIterator.next();</span> |
| <span class="source-line-no">1132</span><span id="line-1132"> int len = read(channel, buffer, offset, reader);</span> |
| <span class="source-line-no">1133</span><span id="line-1133"> if (len > 0) {</span> |
| <span class="source-line-no">1134</span><span id="line-1134"> total += len;</span> |
| <span class="source-line-no">1135</span><span id="line-1135"> offset += len;</span> |
| <span class="source-line-no">1136</span><span id="line-1136"> }</span> |
| <span class="source-line-no">1137</span><span id="line-1137"> if (buffer.hasRemaining()) {</span> |
| <span class="source-line-no">1138</span><span id="line-1138"> // reset</span> |
| <span class="source-line-no">1139</span><span id="line-1139"> curItem = buffer;</span> |
| <span class="source-line-no">1140</span><span id="line-1140"> curItemIndex = (curItemIndex - 1);</span> |
| <span class="source-line-no">1141</span><span id="line-1141"> break;</span> |
| <span class="source-line-no">1142</span><span id="line-1142"> }</span> |
| <span class="source-line-no">1143</span><span id="line-1143"> }</span> |
| <span class="source-line-no">1144</span><span id="line-1144"> return total;</span> |
| <span class="source-line-no">1145</span><span id="line-1145"> }</span> |
| <span class="source-line-no">1146</span><span id="line-1146"></span> |
| <span class="source-line-no">1147</span><span id="line-1147"> @Override</span> |
| <span class="source-line-no">1148</span><span id="line-1148"> public int read(ReadableByteChannel channel) throws IOException {</span> |
| <span class="source-line-no">1149</span><span id="line-1149"> return internalRead(channel, 0, CHANNEL_READER);</span> |
| <span class="source-line-no">1150</span><span id="line-1150"> }</span> |
| <span class="source-line-no">1151</span><span id="line-1151"></span> |
| <span class="source-line-no">1152</span><span id="line-1152"> @Override</span> |
| <span class="source-line-no">1153</span><span id="line-1153"> public int read(FileChannel channel, long offset) throws IOException {</span> |
| <span class="source-line-no">1154</span><span id="line-1154"> return internalRead(channel, offset, FILE_READER);</span> |
| <span class="source-line-no">1155</span><span id="line-1155"> }</span> |
| <span class="source-line-no">1156</span><span id="line-1156"></span> |
| <span class="source-line-no">1157</span><span id="line-1157"> @Override</span> |
| <span class="source-line-no">1158</span><span id="line-1158"> public int write(FileChannel channel, long offset) throws IOException {</span> |
| <span class="source-line-no">1159</span><span id="line-1159"> checkRefCount();</span> |
| <span class="source-line-no">1160</span><span id="line-1160"> int total = 0;</span> |
| <span class="source-line-no">1161</span><span id="line-1161"> while (buffsIterator.hasNext()) {</span> |
| <span class="source-line-no">1162</span><span id="line-1162"> ByteBuffer buffer = buffsIterator.next();</span> |
| <span class="source-line-no">1163</span><span id="line-1163"> while (buffer.hasRemaining()) {</span> |
| <span class="source-line-no">1164</span><span id="line-1164"> int len = channel.write(buffer, offset);</span> |
| <span class="source-line-no">1165</span><span id="line-1165"> total += len;</span> |
| <span class="source-line-no">1166</span><span id="line-1166"> offset += len;</span> |
| <span class="source-line-no">1167</span><span id="line-1167"> }</span> |
| <span class="source-line-no">1168</span><span id="line-1168"> }</span> |
| <span class="source-line-no">1169</span><span id="line-1169"> return total;</span> |
| <span class="source-line-no">1170</span><span id="line-1170"> }</span> |
| <span class="source-line-no">1171</span><span id="line-1171"></span> |
| <span class="source-line-no">1172</span><span id="line-1172"> @Override</span> |
| <span class="source-line-no">1173</span><span id="line-1173"> public ByteBuffer[] nioByteBuffers() {</span> |
| <span class="source-line-no">1174</span><span id="line-1174"> checkRefCount();</span> |
| <span class="source-line-no">1175</span><span id="line-1175"> return this.items;</span> |
| <span class="source-line-no">1176</span><span id="line-1176"> }</span> |
| <span class="source-line-no">1177</span><span id="line-1177"></span> |
| <span class="source-line-no">1178</span><span id="line-1178"> @Override</span> |
| <span class="source-line-no">1179</span><span id="line-1179"> public boolean equals(Object obj) {</span> |
| <span class="source-line-no">1180</span><span id="line-1180"> if (!(obj instanceof MultiByteBuff)) return false;</span> |
| <span class="source-line-no">1181</span><span id="line-1181"> if (this == obj) return true;</span> |
| <span class="source-line-no">1182</span><span id="line-1182"> MultiByteBuff that = (MultiByteBuff) obj;</span> |
| <span class="source-line-no">1183</span><span id="line-1183"> if (this.capacity() != that.capacity()) return false;</span> |
| <span class="source-line-no">1184</span><span id="line-1184"> if (</span> |
| <span class="source-line-no">1185</span><span id="line-1185"> ByteBuff.compareTo(this, this.position(), this.limit(), that, that.position(), that.limit())</span> |
| <span class="source-line-no">1186</span><span id="line-1186"> == 0</span> |
| <span class="source-line-no">1187</span><span id="line-1187"> ) {</span> |
| <span class="source-line-no">1188</span><span id="line-1188"> return true;</span> |
| <span class="source-line-no">1189</span><span id="line-1189"> }</span> |
| <span class="source-line-no">1190</span><span id="line-1190"> return false;</span> |
| <span class="source-line-no">1191</span><span id="line-1191"> }</span> |
| <span class="source-line-no">1192</span><span id="line-1192"></span> |
| <span class="source-line-no">1193</span><span id="line-1193"> @Override</span> |
| <span class="source-line-no">1194</span><span id="line-1194"> public int hashCode() {</span> |
| <span class="source-line-no">1195</span><span id="line-1195"> int hash = 0;</span> |
| <span class="source-line-no">1196</span><span id="line-1196"> for (ByteBuffer b : this.items) {</span> |
| <span class="source-line-no">1197</span><span id="line-1197"> hash += b.hashCode();</span> |
| <span class="source-line-no">1198</span><span id="line-1198"> }</span> |
| <span class="source-line-no">1199</span><span id="line-1199"> return hash;</span> |
| <span class="source-line-no">1200</span><span id="line-1200"> }</span> |
| <span class="source-line-no">1201</span><span id="line-1201"></span> |
| <span class="source-line-no">1202</span><span id="line-1202"> @Override</span> |
| <span class="source-line-no">1203</span><span id="line-1203"> public MultiByteBuff retain() {</span> |
| <span class="source-line-no">1204</span><span id="line-1204"> refCnt.retain();</span> |
| <span class="source-line-no">1205</span><span id="line-1205"> return this;</span> |
| <span class="source-line-no">1206</span><span id="line-1206"> }</span> |
| <span class="source-line-no">1207</span><span id="line-1207">}</span> |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| </pre> |
| </div> |
| </main> |
| </body> |
| </html> |
