// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef IOS_CHROME_BROWSER_SEGMENTATION_PLATFORM_MODEL_UKM_DATABASE_CLIENT_H_
#define IOS_CHROME_BROWSER_SEGMENTATION_PLATFORM_MODEL_UKM_DATABASE_CLIENT_H_
#import "base/check.h"
#import "base/memory/raw_ptr.h"
#import "base/no_destructor.h"
#import "base/synchronization/lock.h"
#import "base/thread_annotations.h"
#import "components/segmentation_platform/internal/signals/ukm_observer.h"
#import "components/segmentation_platform/internal/ukm_data_manager.h"
#import "ios/chrome/browser/shared/model/profile/profile_ios.h"
namespace ukm {
class UkmRecorderImpl;
}
namespace segmentation_platform {
class UkmDataManager;
class UkmObserver;
// Provides UKM functionality to the segmentation platform service(s).
class UkmDatabaseClient {
public:
UkmDatabaseClient();
~UkmDatabaseClient();
UkmDatabaseClient(const UkmDatabaseClient&) = delete;
UkmDatabaseClient& operator=(const UkmDatabaseClient&) = delete;
// Must be called before any profiles (segmentation services) are created.
// `in_memory_database` is used in tests only to create in memory databases to
// make tests faster.
void PreProfileInit(bool in_memory_database);
void StartObservation();
// Must be called after profiles are destroyed, but before metrics service is
// destroyed.
void PostMessageLoopRun();
// UkmDataManager will be valid for the lifetime of all the profiles. It is
// created before profiles are created at startup. It is safe to use this
// pointer till ProfileManagerDestroying() is called.
segmentation_platform::UkmDataManager* GetUkmDataManager();
// UKM observer will use the test recorder to observe metrics.
void set_ukm_recorder_for_testing(ukm::UkmRecorderImpl* ukm_recorder) {
DCHECK(!ukm_observer_);
ukm_recorder_for_testing_ = ukm_recorder;
}
UkmObserver* ukm_observer_for_testing() { return ukm_observer_.get(); }
void TearDownForTesting();
private:
raw_ptr<ukm::UkmRecorderImpl> ukm_recorder_for_testing_;
std::unique_ptr<UkmObserver> ukm_observer_;
std::unique_ptr<UkmDataManager> ukm_data_manager_;
};
// Class to own the UkmDatabaseClient. Supports overriding instances in tests.
class UkmDatabaseClientHolder {
public:
// Always returns the main client instance, unless a client was set for
// testing. Can be called with nullptr to get the main instance when `profile`
// is not created.
static UkmDatabaseClient& GetClientInstance(ChromeBrowserState* profile);
// Sets or removes the instance used by `profile` for testing. Thread safe,
// and GetClientInstance() will return the new client based on the profile.
// Note that if GetClientInstance() is called with nullptr, the main instance
// will still be returned.
static void SetUkmClientForTesting(ChromeBrowserState* profile,
UkmDatabaseClient* client);
UkmDatabaseClientHolder(const UkmDatabaseClientHolder&) = delete;
UkmDatabaseClientHolder& operator=(const UkmDatabaseClientHolder&) = delete;
private:
friend base::NoDestructor<UkmDatabaseClientHolder>;
static UkmDatabaseClientHolder& GetInstance();
UkmDatabaseClientHolder();
~UkmDatabaseClientHolder();
void SetUkmClientForTestingInternal(ChromeBrowserState* profile,
UkmDatabaseClient* client);
base::Lock lock_;
std::map<raw_ptr<ChromeBrowserState>, raw_ptr<UkmDatabaseClient>>
clients_for_testing_ GUARDED_BY(lock_);
std::unique_ptr<UkmDatabaseClient> main_client_;
};
} // namespace segmentation_platform
#endif // IOS_CHROME_BROWSER_SEGMENTATION_PLATFORM_MODEL_UKM_DATABASE_CLIENT_H_