12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- // 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/assist_ranker/assist_ranker_service_impl.h"
- #include "base/memory/weak_ptr.h"
- #include "components/assist_ranker/binary_classifier_predictor.h"
- #include "components/assist_ranker/ranker_model_loader_impl.h"
- #include "services/network/public/cpp/shared_url_loader_factory.h"
- #include "url/gurl.h"
- namespace assist_ranker {
- AssistRankerServiceImpl::AssistRankerServiceImpl(
- base::FilePath base_path,
- scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory)
- : url_loader_factory_(std::move(url_loader_factory)),
- base_path_(std::move(base_path)) {}
- AssistRankerServiceImpl::~AssistRankerServiceImpl() {}
- base::WeakPtr<BinaryClassifierPredictor>
- AssistRankerServiceImpl::FetchBinaryClassifierPredictor(
- const PredictorConfig& config) {
- DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
- const std::string& model_name = config.model_name;
- auto predictor_it = predictor_map_.find(model_name);
- if (predictor_it != predictor_map_.end()) {
- DVLOG(1) << "Predictor " << model_name << " already initialized.";
- return base::AsWeakPtr(
- static_cast<BinaryClassifierPredictor*>(predictor_it->second.get()));
- }
- // The predictor does not exist yet, so we create one.
- DVLOG(1) << "Initializing predictor: " << model_name;
- std::unique_ptr<BinaryClassifierPredictor> predictor =
- BinaryClassifierPredictor::Create(config, GetModelPath(model_name),
- url_loader_factory_);
- base::WeakPtr<BinaryClassifierPredictor> weak_ptr =
- base::AsWeakPtr(predictor.get());
- predictor_map_[model_name] = std::move(predictor);
- return weak_ptr;
- }
- base::FilePath AssistRankerServiceImpl::GetModelPath(
- const std::string& model_filename) {
- if (base_path_.empty())
- return base::FilePath();
- return base_path_.AppendASCII(model_filename);
- }
- } // namespace assist_ranker
|