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
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_HTTP_HEADER_PROVIDER_H_
#define CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_HTTP_HEADER_PROVIDER_H_
#include <set>
#include <string>
#include "base/basictypes.h"
#include "base/gtest_prod_util.h"
#include "base/metrics/field_trial.h"
#include "base/synchronization/lock.h"
#include "components/variations/variations_associated_data.h"
namespace content {
class ResourceContext;
}
namespace net {
class HttpRequestHeaders;
}
class GURL;
class Profile;
class ProfileIOData;
template <typename T> struct DefaultSingletonTraits;
namespace chrome_variations {
// A helper class for maintaining Chrome experiments and metrics state
// transmitted in custom HTTP request headers.
// This class is a thread-safe singleton.
class VariationsHttpHeaderProvider : base::FieldTrialList::Observer {
public:
static VariationsHttpHeaderProvider* GetInstance();
// Adds Chrome experiment and metrics state as custom headers to |headers|.
// Some headers may not be set given the |incognito| mode or whether
// the user has |uma_enabled|. Also, we never transmit headers to non-Google
// sites, which is checked based on the destination |url|.
void AppendHeaders(const GURL& url,
bool incognito,
bool uma_enabled,
net::HttpRequestHeaders* headers);
// Sets *additional* variation ids and trigger variation ids to be encoded in
// the X-Client-Data request header. This is intended for development use to
// force a server side experiment id. |variation_ids| should be a
// comma-separated string of numeric experiment ids. If an id is prefixed
// with "t" it will be treated as a trigger experiment id.
bool SetDefaultVariationIds(const std::string& variation_ids);
private:
friend struct DefaultSingletonTraits<VariationsHttpHeaderProvider>;
FRIEND_TEST_ALL_PREFIXES(VariationsHttpHeaderProviderTest,
ShouldAppendHeaders);
FRIEND_TEST_ALL_PREFIXES(VariationsHttpHeaderProviderTest,
SetDefaultVariationIds_Valid);
FRIEND_TEST_ALL_PREFIXES(VariationsHttpHeaderProviderTest,
SetDefaultVariationIds_Invalid);
FRIEND_TEST_ALL_PREFIXES(VariationsHttpHeaderProviderTest,
OnFieldTrialGroupFinalized);
VariationsHttpHeaderProvider();
virtual ~VariationsHttpHeaderProvider();
// base::FieldTrialList::Observer implementation.
// This will add the variation ID associated with |trial_name| and
// |group_name| to the variation ID cache.
virtual void OnFieldTrialGroupFinalized(
const std::string& trial_name,
const std::string& group_name) OVERRIDE;
// Prepares the variation IDs cache with initial values if not already done.
// This method also registers the caller with the FieldTrialList to receive
// new variation IDs.
void InitVariationIDsCacheIfNeeded();
// Takes whatever is currently in |variation_ids_set_| and recreates
// |variation_ids_header_| with it. Assumes the the |lock_| is currently
// held.
void UpdateVariationIDsHeaderValue();
// Checks whether variation headers should be appended to requests to the
// specified |url|. Returns true for google.<TLD> and youtube.<TLD> URLs.
static bool ShouldAppendHeaders(const GURL& url);
// Guards |variation_ids_cache_initialized_|, |variation_ids_set_| and
// |variation_ids_header_|.
base::Lock lock_;
// Whether or not we've initialized the cache.
bool variation_ids_cache_initialized_;
// Keep a cache of variation IDs that are transmitted in headers to Google.
// This consists of a list of valid IDs, and the actual transmitted header.
std::set<chrome_variations::VariationID> variation_ids_set_;
std::set<chrome_variations::VariationID> variation_trigger_ids_set_;
// Provides the google experiment ids forced from command line.
std::set<chrome_variations::VariationID> default_variation_ids_set_;
std::set<chrome_variations::VariationID> default_trigger_id_set_;
std::string variation_ids_header_;
DISALLOW_COPY_AND_ASSIGN(VariationsHttpHeaderProvider);
};
} // namespace chrome_variations
#endif // CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_HTTP_HEADER_PROVIDER_H_
|
