writethrottlefilter.html

MINA+API 关于MINA的API

<a name="227" href="#227">227</a>         <strong class="jxr_keyword">return</strong> maxServiceScheduledWriteBytes;
<a name="228" href="#228">228</a>     }
<a name="229" href="#229">229</a> 
<a name="230" href="#230">230</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> setMaxServiceScheduledWriteBytes(<strong class="jxr_keyword">long</strong> maxServiceScheduledWriteBytes) {
<a name="231" href="#231">231</a>         <strong class="jxr_keyword">if</strong> (maxServiceScheduledWriteBytes &lt; 0) {
<a name="232" href="#232">232</a>             maxServiceScheduledWriteBytes = 0;
<a name="233" href="#233">233</a>         }
<a name="234" href="#234">234</a>         <strong class="jxr_keyword">this</strong>.maxServiceScheduledWriteBytes = maxServiceScheduledWriteBytes;
<a name="235" href="#235">235</a>     }
<a name="236" href="#236">236</a> 
<a name="237" href="#237">237</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">int</strong> getMaxGlobalScheduledWriteMessages() {
<a name="238" href="#238">238</a>         <strong class="jxr_keyword">return</strong> maxGlobalScheduledWriteMessages;
<a name="239" href="#239">239</a>     }
<a name="240" href="#240">240</a> 
<a name="241" href="#241">241</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> setMaxGlobalScheduledWriteMessages(<strong class="jxr_keyword">int</strong> maxGlobalScheduledWriteMessages) {
<a name="242" href="#242">242</a>         <strong class="jxr_keyword">if</strong> (maxGlobalScheduledWriteMessages &lt; 0) {
<a name="243" href="#243">243</a>             maxGlobalScheduledWriteMessages = 0;
<a name="244" href="#244">244</a>         }
<a name="245" href="#245">245</a>         <strong class="jxr_keyword">this</strong>.maxGlobalScheduledWriteMessages = maxGlobalScheduledWriteMessages;
<a name="246" href="#246">246</a>     }
<a name="247" href="#247">247</a> 
<a name="248" href="#248">248</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">long</strong> getMaxGlobalScheduledWriteBytes() {
<a name="249" href="#249">249</a>         <strong class="jxr_keyword">return</strong> maxGlobalScheduledWriteBytes;
<a name="250" href="#250">250</a>     }
<a name="251" href="#251">251</a> 
<a name="252" href="#252">252</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> setMaxGlobalScheduledWriteBytes(<strong class="jxr_keyword">long</strong> maxGlobalScheduledWriteBytes) {
<a name="253" href="#253">253</a>         <strong class="jxr_keyword">if</strong> (maxGlobalScheduledWriteBytes &lt; 0) {
<a name="254" href="#254">254</a>             maxGlobalScheduledWriteBytes = 0;
<a name="255" href="#255">255</a>         }
<a name="256" href="#256">256</a>         <strong class="jxr_keyword">this</strong>.maxGlobalScheduledWriteBytes = maxGlobalScheduledWriteBytes;
<a name="257" href="#257">257</a>     }
<a name="258" href="#258">258</a> 
<a name="259" href="#259">259</a>     @Override
<a name="260" href="#260">260</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> onPreAdd(
<a name="261" href="#261">261</a>             <a href="../../../../../org/apache/mina/common/IoFilterChain.html">IoFilterChain</a> parent, String name, <a href="../../../../../org/apache/mina/common/IoFilter.html">NextFilter</a> nextFilter) <strong class="jxr_keyword">throws</strong> Exception {
<a name="262" href="#262">262</a>         <strong class="jxr_keyword">if</strong> (parent.contains(WriteThrottleFilter.<strong class="jxr_keyword">class</strong>)) {
<a name="263" href="#263">263</a>             <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalStateException(
<a name="264" href="#264">264</a>                     <span class="jxr_string">"Only one "</span> + WriteThrottleFilter.<strong class="jxr_keyword">class</strong>.getName() + <span class="jxr_string">" is allowed per chain."</span>);
<a name="265" href="#265">265</a>         }
<a name="266" href="#266">266</a>     }
<a name="267" href="#267">267</a> 
<a name="268" href="#268">268</a>     @Override
<a name="269" href="#269">269</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> filterWrite(<a href="../../../../../org/apache/mina/common/IoFilter.html">NextFilter</a> nextFilter, <a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session,
<a name="270" href="#270">270</a>             <a href="../../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> writeRequest) <strong class="jxr_keyword">throws</strong> Exception {
<a name="271" href="#271">271</a>         
<a name="272" href="#272">272</a>         <a href="../../../../../org/apache/mina/filter/traffic/WriteThrottlePolicy.html">WriteThrottlePolicy</a> policy = getPolicy();
<a name="273" href="#273">273</a>         <strong class="jxr_keyword">if</strong> (policy != WriteThrottlePolicy.OFF) {
<a name="274" href="#274">274</a>             <strong class="jxr_keyword">if</strong> (!readyToWrite(session)) {
<a name="275" href="#275">275</a>                 <strong class="jxr_keyword">switch</strong> (policy) {
<a name="276" href="#276">276</a>                 <strong class="jxr_keyword">case</strong> FAIL:
<a name="277" href="#277">277</a>                     log(session);
<a name="278" href="#278">278</a>                     fail(session, writeRequest);
<a name="279" href="#279">279</a>                     <strong class="jxr_keyword">break</strong>;
<a name="280" href="#280">280</a>                 <strong class="jxr_keyword">case</strong> BLOCK:
<a name="281" href="#281">281</a>                     log(session);
<a name="282" href="#282">282</a>                     block(session);
<a name="283" href="#283">283</a>                     <strong class="jxr_keyword">break</strong>;
<a name="284" href="#284">284</a>                 <strong class="jxr_keyword">case</strong> LOG:
<a name="285" href="#285">285</a>                     log(session);
<a name="286" href="#286">286</a>                     <strong class="jxr_keyword">break</strong>;
<a name="287" href="#287">287</a>                 }
<a name="288" href="#288">288</a>             }
<a name="289" href="#289">289</a>         }
<a name="290" href="#290">290</a>         
<a name="291" href="#291">291</a>         nextFilter.filterWrite(session, writeRequest);
<a name="292" href="#292">292</a>     }
<a name="293" href="#293">293</a>     
<a name="294" href="#294">294</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">boolean</strong> readyToWrite(<a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session) {
<a name="295" href="#295">295</a>         <strong class="jxr_keyword">if</strong> (session.isClosing()) {
<a name="296" href="#296">296</a>             <strong class="jxr_keyword">return</strong> <strong class="jxr_keyword">true</strong>;
<a name="297" href="#297">297</a>         }
<a name="298" href="#298">298</a> 
<a name="299" href="#299">299</a>         <strong class="jxr_keyword">int</strong>  mSession = maxSessionScheduledWriteMessages;
<a name="300" href="#300">300</a>         <strong class="jxr_keyword">long</strong> bSession = maxSessionScheduledWriteBytes;
<a name="301" href="#301">301</a>         <strong class="jxr_keyword">int</strong>  mService = maxServiceScheduledWriteMessages;
<a name="302" href="#302">302</a>         <strong class="jxr_keyword">long</strong> bService = maxServiceScheduledWriteBytes;
<a name="303" href="#303">303</a>         <strong class="jxr_keyword">int</strong>  mGlobal  = maxGlobalScheduledWriteMessages;
<a name="304" href="#304">304</a>         <strong class="jxr_keyword">long</strong> bGlobal  = maxGlobalScheduledWriteBytes;
<a name="305" href="#305">305</a>         
<a name="306" href="#306">306</a>         <strong class="jxr_keyword">return</strong> (mSession == 0 || session.getScheduledWriteMessages() &lt; mSession) &amp;&amp;
<a name="307" href="#307">307</a>                (bSession == 0 || session.getScheduledWriteBytes() &lt; bSession) &amp;&amp;
<a name="308" href="#308">308</a>                (mService == 0 || session.getService().getScheduledWriteMessages() &lt; mService) &amp;&amp;
<a name="309" href="#309">309</a>                (bService == 0 || session.getService().getScheduledWriteBytes() &lt; bService) &amp;&amp;
<a name="310" href="#310">310</a>                (mGlobal  == 0 || getGlobalScheduledWriteMessages(session.getService()) &lt; mGlobal) &amp;&amp;
<a name="311" href="#311">311</a>                (bGlobal  == 0 || getGlobalScheduledWriteBytes(session.getService()) &lt; bGlobal);
<a name="312" href="#312">312</a>     }
<a name="313" href="#313">313</a>     
<a name="314" href="#314">314</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> log(<a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session) {
<a name="315" href="#315">315</a>         <strong class="jxr_keyword">long</strong> currentTime = System.currentTimeMillis();
<a name="316" href="#316">316</a>         
<a name="317" href="#317">317</a>         <em class="jxr_comment">// Prevent log flood by logging every 3 seconds.</em>
<a name="318" href="#318">318</a>         <strong class="jxr_keyword">boolean</strong> log;
<a name="319" href="#319">319</a>         <strong class="jxr_keyword">synchronized</strong> (logLock) {
<a name="320" href="#320">320</a>             <strong class="jxr_keyword">if</strong> (currentTime - lastLogTime &gt; 3000) {
<a name="321" href="#321">321</a>                 lastLogTime = currentTime;
<a name="322" href="#322">322</a>                 log = <strong class="jxr_keyword">true</strong>;
<a name="323" href="#323">323</a>             } <strong class="jxr_keyword">else</strong> {
<a name="324" href="#324">324</a>                 log = false;
<a name="325" href="#325">325</a>             }
<a name="326" href="#326">326</a>         }
<a name="327" href="#327">327</a>         
<a name="328" href="#328">328</a>         <strong class="jxr_keyword">if</strong> (log) {
<a name="329" href="#329">329</a>             logger.warn(getMessage(session));
<a name="330" href="#330">330</a>         }
<a name="331" href="#331">331</a>     }
<a name="332" href="#332">332</a>     
<a name="333" href="#333">333</a>     @Override
<a name="334" href="#334">334</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> messageSent(
<a name="335" href="#335">335</a>             <a href="../../../../../org/apache/mina/common/IoFilter.html">NextFilter</a> nextFilter, <a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session, <a href="../../../../../org/apache/mina/common/WriteRequest.html">WriteRequest</a> writeRequest) <strong class="jxr_keyword">throws</strong> Exception {
<a name="336" href="#336">336</a>         notifyWaitingWriters();
<a name="337" href="#337">337</a>         nextFilter.messageSent(session, writeRequest);
<a name="338" href="#338">338</a>     }
<a name="339" href="#339">339</a> 
<a name="340" href="#340">340</a>     @Override
<a name="341" href="#341">341</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> exceptionCaught(<a href="../../../../../org/apache/mina/common/IoFilter.html">NextFilter</a> nextFilter, <a href="../../../../../org/apache/mina/common/IoSession.html">IoSession</a> session,
<a name="342" href="#342">342</a>             Throwable cause) <strong class="jxr_keyword">throws</strong> Exception {
<a name="343" href="#343">343</a>         <strong class="jxr_keyword">try</strong> {
<a name="344" href="#344">344</a>             nextFilter.exceptionCaught(session, cause);