Move ranker_model_loader to a new component.

components/ranker/ranker_model_loader.cc

 // Copyright 2017 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 "components/translate/core/browser/ranker_model_loader.h"
+#include "components/ranker/ranker_model_loader.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/files/important_file_writer.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/profiler/scoped_tracker.h"
 #include "base/sequenced_task_runner.h"
 #include "base/strings/string_util.h"
 #include "base/task_runner_util.h"
 #include "base/task_scheduler/post_task.h"
 #include "base/threading/sequenced_task_runner_handle.h"
-#include "components/translate/core/browser/proto/ranker_model.pb.h"
-#include "components/translate/core/browser/ranker_model.h"
-#include "components/translate/core/browser/translate_url_fetcher.h"
+#include "components/ranker/proto/ranker_model.pb.h"
+#include "components/ranker/ranker_model.h"
+#include "components/ranker/ranker_url_fetcher.h"
 
-namespace translate {
+namespace ranker {
 namespace {
 
-using chrome_intelligence::RankerModel;
-using chrome_intelligence::RankerModelProto;
-
-constexpr int kUrlFetcherId = 2;
-
 // The minimum duration, in minutes, between download attempts.
 constexpr int kMinRetryDelayMins = 3;
 
 // Suffixes for the various histograms produced by the backend.
 const char kWriteTimerHistogram[] = ".Timer.WriteModel";
 const char kReadTimerHistogram[] = ".Timer.ReadModel";
 const char kDownloadTimerHistogram[] = ".Timer.DownloadModel";
 const char kParsetimerHistogram[] = ".Timer.ParseModel";
 const char kModelStatusHistogram[] = ".Model.Status";
 
@@ skipping 55 matching lines @@
     GURL model_url,
     std::string uma_prefix)
     : background_task_runner_(base::CreateSequencedTaskRunnerWithTraits(
           {base::MayBlock(), base::TaskPriority::BACKGROUND,
            base::TaskShutdownBehavior::SKIP_ON_SHUTDOWN})),
       validate_model_cb_(std::move(validate_model_cb)),
       on_model_available_cb_(std::move(on_model_available_cb)),
       model_path_(std::move(model_path)),
       model_url_(std::move(model_url)),
       uma_prefix_(std::move(uma_prefix)),
-      url_fetcher_(base::MakeUnique<TranslateURLFetcher>(kUrlFetcherId)),
+      url_fetcher_(base::MakeUnique<RankerURLFetcher>()),
       weak_ptr_factory_(this) {}
 
 RankerModelLoader::~RankerModelLoader() {
   DCHECK(sequence_checker_.CalledOnValidSequence());
 }
 
 void RankerModelLoader::NotifyOfRankerActivity() {
   DCHECK(sequence_checker_.CalledOnValidSequence());
   switch (state_) {
     case LoaderState::NOT_STARTED:
@@ skipping 105 matching lines @@
   // |url_fetcher_| maintains a request retry counter. If all allowed attempts
   // have already been exhausted, then the loader is finished and has abandoned
   // loading the model.
   if (!request_started) {
     DVLOG(2) << "Model download abandoned.";
     ReportModelStatus(RankerModelStatus::MODEL_LOADING_ABANDONED);
     state_ = LoaderState::FINISHED;
   }
 }
 
-void RankerModelLoader::OnURLFetched(int /* id */,
-                                     bool success,
-                                     const std::string& data) {
+void RankerModelLoader::OnURLFetched(bool success, const std::string& data) {
   DCHECK(sequence_checker_.CalledOnValidSequence());
   DCHECK_EQ(state_, LoaderState::LOADING_FROM_URL);
 
   // Record the duration of the download.
   RecordTimerHistogram(uma_prefix_ + kDownloadTimerHistogram,
                        base::TimeTicks::Now() - load_start_time_);
 
   // On request failure, transition back to IDLE. The loader will retry, or
   // enforce the max download attempts, later.
   if (!success || data.empty()) {
@@ skipping 24 matching lines @@
     background_task_runner_->PostTask(
         FROM_HERE, base::BindOnce(&SaveToFile, model_url_, model_path_,
                                   model->SerializeAsString(), uma_prefix_));
   }
 
   // The loader is finished. Transfer the model to the client.
   state_ = LoaderState::FINISHED;
   on_model_available_cb_.Run(std::move(model));
 }
 
-std::unique_ptr<chrome_intelligence::RankerModel>
-RankerModelLoader::CreateAndValidateModel(const std::string& data) {
+std::unique_ptr<RankerModel> RankerModelLoader::CreateAndValidateModel(
+    const std::string& data) {
   DCHECK(sequence_checker_.CalledOnValidSequence());
   MyScopedHistogramTimer timer(uma_prefix_ + kParsetimerHistogram);
   auto model = RankerModel::FromString(data);
   if (ReportModelStatus(model ? validate_model_cb_.Run(*model)
                               : RankerModelStatus::PARSE_FAILED) !=
       RankerModelStatus::OK) {
     return nullptr;
   }
   return model;
 }
 
 RankerModelStatus RankerModelLoader::ReportModelStatus(
     RankerModelStatus model_status) {
   DCHECK(sequence_checker_.CalledOnValidSequence());
   base::HistogramBase* histogram = base::LinearHistogram::FactoryGet(
       uma_prefix_ + kModelStatusHistogram, 1,
       static_cast<int>(RankerModelStatus::MAX),
       static_cast<int>(RankerModelStatus::MAX) + 1,
       base::HistogramBase::kUmaTargetedHistogramFlag);
   if (histogram)
     histogram->Add(static_cast<int>(model_status));
   return model_status;
 }
 
-}  // namespace translate
+}  // namespace ranker