| Min Qin | 75bee1b | 2022-02-05 03:57:34 | [diff] [blame] | 1 | // Copyright 2022 The Chromium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "components/segmentation_platform/internal/segmentation_ukm_helper.h" |
| 6 | |
| 7 | #include "base/bit_cast.h" |
| 8 | #include "base/metrics/field_trial_params.h" |
| 9 | #include "base/strings/string_number_conversions.h" |
| 10 | #include "base/strings/string_split.h" |
| 11 | #include "components/segmentation_platform/internal/stats.h" |
| 12 | #include "components/segmentation_platform/public/config.h" |
| 13 | #include "components/segmentation_platform/public/features.h" |
| 14 | #include "services/metrics/public/cpp/ukm_builders.h" |
| 15 | #include "services/metrics/public/cpp/ukm_recorder.h" |
| 16 | |
| 17 | #define CALL_MEMBER_FN(obj, func) ((obj).*(func)) |
| 18 | #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x)[0]) |
| 19 | |
| 20 | using ukm::builders::Segmentation_ModelExecution; |
| 21 | |
| 22 | namespace { |
| 23 | using UkmMemberFn = |
| 24 | Segmentation_ModelExecution& (Segmentation_ModelExecution::*)(int64_t); |
| 25 | |
| 26 | const UkmMemberFn kSegmentationUkmMethods[] = { |
| 27 | &Segmentation_ModelExecution::SetInput0, |
| 28 | &Segmentation_ModelExecution::SetInput1, |
| 29 | &Segmentation_ModelExecution::SetInput2, |
| 30 | &Segmentation_ModelExecution::SetInput3, |
| 31 | &Segmentation_ModelExecution::SetInput4, |
| 32 | &Segmentation_ModelExecution::SetInput5, |
| 33 | &Segmentation_ModelExecution::SetInput6, |
| 34 | &Segmentation_ModelExecution::SetInput7, |
| 35 | &Segmentation_ModelExecution::SetInput8, |
| 36 | &Segmentation_ModelExecution::SetInput9, |
| 37 | &Segmentation_ModelExecution::SetInput10, |
| 38 | &Segmentation_ModelExecution::SetInput11, |
| 39 | &Segmentation_ModelExecution::SetInput12, |
| 40 | &Segmentation_ModelExecution::SetInput13, |
| 41 | &Segmentation_ModelExecution::SetInput14, |
| 42 | &Segmentation_ModelExecution::SetInput15, |
| 43 | &Segmentation_ModelExecution::SetInput16, |
| 44 | &Segmentation_ModelExecution::SetInput17, |
| 45 | &Segmentation_ModelExecution::SetInput18, |
| 46 | &Segmentation_ModelExecution::SetInput19, |
| 47 | &Segmentation_ModelExecution::SetInput20, |
| 48 | &Segmentation_ModelExecution::SetInput21, |
| 49 | &Segmentation_ModelExecution::SetInput22, |
| 50 | &Segmentation_ModelExecution::SetInput23, |
| 51 | &Segmentation_ModelExecution::SetInput24, |
| 52 | &Segmentation_ModelExecution::SetInput25, |
| 53 | &Segmentation_ModelExecution::SetInput26, |
| 54 | &Segmentation_ModelExecution::SetInput27, |
| 55 | &Segmentation_ModelExecution::SetInput28, |
| 56 | &Segmentation_ModelExecution::SetInput29}; |
| 57 | |
| 58 | // Gets a set of segment IDs that are allowed to upload metrics. |
| 59 | base::flat_set<int> GetSegmentIdsAllowedForReporting() { |
| 60 | std::vector<std::string> segment_ids = base::SplitString( |
| 61 | base::GetFieldTrialParamValueByFeature( |
| 62 | segmentation_platform::features:: |
| 63 | kSegmentationStructuredMetricsFeature, |
| 64 | segmentation_platform::kSegmentIdsAllowedForReportingKey), |
| 65 | ",;", base::WhitespaceHandling::TRIM_WHITESPACE, |
| 66 | base::SplitResult::SPLIT_WANT_NONEMPTY); |
| 67 | base::flat_set<int> result; |
| 68 | for (const auto& id : segment_ids) { |
| 69 | int segment_id; |
| 70 | if (base::StringToInt(id, &segment_id)) |
| 71 | result.emplace(segment_id); |
| 72 | } |
| 73 | return result; |
| 74 | } |
| 75 | |
| 76 | } // namespace |
| 77 | |
| 78 | namespace segmentation_platform { |
|
|
