Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2016 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 "chrome/installer/setup/installer_metrics.h" | |
| 6 | |
| 7 #include "base/files/file.h" | |
| 8 #include "base/files/file_path.h" | |
| 9 #include "base/files/file_util.h" | |
| 10 #include "base/metrics/histogram_base.h" | |
| 11 #include "base/metrics/histogram_persistence.h" | |
| 12 #include "base/metrics/persistent_memory_allocator.h" | |
| 13 #include "chrome/installer/util/util_constants.h" | |
| 14 | |
| 15 namespace installer { | |
| 16 | |
| 17 void BeginPersistentHistogramStorage() { | |
| 18 base::SetPersistentHistogramMemoryAllocator( | |
| 19 new base::LocalPersistentMemoryAllocator( | |
| 20 1 << 20, 0, // 1 MiB | |
| 21 installer::kSetupHistogramAllocatorName)); | |
| 22 base::GetPersistentHistogramMemoryAllocator()->CreateTrackingHistograms( | |
| 23 installer::kSetupHistogramAllocatorName); | |
| 24 } | |
| 25 | |
| 26 void EndPersistentHistogramStorage(const base::FilePath& target_path) { | |
| 27 // For atomicity, first write to a temporary file and then rename it. | |
| 28 // The ImportantFileWriter would be good for this except it supports only | |
| 29 // std::string for its data. | |
| 30 base::PersistentMemoryAllocator* allocator = | |
| 31 base::GetPersistentHistogramMemoryAllocator(); | |
| 32 allocator->UpdateTrackingHistograms(); | |
| 33 | |
| 34 base::FilePath file_path = target_path | |
| 35 .AppendASCII(allocator->Name()) | |
| 36 .AddExtension(L".pma"); | |
| 37 base::FilePath tmp_file_path; | |
| 38 base::DeleteFile(file_path, false); | |
| 39 | |
| 40 if (base::CreateTemporaryFileInDir(file_path.DirName(), &tmp_file_path)) { | |
| 41 base::File histograms(tmp_file_path, | |
| 42 base::File::FLAG_OPEN | base::File::FLAG_WRITE); | |
| 43 if (histograms.IsValid()) { | |
| 44 // Allocator doesn't support more than 2GB so can never overflow. | |
|
grt (UTC plus 2)
2016/02/15 15:42:39
1GB? (kSegmentMaxSize is 1 << 30)
bcwhite
2016/02/15 19:22:10
Still technically true! :-D
Done.
| |
| 45 int used = static_cast<int>(allocator->used()); | |
| 46 if (histograms.Write(0, static_cast<const char*>(allocator->data()), | |
| 47 used) == used && | |
| 48 histograms.Flush()) { | |
| 49 histograms.Close(); | |
| 50 if (base::ReplaceFile(tmp_file_path, file_path, nullptr)) { | |
| 51 VLOG(1) << "Persistent histograms saved in file: " | |
| 52 << file_path.value(); | |
| 53 } | |
| 54 } | |
| 55 histograms.Close(); | |
| 56 } | |
| 57 base::DeleteFile(tmp_file_path, false); | |
| 58 } | |
| 59 } | |
| 60 | |
| 61 } // namespace installer | |
| OLD | NEW |