Fix premature remembering of saved entity idswebstat-0.2.1

Don't persist entity ids saved to the DB until the transaction is
committed. Prevents the issue where a later DB operation fails, the
transaction is rolled back, but we still think the entity has been
saved.

1 files changed, 3 insertions, 2 deletions

diff --git a/src/ingestor.hpp b/src/ingestor.hpp
index a20071e..8be7360 100644
--- a/src/ingestor.hpp
+++ b/src/ingestor.hpp
@@ -60,13 +60,15 @@ namespace WebStat {
 		size_t linesParsed = 0;
 		size_t linesDiscarded = 0;
 		size_t linesParked = 0;
+		mutable std::flat_set<Crc32Value> existingEntities;
 
 		using JobLastRunTime = std::chrono::system_clock::time_point;
 		JobLastRunTime lastRunIngestParkedLines;
 
 	private:
 		static constexpr size_t MAX_NEW_ENTITIES = 6;
-		void storeEntities(DB::Connection *, std::span<const std::optional<Entity>>) const;
+		using NewEntityIds = std::array<std::optional<Crc32Value>, MAX_NEW_ENTITIES>;
+		NewEntityIds storeEntities(DB::Connection *, std::span<const std::optional<Entity>>) const;
 		using NewEntities = std::array<std::optional<Entity>, MAX_NEW_ENTITIES>;
 		template<typename... T> NewEntities newEntities(const std::tuple<T...> &) const;
 		void handleCurlOperations();
@@ -75,7 +77,6 @@ namespace WebStat {
 		void jobIngestParkedLine(const std::filesystem::path &, uintmax_t size);
 
 		using CurlOperations = std::map<CURL *, std::unique_ptr<CurlOperation>>;
-		mutable std::flat_set<Crc32Value> existingEntities;
 		uint32_t hostnameId;
 		CurlMultiPtr curl;
 		mutable CurlOperations curlOperations;