Made the last scan step (checking for removed files) faster on big libraries
This commit is contained in:
@@ -228,6 +228,32 @@ namespace lms::db
|
||||
.resultValue();
|
||||
}
|
||||
|
||||
void Track::find(Session& session, TrackId& lastRetrievedTrack, std::size_t batchSize, bool& moreResults, const std::function<void(const Track::pointer&)>& func)
|
||||
{
|
||||
session.checkReadTransaction();
|
||||
|
||||
auto collection{ session.getDboSession().find<Track>()
|
||||
.orderBy("id")
|
||||
.where("id > ?").bind(lastRetrievedTrack)
|
||||
.limit(static_cast<int>(batchSize) + 1)
|
||||
.resultList() };
|
||||
|
||||
moreResults = false;
|
||||
|
||||
std::size_t count{};
|
||||
for (auto itResult{ collection.begin() }; itResult != collection.end(); ++itResult)
|
||||
{
|
||||
if (count++ == batchSize)
|
||||
{
|
||||
moreResults = true;
|
||||
break;
|
||||
}
|
||||
|
||||
func(*itResult);
|
||||
lastRetrievedTrack = (*itResult)->getId();
|
||||
}
|
||||
}
|
||||
|
||||
bool Track::exists(Session& session, TrackId id)
|
||||
{
|
||||
session.checkReadTransaction();
|
||||
|
||||
@@ -110,7 +110,8 @@ namespace lms::db
|
||||
static std::size_t getCount(Session& session);
|
||||
static pointer findByPath(Session& session, const std::filesystem::path& p);
|
||||
static pointer find(Session& session, TrackId id);
|
||||
static bool exists(Session& session, TrackId id);
|
||||
static void find(Session& session, TrackId& lastRetrievedTrack, std::size_t batchSize, bool& moreResults, const std::function<void(const Track::pointer&)>& func);
|
||||
static bool exists(Session& session, TrackId id);
|
||||
static std::vector<pointer> findByRecordingMBID(Session& session, const core::UUID& MBID);
|
||||
static std::vector<pointer> findByMBID(Session& session, const core::UUID& MBID);
|
||||
static RangeResults<TrackId> findSimilarTrackIds(Session& session, const std::vector<TrackId>& trackIds, std::optional<Range> range = std::nullopt);
|
||||
|
||||
@@ -63,6 +63,62 @@ namespace lms::db::tests
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(DatabaseFixture, Track_findByRangedIdBased)
|
||||
{
|
||||
ScopedTrack track1{ session, "MyTrackFile1" };
|
||||
ScopedTrack track2{ session, "MyTrackFile1" };
|
||||
ScopedTrack track3{ session, "MyTrackFile1" };
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
|
||||
bool moreResults;
|
||||
TrackId lastRetrievedTrackId;
|
||||
std::vector<Track::pointer> visitedTracks;
|
||||
Track::find(session, lastRetrievedTrackId, 10, moreResults, [&](const Track::pointer& track)
|
||||
{
|
||||
visitedTracks.push_back(track);
|
||||
});
|
||||
ASSERT_EQ(visitedTracks.size(), 3);
|
||||
EXPECT_EQ(visitedTracks[0]->getId(), track1.getId());
|
||||
EXPECT_EQ(visitedTracks[1]->getId(), track2.getId());
|
||||
EXPECT_EQ(visitedTracks[2]->getId(), track3.getId());
|
||||
EXPECT_FALSE(moreResults);
|
||||
EXPECT_EQ(lastRetrievedTrackId, track3.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
|
||||
bool moreResults;
|
||||
TrackId lastRetrievedTrackId{ track1.getId() };
|
||||
std::vector<Track::pointer> visitedTracks;
|
||||
Track::find(session, lastRetrievedTrackId, 1, moreResults, [&](const Track::pointer& track)
|
||||
{
|
||||
visitedTracks.push_back(track);
|
||||
});
|
||||
ASSERT_EQ(visitedTracks.size(), 1);
|
||||
EXPECT_EQ(visitedTracks[0]->getId(), track2.getId());
|
||||
EXPECT_TRUE(moreResults);
|
||||
EXPECT_EQ(lastRetrievedTrackId, track2.getId());
|
||||
}
|
||||
|
||||
{
|
||||
auto transaction{ session.createReadTransaction() };
|
||||
|
||||
bool moreResults;
|
||||
TrackId lastRetrievedTrackId{ track1.getId() };
|
||||
std::vector<Track::pointer> visitedTracks;
|
||||
Track::find(session, lastRetrievedTrackId, 0, moreResults, [&](const Track::pointer& track)
|
||||
{
|
||||
visitedTracks.push_back(track);
|
||||
});
|
||||
ASSERT_EQ(visitedTracks.size(), 0);
|
||||
EXPECT_TRUE(moreResults);
|
||||
EXPECT_EQ(lastRetrievedTrackId, track1.getId());
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(DatabaseFixture, Track_MediaLibrary)
|
||||
{
|
||||
ScopedTrack track{ session, "MyTrackFile" };
|
||||
|
||||
Reference in New Issue
Block a user