| // 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/bind.h" |
| #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_cache_observer.h" |
| #include "net/reporting/reporting_context.h" |
| #include "net/reporting/reporting_policy.h" |
| #include "net/reporting/reporting_report.h" |
| |
| namespace net { |
| |
| namespace { |
| |
| class ReportingGarbageCollectorImpl : public ReportingGarbageCollector, |
| public ReportingCacheObserver { |
| public: |
| ReportingGarbageCollectorImpl(ReportingContext* context) |
| : context_(context), timer_(std::make_unique<base::OneShotTimer>()) { |
| context_->AddCacheObserver(this); |
| } |
| |
| // ReportingGarbageCollector implementation: |
| |
| ~ReportingGarbageCollectorImpl() override { |
| context_->RemoveCacheObserver(this); |
| } |
| |
| void SetTimerForTesting(std::unique_ptr<base::OneShotTimer> timer) override { |
| timer_ = std::move(timer); |
| } |
| |
| // ReportingObserver implementation: |
| void OnReportsUpdated() override { |
| if (timer_->IsRunning()) |
| return; |
| |
| timer_->Start(FROM_HERE, context_->policy().garbage_collection_interval, |
| base::BindOnce(&ReportingGarbageCollectorImpl::CollectGarbage, |
| base::Unretained(this))); |
| } |
| |
| private: |
| // TODO(crbug.com/912622): Garbage collect clients, reports with no matching |
| // endpoints. |
| 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_->RemoveCacheObserver(this); |
| context_->cache()->RemoveReports(failed_reports, |
| ReportingReport::Outcome::ERASED_FAILED); |
| context_->cache()->RemoveReports(expired_reports, |
| ReportingReport::Outcome::ERASED_EXPIRED); |
| context_->AddCacheObserver(this); |
| } |
| |
| ReportingContext* context_; |
| std::unique_ptr<base::OneShotTimer> timer_; |
| }; |
| |
| } // namespace |
| |
| // static |
| std::unique_ptr<ReportingGarbageCollector> ReportingGarbageCollector::Create( |
| ReportingContext* context) { |
| return std::make_unique<ReportingGarbageCollectorImpl>(context); |
| } |
| |
| ReportingGarbageCollector::~ReportingGarbageCollector() = default; |
| |
| } // namespace net |