streamwritefilter.html

MINA+API 关于MINA的API

<a name="105" href="#105">105</a> 
<a name="106" href="#106">106</a>             <a href="../../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> buffer = getNextBuffer(inputStream);
<a name="107" href="#107">107</a>             <strong class="jxr_keyword">if</strong> (buffer == <strong class="jxr_keyword">null</strong>) {
<a name="108" href="#108">108</a>                 <em class="jxr_comment">// End of stream reached.</em>
<a name="109" href="#109">109</a>                 writeRequest.getFuture().setWritten();
<a name="110" href="#110">110</a>                 nextFilter.messageSent(session, writeRequest);
<a name="111" href="#111">111</a>             } <strong class="jxr_keyword">else</strong> {
<a name="112" href="#112">112</a>                 session.setAttribute(CURRENT_STREAM, inputStream);
<a name="113" href="#113">113</a>                 session.setAttribute(CURRENT_WRITE_REQUEST, writeRequest);
<a name="114" href="#114">114</a> 
<a name="115" href="#115">115</a>                 nextFilter.filterWrite(session, <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/common/DefaultWriteRequest.html">DefaultWriteRequest</a>(
<a name="116" href="#116">116</a>                         buffer));
<a name="117" href="#117">117</a>             }
<a name="118" href="#118">118</a> 
<a name="119" href="#119">119</a>         } <strong class="jxr_keyword">else</strong> {
<a name="120" href="#120">120</a>             nextFilter.filterWrite(session, writeRequest);
<a name="121" href="#121">121</a>         }
<a name="122" href="#122">122</a>     }
<a name="123" href="#123">123</a> 
<a name="124" href="#124">124</a>     @SuppressWarnings(<span class="jxr_string">"unchecked"</span>)
<a name="125" href="#125">125</a>     <strong class="jxr_keyword">private</strong> Queue&lt;WriteRequest&gt; getWriteRequestQueue(<a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session) {
<a name="126" href="#126">126</a>         <strong class="jxr_keyword">return</strong> (Queue&lt;WriteRequest&gt;) session.getAttribute(WRITE_REQUEST_QUEUE);
<a name="127" href="#127">127</a>     }
<a name="128" href="#128">128</a> 
<a name="129" href="#129">129</a>     @SuppressWarnings(<span class="jxr_string">"unchecked"</span>)
<a name="130" href="#130">130</a>     <strong class="jxr_keyword">private</strong> Queue&lt;WriteRequest&gt; removeWriteRequestQueue(<a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session) {
<a name="131" href="#131">131</a>         <strong class="jxr_keyword">return</strong> (Queue&lt;WriteRequest&gt;) session.removeAttribute(WRITE_REQUEST_QUEUE);
<a name="132" href="#132">132</a>     }
<a name="133" href="#133">133</a>     @Override
<a name="134" href="#134">134</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> messageSent(<a href="../../../../../org/apache/mina/common/IoFilter.html">NextFilter</a> nextFilter, <a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session,
<a name="135" href="#135">135</a>                             <a href="../../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> writeRequest) <strong class="jxr_keyword">throws</strong> Exception {
<a name="136" href="#136">136</a>         InputStream inputStream = (InputStream) session
<a name="137" href="#137">137</a>                 .getAttribute(CURRENT_STREAM);
<a name="138" href="#138">138</a> 
<a name="139" href="#139">139</a>         <strong class="jxr_keyword">if</strong> (inputStream == <strong class="jxr_keyword">null</strong>) {
<a name="140" href="#140">140</a>             nextFilter.messageSent(session, writeRequest);
<a name="141" href="#141">141</a>         } <strong class="jxr_keyword">else</strong> {
<a name="142" href="#142">142</a>             <a href="../../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> buffer = getNextBuffer(inputStream);
<a name="143" href="#143">143</a> 
<a name="144" href="#144">144</a>             <strong class="jxr_keyword">if</strong> (buffer == <strong class="jxr_keyword">null</strong>) {
<a name="145" href="#145">145</a>                 <em class="jxr_comment">// End of stream reached.</em>
<a name="146" href="#146">146</a>                 session.removeAttribute(CURRENT_STREAM);
<a name="147" href="#147">147</a>                 <a href="../../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> currentWriteRequest = (WriteRequest) session
<a name="148" href="#148">148</a>                         .removeAttribute(CURRENT_WRITE_REQUEST);
<a name="149" href="#149">149</a> 
<a name="150" href="#150">150</a>                 <em class="jxr_comment">// Write queued WriteRequests.</em>
<a name="151" href="#151">151</a>                 Queue&lt;WriteRequest&gt; queue = removeWriteRequestQueue(session);
<a name="152" href="#152">152</a>                 <strong class="jxr_keyword">if</strong> (queue != <strong class="jxr_keyword">null</strong>) {
<a name="153" href="#153">153</a>                     <a href="../../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> wr = queue.poll();
<a name="154" href="#154">154</a>                     <strong class="jxr_keyword">while</strong> (wr != <strong class="jxr_keyword">null</strong>) {
<a name="155" href="#155">155</a>                         filterWrite(nextFilter, session, wr);
<a name="156" href="#156">156</a>                         wr = queue.poll();
<a name="157" href="#157">157</a>                     }
<a name="158" href="#158">158</a>                 }
<a name="159" href="#159">159</a> 
<a name="160" href="#160">160</a>                 currentWriteRequest.getFuture().setWritten();
<a name="161" href="#161">161</a>                 nextFilter.messageSent(session, currentWriteRequest);
<a name="162" href="#162">162</a>             } <strong class="jxr_keyword">else</strong> {
<a name="163" href="#163">163</a>                 nextFilter.filterWrite(session, <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/common/DefaultWriteRequest.html">DefaultWriteRequest</a>(
<a name="164" href="#164">164</a>                         buffer));
<a name="165" href="#165">165</a>             }
<a name="166" href="#166">166</a>         }
<a name="167" href="#167">167</a>     }
<a name="168" href="#168">168</a> 
<a name="169" href="#169">169</a>     <strong class="jxr_keyword">private</strong> <a href="../../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> getNextBuffer(InputStream is) <strong class="jxr_keyword">throws</strong> IOException {
<a name="170" href="#170">170</a>         byte[] bytes = <strong class="jxr_keyword">new</strong> byte[writeBufferSize];
<a name="171" href="#171">171</a> 
<a name="172" href="#172">172</a>         <strong class="jxr_keyword">int</strong> off = 0;
<a name="173" href="#173">173</a>         <strong class="jxr_keyword">int</strong> n = 0;
<a name="174" href="#174">174</a>         <strong class="jxr_keyword">while</strong> (off &lt; bytes.length
<a name="175" href="#175">175</a>                 &amp;&amp; (n = is.read(bytes, off, bytes.length - off)) != -1) {
<a name="176" href="#176">176</a>             off += n;
<a name="177" href="#177">177</a>         }
<a name="178" href="#178">178</a> 
<a name="179" href="#179">179</a>         <strong class="jxr_keyword">if</strong> (n == -1 &amp;&amp; off == 0) {
<a name="180" href="#180">180</a>             <strong class="jxr_keyword">return</strong> <strong class="jxr_keyword">null</strong>;
<a name="181" href="#181">181</a>         }
<a name="182" href="#182">182</a> 
<a name="183" href="#183">183</a>         <a href="../../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> buffer = IoBuffer.wrap(bytes, 0, off);
<a name="184" href="#184">184</a> 
<a name="185" href="#185">185</a>         <strong class="jxr_keyword">return</strong> buffer;
<a name="186" href="#186">186</a>     }
<a name="187" href="#187">187</a> 
<a name="188" href="#188">188</a>     <em class="jxr_javadoccomment">/**</em>
<a name="189" href="#189">189</a> <em class="jxr_javadoccomment">     * Returns the size of the write buffer in bytes. Data will be read from the</em>
<a name="190" href="#190">190</a> <em class="jxr_javadoccomment">     * stream in chunks of this size and then written to the next filter.</em>
<a name="191" href="#191">191</a> <em class="jxr_javadoccomment">     *</em>
<a name="192" href="#192">192</a> <em class="jxr_javadoccomment">     * @return the write buffer size.</em>
<a name="193" href="#193">193</a> <em class="jxr_javadoccomment">     */</em>
<a name="194" href="#194">194</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">int</strong> getWriteBufferSize() {
<a name="195" href="#195">195</a>         <strong class="jxr_keyword">return</strong> writeBufferSize;
<a name="196" href="#196">196</a>     }
<a name="197" href="#197">197</a> 
<a name="198" href="#198">198</a>     <em class="jxr_javadoccomment">/**</em>
<a name="199" href="#199">199</a> <em class="jxr_javadoccomment">     * Sets the size of the write buffer in bytes. Data will be read from the</em>
<a name="200" href="#200">200</a> <em class="jxr_javadoccomment">     * stream in chunks of this size and then written to the next filter.</em>
<a name="201" href="#201">201</a> <em class="jxr_javadoccomment">     *</em>
<a name="202" href="#202">202</a> <em class="jxr_javadoccomment">     * @throws IllegalArgumentException if the specified size is &amp;lt; 1.</em>
<a name="203" href="#203">203</a> <em class="jxr_javadoccomment">     */</em>
<a name="204" href="#204">204</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> setWriteBufferSize(<strong class="jxr_keyword">int</strong> writeBufferSize) {
<a name="205" href="#205">205</a>         <strong class="jxr_keyword">if</strong> (writeBufferSize &lt; 1) {
<a name="206" href="#206">206</a>             <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalArgumentException(
<a name="207" href="#207">207</a>                     <span class="jxr_string">"writeBufferSize must be at least 1"</span>);
<a name="208" href="#208">208</a>         }
<a name="209" href="#209">209</a>         <strong class="jxr_keyword">this</strong>.writeBufferSize = writeBufferSize;
<a name="210" href="#210">210</a>     }
<a name="211" href="#211">211</a> 
<a name="212" href="#212">212</a> }
</pre>
<hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>
</html>