httprecorder.html

用JAVA编写的,在做实验的时候留下来的,本来想删的,但是传上来,大家分享吧

<a name="169" href="#169">169</a> 
<a name="170" href="#170">170</a>     <em>/**<em>*</em></em>
<a name="171" href="#171">171</a> <em>     * Close all streams.</em>
<a name="172" href="#172">172</a> <em>     */</em>
<a name="173" href="#173">173</a>     <strong>public</strong> <strong>void</strong> close() {
<a name="174" href="#174">174</a>         logger.fine(Thread.currentThread().getName() + <span class="string">" closing"</span>);
<a name="175" href="#175">175</a>         <strong>try</strong> {
<a name="176" href="#176">176</a>             <strong>this</strong>.ris.close();
<a name="177" href="#177">177</a>         } <strong>catch</strong> (IOException e) {
<a name="178" href="#178">178</a>             <em class="comment">// TODO: Can we not let the exception out of here and report it</em>
<a name="179" href="#179">179</a>             <em class="comment">// higher up in the caller?</em>
<a name="180" href="#180">180</a>             DevUtils.logger.log(Level.SEVERE, <span class="string">"close() ris"</span> +
<a name="181" href="#181">181</a>                 DevUtils.extraInfo(), e);
<a name="182" href="#182">182</a>         }
<a name="183" href="#183">183</a>         <strong>try</strong> {
<a name="184" href="#184">184</a>             <strong>this</strong>.ros.close();
<a name="185" href="#185">185</a>         } <strong>catch</strong> (IOException e) {
<a name="186" href="#186">186</a>             DevUtils.logger.log(Level.SEVERE, <span class="string">"close() ros"</span> +
<a name="187" href="#187">187</a>                 DevUtils.extraInfo(), e);
<a name="188" href="#188">188</a>         }
<a name="189" href="#189">189</a>     }
<a name="190" href="#190">190</a> 
<a name="191" href="#191">191</a>     <em>/**<em>*</em></em>
<a name="192" href="#192">192</a> <em>     * Return the internal RecordingInputStream</em>
<a name="193" href="#193">193</a> <em>     *</em>
<a name="194" href="#194">194</a> <em>     * @return A RIS.</em>
<a name="195" href="#195">195</a> <em>     */</em>
<a name="196" href="#196">196</a>     <strong>public</strong> <a href="../../../org/archive/io/RecordingInputStream.html">RecordingInputStream</a> getRecordedInput() {
<a name="197" href="#197">197</a>         <strong>return</strong> <strong>this</strong>.ris;
<a name="198" href="#198">198</a>     }
<a name="199" href="#199">199</a> 
<a name="200" href="#200">200</a>     <em>/**<em>*</em></em>
<a name="201" href="#201">201</a> <em>     * @return The RecordingOutputStream.</em>
<a name="202" href="#202">202</a> <em>     */</em>
<a name="203" href="#203">203</a>     <strong>public</strong> <a href="../../../org/archive/io/RecordingOutputStream.html">RecordingOutputStream</a> getRecordedOutput() {
<a name="204" href="#204">204</a>         <strong>return</strong> <strong>this</strong>.ros;
<a name="205" href="#205">205</a>     }
<a name="206" href="#206">206</a> 
<a name="207" href="#207">207</a>     <em>/**<em>*</em></em>
<a name="208" href="#208">208</a> <em>     * Mark current position as the point where the HTTP headers end.</em>
<a name="209" href="#209">209</a> <em>     */</em>
<a name="210" href="#210">210</a>     <strong>public</strong> <strong>void</strong> markContentBegin() {
<a name="211" href="#211">211</a>         <strong>this</strong>.ris.markContentBegin();
<a name="212" href="#212">212</a>     }
<a name="213" href="#213">213</a> 
<a name="214" href="#214">214</a>     <strong>public</strong> <strong>long</strong> getResponseContentLength() {
<a name="215" href="#215">215</a>         <strong>return</strong> <strong>this</strong>.ris.getResponseContentLength();
<a name="216" href="#216">216</a>     }
<a name="217" href="#217">217</a> 
<a name="218" href="#218">218</a>     <em>/**<em>*</em></em>
<a name="219" href="#219">219</a> <em>     * Close both input and output recorders.</em>
<a name="220" href="#220">220</a> <em>     *</em>
<a name="221" href="#221">221</a> <em>     * Recorders are the output streams to which we are recording.</em>
<a name="222" href="#222">222</a> <em>     * {@link #close()} closes the stream that is being recorded and the</em>
<a name="223" href="#223">223</a> <em>     * recorder. This method explicitly closes the recorder only.</em>
<a name="224" href="#224">224</a> <em>     */</em>
<a name="225" href="#225">225</a>     <strong>public</strong> <strong>void</strong> closeRecorders() {
<a name="226" href="#226">226</a>         <strong>try</strong> {
<a name="227" href="#227">227</a>             <strong>this</strong>.ris.closeRecorder();
<a name="228" href="#228">228</a>             <strong>this</strong>.ros.closeRecorder();
<a name="229" href="#229">229</a>         } <strong>catch</strong> (IOException e) {
<a name="230" href="#230">230</a>             DevUtils.warnHandle(e, <span class="string">"Convert to runtime exception?"</span>);
<a name="231" href="#231">231</a>         }
<a name="232" href="#232">232</a>     }
<a name="233" href="#233">233</a> 
<a name="234" href="#234">234</a>     <em>/**<em>*</em></em>
<a name="235" href="#235">235</a> <em>     * Cleanup backing files.</em>
<a name="236" href="#236">236</a> <em>     *</em>
<a name="237" href="#237">237</a> <em>     * Call when completely done w/ recorder.  Removes any backing files that</em>
<a name="238" href="#238">238</a> <em>     * may have been dropped.</em>
<a name="239" href="#239">239</a> <em>     */</em>
<a name="240" href="#240">240</a>     <strong>public</strong> <strong>void</strong> cleanup() {
<a name="241" href="#241">241</a>         <strong>this</strong>.close();
<a name="242" href="#242">242</a>         <strong>this</strong>.delete(<strong>this</strong>.backingFileBasename + RECORDING_OUTPUT_STREAM_SUFFIX);
<a name="243" href="#243">243</a>         <strong>this</strong>.delete(<strong>this</strong>.backingFileBasename + RECORDING_INPUT_STREAM_SUFFIX);
<a name="244" href="#244">244</a>     }
<a name="245" href="#245">245</a> 
<a name="246" href="#246">246</a>     <em>/**<em>*</em></em>
<a name="247" href="#247">247</a> <em>     * Delete file if exists.</em>
<a name="248" href="#248">248</a> <em>     *</em>
<a name="249" href="#249">249</a> <em>     * @param name Filename to delete.</em>
<a name="250" href="#250">250</a> <em>     */</em>
<a name="251" href="#251">251</a>     <strong>private</strong> <strong>void</strong> delete(String name) {
<a name="252" href="#252">252</a>         File f = <strong>new</strong> File(name);
<a name="253" href="#253">253</a>         <strong>if</strong> (f.exists()) {
<a name="254" href="#254">254</a>             f.delete();
<a name="255" href="#255">255</a>         }
<a name="256" href="#256">256</a>     }
<a name="257" href="#257">257</a> 
<a name="258" href="#258">258</a>     <em>/**<em>*</em></em>
<a name="259" href="#259">259</a> <em>     * Get the current threads' HttpRecorder.</em>
<a name="260" href="#260">260</a> <em>     *</em>
<a name="261" href="#261">261</a> <em>     * @return This threads' HttpRecorder.  Returns null if can't find a</em>
<a name="262" href="#262">262</a> <em>     * HttpRecorder in current instance.</em>
<a name="263" href="#263">263</a> <em>     */</em>
<a name="264" href="#264">264</a>     <strong>public</strong> <strong>static</strong> <a href="../../../org/archive/util/HttpRecorder.html">HttpRecorder</a> getHttpRecorder() {
<a name="265" href="#265">265</a>         <a href="../../../org/archive/util/HttpRecorder.html">HttpRecorder</a> recorder = <strong>null</strong>;
<a name="266" href="#266">266</a>         Thread thread = Thread.currentThread();
<a name="267" href="#267">267</a>         <strong>if</strong> (thread instanceof HttpRecorderMarker) {
<a name="268" href="#268">268</a>             recorder = ((HttpRecorderMarker)thread).getHttpRecorder();
<a name="269" href="#269">269</a>         }
<a name="270" href="#270">270</a>         <strong>return</strong> recorder;
<a name="271" href="#271">271</a>     }
<a name="272" href="#272">272</a> 
<a name="273" href="#273">273</a>     <em>/**<em>*</em></em>
<a name="274" href="#274">274</a> <em>     * @param characterEncoding Character encoding of recording.</em>
<a name="275" href="#275">275</a> <em>     */</em>
<a name="276" href="#276">276</a>     <strong>public</strong> <strong>void</strong> setCharacterEncoding(String characterEncoding) {
<a name="277" href="#277">277</a>         <strong>this</strong>.characterEncoding = characterEncoding;
<a name="278" href="#278">278</a>     }
<a name="279" href="#279">279</a> 
<a name="280" href="#280">280</a>     <em>/**<em>*</em></em>
<a name="281" href="#281">281</a> <em>     * @return Returns the characterEncoding.</em>
<a name="282" href="#282">282</a> <em>     */</em>
<a name="283" href="#283">283</a>     <strong>public</strong> String getCharacterEncoding() {
<a name="284" href="#284">284</a>         <strong>return</strong> <strong>this</strong>.characterEncoding;
<a name="285" href="#285">285</a>     }
<a name="286" href="#286">286</a> 
<a name="287" href="#287">287</a>     <em>/**<em>*</em></em>
<a name="288" href="#288">288</a> <em>     * @return A ReplayCharSequence.  Call close on the RCS when done w/ it.</em>
<a name="289" href="#289">289</a> <em>     * Will return indeterminate results if the underlying recording streams</em>
<a name="290" href="#290">290</a> <em>     * have not been closed first.</em>
<a name="291" href="#291">291</a> <em>     * @throws IOException</em>
<a name="292" href="#292">292</a> <em>     * @throws IOException</em>
<a name="293" href="#293">293</a> <em>     */</em>
<a name="294" href="#294">294</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayCharSequence.html">ReplayCharSequence</a> getReplayCharSequence() throws IOException {
<a name="295" href="#295">295</a>         <strong>return</strong> getRecordedInput().
<a name="296" href="#296">296</a>             getReplayCharSequence(<strong>this</strong>.characterEncoding);
<a name="297" href="#297">297</a>     }
<a name="298" href="#298">298</a>     
<a name="299" href="#299">299</a>     <em>/**<em>*</em></em>
<a name="300" href="#300">300</a> <em>     * @return A replay input stream.</em>
<a name="301" href="#301">301</a> <em>     * @throws IOException</em>
<a name="302" href="#302">302</a> <em>     */</em>
<a name="303" href="#303">303</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getReplayInputStream() throws IOException {
<a name="304" href="#304">304</a>         <strong>return</strong> getRecordedInput().getReplayInputStream();
<a name="305" href="#305">305</a>     }
<a name="306" href="#306">306</a>     
<a name="307" href="#307">307</a>     <em>/**<em>*</em></em>
<a name="308" href="#308">308</a> <em>     * Record the input stream for later playback by an extractor, etc.</em>
<a name="309" href="#309">309</a> <em>     * This is convenience method used to setup an artificial HttpRecorder</em>
<a name="310" href="#310">310</a> <em>     * scenario used in unit tests, etc.</em>
<a name="311" href="#311">311</a> <em>     * @param dir Directory to write backing file to.</em>
<a name="312" href="#312">312</a> <em>     * @param basename of what we're recording.</em>
<a name="313" href="#313">313</a> <em>     * @param in Stream to read.</em>
<a name="314" href="#314">314</a> <em>     * @param encoding Stream encoding.</em>
<a name="315" href="#315">315</a> <em>     * @throws IOException</em>
<a name="316" href="#316">316</a> <em>     * @return An {@link org.archive.util.HttpRecorder}.</em>
<a name="317" href="#317">317</a> <em>     */</em>
<a name="318" href="#318">318</a>     <strong>public</strong> <strong>static</strong> <a href="../../../org/archive/util/HttpRecorder.html">HttpRecorder</a> wrapInputStreamWithHttpRecord(File dir,
<a name="319" href="#319">319</a>         String basename, InputStream in, String encoding)
<a name="320" href="#320">320</a>     throws IOException {
<a name="321" href="#321">321</a>         <a href="../../../org/archive/util/HttpRecorder.html">HttpRecorder</a> rec = <strong>new</strong> <a href="../../../org/archive/util/HttpRecorder.html">HttpRecorder</a>(dir, basename);
<a name="322" href="#322">322</a>         <strong>if</strong> (encoding != <strong>null</strong> &amp;&amp; encoding.length() > 0) {
<a name="323" href="#323">323</a>             rec.setCharacterEncoding(encoding);
<a name="324" href="#324">324</a>         }
<a name="325" href="#325">325</a>         <em class="comment">// Do not use FastBufferedInputStream here.  It does not</em>
<a name="326" href="#326">326</a>         <em class="comment">// support mark.</em>
<a name="327" href="#327">327</a>         InputStream is = rec.inputWrap(<strong>new</strong> BufferedInputStream(in));
<a name="328" href="#328">328</a>         <strong>final</strong> <strong>int</strong> BUFFER_SIZE = 1024 * 4;
<a name="329" href="#329">329</a>         byte [] buffer = <strong>new</strong> byte[BUFFER_SIZE];
<a name="330" href="#330">330</a>         <strong>while</strong>(<strong>true</strong>) {
<a name="331" href="#331">331</a>             <em class="comment">// Just read it all down.</em>
<a name="332" href="#332">332</a>             <strong>int</strong> x = is.read(buffer);
<a name="333" href="#333">333</a>             <strong>if</strong> (x == -1) {
<a name="334" href="#334">334</a>                 <strong>break</strong>;
<a name="335" href="#335">335</a>             }
<a name="336" href="#336">336</a>         }
<a name="337" href="#337">337</a>         is.close();
<a name="338" href="#338">338</a>         <strong>return</strong> rec;
<a name="339" href="#339">339</a>     }
<a name="340" href="#340">340</a> }
</pre>
<hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>
</html>