mdcinjectionfiltertest.html

MINA+API 关于MINA的API

<a name="124" href="#124">124</a>         chain.addLast(<span class="jxr_string">"protocol"</span>, <strong class="jxr_keyword">new</strong> ProtocolCodecFilter(<strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyProtocolCodecFactory</a>()));
<a name="125" href="#125">125</a>         test(chain);
<a name="126" href="#126">126</a>     }
<a name="127" href="#127">127</a> 
<a name="128" href="#128">128</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testMultipleFilters() <strong class="jxr_keyword">throws</strong> IOException, InterruptedException {
<a name="129" href="#129">129</a>         DefaultIoFilterChainBuilder chain = <strong class="jxr_keyword">new</strong> DefaultIoFilterChainBuilder();
<a name="130" href="#130">130</a>         MdcInjectionFilter mdcInjectionFilter = <strong class="jxr_keyword">new</strong> MdcInjectionFilter();
<a name="131" href="#131">131</a>         chain.addLast(<span class="jxr_string">"executor"</span> , <strong class="jxr_keyword">new</strong> ExecutorFilter());
<a name="132" href="#132">132</a>         chain.addLast(<span class="jxr_string">"mdc-injector"</span>, mdcInjectionFilter);
<a name="133" href="#133">133</a>         chain.addLast(<span class="jxr_string">"profiler"</span>, <strong class="jxr_keyword">new</strong> ProfilerTimerFilter());
<a name="134" href="#134">134</a>         chain.addLast(<span class="jxr_string">"dummy"</span>, <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyIoFilter</a>());
<a name="135" href="#135">135</a>         chain.addLast(<span class="jxr_string">"logger"</span>, <strong class="jxr_keyword">new</strong> LoggingFilter());
<a name="136" href="#136">136</a>         chain.addLast(<span class="jxr_string">"protocol"</span>, <strong class="jxr_keyword">new</strong> ProtocolCodecFilter(<strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyProtocolCodecFactory</a>()));
<a name="137" href="#137">137</a>         test(chain);
<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>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testTwoExecutorFilters() <strong class="jxr_keyword">throws</strong> IOException, InterruptedException {
<a name="142" href="#142">142</a>         DefaultIoFilterChainBuilder chain = <strong class="jxr_keyword">new</strong> DefaultIoFilterChainBuilder();
<a name="143" href="#143">143</a>         MdcInjectionFilter mdcInjectionFilter = <strong class="jxr_keyword">new</strong> MdcInjectionFilter();
<a name="144" href="#144">144</a>         chain.addLast(<span class="jxr_string">"executor1"</span> , <strong class="jxr_keyword">new</strong> ExecutorFilter());
<a name="145" href="#145">145</a>         chain.addLast(<span class="jxr_string">"mdc-injector1"</span>, mdcInjectionFilter);
<a name="146" href="#146">146</a>         chain.addLast(<span class="jxr_string">"protocol"</span>, <strong class="jxr_keyword">new</strong> ProtocolCodecFilter(<strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyProtocolCodecFactory</a>()));
<a name="147" href="#147">147</a>         chain.addLast(<span class="jxr_string">"dummy"</span>, <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyIoFilter</a>());
<a name="148" href="#148">148</a>         chain.addLast(<span class="jxr_string">"executor2"</span> , <strong class="jxr_keyword">new</strong> ExecutorFilter());
<a name="149" href="#149">149</a>         chain.addLast(<span class="jxr_string">"mdc-injector2"</span>, mdcInjectionFilter);
<a name="150" href="#150">150</a>         test(chain);
<a name="151" href="#151">151</a>     }
<a name="152" href="#152">152</a> 
<a name="153" href="#153">153</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testOnlyRemoteAddress() <strong class="jxr_keyword">throws</strong> IOException, InterruptedException {
<a name="154" href="#154">154</a>         DefaultIoFilterChainBuilder chain = <strong class="jxr_keyword">new</strong> DefaultIoFilterChainBuilder();
<a name="155" href="#155">155</a>         chain.addFirst(<span class="jxr_string">"mdc-injector"</span>, <strong class="jxr_keyword">new</strong> MdcInjectionFilter(
<a name="156" href="#156">156</a>             MdcInjectionFilter.MdcKey.remoteAddress));
<a name="157" href="#157">157</a>         chain.addLast(<span class="jxr_string">"dummy"</span>, <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyIoFilter</a>());
<a name="158" href="#158">158</a>         chain.addLast(<span class="jxr_string">"protocol"</span>, <strong class="jxr_keyword">new</strong> ProtocolCodecFilter(<strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">DummyProtocolCodecFactory</a>()));
<a name="159" href="#159">159</a>         <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">SimpleIoHandler</a> simpleIoHandler = <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">SimpleIoHandler</a>();
<a name="160" href="#160">160</a>         acceptor.setHandler(simpleIoHandler);
<a name="161" href="#161">161</a>         acceptor.bind(<strong class="jxr_keyword">new</strong> InetSocketAddress(PORT));
<a name="162" href="#162">162</a>         acceptor.setFilterChainBuilder(chain);
<a name="163" href="#163">163</a>         <em class="jxr_comment">// create some clients</em>
<a name="164" href="#164">164</a>         NioSocketConnector connector = <strong class="jxr_keyword">new</strong> NioSocketConnector();
<a name="165" href="#165">165</a>         connector.setHandler(<strong class="jxr_keyword">new</strong> IoHandlerAdapter());
<a name="166" href="#166">166</a>         SocketAddress remoteAddressClients[] = <strong class="jxr_keyword">new</strong> SocketAddress[2];
<a name="167" href="#167">167</a>         remoteAddressClients[0] = connectAndWrite(connector,0);
<a name="168" href="#168">168</a>         remoteAddressClients[1] = connectAndWrite(connector,1);
<a name="169" href="#169">169</a>         <em class="jxr_comment">// wait until Iohandler has received all events</em>
<a name="170" href="#170">170</a>         simpleIoHandler.messageSentLatch.await();
<a name="171" href="#171">171</a>         simpleIoHandler.sessionIdleLatch.await();
<a name="172" href="#172">172</a>         simpleIoHandler.sessionClosedLatch.await();
<a name="173" href="#173">173</a>         <em class="jxr_comment">// make a copy to prevent ConcurrentModificationException</em>
<a name="174" href="#174">174</a>         List&lt;LoggingEvent&gt; events = <strong class="jxr_keyword">new</strong> ArrayList&lt;LoggingEvent&gt;(appender.events);
<a name="175" href="#175">175</a>         <em class="jxr_comment">// verify that all logging events have correct MDC</em>
<a name="176" href="#176">176</a>         <strong class="jxr_keyword">for</strong> (LoggingEvent event : events) {
<a name="177" href="#177">177</a>             <strong class="jxr_keyword">for</strong> (MdcInjectionFilter.MdcKey mdcKey : MdcInjectionFilter.MdcKey.values()) {
<a name="178" href="#178">178</a>               String key = mdcKey.name();
<a name="179" href="#179">179</a>               Object value = event.getMDC(key);
<a name="180" href="#180">180</a>               <strong class="jxr_keyword">if</strong> (mdcKey == MdcInjectionFilter.MdcKey.remoteAddress) {
<a name="181" href="#181">181</a>                   assertNotNull(
<a name="182" href="#182">182</a>                       <span class="jxr_string">"MDC[remoteAddress] not set for ["</span> + event.getMessage() + <span class="jxr_string">"]"</span>, value);
<a name="183" href="#183">183</a>               } <strong class="jxr_keyword">else</strong> {
<a name="184" href="#184">184</a>                   assertNull(<span class="jxr_string">"MDC["</span> + key + <span class="jxr_string">"] set for ["</span> + event.getMessage() + <span class="jxr_string">"]"</span>, value);
<a name="185" href="#185">185</a>               }
<a name="186" href="#186">186</a>             }
<a name="187" href="#187">187</a>         }
<a name="188" href="#188">188</a>     }
<a name="189" href="#189">189</a> 
<a name="190" href="#190">190</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> test(DefaultIoFilterChainBuilder chain) <strong class="jxr_keyword">throws</strong> IOException, InterruptedException {
<a name="191" href="#191">191</a>         <em class="jxr_comment">// configure the server</em>
<a name="192" href="#192">192</a>         <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">SimpleIoHandler</a> simpleIoHandler = <strong class="jxr_keyword">new</strong> <a href="../../../../../org/apache/mina/filter/logging/MdcInjectionFilterTest.html">SimpleIoHandler</a>();
<a name="193" href="#193">193</a>         acceptor.setHandler(simpleIoHandler);
<a name="194" href="#194">194</a>         acceptor.bind(<strong class="jxr_keyword">new</strong> InetSocketAddress(PORT));
<a name="195" href="#195">195</a>         acceptor.setFilterChainBuilder(chain);
<a name="196" href="#196">196</a>         <em class="jxr_comment">// create some clients</em>
<a name="197" href="#197">197</a>         NioSocketConnector connector = <strong class="jxr_keyword">new</strong> NioSocketConnector();
<a name="198" href="#198">198</a>         connector.setHandler(<strong class="jxr_keyword">new</strong> IoHandlerAdapter());
<a name="199" href="#199">199</a>         SocketAddress remoteAddressClients[] = <strong class="jxr_keyword">new</strong> SocketAddress[2];
<a name="200" href="#200">200</a>         remoteAddressClients[0] = connectAndWrite(connector,0);
<a name="201" href="#201">201</a>         remoteAddressClients[1] = connectAndWrite(connector,1);
<a name="202" href="#202">202</a>         <em class="jxr_comment">// wait until Iohandler has received all events</em>
<a name="203" href="#203">203</a>         simpleIoHandler.messageSentLatch.await();
<a name="204" href="#204">204</a>         simpleIoHandler.sessionIdleLatch.await();
<a name="205" href="#205">205</a>         simpleIoHandler.sessionClosedLatch.await();
<a name="206" href="#206">206</a> 
<a name="207" href="#207">207</a>         <em class="jxr_comment">// make a copy to prevent ConcurrentModificationException</em>
<a name="208" href="#208">208</a>         List&lt;LoggingEvent&gt; events = <strong class="jxr_keyword">new</strong> ArrayList&lt;LoggingEvent&gt;(appender.events);
<a name="209" href="#209">209</a> 
<a name="210" href="#210">210</a>         <em class="jxr_comment">// verify that all logging events have correct MDC</em>
<a name="211" href="#211">211</a>         <strong class="jxr_keyword">for</strong> (LoggingEvent event : events) {
<a name="212" href="#212">212</a>             <strong class="jxr_keyword">if</strong> (!ExecutorFilter.<strong class="jxr_keyword">class</strong>.getName().equals(event.getLoggerName())) {
<a name="213" href="#213">213</a>                 Object remoteAddress = event.getMDC(<span class="jxr_string">"remoteAddress"</span>);
<a name="214" href="#214">214</a>                 assertNotNull(
<a name="215" href="#215">215</a>                     <span class="jxr_string">"MDC[remoteAddress] not set for ["</span> + event.getMessage() + <span class="jxr_string">"]"</span>,
<a name="216" href="#216">216</a>                     remoteAddress);
<a name="217" href="#217">217</a>                 assertNotNull(
<a name="218" href="#218">218</a>                     <span class="jxr_string">"MDC[remotePort] not set for ["</span> + event.getMessage() + <span class="jxr_string">"]"</span>,
<a name="219" href="#219">219</a>                     event.getMDC(<span class="jxr_string">"remotePort"</span>));
<a name="220" href="#220">220</a>                 assertEquals(
<a name="221" href="#221">221</a>                     <span class="jxr_string">"every event should have MDC[handlerClass]"</span>,
<a name="222" href="#222">222</a>                     SimpleIoHandler.<strong class="jxr_keyword">class</strong>.getName(),
<a name="223" href="#223">223</a>                     event.getMDC(<span class="jxr_string">"handlerClass"</span>) );
<a name="224" href="#224">224</a>             }
<a name="225" href="#225">225</a>         }
<a name="226" href="#226">226</a>         <em class="jxr_comment">// asert we have received all expected logging events for each client</em>
<a name="227" href="#227">227</a>         <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = 0; i &lt; remoteAddressClients.length; i++) {
<a name="228" href="#228">228</a>             SocketAddress remoteAddressClient = remoteAddressClients[i];
<a name="229" href="#229">229</a>             assertEventExists(events, <span class="jxr_string">"sessionCreated"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="230" href="#230">230</a>             assertEventExists(events, <span class="jxr_string">"sessionOpened"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="231" href="#231">231</a>             assertEventExists(events, <span class="jxr_string">"decode"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="232" href="#232">232</a>             assertEventExists(events, <span class="jxr_string">"messageReceived-1"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="233" href="#233">233</a>             assertEventExists(events, <span class="jxr_string">"messageReceived-2"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="234" href="#234">234</a>             assertEventExists(events, <span class="jxr_string">"encode"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="235" href="#235">235</a>             assertEventExists(events, <span class="jxr_string">"exceptionCaught"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="236" href="#236">236</a>             assertEventExists(events, <span class="jxr_string">"messageSent-1"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="237" href="#237">237</a>             assertEventExists(events, <span class="jxr_string">"messageSent-2"</span>, remoteAddressClient, <strong class="jxr_keyword">null</strong>);
<a name="238" href="#238">238</a>             assertEventExists(events, <span class="jxr_string">"sessionIdle"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="239" href="#239">239</a>             assertEventExists(events, <span class="jxr_string">"sessionClosed"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="240" href="#240">240</a>             assertEventExists(events, <span class="jxr_string">"sessionClosed"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="241" href="#241">241</a>             assertEventExists(events, <span class="jxr_string">"DummyIoFilter.sessionOpened"</span>, remoteAddressClient, <span class="jxr_string">"user-"</span> + i);
<a name="242" href="#242">242</a>         }
<a name="243" href="#243">243</a>     }
<a name="244" href="#244">244</a> 
<a name="245" href="#245">245</a>     <strong class="jxr_keyword">private</strong> SocketAddress connectAndWrite(NioSocketConnector connector, <strong class="jxr_keyword">int</strong> clientNr) {
<a name="246" href="#246">246</a>         ConnectFuture connectFuture = connector.connect(<strong class="jxr_keyword">new</strong> InetSocketAddress(<span class="jxr_string">"localhost"</span>,PORT));
<a name="247" href="#247">247</a>         connectFuture.awaitUninterruptibly(TIMEOUT);
<a name="248" href="#248">248</a>         IoBuffer message = IoBuffer.allocate(4).putInt(clientNr).flip();
<a name="249" href="#249">249</a>         IoSession session = connectFuture.getSession();
<a name="250" href="#250">250</a>         session.write(message).awaitUninterruptibly(TIMEOUT);
<a name="251" href="#251">251</a>         <strong class="jxr_keyword">return</strong> session.getLocalAddress();
<a name="252" href="#252">252</a>     }
<a name="253" href="#253">253</a> 
<a name="254" href="#254">254</a>     <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">void</strong> assertEventExists(List&lt;LoggingEvent&gt; events,
<a name="255" href="#255">255</a>                                    String message,
<a name="256" href="#256">256</a>                                    SocketAddress address,