util/network/iovec.h


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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

#pragma once
 
#include <util/stream/output.h> 
#include <util/system/types.h> 
#include <util/system/yassert.h> 
 
class TContIOVector { 
    using TPart = IOutputStream::TPart;
 
public: 
    inline TContIOVector(TPart* parts, size_t count) 
        : Parts_(parts) 
        , Count_(count) 
    { 
    } 
 
    inline void Proceed(size_t len) noexcept {
        while (Count_) { 
            if (len < Parts_->len) { 
                Parts_->len -= len; 
                Parts_->buf = (const char*)Parts_->buf + len; 
 
                return; 
            } else { 
                len -= Parts_->len; 
                --Count_; 
                ++Parts_; 
            } 
        } 
 
        if (len) { 
            Y_ASSERT(0 && "non zero length left");
        } 
    } 
 
    inline const TPart* Parts() const noexcept {
        return Parts_; 
    } 
 
    inline size_t Count() const noexcept {
        return Count_; 
    } 
 
    static inline size_t Bytes(const TPart* parts, size_t count) noexcept {
        size_t ret = 0; 
 
        for (size_t i = 0; i < count; ++i) { 
            ret += parts[i].len; 
        } 
 
        return ret; 
    } 

    inline size_t Bytes() const noexcept {
        return Bytes(Parts_, Count_); 
    } 
 
    inline bool Complete() const noexcept {
        return !Count(); 
    } 
 
private: 
    TPart* Parts_; 
    size_t Count_; 
};