Amebis/stdex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stream: add support for collection reading and writing

Browse Source 
Signed-off-by: Simon Rozman <simon@rozman.si>
This commit is contained in:
Simon Rozman 2023-10-01 23:23:16 +02:00
parent 838dfabbda
commit 86e7ed3690
1 changed files with 90 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
include/stdex/stream.hpp
View File

@ -28,6 +28,7 @@
#include <condition_variable>
#include <list>
#include <memory>
#include <set>
#include <string>
#include <thread>
#include <vector>
@ -652,6 +653,95 @@ namespace stdex
template <class T>
inline basic& operator <<(_In_ const T* data) { return write_str(data); }
template <class _Ty, class _Alloc = std::allocator<_Ty>>
basic& operator <<(_In_ const std::vector<_Ty, _Alloc>& data)
{
size_t num = data.size();
if (num > UINT32_MAX) _Unlikely_
throw std::invalid_argument("collection too big");
*this << static_cast<uint32_t>(num);
for (auto& el : data)
*this << el;
return *this;
}
template <class _Ty, class _Alloc = std::allocator<_Ty>>
basic& operator >>(_Out_ std::vector<_Ty, _Alloc>& data)
{
data.clear();
uint32_t num;
*this >> num;
if (!ok()) _Unlikely_
return *this;
data.reserve(num);
for (uint32_t i = 0; i < num; ++i) {
_Ty el;
*this >> el;
if (!ok()) _Unlikely_
return *this;
data.push_back(std::move(el));
}
}
template <class _Kty, class _Pr = std::less<_Kty>, class _Alloc = std::allocator<_Kty>>
basic& operator <<(_In_ const std::set<_Kty, _Pr, _Alloc>& data)
{
size_t num = data.size();
if (num > UINT32_MAX) _Unlikely_
throw std::invalid_argument("collection too big");
*this << static_cast<uint32_t>(num);
for (auto& el : data)
*this << el;
return *this;
}
template <class _Kty, class _Pr = std::less<_Kty>, class _Alloc = std::allocator<_Kty>>
basic& operator >>(_Out_ std::set<_Kty, _Pr, _Alloc>& data)
{
data.clear();
uint32_t num;
*this >> num;
if (!ok()) _Unlikely_
return *this;
for (uint32_t i = 0; i < num; ++i) {
_Kty el;
*this >> el;
if (!ok()) _Unlikely_
return *this;
data.insert(std::move(el));
}
}
template <class _Kty, class _Pr = std::less<_Kty>, class _Alloc = std::allocator<_Kty>>
basic& operator <<(_In_ const std::multiset<_Kty, _Pr, _Alloc>& data)
{
size_t num = data.size();
if (num > UINT32_MAX) _Unlikely_
throw std::invalid_argument("collection too big");
*this << static_cast<uint32_t>(num);
for (auto& el : data)
*this << el;
return *this;
}
template <class _Kty, class _Pr = std::less<_Kty>, class _Alloc = std::allocator<_Kty>>
basic& operator >>(_Out_ std::multiset<_Kty, _Pr, _Alloc>& data)
{
data.clear();
uint32_t num;
*this >> num;
if (!ok()) _Unlikely_
return *this;
for (uint32_t i = 0; i < num; ++i) {
_Kty el;
*this >> el;
if (!ok()) _Unlikely_
return *this;
data.insert(std::move(el));
}
return *this;
}
protected:
state_t m_state;
};