Persson-dev/Simple-Protocol-Lib
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

add file input/output

Browse Source
This commit is contained in:
Simon Pribylski
2025-02-26 00:07:06 +01:00
parent a2eb10ec6d
commit 37ff881819
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
include/sp/io/FileIO.h
View File

@@ -10,23 +10,30 @@ struct FileTag {};
template <> template <>
class IOInterface<FileTag> { class IOInterface<FileTag> {
private: private:
std::ofstream m_FileOutput; std::unique_ptr<std::ifstream> m_FileInput;
std::ifstream m_FileInput; std::unique_ptr<std::ofstream> m_FileOutput;
public: public:
IOInterface(const std::string& filePath) : m_FileOutput(filePath), m_FileInput(filePath) {} IOInterface(const std::string& fileInput, const std::string& fileOutput) {
if (!fileInput.empty())
m_FileInput = std::make_unique<std::ifstream>(fileInput);
if (!fileOutput.empty())
m_FileOutput = std::make_unique<std::ofstream>(fileOutput);
}
IOInterface(IOInterface&& other) : m_FileOutput(std::move(other.m_FileOutput)), m_FileInput(std::move(other.m_FileInput)) {} IOInterface(IOInterface&& other) : m_FileOutput(std::move(other.m_FileOutput)), m_FileInput(std::move(other.m_FileInput)) {}
DataBuffer Read(std::size_t a_Amount) { DataBuffer Read(std::size_t a_Amount) {
DataBuffer buffer; DataBuffer buffer;
buffer.Resize(a_Amount); buffer.Resize(a_Amount);
m_FileInput.read(reinterpret_cast<char*>(buffer.data()), a_Amount); assert(m_FileInput != nullptr);
m_FileInput->read(reinterpret_cast<char*>(buffer.data()), a_Amount);
return buffer; return buffer;
} }
void Write(const sp::DataBuffer& a_Data) { void Write(const sp::DataBuffer& a_Data) {
m_FileOutput.write(reinterpret_cast<const char*>(a_Data.data()), a_Data.GetSize()); assert(m_FileOutput != nullptr);
m_FileOutput.flush(); m_FileOutput->write(reinterpret_cast<const char*>(a_Data.data()), a_Data.GetSize());
m_FileOutput->flush();
} }
}; };

2
test/test_file.cpp
View File

@@ -23,7 +23,7 @@ using FileStream = sp::IOStream<sp::FileTag, sp::PacketDispatcher, sp::PacketFac
int main() { int main() {
auto handler = std::make_shared<CustomPacketHandler>(); auto handler = std::make_shared<CustomPacketHandler>();
FileStream stream(sp::FileIO{"test.txt"}); FileStream stream(sp::FileIO{"test.txt", "text.txt"});
stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler); stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler);
stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler); stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler);