chromium/components/segmentation_platform/internal/database/segment_info_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/database/segment_info_cache.h"

#include <memory>
#include <optional>

#include "base/functional/callback.h"
#include "components/segmentation_platform/internal/proto/model_prediction.pb.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"

namespace segmentation_platform {

SegmentInfoCache::SegmentInfoCache() = default;

SegmentInfoCache::~SegmentInfoCache() = default;

const SegmentInfo* SegmentInfoCache::GetSegmentInfo(
    SegmentId segment_id,
    ModelSource model_source) const { … }

std::unique_ptr<SegmentInfoCache::SegmentInfoList>
SegmentInfoCache::GetSegmentInfoForSegments(
    const base::flat_set<SegmentId>& segment_ids,
    ModelSource model_source) const { … }

std::unique_ptr<SegmentInfoCache::SegmentInfoList>
SegmentInfoCache::GetSegmentInfoForBothModels(
    const base::flat_set<SegmentId>& segment_ids) const { … }

void SegmentInfoCache::UpdateSegmentInfo(
    SegmentId segment_id,
    ModelSource model_source,
    std::optional<SegmentInfo> segment_info) { … }

}  // namespace segmentation_platform