util/stream/length.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#include "length.h" 
 
size_t TLengthLimitedInput::DoRead(void* buf, size_t len) { 
    const size_t toRead = (size_t)Min<ui64>(Length_, len); 
    const size_t ret = Slave_->Read(buf, toRead); 
 
    Length_ -= ret; 
 
    return ret; 
} 
 
size_t TLengthLimitedInput::DoSkip(size_t len) { 
    len = (size_t)Min<ui64>((ui64)len, Length_); 
    len = Slave_->Skip(len); 
    Length_ -= len; 
    return len; 
} 
 
size_t TCountingInput::DoRead(void* buf, size_t len) { 
    const size_t ret = Slave_->Read(buf, len); 
    Count_ += ret; 
    return ret; 
} 
 
size_t TCountingInput::DoSkip(size_t len) { 
    const size_t ret = Slave_->Skip(len); 
    Count_ += ret; 
    return ret; 
} 

size_t TCountingInput::DoReadTo(TString& st, char ch) {
    const size_t ret = Slave_->ReadTo(st, ch);
    Count_ += ret;
    return ret;
}

ui64 TCountingInput::DoReadAll(IOutputStream& out) {
    const ui64 ret = Slave_->ReadAll(out);
    Count_ += ret;
    return ret;
}

void TCountingOutput::DoWrite(const void* buf, size_t len) {
    Slave_->Write(buf, len);

    Count_ += len;
}