abstractpollingioprocessor.html

MINA+API 关于MINA的API

<a name="527" href="#527">527</a>                 <strong class="jxr_keyword">if</strong> (message instanceof IoBuffer) {
<a name="528" href="#528">528</a>                     localWrittenBytes = writeBuffer(
<a name="529" href="#529">529</a>                             session, req, hasFragmentation,
<a name="530" href="#530">530</a>                             maxWrittenBytes - writtenBytes);
<a name="531" href="#531">531</a>                 } <strong class="jxr_keyword">else</strong> <strong class="jxr_keyword">if</strong> (message instanceof FileRegion) {
<a name="532" href="#532">532</a>                     localWrittenBytes = writeFile(
<a name="533" href="#533">533</a>                             session, req, hasFragmentation,
<a name="534" href="#534">534</a>                             maxWrittenBytes - writtenBytes);
<a name="535" href="#535">535</a>                 } <strong class="jxr_keyword">else</strong> {
<a name="536" href="#536">536</a>                 	<strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalStateException(<span class="jxr_string">"Don't know how to handle message of type '"</span> + message.getClass().getName() + <span class="jxr_string">"'.  Are you missing a protocol encoder?"</span>);
<a name="537" href="#537">537</a>                 }
<a name="538" href="#538">538</a>                 
<a name="539" href="#539">539</a>                 writtenBytes += localWrittenBytes;
<a name="540" href="#540">540</a>     
<a name="541" href="#541">541</a>                 <strong class="jxr_keyword">if</strong> (localWrittenBytes == 0 || writtenBytes &gt;= maxWrittenBytes) {
<a name="542" href="#542">542</a>                     <em class="jxr_comment">// Kernel buffer is full or wrote too much.</em>
<a name="543" href="#543">543</a>                     setInterestedInWrite(session, <strong class="jxr_keyword">true</strong>);
<a name="544" href="#544">544</a>                     <strong class="jxr_keyword">return</strong> false;
<a name="545" href="#545">545</a>                 }
<a name="546" href="#546">546</a>             } <strong class="jxr_keyword">while</strong> (writtenBytes &lt; maxWrittenBytes);
<a name="547" href="#547">547</a>         } <strong class="jxr_keyword">catch</strong> (Exception e) {
<a name="548" href="#548">548</a>             session.getFilterChain().fireExceptionCaught(e);
<a name="549" href="#549">549</a>             <strong class="jxr_keyword">return</strong> false;
<a name="550" href="#550">550</a>         }
<a name="551" href="#551">551</a> 
<a name="552" href="#552">552</a>         <strong class="jxr_keyword">return</strong> <strong class="jxr_keyword">true</strong>;
<a name="553" href="#553">553</a>     }
<a name="554" href="#554">554</a> 
<a name="555" href="#555">555</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">int</strong> writeBuffer(T session, <a href="../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> req,
<a name="556" href="#556">556</a>             <strong class="jxr_keyword">boolean</strong> hasFragmentation, <strong class="jxr_keyword">int</strong> maxLength) <strong class="jxr_keyword">throws</strong> Exception {
<a name="557" href="#557">557</a>         <a href="../../../../org/apache/mina/common/IoBuffer.html">IoBuffer</a> buf = (IoBuffer) req.getMessage();
<a name="558" href="#558">558</a>         <strong class="jxr_keyword">int</strong> localWrittenBytes = 0;
<a name="559" href="#559">559</a>         <strong class="jxr_keyword">if</strong> (buf.hasRemaining()) {
<a name="560" href="#560">560</a>             <strong class="jxr_keyword">int</strong> length;
<a name="561" href="#561">561</a>             <strong class="jxr_keyword">if</strong> (hasFragmentation) {
<a name="562" href="#562">562</a>                 length = Math.min(buf.remaining(), maxLength);
<a name="563" href="#563">563</a>             } <strong class="jxr_keyword">else</strong> {
<a name="564" href="#564">564</a>                 length = buf.remaining();
<a name="565" href="#565">565</a>             }
<a name="566" href="#566">566</a>             <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = WRITE_SPIN_COUNT; i &gt; 0; i --) {
<a name="567" href="#567">567</a>                 localWrittenBytes = write(session, buf, length);
<a name="568" href="#568">568</a>                 <strong class="jxr_keyword">if</strong> (localWrittenBytes != 0) {
<a name="569" href="#569">569</a>                     <strong class="jxr_keyword">break</strong>;
<a name="570" href="#570">570</a>                 }
<a name="571" href="#571">571</a>             }
<a name="572" href="#572">572</a>         }
<a name="573" href="#573">573</a> 
<a name="574" href="#574">574</a>         <strong class="jxr_keyword">if</strong> (!buf.hasRemaining() ||
<a name="575" href="#575">575</a>                 (!hasFragmentation &amp;&amp; localWrittenBytes != 0)) {
<a name="576" href="#576">576</a>             <em class="jxr_comment">// Buffer has been sent, clear the current request.</em>
<a name="577" href="#577">577</a>             buf.reset();
<a name="578" href="#578">578</a>             fireMessageSent(session, req);
<a name="579" href="#579">579</a>         }
<a name="580" href="#580">580</a>         <strong class="jxr_keyword">return</strong> localWrittenBytes;
<a name="581" href="#581">581</a>     }
<a name="582" href="#582">582</a> 
<a name="583" href="#583">583</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">int</strong> writeFile(T session, <a href="../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> req,
<a name="584" href="#584">584</a>             <strong class="jxr_keyword">boolean</strong> hasFragmentation, <strong class="jxr_keyword">int</strong> maxLength) <strong class="jxr_keyword">throws</strong> Exception {
<a name="585" href="#585">585</a>         <strong class="jxr_keyword">int</strong> localWrittenBytes;
<a name="586" href="#586">586</a>         <a href="../../../../org/apache/mina/common/FileRegion.html">FileRegion</a> region = (FileRegion) req.getMessage();
<a name="587" href="#587">587</a>         <strong class="jxr_keyword">if</strong> (region.getCount() &gt; 0) {
<a name="588" href="#588">588</a>             <strong class="jxr_keyword">int</strong> length;
<a name="589" href="#589">589</a>             <strong class="jxr_keyword">if</strong> (hasFragmentation) {
<a name="590" href="#590">590</a>                 length = (<strong class="jxr_keyword">int</strong>) Math.min(region.getCount(), maxLength);
<a name="591" href="#591">591</a>             } <strong class="jxr_keyword">else</strong> {
<a name="592" href="#592">592</a>                 length = (<strong class="jxr_keyword">int</strong>) Math.min(Integer.MAX_VALUE, region.getCount());
<a name="593" href="#593">593</a>             }
<a name="594" href="#594">594</a>             localWrittenBytes = transferFile(session, region, length);
<a name="595" href="#595">595</a>             region.setPosition(region.getPosition() + localWrittenBytes);
<a name="596" href="#596">596</a>             
<a name="597" href="#597">597</a>             <em class="jxr_comment">// Fix for Java bug on Linux http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=5103988</em>
<a name="598" href="#598">598</a>             <em class="jxr_comment">// If there's still data to be written in the FileRegion, return 0 indicating that we need</em>
<a name="599" href="#599">599</a>             <em class="jxr_comment">// to pause until writing may resume.</em>
<a name="600" href="#600">600</a>             <strong class="jxr_keyword">if</strong> (localWrittenBytes &gt; 0 &amp;&amp; region.getCount() &gt; 0) {
<a name="601" href="#601">601</a>                 <strong class="jxr_keyword">return</strong> 0;
<a name="602" href="#602">602</a>             }
<a name="603" href="#603">603</a>         } <strong class="jxr_keyword">else</strong> {
<a name="604" href="#604">604</a>             localWrittenBytes = 0;
<a name="605" href="#605">605</a>         }
<a name="606" href="#606">606</a> 
<a name="607" href="#607">607</a>         <strong class="jxr_keyword">if</strong> (region.getCount() &lt;= 0 ||
<a name="608" href="#608">608</a>                 (!hasFragmentation &amp;&amp; localWrittenBytes != 0)) {
<a name="609" href="#609">609</a>             fireMessageSent(session, req);
<a name="610" href="#610">610</a>         }
<a name="611" href="#611">611</a>         <strong class="jxr_keyword">return</strong> localWrittenBytes;
<a name="612" href="#612">612</a>     }
<a name="613" href="#613">613</a> 
<a name="614" href="#614">614</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> fireMessageSent(T session, <a href="../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> req) {
<a name="615" href="#615">615</a>         session.setCurrentWriteRequest(<strong class="jxr_keyword">null</strong>);
<a name="616" href="#616">616</a>         session.getFilterChain().fireMessageSent(req);
<a name="617" href="#617">617</a>     }
<a name="618" href="#618">618</a> 
<a name="619" href="#619">619</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> updateTrafficMask() {
<a name="620" href="#620">620</a>         <strong class="jxr_keyword">for</strong> (; ;) {
<a name="621" href="#621">621</a>             T session = trafficControllingSessions.poll();
<a name="622" href="#622">622</a> 
<a name="623" href="#623">623</a>             <strong class="jxr_keyword">if</strong> (session == <strong class="jxr_keyword">null</strong>) {
<a name="624" href="#624">624</a>                 <strong class="jxr_keyword">break</strong>;
<a name="625" href="#625">625</a>             }
<a name="626" href="#626">626</a> 
<a name="627" href="#627">627</a>             <a href="../../../../org/apache/mina/common/AbstractPollingIoProcessor.html">SessionState</a> state = state(session);
<a name="628" href="#628">628</a>             <strong class="jxr_keyword">switch</strong> (state) {
<a name="629" href="#629">629</a>             <strong class="jxr_keyword">case</strong> OPEN:
<a name="630" href="#630">630</a>                 updateTrafficMaskNow(session);
<a name="631" href="#631">631</a>                 <strong class="jxr_keyword">break</strong>;
<a name="632" href="#632">632</a>             <strong class="jxr_keyword">case</strong> CLOSED:
<a name="633" href="#633">633</a>                 <strong class="jxr_keyword">break</strong>;
<a name="634" href="#634">634</a>             <strong class="jxr_keyword">case</strong> PREPARING:
<a name="635" href="#635">635</a>                 <em class="jxr_comment">// Retry later if session is not yet fully initialized.</em>
<a name="636" href="#636">636</a>                 <em class="jxr_comment">// (In case that Session.suspend??() or session.resume??() is</em>
<a name="637" href="#637">637</a>                 <em class="jxr_comment">// called before addSession() is processed)</em>
<a name="638" href="#638">638</a>                 scheduleTrafficControl(session);
<a name="639" href="#639">639</a>                 <strong class="jxr_keyword">return</strong>;
<a name="640" href="#640">640</a>             <strong class="jxr_keyword">default</strong>:
<a name="641" href="#641">641</a>                 <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalStateException(String.valueOf(state));
<a name="642" href="#642">642</a>             }
<a name="643" href="#643">643</a>         }
<a name="644" href="#644">644</a>     }
<a name="645" href="#645">645</a> 
<a name="646" href="#646">646</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> updateTrafficMaskNow(T session) {
<a name="647" href="#647">647</a>         <em class="jxr_comment">// The normal is OP_READ and, if there are write requests in the</em>
<a name="648" href="#648">648</a>         <em class="jxr_comment">// session's write queue, set OP_WRITE to trigger flushing.</em>
<a name="649" href="#649">649</a>         <strong class="jxr_keyword">int</strong> mask = session.getTrafficMask().getInterestOps();
<a name="650" href="#650">650</a>         <strong class="jxr_keyword">try</strong> {
<a name="651" href="#651">651</a>             setInterestedInRead(session, (mask &amp; SelectionKey.OP_READ) != 0);
<a name="652" href="#652">652</a>         } <strong class="jxr_keyword">catch</strong> (Exception e) {
<a name="653" href="#653">653</a>             session.getFilterChain().fireExceptionCaught(e);
<a name="654" href="#654">654</a>         }
<a name="655" href="#655">655</a>         <strong class="jxr_keyword">try</strong> {
<a name="656" href="#656">656</a>             setInterestedInWri