components/segmentation_platform/internal/selection/cached_result_writer.h

// 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 COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SELECTION_CACHED_RESULT_WRITER_H_
#define COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SELECTION_CACHED_RESULT_WRITER_H_

#include <string>

#include "base/functional/callback_helpers.h"
#include "base/memory/raw_ptr.h"
#include "base/time/clock.h"
#include "components/segmentation_platform/internal/platform_options.h"
#include "components/segmentation_platform/internal/selection/client_result_prefs.h"
#include "components/segmentation_platform/public/result.h"
namespace segmentation_platform {
struct Config;
class ClientResultPrefs;

// CachedResultWriter layer writes client results to prefs if the result in
// prefs for the given client has expired or not present. It also get the fresh
// results for the  client after model execution inorder to update it in prefs.
class CachedResultWriter {
 public:
  CachedResultWriter(std::unique_ptr<ClientResultPrefs> prefs,
                     base::Clock* clock);

  ~CachedResultWriter();

  // Disallow copy/assign.
  CachedResultWriter(CachedResultWriter&) = delete;
  CachedResultWriter& operator=(CachedResultWriter&) = delete;

  // Updates the prefs only if the previous result in the pref is expired or
  // unavailable or `force_refresh_results` is set as true. In both the above
  // cases, fetch result for model either from database or running the model.
  void UpdatePrefsIfExpired(Config* config,
                            proto::ClientResult client_result,
                            const PlatformOptions& platform_options);

 private:
  // Checks the following to determine whether to update pref with new result.
  // 1. Previous model results for client are either expired or unavailable.
  // 2. `force_refresh_results` option is set to true.
  bool IsPrefUpdateRequiredForClient(Config* config,
                                     const PlatformOptions& platform_options);

  // Updates the supplied `client_result` as new result for the client in prefs.
  void UpdateNewClientResultToPrefs(Config* config,
                                    const proto::ClientResult& client_result);

  // Helper class to read/write results to the prefs.
  std::unique_ptr<ClientResultPrefs> result_prefs_;

  // The time provider.
  const raw_ptr<base::Clock> clock_;

  base::WeakPtrFactory<CachedResultWriter> weak_ptr_factory_{this};
};

}  // namespace segmentation_platform

#endif  // COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SELECTION_CACHED_RESULT_WRITER_H_