feat(datastore): add getter for uint32_t

This commit is contained in:
fallenoak 2023-01-08 14:03:40 -06:00
parent 42384cbf77
commit 28e59a11ef
No known key found for this signature in database
GPG key ID: 7628F8E61AEA070D
3 changed files with 29 additions and 0 deletions

View file

@ -58,6 +58,22 @@ CDataStore& CDataStore::Get(uint8_t& val) {
return *this;
}
CDataStore& CDataStore::Get(uint32_t& val) {
STORM_ASSERT(this->IsFinal());
auto bytes = sizeof(val);
if (this->FetchRead(this->m_read, bytes)) {
auto ofs = this->m_read - this->m_base;
auto ptr = &this->m_data[ofs];
val = *reinterpret_cast<uint32_t*>(ptr);
this->m_read += bytes;
}
return *this;
}
void CDataStore::GetBufferParams(const void** data, uint32_t* size, uint32_t* alloc) const {
if (data) {
*data = this->m_data;

View file

@ -30,6 +30,7 @@ class CDataStore {
int32_t FetchRead(uint32_t pos, uint32_t bytes);
int32_t FetchWrite(uint32_t pos, uint32_t bytes, const char* fileName, int32_t lineNumber);
CDataStore& Get(uint8_t& val);
CDataStore& Get(uint32_t& val);
CDataStore& GetDataInSitu(void*& val, uint32_t bytes);
int32_t IsFinal();
CDataStore& Put(uint8_t val);

View file

@ -20,4 +20,16 @@ TEST_CASE("CDataStore::Get", "[datastore]") {
REQUIRE(readVal == writeVal);
}
SECTION("gets uint32_t") {
uint32_t writeVal = 0x12345678;
uint32_t readVal = 0x12345678;
CDataStore msg;
msg.Put(writeVal);
msg.Finalize();
msg.Get(readVal);
REQUIRE(readVal == writeVal);
}
}