1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
// Copyright 2021 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/selection/segmentation_result_prefs.h"
#include "base/json/values_util.h"
#include "components/prefs/pref_service.h"
#include "components/prefs/scoped_user_pref_update.h"
#include "components/segmentation_platform/internal/constants.h"
namespace segmentation_platform {
SelectedSegment::SelectedSegment(SegmentId segment_id,
std::optional<float> rank)
: segment_id(segment_id), rank(rank), in_use(false) {}
SelectedSegment::~SelectedSegment() = default;
SegmentationResultPrefs::SegmentationResultPrefs(PrefService* pref_service)
: prefs_(pref_service) {}
void SegmentationResultPrefs::SaveSegmentationResultToPref(
const std::string& result_key,
const std::optional<SelectedSegment>& selected_segment) {
ScopedDictPrefUpdate update(prefs_, kSegmentationResultPref);
base::Value::Dict& dictionary = update.Get();
if (!selected_segment.has_value()) {
dictionary.Remove(result_key);
return;
}
base::Value::Dict segmentation_result;
segmentation_result.Set("segment_id", selected_segment->segment_id);
if (selected_segment->rank)
segmentation_result.Set("segment_rank", *selected_segment->rank);
segmentation_result.Set("in_use", selected_segment->in_use);
segmentation_result.Set("selection_time",
base::TimeToValue(selected_segment->selection_time));
dictionary.Set(result_key, std::move(segmentation_result));
}
std::optional<SelectedSegment>
SegmentationResultPrefs::ReadSegmentationResultFromPref(
const std::string& result_key) {
const base::Value::Dict& dictionary =
prefs_->GetDict(kSegmentationResultPref);
const base::Value* value = dictionary.Find(result_key);
if (!value)
return std::nullopt;
const base::Value::Dict& segmentation_result = value->GetDict();
std::optional<int> segment_id = segmentation_result.FindInt("segment_id");
std::optional<float> rank = segmentation_result.FindDouble("segment_rank");
std::optional<bool> in_use = segmentation_result.FindBool("in_use");
std::optional<base::Time> selection_time =
base::ValueToTime(segmentation_result.Find("selection_time"));
SelectedSegment selected_segment(static_cast<SegmentId>(segment_id.value()),
rank);
if (in_use.has_value())
selected_segment.in_use = in_use.value();
if (selection_time.has_value())
selected_segment.selection_time = selection_time.value();
return selected_segment;
}
} // namespace segmentation_platform
|
