base: Make RefCountedBytes::TakeVector return ref ptr instead of raw ptr

chrome/browser/printing/print_preview_data_service.cc

 // 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.
 
 #include "chrome/browser/printing/print_preview_data_service.h"
 
 #include "base/macros.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/memory/singleton.h"
 #include "base/stl_util.h"
@@ skipping 23 matching lines @@
                               scoped_refptr<base::RefCountedBytes>* data) {
     if (IsInvalidIndex(index))
       return;
 
     PreviewPageDataMap::iterator it = page_data_map_.find(index);
     if (it != page_data_map_.end())
       *data = it->second.get();
   }
 
   // Set/Update the preview data entry for the specified |index|.
-  void SetPreviewDataForIndex(int index, const base::RefCountedBytes* data) {
+  void SetPreviewDataForIndex(int index,
+                              scoped_refptr<base::RefCountedBytes> data) {
     if (IsInvalidIndex(index))
       return;
 
-    page_data_map_[index] = const_cast<base::RefCountedBytes*>(data);
+    page_data_map_[index] = std::move(data);
   }
 
   // Returns the available draft page count.
   int GetAvailableDraftPageCount() {
     int page_data_map_size = page_data_map_.size();
     if (ContainsKey(page_data_map_, printing::COMPLETE_PREVIEW_DOCUMENT_INDEX))
       page_data_map_size--;
     return page_data_map_size;
   }
 
@@ skipping 37 matching lines @@
     scoped_refptr<base::RefCountedBytes>* data_bytes) {
   *data_bytes = nullptr;
   PreviewDataStoreMap::const_iterator it = data_store_map_.find(preview_ui_id);
   if (it != data_store_map_.end())
     it->second->GetPreviewDataForIndex(index, data_bytes);
 }
 
 void PrintPreviewDataService::SetDataEntry(
     int32_t preview_ui_id,
     int index,
-    const base::RefCountedBytes* data_bytes) {
-  if (!ContainsKey(data_store_map_, preview_ui_id))
-    data_store_map_[preview_ui_id] = new PrintPreviewDataStore();
+    scoped_refptr<base::RefCountedBytes> data_bytes) {
+  if (!ContainsKey(data_store_map_, preview_ui_id)) {
+    data_store_map_[preview_ui_id] =
+        make_scoped_refptr(new PrintPreviewDataStore());
+  }
 
-  data_store_map_[preview_ui_id]->SetPreviewDataForIndex(index, data_bytes);
+  data_store_map_[preview_ui_id]->SetPreviewDataForIndex(index,
+                                                         std::move(data_bytes));
 }
 
 void PrintPreviewDataService::RemoveEntry(int32_t preview_ui_id) {
   data_store_map_.erase(preview_ui_id);
 }
 
 int PrintPreviewDataService::GetAvailableDraftPageCount(int32_t preview_ui_id) {
   PreviewDataStoreMap::const_iterator it = data_store_map_.find(preview_ui_id);
   return (it == data_store_map_.end()) ?
       0 : it->second->GetAvailableDraftPageCount();
 }