recordingoutputstream.html

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

<a name="472" href="#472">472</a>             <strong>if</strong> (<strong>this</strong>.digest == <strong>null</strong> ||
<a name="473" href="#473">473</a>                     !<strong>this</strong>.digest.getAlgorithm().equals(algorithm)) {
<a name="474" href="#474">474</a>                 setDigest(MessageDigest.getInstance(algorithm));
<a name="475" href="#475">475</a>             }
<a name="476" href="#476">476</a>         } <strong>catch</strong> (NoSuchAlgorithmException e) {
<a name="477" href="#477">477</a>             e.printStackTrace();
<a name="478" href="#478">478</a>         }
<a name="479" href="#479">479</a>     }
<a name="480" href="#480">480</a> 
<a name="481" href="#481">481</a>     <em>/**<em>*</em></em>
<a name="482" href="#482">482</a> <em>     * Sets a digest function which may be applied to recorded data.</em>
<a name="483" href="#483">483</a> <em>     *</em>
<a name="484" href="#484">484</a> <em>     * As usually only a subset of the recorded data should</em>
<a name="485" href="#485">485</a> <em>     * be fed to the digest, you must also call startDigest()</em>
<a name="486" href="#486">486</a> <em>     * to begin digesting.</em>
<a name="487" href="#487">487</a> <em>     *</em>
<a name="488" href="#488">488</a> <em>     * @param md Message digest function to use.</em>
<a name="489" href="#489">489</a> <em>     */</em>
<a name="490" href="#490">490</a>     <strong>public</strong> <strong>void</strong> setDigest(MessageDigest md) {
<a name="491" href="#491">491</a>         <strong>this</strong>.digest = md;
<a name="492" href="#492">492</a>     }
<a name="493" href="#493">493</a> 
<a name="494" href="#494">494</a>     <em>/**<em>*</em></em>
<a name="495" href="#495">495</a> <em>     * Return the digest value for any recorded, digested data. Call</em>
<a name="496" href="#496">496</a> <em>     * only after all data has been recorded; otherwise, the running</em>
<a name="497" href="#497">497</a> <em>     * digest state is ruined.</em>
<a name="498" href="#498">498</a> <em>     *</em>
<a name="499" href="#499">499</a> <em>     * @return the digest final value</em>
<a name="500" href="#500">500</a> <em>     */</em>
<a name="501" href="#501">501</a>     <strong>public</strong> byte[] getDigestValue() {
<a name="502" href="#502">502</a>         <strong>if</strong>(<strong>this</strong>.digest == <strong>null</strong>) {
<a name="503" href="#503">503</a>             <strong>return</strong> <strong>null</strong>;
<a name="504" href="#504">504</a>         }
<a name="505" href="#505">505</a>         <strong>return</strong> <strong>this</strong>.digest.digest();
<a name="506" href="#506">506</a>     }
<a name="507" href="#507">507</a> 
<a name="508" href="#508">508</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayCharSequence.html">ReplayCharSequence</a> getReplayCharSequence() throws IOException {
<a name="509" href="#509">509</a>         <strong>return</strong> getReplayCharSequence(<strong>null</strong>);
<a name="510" href="#510">510</a>     }
<a name="511" href="#511">511</a> 
<a name="512" href="#512">512</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayCharSequence.html">ReplayCharSequence</a> getReplayCharSequence(String characterEncoding) 
<a name="513" href="#513">513</a>     throws IOException {
<a name="514" href="#514">514</a>         <strong>return</strong> getReplayCharSequence(characterEncoding, <strong>this</strong>.contentBeginMark);
<a name="515" href="#515">515</a>     }
<a name="516" href="#516">516</a>     
<a name="517" href="#517">517</a>     <em>/**<em>*</em></em>
<a name="518" href="#518">518</a> <em>     * @param characterEncoding Encoding of recorded stream.</em>
<a name="519" href="#519">519</a> <em>     * @return A ReplayCharSequence  Will return null if an IOException.  Call</em>
<a name="520" href="#520">520</a> <em>     * close on returned RCS when done.</em>
<a name="521" href="#521">521</a> <em>     * @throws IOException</em>
<a name="522" href="#522">522</a> <em>     */</em>
<a name="523" href="#523">523</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayCharSequence.html">ReplayCharSequence</a> getReplayCharSequence(String characterEncoding, 
<a name="524" href="#524">524</a>             <strong>long</strong> startOffset) throws IOException {
<a name="525" href="#525">525</a>         <em class="comment">// TODO: handled transfer-encoding: chunked content-bodies properly</em>
<a name="526" href="#526">526</a>         <strong>float</strong> maxBytesPerChar = IoUtils.encodingMaxBytesPerChar(characterEncoding);
<a name="527" href="#527">527</a>         <strong>if</strong>(maxBytesPerChar&lt;=1) {
<a name="528" href="#528">528</a>             <em class="comment">// single</em>
<a name="529" href="#529">529</a>             <em class="comment">// TODO: take into account single-byte encoding may be non-default</em>
<a name="530" href="#530">530</a>             <strong>return</strong> <strong>new</strong> <a href="../../../org/archive/io/ByteReplayCharSequence.html">ByteReplayCharSequence</a>(
<a name="531" href="#531">531</a>                     <strong>this</strong>.buffer, 
<a name="532" href="#532">532</a>                     <strong>this</strong>.size, 
<a name="533" href="#533">533</a>                     startOffset,
<a name="534" href="#534">534</a>                     <strong>this</strong>.backingFilename);
<a name="535" href="#535">535</a>         } <strong>else</strong> {
<a name="536" href="#536">536</a>             <em class="comment">// multibyte </em>
<a name="537" href="#537">537</a>             <strong>if</strong>(<strong>this</strong>.size &lt;= <strong>this</strong>.buffer.length) {
<a name="538" href="#538">538</a>                 <em class="comment">// raw data is all in memory; do in memory</em>
<a name="539" href="#539">539</a>                 <strong>return</strong> <strong>new</strong> <a href="../../../org/archive/io/MultiByteReplayCharSequence.html">MultiByteReplayCharSequence</a>(
<a name="540" href="#540">540</a>                         <strong>this</strong>.buffer, 
<a name="541" href="#541">541</a>                         <strong>this</strong>.size, 
<a name="542" href="#542">542</a>                         startOffset,
<a name="543" href="#543">543</a>                         characterEncoding);
<a name="544" href="#544">544</a>                 
<a name="545" href="#545">545</a>             } <strong>else</strong> {
<a name="546" href="#546">546</a>                 <em class="comment">// raw data overflows to disk; use temp file</em>
<a name="547" href="#547">547</a>                 <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> ris = getReplayInputStream(startOffset);
<a name="548" href="#548">548</a>                 <a href="../../../org/archive/io/ReplayCharSequence.html">ReplayCharSequence</a> rcs = <strong>new</strong> <a href="../../../org/archive/io/MultiByteReplayCharSequence.html">MultiByteReplayCharSequence</a>(
<a name="549" href="#549">549</a>                         ris, 
<a name="550" href="#550">550</a>                         <strong>this</strong>.backingFilename,
<a name="551" href="#551">551</a>                         characterEncoding);
<a name="552" href="#552">552</a>                 ris.close(); 
<a name="553" href="#553">553</a>                 <strong>return</strong> rcs;
<a name="554" href="#554">554</a>             }
<a name="555" href="#555">555</a>             
<a name="556" href="#556">556</a>         }
<a name="557" href="#557">557</a>         
<a name="558" href="#558">558</a>     }
<a name="559" href="#559">559</a> 
<a name="560" href="#560">560</a>     <strong>public</strong> <strong>long</strong> getResponseContentLength() {
<a name="561" href="#561">561</a>         <strong>return</strong> <strong>this</strong>.size - <strong>this</strong>.contentBeginMark;
<a name="562" href="#562">562</a>     }
<a name="563" href="#563">563</a> 
<a name="564" href="#564">564</a>     <em>/**<em>*</em></em>
<a name="565" href="#565">565</a> <em>     * @return True if this ROS is open.</em>
<a name="566" href="#566">566</a> <em>     */</em>
<a name="567" href="#567">567</a>     <strong>public</strong> <strong>boolean</strong> isOpen() {
<a name="568" href="#568">568</a>         <strong>return</strong> <strong>this</strong>.out != <strong>null</strong>;
<a name="569" href="#569">569</a>     }
<a name="570" href="#570">570</a>     
<a name="571" href="#571">571</a>     <em>/**<em>*</em></em>
<a name="572" href="#572">572</a> <em>     * When used alongside a mark-supporting RecordingInputStream, remember</em>
<a name="573" href="#573">573</a> <em>     * a position reachable by a future reset().</em>
<a name="574" href="#574">574</a> <em>     */</em>
<a name="575" href="#575">575</a>     <strong>public</strong> <strong>void</strong> mark() {
<a name="576" href="#576">576</a>         <em class="comment">// remember this position for subsequent reset()</em>
<a name="577" href="#577">577</a>         <strong>this</strong>.markPosition = position; 
<a name="578" href="#578">578</a>     }
<a name="579" href="#579">579</a>     
<a name="580" href="#580">580</a>     <em>/**<em>*</em></em>
<a name="581" href="#581">581</a> <em>     * When used alongside a mark-supporting RecordingInputStream, reset </em>
<a name="582" href="#582">582</a> <em>     * the position to that saved by previous mark(). Until the position </em>
<a name="583" href="#583">583</a> <em>     * again reached "new" material, none of the bytes pushed to this </em>
<a name="584" href="#584">584</a> <em>     * stream will be digested or recorded. </em>
<a name="585" href="#585">585</a> <em>     */</em>
<a name="586" href="#586">586</a>     <strong>public</strong> <strong>void</strong> reset() {
<a name="587" href="#587">587</a>         <em class="comment">// take note of furthest-position-reached to avoid double-recording</em>
<a name="588" href="#588">588</a>         maxPosition = Math.max(maxPosition, position); 
<a name="589" href="#589">589</a>         <em class="comment">// reset to previous position</em>
<a name="590" href="#590">590</a>         position = markPosition;
<a name="591" href="#591">591</a>     }
<a name="592" href="#592">592</a>     
<a name="593" href="#593">593</a>     <em>/**<em>*</em></em>
<a name="594" href="#594">594</a> <em>     * Set limits on length, time, and rate to enforce.</em>
<a name="595" href="#595">595</a> <em>     * </em>
<a name="596" href="#596">596</a> <em>     * @param length</em>
<a name="597" href="#597">597</a> <em>     * @param milliseconds</em>
<a name="598" href="#598">598</a> <em>     * @param rateKBps</em>
<a name="599" href="#599">599</a> <em>     */</em>
<a name="600" href="#600">600</a>     <strong>public</strong> <strong>void</strong> setLimits(<strong>long</strong> length, <strong>long</strong> milliseconds, <strong>long</strong> rateKBps) {
<a name="601" href="#601">601</a>         maxLength = (length>0) ? length : Long.MAX_VALUE;
<a name="602" href="#602">602</a>         timeoutMs = (milliseconds>0) ? milliseconds : Long.MAX_VALUE;
<a name="603" href="#603">603</a>         maxRateBytesPerMs = (rateKBps>0) ? rateKBps*1024/1000 : Long.MAX_VALUE;
<a name="604" href="#604">604</a>     }
<a name="605" href="#605">605</a>     
<a name="606" href="#606">606</a>     <em>/**<em>*</em></em>
<a name="607" href="#607">607</a> <em>     * Reset limits to effectively-unlimited defaults</em>
<a name="608" href="#608">608</a> <em>     */</em>
<a name="609" href="#609">609</a>     <strong>public</strong> <strong>void</strong> resetLimits() {
<a name="610" href="#610">610</a>         maxLength = Long.MAX_VALUE;
<a name="611" href="#611">611</a>         timeoutMs = Long.MAX_VALUE;
<a name="612" href="#612">612</a>         maxRateBytesPerMs = Long.MAX_VALUE;
<a name="613" href="#613">613</a>     }
<a name="614" href="#614">614</a>     
<a name="615" href="#615">615</a>     <em>/**<em>*</em></em>
<a name="616" href="#616">616</a> <em>     * Return number of bytes that could be recorded without hitting </em>
<a name="617" href="#617">617</a> <em>     * length limit</em>
<a name="618" href="#618">618</a> <em>     * </em>
<a name="619" href="#619">619</a> <em>     * @return long byte count</em>
<a name="620" href="#620">620</a> <em>     */</em>
<a name="621" href="#621">621</a>     <strong>public</strong> <strong>long</strong> getRemainingLength() {
<a name="622" href="#622">622</a>         <strong>return</strong> maxLength - position; 
<a name="623" href="#623">623</a>     }
<a name="624" href="#624">624</a> }
</pre>
<hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>
</html>