abstractpollingioprocessor.html

MINA+API 关于MINA的API

<a name="393" href="#393">393</a>     }
<a name="394" href="#394">394</a> 
<a name="395" href="#395">395</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> read(T session) {
<a name="396" href="#396">396</a>         <a href="../../../../org/apache/mina/common/IoSessionConfig.html">IoSessionConfig</a> config = session.getConfig();
<a name="397" href="#397">397</a>         <a href="../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> buf = IoBuffer.allocate(config.getReadBufferSize());
<a name="398" href="#398">398</a> 
<a name="399" href="#399">399</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> hasFragmentation =
<a name="400" href="#400">400</a>             session.getTransportMetadata().hasFragmentation();
<a name="401" href="#401">401</a> 
<a name="402" href="#402">402</a>         <strong class="jxr_keyword">try</strong> {
<a name="403" href="#403">403</a>             <strong class="jxr_keyword">int</strong> readBytes = 0;
<a name="404" href="#404">404</a>             <strong class="jxr_keyword">int</strong> ret;
<a name="405" href="#405">405</a> 
<a name="406" href="#406">406</a>             <strong class="jxr_keyword">try</strong> {
<a name="407" href="#407">407</a>                 <strong class="jxr_keyword">if</strong> (hasFragmentation) {
<a name="408" href="#408">408</a>                     <strong class="jxr_keyword">while</strong> ((ret = read(session, buf)) &gt; 0) {
<a name="409" href="#409">409</a>                         readBytes += ret;
<a name="410" href="#410">410</a>                         <strong class="jxr_keyword">if</strong> (!buf.hasRemaining()) {
<a name="411" href="#411">411</a>                             <strong class="jxr_keyword">break</strong>;
<a name="412" href="#412">412</a>                         }
<a name="413" href="#413">413</a>                     }
<a name="414" href="#414">414</a>                 } <strong class="jxr_keyword">else</strong> {
<a name="415" href="#415">415</a>                     ret = read(session, buf);
<a name="416" href="#416">416</a>                     <strong class="jxr_keyword">if</strong> (ret &gt; 0) {
<a name="417" href="#417">417</a>                         readBytes = ret;
<a name="418" href="#418">418</a>                     }
<a name="419" href="#419">419</a>                 }
<a name="420" href="#420">420</a>             } <strong class="jxr_keyword">finally</strong> {
<a name="421" href="#421">421</a>                 buf.flip();
<a name="422" href="#422">422</a>             }
<a name="423" href="#423">423</a> 
<a name="424" href="#424">424</a>             <strong class="jxr_keyword">if</strong> (readBytes &gt; 0) {
<a name="425" href="#425">425</a>                 session.getFilterChain().fireMessageReceived(buf);
<a name="426" href="#426">426</a>                 buf = <strong class="jxr_keyword">null</strong>;
<a name="427" href="#427">427</a> 
<a name="428" href="#428">428</a>                 <strong class="jxr_keyword">if</strong> (hasFragmentation) {
<a name="429" href="#429">429</a>                     <strong class="jxr_keyword">if</strong> (readBytes &lt;&lt; 1 &lt; config.getReadBufferSize()) {
<a name="430" href="#430">430</a>                         session.decreaseReadBufferSize();
<a name="431" href="#431">431</a>                     } <strong class="jxr_keyword">else</strong> <strong class="jxr_keyword">if</strong> (readBytes == config.getReadBufferSize()) {
<a name="432" href="#432">432</a>                         session.increaseReadBufferSize();
<a name="433" href="#433">433</a>                     }
<a name="434" href="#434">434</a>                 }
<a name="435" href="#435">435</a>             }
<a name="436" href="#436">436</a>             <strong class="jxr_keyword">if</strong> (ret &lt; 0) {
<a name="437" href="#437">437</a>                 scheduleRemove(session);
<a name="438" href="#438">438</a>             }
<a name="439" href="#439">439</a>         } <strong class="jxr_keyword">catch</strong> (Throwable e) {
<a name="440" href="#440">440</a>             <strong class="jxr_keyword">if</strong> (e instanceof IOException) {
<a name="441" href="#441">441</a>                 scheduleRemove(session);
<a name="442" href="#442">442</a>             }
<a name="443" href="#443">443</a>             session.getFilterChain().fireExceptionCaught(e);
<a name="444" href="#444">444</a>         }
<a name="445" href="#445">445</a>     }
<a name="446" href="#446">446</a> 
<a name="447" href="#447">447</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> notifyIdleSessions() <strong class="jxr_keyword">throws</strong> Exception {
<a name="448" href="#448">448</a>         <em class="jxr_comment">// process idle sessions</em>
<a name="449" href="#449">449</a>         <strong class="jxr_keyword">long</strong> currentTime = System.currentTimeMillis();
<a name="450" href="#450">450</a>         <strong class="jxr_keyword">if</strong> (currentTime - lastIdleCheckTime &gt;= 1000) {
<a name="451" href="#451">451</a>             lastIdleCheckTime = currentTime;
<a name="452" href="#452">452</a>             IdleStatusChecker.notifyIdleness(allSessions(), currentTime);
<a name="453" href="#453">453</a>         }
<a name="454" href="#454">454</a>     }
<a name="455" href="#455">455</a> 
<a name="456" href="#456">456</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> flush() {
<a name="457" href="#457">457</a>         <strong class="jxr_keyword">for</strong> (; ;) {
<a name="458" href="#458">458</a>             T session = flushingSessions.poll();
<a name="459" href="#459">459</a> 
<a name="460" href="#460">460</a>             <strong class="jxr_keyword">if</strong> (session == <strong class="jxr_keyword">null</strong>) {
<a name="461" href="#461">461</a>                 <strong class="jxr_keyword">break</strong>;
<a name="462" href="#462">462</a>             }
<a name="463" href="#463">463</a> 
<a name="464" href="#464">464</a>             session.setScheduledForFlush(false);
<a name="465" href="#465">465</a>             <a href="../../../../org/apache/mina/common/AbstractPollingIoProcessor.html">SessionState</a> state = state(session);
<a name="466" href="#466">466</a>             <strong class="jxr_keyword">switch</strong> (state) {
<a name="467" href="#467">467</a>             <strong class="jxr_keyword">case</strong> OPEN:
<a name="468" href="#468">468</a>                 <strong class="jxr_keyword">try</strong> {
<a name="469" href="#469">469</a>                     <strong class="jxr_keyword">boolean</strong> flushedAll = flushNow(session);
<a name="470" href="#470">470</a>                     <strong class="jxr_keyword">if</strong> (flushedAll &amp;&amp; !session.getWriteRequestQueue().isEmpty(session) &amp;&amp;
<a name="471" href="#471">471</a>                         !session.isScheduledForFlush()) {
<a name="472" href="#472">472</a>                         scheduleFlush(session);
<a name="473" href="#473">473</a>                     }
<a name="474" href="#474">474</a>                 } <strong class="jxr_keyword">catch</strong> (Exception e) {
<a name="475" href="#475">475</a>                     scheduleRemove(session);
<a name="476" href="#476">476</a>                     session.getFilterChain().fireExceptionCaught(e);
<a name="477" href="#477">477</a>                 }
<a name="478" href="#478">478</a>                 <strong class="jxr_keyword">break</strong>;
<a name="479" href="#479">479</a>             <strong class="jxr_keyword">case</strong> CLOSED:
<a name="480" href="#480">480</a>                 <em class="jxr_comment">// Skip if the channel is already closed.</em>
<a name="481" href="#481">481</a>                 <strong class="jxr_keyword">break</strong>;
<a name="482" href="#482">482</a>             <strong class="jxr_keyword">case</strong> PREPARING:
<a name="483" href="#483">483</a>                 <em class="jxr_comment">// Retry later if session is not yet fully initialized.</em>
<a name="484" href="#484">484</a>                 <em class="jxr_comment">// (In case that Session.write() is called before addSession() is processed)</em>
<a name="485" href="#485">485</a>                 scheduleFlush(session);
<a name="486" href="#486">486</a>                 <strong class="jxr_keyword">return</strong>;
<a name="487" href="#487">487</a>             <strong class="jxr_keyword">default</strong>:
<a name="488" href="#488">488</a>                 <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalStateException(String.valueOf(state));
<a name="489" href="#489">489</a>             }
<a name="490" href="#490">490</a>         }
<a name="491" href="#491">491</a>     }
<a name="492" href="#492">492</a> 
<a name="493" href="#493">493</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">boolean</strong> flushNow(T session) {
<a name="494" href="#494">494</a>         <strong class="jxr_keyword">if</strong> (!session.isConnected()) {
<a name="495" href="#495">495</a>             scheduleRemove(session);
<a name="496" href="#496">496</a>             <strong class="jxr_keyword">return</strong> false;
<a name="497" href="#497">497</a>         }
<a name="498" href="#498">498</a>         
<a name="499" href="#499">499</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> hasFragmentation = 
<a name="500" href="#500">500</a>             session.getTransportMetadata().hasFragmentation();
<a name="501" href="#501">501</a> 
<a name="502" href="#502">502</a>         <strong class="jxr_keyword">try</strong> {
<a name="503" href="#503">503</a>             <em class="jxr_comment">// Clear OP_WRITE</em>
<a name="504" href="#504">504</a>             setInterestedInWrite(session, false);
<a name="505" href="#505">505</a>     
<a name="506" href="#506">506</a>             <a href="../../../../org/apache/mina/common/WriteRequestQueue.html">WriteRequestQueue</a> writeRequestQueue = session.getWriteRequestQueue();
<a name="507" href="#507">507</a>     
<a name="508" href="#508">508</a>             <em class="jxr_comment">// Set limitation for the number of written bytes for read-write</em>
<a name="509" href="#509">509</a>             <em class="jxr_comment">// fairness.  I used maxReadBufferSize * 3 / 2, which yields best</em>
<a name="510" href="#510">510</a>             <em class="jxr_comment">// performance in my experience while not breaking fairness much.</em>
<a name="511" href="#511">511</a>             <strong class="jxr_keyword">int</strong> maxWrittenBytes = session.getConfig().getMaxReadBufferSize() +
<a name="512" href="#512">512</a>                                   (session.getConfig().getMaxReadBufferSize() &gt;&gt;&gt; 1);
<a name="513" href="#513">513</a>             <strong class="jxr_keyword">int</strong> writtenBytes = 0;
<a name="514" href="#514">514</a>             <strong class="jxr_keyword">do</strong> {
<a name="515" href="#515">515</a>                 <em class="jxr_comment">// Check for pending writes.</em>
<a name="516" href="#516">516</a>                 <a href="../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> req = session.getCurrentWriteRequest();
<a name="517" href="#517">517</a>                 <strong class="jxr_keyword">if</strong> (req == <strong class="jxr_keyword">null</strong>) {
<a name="518" href="#518">518</a>                     req = writeRequestQueue.poll(session);
<a name="519" href="#519">519</a>                     <strong class="jxr_keyword">if</strong> (req == <strong class="jxr_keyword">null</strong>) {
<a name="520" href="#520">520</a>                         <strong class="jxr_keyword">break</strong>;
<a name="521" href="#521">521</a>                     }
<a name="522" href="#522">522</a>                     session.setCurrentWriteRequest(req);
<a name="523" href="#523">523</a>                 }
<a name="524" href="#524">524</a>     
<a name="525" href="#525">525</a>                 <strong class="jxr_keyword">int</strong> localWrittenBytes = 0;
<a name="526" href="#526">526</a>                 Object message = req.getMessage();