chromium/components/segmentation_platform/embedder/default_model/tab_resumption_ranker.cc

// 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.

#include "components/segmentation_platform/embedder/default_model/tab_resumption_ranker.h"

#include "base/task/sequenced_task_runner.h"
#include "components/segmentation_platform/embedder/input_delegate/tab_session_source.h"
#include "components/segmentation_platform/internal/database/ukm_types.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/public/constants.h"
#include "components/segmentation_platform/public/features.h"
#include "components/segmentation_platform/public/model_provider.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"
#include "services/metrics/public/cpp/ukm_builders.h"

namespace segmentation_platform {

namespace {
SegmentId;

constexpr SegmentId kSegmentId = …;
constexpr uint64_t kTabResumptionRankerVersion = …;

#define DECL_SQL_FEATURE_SINGLE_UKM_METRIC(name, query, event_name,        \
                                           metric_name) …

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

DECL_SQL_FEATURE_SINGLE_UKM_METRIC(…);

constexpr std::array<MetadataWriter::SqlFeature, 11> kSqlFeatures = …;

constexpr std::array<MetadataWriter::CustomInput::Arg, 1> kBindValueArg{ … };
constexpr MetadataWriter::CustomInput kBindValue{ … };

constexpr std::array<MetadataWriter::UMAFeature, 1> kOutput = …;

}  // namespace

// static
std::unique_ptr<Config> TabResumptionRanker::GetConfig() { … }

TabResumptionRanker::TabResumptionRanker()
    : … { … }
TabResumptionRanker::~TabResumptionRanker() = default;

std::unique_ptr<DefaultModelProvider::ModelConfig>
TabResumptionRanker::GetModelConfig() { … }

void TabResumptionRanker::ExecuteModelWithInput(
    const ModelProvider::Request& inputs,
    ExecutionCallback callback) { … }

}  // namespace segmentation_platform