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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// 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.
#ifndef COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SCHEDULER_EXECUTION_SERVICE_H_
#define COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SCHEDULER_EXECUTION_SERVICE_H_
#include <memory>
#include <vector>
#include "base/containers/flat_set.h"
#include "base/memory/raw_ptr.h"
#include "base/task/sequenced_task_runner.h"
#include "base/time/clock.h"
#include "components/segmentation_platform/internal/data_collection/training_data_collector.h"
#include "components/segmentation_platform/internal/database/cached_result_provider.h"
#include "components/segmentation_platform/internal/execution/execution_request.h"
#include "components/segmentation_platform/internal/execution/model_manager_impl.h"
#include "components/segmentation_platform/internal/scheduler/model_execution_scheduler.h"
#include "components/segmentation_platform/public/input_context.h"
#include "components/segmentation_platform/public/proto/segmentation_platform.pb.h"
class PrefService;
namespace segmentation_platform {
namespace processing {
class FeatureListQueryProcessor;
class InputDelegateHolder;
}
struct PlatformOptions;
class ModelExecutor;
class ModelProviderFactory;
class SignalHandler;
class StorageService;
// Handles feature processing and model execution.
class ExecutionService {
public:
ExecutionService();
~ExecutionService();
ExecutionService(const ExecutionService&) = delete;
ExecutionService& operator=(const ExecutionService&) = delete;
void InitForTesting(
std::unique_ptr<processing::FeatureListQueryProcessor> feature_processor,
std::unique_ptr<ModelExecutor> executor,
std::unique_ptr<ModelExecutionScheduler> scheduler,
ModelManager* model_manager);
void Initialize(
StorageService* storage_service,
SignalHandler* signal_handler,
base::Clock* clock,
scoped_refptr<base::SequencedTaskRunner> task_runner,
const base::flat_set<SegmentId>& legacy_output_segment_ids,
ModelProviderFactory* model_provider_factory,
std::vector<raw_ptr<ModelExecutionScheduler::Observer,
VectorExperimental>>&& observers,
const PlatformOptions& platform_options,
std::unique_ptr<processing::InputDelegateHolder> input_delegate_holder,
PrefService* profile_prefs,
CachedResultProvider* cached_result_provider);
// Returns the training data collector.
TrainingDataCollector* training_data_collector() {
return training_data_collector_.get();
}
// DEPRECATED: New multi output supporting models doesn't use it.
// Called whenever a new or updated model is available. Must be a valid
// SegmentInfo with valid metadata and features.
void OnNewModelInfoReadyLegacy(const proto::SegmentInfo& segment_info);
// Gets the model provider for execution.
ModelProvider* GetModelProvider(SegmentId segment_id,
ModelSource model_source);
void RequestModelExecution(std::unique_ptr<ExecutionRequest> request);
void OverwriteModelExecutionResult(
proto::SegmentId segment_id,
const std::pair<float, ModelExecutionStatus>& result);
// Refreshes model results for all eligible models.
void RefreshModelResults();
// Executes daily maintenance and collection tasks.
void RunDailyTasks(bool is_startup);
processing::FeatureListQueryProcessor* feature_processor() {
return feature_list_query_processor_.get();
}
void set_training_data_collector_for_testing(
std::unique_ptr<TrainingDataCollector> collector) {
training_data_collector_ = std::move(collector);
}
private:
raw_ptr<StorageService> storage_service_ = nullptr;
// Training/inference input data generation.
std::unique_ptr<processing::FeatureListQueryProcessor>
feature_list_query_processor_;
// Traing data collection logic.
std::unique_ptr<TrainingDataCollector> training_data_collector_;
// Utility to execute model and return result.
std::unique_ptr<ModelExecutor> model_executor_;
// TODO(ritikagup) : Remoove this and use model manager from storage service.
// Model execution.
raw_ptr<ModelManager> model_manager_;
// Model execution scheduling logic.
std::unique_ptr<ModelExecutionScheduler> model_execution_scheduler_;
};
} // namespace segmentation_platform
#endif // COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_SCHEDULER_EXECUTION_SERVICE_H_
|
