BigFile access

Bigfoot/Sources/Engine/BigFile/BigFile.cpp

@@ -8,6 +8,11 @@
 
 #include <Engine/EngineAssertHandler.hpp>
 
+#include <EASTL/algorithm.h>
+#include <EASTL/sort.h>
+
+#include <fmt/format.h>
+
 namespace Bigfoot
 {
 #ifdef BIGFOOT_TOOLMODE
@@ -60,45 +65,110 @@ BigFile::~BigFile()
 
 /****************************************************************************************/
 
-eastl::vector<std::byte> BigFile::GetAssetFlatbuffer(const UUID& p_uuid) const
+eastl::vector<eastl::vector<std::byte>> BigFile::GetAssetFlatbuffers(const eastl::span<const UUID> p_uuids) const
 {
-    BigFile::Request assetRequest {*this,
-                                   "SELECT Asset.Asset FROM Asset "
-                                   "JOIN AssetMetadata ON AssetMetadata.UUID = Asset.UUID "
-                                   "WHERE Asset.UUID = ?1;"};
-    assetRequest.Bind(1, p_uuid.ToBytes());
-
-    if (!assetRequest.Step())
+    eastl::vector<eastl::vector<std::byte>> flatbuffers;
+    if (p_uuids.empty())
     {
-        return {};
+        return flatbuffers;
     }
 
-    const eastl::span<const std::byte> flatbuffer = assetRequest.Get(0);
-    return eastl::vector<std::byte> {flatbuffer.begin(), flatbuffer.end()};
+    eastl::vector<UUID> uniqueUUIDs {p_uuids.begin(), p_uuids.end()};
+    eastl::sort(uniqueUUIDs.begin(), uniqueUUIDs.end());
+    uniqueUUIDs.erase(eastl::unique(uniqueUUIDs.begin(), uniqueUUIDs.end()), uniqueUUIDs.end());
+
+    eastl::string placeholders;
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        if (i > 0)
+        {
+            placeholders += ", ";
+        }
+        placeholders += "?";
+        placeholders += std::to_string(i + 1).c_str();
+    }
+
+    const std::string sqlRequest = fmt::format("SELECT Asset.Asset FROM Asset "
+                                                     "JOIN AssetMetadata ON AssetMetadata.UUID = Asset.UUID "
+                                                     "WHERE Asset.UUID IN ({});",
+                                                     placeholders);
+
+    BigFile::Request assetRequest {*this, sqlRequest.c_str()};
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        assetRequest.Bind(static_cast<std::uint32_t>(i + 1), uniqueUUIDs[i].ToBytes());
+    }
+
+    flatbuffers.reserve(uniqueUUIDs.size());
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        if (!assetRequest.Step())
+        {
+            break;
+        }
+
+        const eastl::span<const std::byte> flatbuffer = assetRequest.Get(0);
+        flatbuffers.emplace_back(flatbuffer.begin(), flatbuffer.end());
+    }
+
+    return flatbuffers;
 }
 
 /****************************************************************************************/
 
-AssetMetadata BigFile::GetAssetMetadata(const UUID& p_uuid) const
+eastl::vector<AssetMetadata> BigFile::GetAssetMetadatas(const eastl::span<const UUID> p_uuids) const
 {
-    BigFile::Request assetRequest {
-      *this,
-      "SELECT UUID, Name, TypeID, TypeName, CreateTime, ModificationTime FROM AssetMetadata WHERE UUID = ?1;"};
-    assetRequest.Bind(1, p_uuid.ToBytes());
-
-    if (!assetRequest.Step())
+    eastl::vector<AssetMetadata> metadatas;
+    if (p_uuids.empty())
     {
-        return {};
+        return metadatas;
     }
 
-    return AssetMetadata {.m_uuid = UUID {std::span<const std::byte, UUID::UUID_BYTE_SIZE> {
-                            static_cast<eastl::span<const std::byte>>(assetRequest.Get(0)).data(),
-                            UUID::UUID_BYTE_SIZE}},
-                          .m_name = eastl::string {static_cast<eastl::string_view>(assetRequest.Get(1))},
-                          .m_typeID = assetRequest.Get(2),
-                          .m_typeName = eastl::string {static_cast<eastl::string_view>(assetRequest.Get(3))},
-                          .m_creationTime = Time {static_cast<std::uint64_t>(assetRequest.Get(4))},
-                          .m_modificationTime = Time {static_cast<std::uint64_t>(assetRequest.Get(5))}};
+    eastl::vector<UUID> uniqueUUIDs {p_uuids.begin(), p_uuids.end()};
+    eastl::sort(uniqueUUIDs.begin(), uniqueUUIDs.end());
+    uniqueUUIDs.erase(eastl::unique(uniqueUUIDs.begin(), uniqueUUIDs.end()), uniqueUUIDs.end());
+
+    eastl::string placeholders;
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        if (i > 0)
+        {
+            placeholders += ", ";
+        }
+        placeholders += "?";
+        placeholders += std::to_string(i + 1).c_str();
+    }
+
+    const std::string sqlRequest = fmt::format("SELECT UUID, Name, TypeID, TypeName, CreateTime, ModificationTime "
+                                              "FROM AssetMetadata WHERE UUID IN ({});",
+                                              placeholders);
+
+    BigFile::Request assetRequest {*this, sqlRequest.c_str()};
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        assetRequest.Bind(static_cast<std::uint32_t>(i + 1), uniqueUUIDs[i].ToBytes());
+    }
+
+    metadatas.reserve(uniqueUUIDs.size());
+    for (std::size_t i = 0; i < uniqueUUIDs.size(); ++i)
+    {
+        if (!assetRequest.Step())
+        {
+            break;
+        }
+
+        metadatas.push_back(
+          AssetMetadata {.m_uuid = UUID {std::span<const std::byte, UUID::UUID_BYTE_SIZE> {
+                           static_cast<eastl::span<const std::byte>>(assetRequest.Get(0)).data(),
+                           UUID::UUID_BYTE_SIZE}},
+                         .m_name = eastl::string {static_cast<eastl::string_view>(assetRequest.Get(1))},
+                         .m_typeID = assetRequest.Get(2),
+                         .m_typeName = eastl::string {static_cast<eastl::string_view>(assetRequest.Get(3))},
+                         .m_creationTime = Time {static_cast<std::uint64_t>(assetRequest.Get(4))},
+                         .m_modificationTime = Time {static_cast<std::uint64_t>(assetRequest.Get(5))}});
+    }
+
+    return metadatas;
 }
 
 /****************************************************************************************/