recordinginputstream.html

来自「网络爬虫开源代码」· HTML 代码 · 共 383 行 · 第 1/3 页

<a name="120" href="#120">120</a>         <strong>return</strong> count;
<a name="121" href="#121">121</a>     }
<a name="122" href="#122">122</a> 
<a name="123" href="#123">123</a>     <strong>public</strong> <strong>int</strong> read(byte[] b) throws IOException {
<a name="124" href="#124">124</a>     	    <strong>if</strong> (!isOpen()) {
<a name="125" href="#125">125</a>     	    	    <strong>throw</strong> <strong>new</strong> IOException(<span class="string">"Stream closed "</span> +
<a name="126" href="#126">126</a>     			    Thread.currentThread().getName());
<a name="127" href="#127">127</a>     	    }
<a name="128" href="#128">128</a>     	    <strong>int</strong> count = <strong>this</strong>.in.read(b);
<a name="129" href="#129">129</a>         <strong>if</strong> (count > 0) {
<a name="130" href="#130">130</a>             assert <strong>this</strong>.recordingOutputStream != <strong>null</strong>: <span class="string">"ROS is null "</span> +
<a name="131" href="#131">131</a>                 Thread.currentThread().getName();
<a name="132" href="#132">132</a>             <strong>this</strong>.recordingOutputStream.write(b,0,count);
<a name="133" href="#133">133</a>         }
<a name="134" href="#134">134</a>         <strong>return</strong> count;
<a name="135" href="#135">135</a>     }
<a name="136" href="#136">136</a> 
<a name="137" href="#137">137</a>     <strong>public</strong> <strong>void</strong> close() throws IOException {
<a name="138" href="#138">138</a>         <strong>if</strong> (logger.isLoggable(Level.FINE)) {
<a name="139" href="#139">139</a>             logger.fine(Thread.currentThread().getName() + <span class="string">" closing "</span> +
<a name="140" href="#140">140</a>                     <strong>this</strong>.in + <span class="string">", "</span> + Thread.currentThread().getName());
<a name="141" href="#141">141</a>         }
<a name="142" href="#142">142</a>         <strong>if</strong> (<strong>this</strong>.in != <strong>null</strong>) {
<a name="143" href="#143">143</a>             <strong>this</strong>.in.close();
<a name="144" href="#144">144</a>             <strong>this</strong>.in = <strong>null</strong>;
<a name="145" href="#145">145</a>         }
<a name="146" href="#146">146</a>         <strong>this</strong>.recordingOutputStream.close();
<a name="147" href="#147">147</a>     }
<a name="148" href="#148">148</a> 
<a name="149" href="#149">149</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getReplayInputStream() throws IOException {
<a name="150" href="#150">150</a>         <strong>return</strong> <strong>this</strong>.recordingOutputStream.getReplayInputStream();
<a name="151" href="#151">151</a>     }
<a name="152" href="#152">152</a> 
<a name="153" href="#153">153</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getContentReplayInputStream() throws IOException {
<a name="154" href="#154">154</a>         <strong>return</strong> <strong>this</strong>.recordingOutputStream.getContentReplayInputStream();
<a name="155" href="#155">155</a>     }
<a name="156" href="#156">156</a> 
<a name="157" href="#157">157</a>     <strong>public</strong> <strong>long</strong> readFully() throws IOException {
<a name="158" href="#158">158</a>         <strong>while</strong>(read(drainBuffer) != -1) {
<a name="159" href="#159">159</a>             <em class="comment">// Empty out stream.</em>
<a name="160" href="#160">160</a>             <strong>continue</strong>;
<a name="161" href="#161">161</a>         }
<a name="162" href="#162">162</a>         <strong>return</strong> <strong>this</strong>.recordingOutputStream.getSize();
<a name="163" href="#163">163</a>     }
<a name="164" href="#164">164</a> 
<a name="165" href="#165">165</a>     <em>/**<em>*</em></em>
<a name="166" href="#166">166</a> <em>     * Read all of a stream (Or read until we timeout or have read to the max).</em>
<a name="167" href="#167">167</a> <em>     * @param softMaxLength Maximum length to read; if zero or &lt; 0, then no </em>
<a name="168" href="#168">168</a> <em>     * limit. If met, return normally. </em>
<a name="169" href="#169">169</a> <em>     * @param hardMaxLength Maximum length to read; if zero or &lt; 0, then no </em>
<a name="170" href="#170">170</a> <em>     * limit. If exceeded, throw RecorderLengthExceededException</em>
<a name="171" href="#171">171</a> <em>     * @param timeout Timeout in milliseconds for total read; if zero or</em>
<a name="172" href="#172">172</a> <em>     * negative, timeout is &lt;code>Long.MAX_VALUE&lt;/code>. If exceeded, throw</em>
<a name="173" href="#173">173</a> <em>     * RecorderTimeoutException</em>
<a name="174" href="#174">174</a> <em>     * @param maxBytesPerMs How many bytes per millisecond.</em>
<a name="175" href="#175">175</a> <em>     * @throws IOException failed read.</em>
<a name="176" href="#176">176</a> <em>     * @throws RecorderLengthExceededException</em>
<a name="177" href="#177">177</a> <em>     * @throws RecorderTimeoutException</em>
<a name="178" href="#178">178</a> <em>     * @throws InterruptedException</em>
<a name="179" href="#179">179</a> <em>     */</em>
<a name="180" href="#180">180</a>     <strong>public</strong> <strong>void</strong> readFullyOrUntil(<strong>long</strong> softMaxLength)
<a name="181" href="#181">181</a>         throws IOException, RecorderLengthExceededException,
<a name="182" href="#182">182</a>             RecorderTimeoutException, InterruptedException {
<a name="183" href="#183">183</a>         <em class="comment">// Check we're open before proceeding.</em>
<a name="184" href="#184">184</a>         <strong>if</strong> (!isOpen()) {
<a name="185" href="#185">185</a>             <em class="comment">// TODO: should this be a noisier exception-raising error? </em>
<a name="186" href="#186">186</a>             <strong>return</strong>;
<a name="187" href="#187">187</a>         } 
<a name="188" href="#188">188</a> 
<a name="189" href="#189">189</a>         <strong>long</strong> totalBytes = 0L;
<a name="190" href="#190">190</a>         <strong>long</strong> bytesRead = -1L;
<a name="191" href="#191">191</a>         <strong>long</strong> maxToRead = -1; 
<a name="192" href="#192">192</a>         <strong>while</strong> (<strong>true</strong>) {
<a name="193" href="#193">193</a>             <strong>try</strong> {
<a name="194" href="#194">194</a>                 <em class="comment">// read no more than soft max</em>
<a name="195" href="#195">195</a>                 maxToRead = (softMaxLength &lt;= 0) 
<a name="196" href="#196">196</a>                     ? drainBuffer.length 
<a name="197" href="#197">197</a>                     : Math.min(drainBuffer.length, softMaxLength - totalBytes);
<a name="198" href="#198">198</a>                 <em class="comment">// nor more than hard max</em>
<a name="199" href="#199">199</a>                 maxToRead = Math.min(maxToRead, recordingOutputStream.getRemainingLength());
<a name="200" href="#200">200</a>                 <em class="comment">// but always at least 1 (to trigger hard max exception</em>
<a name="201" href="#201">201</a>                 maxToRead = Math.max(maxToRead, 1);
<a name="202" href="#202">202</a>                 
<a name="203" href="#203">203</a>                 bytesRead = read(drainBuffer,0,(<strong>int</strong>)maxToRead);
<a name="204" href="#204">204</a>                 <strong>if</strong> (bytesRead == -1) {
<a name="205" href="#205">205</a>                     <strong>break</strong>;
<a name="206" href="#206">206</a>                 }
<a name="207" href="#207">207</a>                 totalBytes += bytesRead;
<a name="208" href="#208">208</a> 
<a name="209" href="#209">209</a>                 <strong>if</strong> (Thread.interrupted()) {
<a name="210" href="#210">210</a>                     <strong>throw</strong> <strong>new</strong> InterruptedException(<span class="string">"Interrupted during IO"</span>);
<a name="211" href="#211">211</a>                 }
<a name="212" href="#212">212</a>             } <strong>catch</strong> (SocketTimeoutException e) {
<a name="213" href="#213">213</a>                 <em class="comment">// A socket timeout is just a transient problem, meaning</em>
<a name="214" href="#214">214</a>                 <em class="comment">// nothing was available in the configured  timeout period,</em>
<a name="215" href="#215">215</a>                 <em class="comment">// but something else might become available later.</em>
<a name="216" href="#216">216</a>                 <em class="comment">// Take this opportunity to check the overall </em>
<a name="217" href="#217">217</a>                 <em class="comment">// timeout (below).  One reason for this timeout is </em>
<a name="218" href="#218">218</a>                 <em class="comment">// servers that keep up the connection, 'keep-alive', even</em>
<a name="219" href="#219">219</a>                 <em class="comment">// though we asked them to not keep the connection open.</em>
<a name="220" href="#220">220</a>                 <strong>if</strong> (logger.isLoggable(Level.FINE)) {
<a name="221" href="#221">221</a>                     logger.log(Level.FINE, <span class="string">"socket timeout"</span>, e); 
<a name="222" href="#222">222</a>                 }
<a name="223" href="#223">223</a>                 <em class="comment">// check for overall timeout</em>
<a name="224" href="#224">224</a>                 recordingOutputStream.checkLimits();
<a name="225" href="#225">225</a>             } <strong>catch</strong> (SocketException se) {
<a name="226" href="#226">226</a>                 <strong>throw</strong> se;
<a name="227" href="#227">227</a>             } <strong>catch</strong> (NullPointerException e) {
<a name="228" href="#228">228</a>                 <em class="comment">// [ 896757 ] NPEs in Andy's Th-Fri Crawl.</em>
<a name="229" href="#229">229</a>                 <em class="comment">// A crawl was showing NPE's in this part of the code but can</em>
<a name="230" href="#230">230</a>                 <em class="comment">// not reproduce.  Adding this rethrowing catch block w/</em>
<a name="231" href="#231">231</a>                 <em class="comment">// diagnostics to help should we come across the problem in the</em>
<a name="232" href="#232">232</a>                 <em class="comment">// future.</em>
<a name="233" href="#233">233</a>                 <strong>throw</strong> <strong>new</strong> NullPointerException(<span class="string">"Stream "</span> + <strong>this</strong>.in + <span class="string">", "</span> +
<a name="234" href="#234">234</a>                     e.getMessage() + <span class="string">" "</span> + Thread.currentThread().getName());
<a name="235" href="#235">235</a>             }
<a name="236" href="#236">236</a>             
<a name="237" href="#237">237</a>             <em class="comment">// if have read 'enough', just finish</em>
<a name="238" href="#238">238</a>             <strong>if</strong> (softMaxLength > 0 &amp;&amp; totalBytes >= softMaxLength) {
<a name="239" href="#239">239</a>                 <strong>break</strong>; <em class="comment">// return</em>
<a name="240" href="#240">240</a>             }
<a name="241" href="#241">241</a>         }
<a name="242" href="#242">242</a>     }
<a name="243" href="#243">243</a> 
<a name="244" href="#244">244</a>     <strong>public</strong> <strong>long</strong> getSize() {
<a name="245" href="#245">245</a>         <strong>return</strong> <strong>this</strong>.recordingOutputStream.getSize();
<a name="246" href="#246">246</a>     }
<a name="247" href="#247">247</a>