ioeventqueuethrottle.html

MINA+API 关于MINA的API

<a name="76" href="#76">76</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> setThreshold(<strong class="jxr_keyword">int</strong> threshold) {
<a name="77" href="#77">77</a>          <strong class="jxr_keyword">if</strong> (threshold &lt;= 0) {
<a name="78" href="#78">78</a>              <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalArgumentException(<span class="jxr_string">"threshold: "</span> + threshold);
<a name="79" href="#79">79</a>          }
<a name="80" href="#80">80</a>          <strong class="jxr_keyword">this</strong>.threshold = threshold;
<a name="81" href="#81">81</a>      }
<a name="82" href="#82">82</a>  
<a name="83" href="#83">83</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">boolean</strong> accept(ThreadPoolExecutor executor, <a href="../../../../../org/apache/mina/common/IoEvent.html">IoEvent</a> event) {
<a name="84" href="#84">84</a>          <strong class="jxr_keyword">return</strong> <strong class="jxr_keyword">true</strong>;
<a name="85" href="#85">85</a>      }
<a name="86" href="#86">86</a>      
<a name="87" href="#87">87</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> offered(ThreadPoolExecutor executor, <a href="../../../../../org/apache/mina/common/IoEvent.html">IoEvent</a> event) {
<a name="88" href="#88">88</a>          <strong class="jxr_keyword">int</strong> eventSize = estimateSize(event);
<a name="89" href="#89">89</a>          <strong class="jxr_keyword">int</strong> currentCounter = counter.addAndGet(eventSize);
<a name="90" href="#90">90</a>          logState();
<a name="91" href="#91">91</a>  
<a name="92" href="#92">92</a>          <strong class="jxr_keyword">if</strong> (currentCounter &gt;= threshold) {
<a name="93" href="#93">93</a>              block();
<a name="94" href="#94">94</a>          }
<a name="95" href="#95">95</a>      }
<a name="96" href="#96">96</a>  
<a name="97" href="#97">97</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> polled(ThreadPoolExecutor executor, <a href="../../../../../org/apache/mina/common/IoEvent.html">IoEvent</a> event) {
<a name="98" href="#98">98</a>          <strong class="jxr_keyword">int</strong> eventSize = estimateSize(event);
<a name="99" href="#99">99</a>          <strong class="jxr_keyword">int</strong> currentCounter = counter.addAndGet(-eventSize);
<a name="100" href="#100">100</a> 
<a name="101" href="#101">101</a>         logState();
<a name="102" href="#102">102</a> 
<a name="103" href="#103">103</a>         <strong class="jxr_keyword">if</strong> (currentCounter &lt; threshold) {
<a name="104" href="#104">104</a>             unblock();
<a name="105" href="#105">105</a>         }
<a name="106" href="#106">106</a>     }
<a name="107" href="#107">107</a> 
<a name="108" href="#108">108</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">int</strong> estimateSize(<a href="../../../../../org/apache/mina/common/IoEvent.html">IoEvent</a> event) {
<a name="109" href="#109">109</a>         <strong class="jxr_keyword">int</strong> size = getEventSizeEstimator().estimateSize(event);
<a name="110" href="#110">110</a>         <strong class="jxr_keyword">if</strong> (size &lt; 0) {
<a name="111" href="#111">111</a>             <strong class="jxr_keyword">throw</strong> <strong class="jxr_keyword">new</strong> IllegalStateException(
<a name="112" href="#112">112</a>                     IoEventSizeEstimator.<strong class="jxr_keyword">class</strong>.getSimpleName() + <span class="jxr_string">" returned "</span> +
<a name="113" href="#113">113</a>                     <span class="jxr_string">"a negative value ("</span> + size + <span class="jxr_string">"): "</span> + event);
<a name="114" href="#114">114</a>         }
<a name="115" href="#115">115</a>         <strong class="jxr_keyword">return</strong> size;
<a name="116" href="#116">116</a>     }
<a name="117" href="#117">117</a>     
<a name="118" href="#118">118</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> logState() {
<a name="119" href="#119">119</a>         <strong class="jxr_keyword">if</strong> (logger.isDebugEnabled()) {
<a name="120" href="#120">120</a>             logger.debug(Thread.currentThread().getName() + <span class="jxr_string">" state: "</span> + counter.get() + <span class="jxr_string">" / "</span> + getThreshold());
<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>     <strong class="jxr_keyword">protected</strong> <strong class="jxr_keyword">void</strong> block() {
<a name="125" href="#125">125</a>         <strong class="jxr_keyword">if</strong> (logger.isDebugEnabled()) {
<a name="126" href="#126">126</a>             logger.debug(Thread.currentThread().getName() + <span class="jxr_string">" blocked: "</span> + counter.get() + <span class="jxr_string">" &gt;= "</span> + threshold);
<a name="127" href="#127">127</a>         }
<a name="128" href="#128">128</a>         
<a name="129" href="#129">129</a>         <strong class="jxr_keyword">synchronized</strong> (lock) {
<a name="130" href="#130">130</a>             <strong class="jxr_keyword">while</strong> (counter.get() &gt;= threshold) {
<a name="131" href="#131">131</a>                 waiters ++;
<a name="132" href="#132">132</a>                 <strong class="jxr_keyword">try</strong> {
<a name="133" href="#133">133</a>                     lock.wait();
<a name="134" href="#134">134</a>                 } <strong class="jxr_keyword">catch</strong> (InterruptedException e) {
<a name="135" href="#135">135</a>                     <em class="jxr_comment">// Wait uninterruptably.</em>
<a name="136" href="#136">136</a>                 } <strong class="jxr_keyword">finally</strong> {
<a name="137" href="#137">137</a>                     waiters --;
<a name="138" href="#138">138</a>                 }
<a name="139" href="#139">139</a>             }
<a name="140" href="#140">140</a>         }
<a name="141" href="#141">141</a>         
<a name="142" href="#142">142</a>         <strong class="jxr_keyword">if</strong> (logger.isDebugEnabled()) {
<a name="143" href="#143">143</a>             logger.debug(Thread.currentThread().getName() + <span class="jxr_string">" unblocked: "</span> + counter.get() + <span class="jxr_string">" &lt; "</span> + threshold);
<a name="144" href="#144">144</a>         }
<a name="145" href="#145">145</a>     }
<a name="146" href="#146">146</a>     
<a name="147" href="#147">147</a>     <strong class="jxr_keyword">protected</strong> <strong class="jxr_keyword">void</strong> unblock() {
<a name="148" href="#148">148</a>         <strong class="jxr_keyword">synchronized</strong> (lock) {
<a name="149" href="#149">149</a>             <strong class="jxr_keyword">if</strong> (waiters &gt; 0) {
<a name="150" href="#150">150</a>                 lock.notify();
<a name="151" href="#151">151</a>             }
<a name="152" href="#152">152</a>         }
<a name="153" href="#153">153</a>     }
<a name="154" href="#154">154</a> }
</pre>
<hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>
</html>