mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-28 00:06:09 +03:00
93 lines
2.9 KiB
C++
93 lines
2.9 KiB
C++
// Copyright 2017 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "net/reporting/reporting_garbage_collector.h"
|
|
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "base/time/tick_clock.h"
|
|
#include "base/time/time.h"
|
|
#include "base/timer/timer.h"
|
|
#include "net/reporting/reporting_cache.h"
|
|
#include "net/reporting/reporting_context.h"
|
|
#include "net/reporting/reporting_observer.h"
|
|
#include "net/reporting/reporting_policy.h"
|
|
#include "net/reporting/reporting_report.h"
|
|
|
|
namespace net {
|
|
|
|
namespace {
|
|
|
|
class ReportingGarbageCollectorImpl : public ReportingGarbageCollector,
|
|
public ReportingObserver {
|
|
public:
|
|
ReportingGarbageCollectorImpl(ReportingContext* context)
|
|
: context_(context), timer_(std::make_unique<base::OneShotTimer>()) {
|
|
context_->AddObserver(this);
|
|
}
|
|
|
|
// ReportingGarbageCollector implementation:
|
|
|
|
~ReportingGarbageCollectorImpl() override {
|
|
context_->RemoveObserver(this);
|
|
}
|
|
|
|
void SetTimerForTesting(std::unique_ptr<base::Timer> timer) override {
|
|
timer_ = std::move(timer);
|
|
}
|
|
|
|
// ReportingObserver implementation:
|
|
void OnCacheUpdated() override {
|
|
if (timer_->IsRunning())
|
|
return;
|
|
|
|
timer_->Start(
|
|
FROM_HERE, context_->policy().garbage_collection_interval,
|
|
base::BindRepeating(&ReportingGarbageCollectorImpl::CollectGarbage,
|
|
base::Unretained(this)));
|
|
}
|
|
|
|
private:
|
|
void CollectGarbage() {
|
|
base::TimeTicks now = context_->tick_clock()->NowTicks();
|
|
const ReportingPolicy& policy = context_->policy();
|
|
|
|
std::vector<const ReportingReport*> all_reports;
|
|
context_->cache()->GetReports(&all_reports);
|
|
|
|
std::vector<const ReportingReport*> failed_reports;
|
|
std::vector<const ReportingReport*> expired_reports;
|
|
for (const ReportingReport* report : all_reports) {
|
|
if (report->attempts >= policy.max_report_attempts)
|
|
failed_reports.push_back(report);
|
|
else if (now - report->queued >= policy.max_report_age)
|
|
expired_reports.push_back(report);
|
|
}
|
|
|
|
// Don't restart the timer on the garbage collector's own updates.
|
|
context_->RemoveObserver(this);
|
|
context_->cache()->RemoveReports(failed_reports,
|
|
ReportingReport::Outcome::ERASED_FAILED);
|
|
context_->cache()->RemoveReports(expired_reports,
|
|
ReportingReport::Outcome::ERASED_EXPIRED);
|
|
context_->AddObserver(this);
|
|
}
|
|
|
|
ReportingContext* context_;
|
|
std::unique_ptr<base::Timer> timer_;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
// static
|
|
std::unique_ptr<ReportingGarbageCollector> ReportingGarbageCollector::Create(
|
|
ReportingContext* context) {
|
|
return std::make_unique<ReportingGarbageCollectorImpl>(context);
|
|
}
|
|
|
|
ReportingGarbageCollector::~ReportingGarbageCollector() = default;
|
|
|
|
} // namespace net
|