aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/binsaver/buffered_io.h
diff options
context:
space:
mode:
authorAnton Samokhvalov <pg83@yandex.ru>2022-02-10 16:45:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:15 +0300
commit72cb13b4aff9bc9cf22e49251bc8fd143f82538f (patch)
treeda2c34829458c7d4e74bdfbdf85dff449e9e7fb8 /library/cpp/binsaver/buffered_io.h
parent778e51ba091dc39e7b7fcab2b9cf4dbedfb6f2b5 (diff)
downloadydb-72cb13b4aff9bc9cf22e49251bc8fd143f82538f.tar.gz
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 1 of 2.
Diffstat (limited to 'library/cpp/binsaver/buffered_io.h')
-rw-r--r--library/cpp/binsaver/buffered_io.h78
1 files changed, 39 insertions, 39 deletions
diff --git a/library/cpp/binsaver/buffered_io.h b/library/cpp/binsaver/buffered_io.h
index 75465c9c5c..893f93cddd 100644
--- a/library/cpp/binsaver/buffered_io.h
+++ b/library/cpp/binsaver/buffered_io.h
@@ -1,15 +1,15 @@
#pragma once
-
+
#include <util/system/yassert.h>
#include <util/generic/utility.h>
#include <util/generic/ylimits.h>
#include <string.h>
-struct IBinaryStream {
- virtual ~IBinaryStream() = default;
- ;
+struct IBinaryStream {
+ virtual ~IBinaryStream() = default;
+ ;
- inline i64 Write(const void* userBuffer, i64 size) {
+ inline i64 Write(const void* userBuffer, i64 size) {
if (size <= Max<int>()) {
return WriteImpl(userBuffer, static_cast<int>(size));
} else {
@@ -17,7 +17,7 @@ struct IBinaryStream {
}
}
- inline i64 Read(void* userBuffer, i64 size) {
+ inline i64 Read(void* userBuffer, i64 size) {
if (size <= Max<int>()) {
return ReadImpl(userBuffer, static_cast<int>(size));
} else {
@@ -29,26 +29,26 @@ struct IBinaryStream {
virtual bool IsFailed() const = 0;
private:
- virtual int WriteImpl(const void* userBuffer, int size) = 0;
- virtual int ReadImpl(void* userBuffer, int size) = 0;
+ virtual int WriteImpl(const void* userBuffer, int size) = 0;
+ virtual int ReadImpl(void* userBuffer, int size) = 0;
- i64 LongRead(void* userBuffer, i64 size);
- i64 LongWrite(const void* userBuffer, i64 size);
+ i64 LongRead(void* userBuffer, i64 size);
+ i64 LongWrite(const void* userBuffer, i64 size);
};
template <int N_SIZE = 16384>
-class TBufferedStream {
+class TBufferedStream {
char Buf[N_SIZE];
i64 Pos, BufSize;
- IBinaryStream& Stream;
+ IBinaryStream& Stream;
bool bIsReading, bIsEof, bFailed;
- void ReadComplex(void* userBuffer, i64 size) {
+ void ReadComplex(void* userBuffer, i64 size) {
if (bIsEof) {
memset(userBuffer, 0, size);
return;
}
- char* dst = (char*)userBuffer;
+ char* dst = (char*)userBuffer;
i64 leftBytes = BufSize - Pos;
memcpy(dst, Buf + Pos, leftBytes);
dst += leftBytes;
@@ -70,28 +70,28 @@ class TBufferedStream {
}
}
- void WriteComplex(const void* userBuffer, i64 size) {
+ void WriteComplex(const void* userBuffer, i64 size) {
Flush();
if (size >= N_SIZE) {
Stream.Write(userBuffer, size);
bFailed = Stream.IsFailed();
- } else
+ } else
Write(userBuffer, size);
}
- void operator=(const TBufferedStream&) {
- }
-
+ void operator=(const TBufferedStream&) {
+ }
+
public:
- TBufferedStream(bool bRead, IBinaryStream& stream)
- : Pos(0)
- , BufSize(0)
- , Stream(stream)
- , bIsReading(bRead)
- , bIsEof(false)
- , bFailed(false)
- {
- }
+ TBufferedStream(bool bRead, IBinaryStream& stream)
+ : Pos(0)
+ , BufSize(0)
+ , Stream(stream)
+ , bIsReading(bRead)
+ , bIsEof(false)
+ , bFailed(false)
+ {
+ }
~TBufferedStream() {
if (!bIsReading)
Flush();
@@ -104,10 +104,10 @@ public:
bFailed = Stream.IsFailed();
Pos = 0;
}
- bool IsEof() const {
- return bIsEof;
- }
- inline void Read(void* userBuffer, i64 size) {
+ bool IsEof() const {
+ return bIsEof;
+ }
+ inline void Read(void* userBuffer, i64 size) {
Y_ASSERT(bIsReading);
if (!bIsEof && size + Pos <= BufSize) {
memcpy(userBuffer, Buf + Pos, size);
@@ -116,7 +116,7 @@ public:
}
ReadComplex(userBuffer, size);
}
- inline void Write(const void* userBuffer, i64 size) {
+ inline void Write(const void* userBuffer, i64 size) {
Y_ASSERT(!bIsReading);
if (Pos + size < N_SIZE) {
memcpy(Buf + Pos, userBuffer, size);
@@ -125,10 +125,10 @@ public:
}
WriteComplex(userBuffer, size);
}
- bool IsValid() const {
- return Stream.IsValid();
- }
- bool IsFailed() const {
- return bFailed;
- }
+ bool IsValid() const {
+ return Stream.IsValid();
+ }
+ bool IsFailed() const {
+ return bFailed;
+ }
};