ukm_data_manager_impl.h 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. // Copyright 2022 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_UKM_DATA_MANAGER_IMPL_H_
  5. #define COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_UKM_DATA_MANAGER_IMPL_H_
  6. #include <memory>
  7. #include "base/memory/raw_ptr.h"
  8. #include "base/memory/weak_ptr.h"
  9. #include "base/sequence_checker.h"
  10. #include "components/segmentation_platform/internal/ukm_data_manager.h"
  11. #include "third_party/abseil-cpp/absl/types/optional.h"
  12. namespace segmentation_platform {
  13. class UkmDatabase;
  14. class UkmObserver;
  15. class UrlSignalHandler;
  16. class UkmDataManagerImpl : public UkmDataManager {
  17. public:
  18. UkmDataManagerImpl();
  19. ~UkmDataManagerImpl() override;
  20. UkmDataManagerImpl(UkmDataManagerImpl&) = delete;
  21. UkmDataManagerImpl& operator=(UkmDataManagerImpl&) = delete;
  22. void InitializeForTesting(std::unique_ptr<UkmDatabase> ukm_database,
  23. UkmObserver* ukm_observer);
  24. // UkmDataManager implementation:
  25. void Initialize(const base::FilePath& database_path,
  26. UkmObserver* ukm_observer) override;
  27. bool IsUkmEngineEnabled() override;
  28. void StartObservingUkm(const UkmConfig& config) override;
  29. void PauseOrResumeObservation(bool pause) override;
  30. UrlSignalHandler* GetOrCreateUrlHandler() override;
  31. UkmDatabase* GetUkmDatabase() override;
  32. void OnEntryAdded(ukm::mojom::UkmEntryPtr entry) override;
  33. void OnUkmSourceUpdated(ukm::SourceId source_id,
  34. const std::vector<GURL>& urls) override;
  35. void AddRef() override;
  36. void RemoveRef() override;
  37. private:
  38. // Helper method for initializing this object.
  39. void InitiailizeImpl(std::unique_ptr<UkmDatabase> ukm_database,
  40. UkmObserver* ukm_observer);
  41. void RunCleanupTask();
  42. int ref_count_ = 0;
  43. raw_ptr<UkmObserver> ukm_observer_ = nullptr;
  44. std::unique_ptr<UkmDatabase> ukm_database_;
  45. std::unique_ptr<UrlSignalHandler> url_signal_handler_;
  46. std::unique_ptr<UkmConfig> pending_ukm_config_;
  47. absl::optional<bool> is_ukm_allowed_;
  48. SEQUENCE_CHECKER(sequence_check_);
  49. base::WeakPtrFactory<UkmDataManagerImpl> weak_factory_{this};
  50. };
  51. } // namespace segmentation_platform
  52. #endif // COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_UKM_DATA_MANAGER_IMPL_H_