File: report_request_queue_generator.h

package info (click to toggle)
chromium 139.0.7258.127-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,122,156 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (75 lines) | stat: -rw-r--r-- 3,008 bytes parent folder | download | duplicates (5)
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
// Copyright 2019 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_ENTERPRISE_BROWSER_REPORTING_REPORT_REQUEST_QUEUE_GENERATOR_H_
#define COMPONENTS_ENTERPRISE_BROWSER_REPORTING_REPORT_REQUEST_QUEUE_GENERATOR_H_

#include <memory>

#include "base/functional/callback.h"
#include "base/memory/weak_ptr.h"
#include "build/build_config.h"
#include "components/enterprise/browser/reporting/profile_report_generator.h"
#include "components/enterprise/browser/reporting/report_request.h"
#include "components/policy/proto/device_management_backend.pb.h"

namespace enterprise_reporting {

class ReportingDelegateFactory;

// Generate a report request queue that contains full profile information. The
// request number in the queue is decided by the maximum report size setting.
// TODO(crbug.com/40704763): Unit tests for this class are still in
// chrome/browser/enterprise/reporting.
class ReportRequestQueueGenerator {
 public:
  explicit ReportRequestQueueGenerator(
      ReportingDelegateFactory* delegate_factory);
  ReportRequestQueueGenerator(const ReportRequestQueueGenerator&) = delete;
  ReportRequestQueueGenerator& operator=(const ReportRequestQueueGenerator&) =
      delete;
  ~ReportRequestQueueGenerator();

  // Get the maximum report size.
  size_t GetMaximumReportSizeForTesting() const;

  // Set the maximum report size. The full profile info will be skipped or moved
  // to another new request if its size exceeds the limit.
  void SetMaximumReportSizeForTesting(size_t maximum_report_size);

  // Generate a queue of requests including full profile info based on given
  // basic request. Will invoke `callback` with the queue when done.
  void Generate(std::unique_ptr<ReportRequest> basic_request,
                base::OnceCallback<void(ReportRequestQueue)> callback);

 private:
  using IndexedProfileReport =
      std::pair<int,
                std::unique_ptr<enterprise_management::ChromeUserProfileInfo>>;

  // Generate request with full profile info at `profile_index` according to
  // `basic_request`. Will invoke `callback` when done.
  void GenerateProfileReportWithIndex(
      int profile_index,
      const ReportRequest* basic_request,
      base::OnceCallback<void(IndexedProfileReport)> callback);

  // Called when all Profile reports were generated. `basic_request` is carried
  // over to ensure it remains alive while individual reports are being
  // generated.
  void OnAllProfileReportsGenerated(
      std::unique_ptr<ReportRequest> basic_request,
      base::OnceCallback<void(ReportRequestQueue)> callback,
      std::vector<IndexedProfileReport> indexed_reports);

 private:
  size_t maximum_report_size_;
  ProfileReportGenerator profile_report_generator_;

  base::WeakPtrFactory<ReportRequestQueueGenerator> weak_factory_{this};
};

}  // namespace enterprise_reporting

#endif  // COMPONENTS_ENTERPRISE_BROWSER_REPORTING_REPORT_REQUEST_QUEUE_GENERATOR_H_