chromium/components/segmentation_platform/internal/data_collection/training_data_cache.cc

// Copyright 2022 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/internal/data_collection/training_data_cache.h"

#include "base/rand_util.h"
#include "base/time/time.h"

namespace segmentation_platform {

TrainingData;

TrainingDataCache::TrainingDataCache(SegmentInfoDatabase* segment_info_database)
    : … { … }

TrainingDataCache::~TrainingDataCache() = default;

void TrainingDataCache::StoreInputs(SegmentId segment_id,
                                    ModelSource model_source,
                                    const TrainingData& data,
                                    bool save_to_db) { … }

void TrainingDataCache::GetInputsAndDelete(SegmentId segment_id,
                                           ModelSource model_source,
                                           TrainingRequestId request_id,
                                           TrainingDataCallback callback) { … }

std::optional<TrainingRequestId> TrainingDataCache::GetRequestId(
    SegmentId segment_id,
    ModelSource model_source) { … }

TrainingRequestId TrainingDataCache::GenerateNextId() { … }

}  // namespace segmentation_platform