| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/budget_service/budget_manager.h" | 5 #include "chrome/browser/budget_service/budget_manager.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include "base/callback.h" | 9 #include "base/callback.h" |
| 10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 62 case blink::mojom::BudgetOperationType::INVALID_OPERATION: | 62 case blink::mojom::BudgetOperationType::INVALID_OPERATION: |
| 63 return SiteEngagementScore::kMaxPoints + 1; | 63 return SiteEngagementScore::kMaxPoints + 1; |
| 64 // No default case. | 64 // No default case. |
| 65 } | 65 } |
| 66 NOTREACHED(); | 66 NOTREACHED(); |
| 67 return SiteEngagementScore::kMaxPoints + 1.0; | 67 return SiteEngagementScore::kMaxPoints + 1.0; |
| 68 } | 68 } |
| 69 | 69 |
| 70 void BudgetManager::GetBudget(const url::Origin& origin, | 70 void BudgetManager::GetBudget(const url::Origin& origin, |
| 71 const GetBudgetCallback& callback) { | 71 const GetBudgetCallback& callback) { |
| 72 if (origin.unique() || !content::IsOriginSecure(GURL(origin.Serialize()))) { | 72 if (origin.unique() || !content::IsOriginSecure(origin.GetURL())) { |
| 73 callback.Run(blink::mojom::BudgetServiceErrorType::NOT_SUPPORTED, | 73 callback.Run(blink::mojom::BudgetServiceErrorType::NOT_SUPPORTED, |
| 74 mojo::Array<blink::mojom::BudgetStatePtr>()); | 74 mojo::Array<blink::mojom::BudgetStatePtr>()); |
| 75 return; | 75 return; |
| 76 } | 76 } |
| 77 db_.GetBudgetDetails(origin, callback); | 77 db_.GetBudgetDetails(origin, callback); |
| 78 } | 78 } |
| 79 | 79 |
| 80 void BudgetManager::Reserve(const url::Origin& origin, | 80 void BudgetManager::Reserve(const url::Origin& origin, |
| 81 blink::mojom::BudgetOperationType type, | 81 blink::mojom::BudgetOperationType type, |
| 82 const ReserveCallback& callback) { | 82 const ReserveCallback& callback) { |
| 83 if (origin.unique() || !content::IsOriginSecure(GURL(origin.Serialize()))) { | 83 if (origin.unique() || !content::IsOriginSecure(origin.GetURL())) { |
| 84 callback.Run(blink::mojom::BudgetServiceErrorType::NOT_SUPPORTED, | 84 callback.Run(blink::mojom::BudgetServiceErrorType::NOT_SUPPORTED, |
| 85 false /* success */); | 85 false /* success */); |
| 86 return; | 86 return; |
| 87 } | 87 } |
| 88 db_.SpendBudget(origin, GetCost(type), | 88 db_.SpendBudget(origin, GetCost(type), |
| 89 base::Bind(&BudgetManager::DidReserve, | 89 base::Bind(&BudgetManager::DidReserve, |
| 90 weak_ptr_factory_.GetWeakPtr(), origin, callback)); | 90 weak_ptr_factory_.GetWeakPtr(), origin, callback)); |
| 91 } | 91 } |
| 92 | 92 |
| 93 void BudgetManager::Consume(const url::Origin& origin, | 93 void BudgetManager::Consume(const url::Origin& origin, |
| 94 blink::mojom::BudgetOperationType type, | 94 blink::mojom::BudgetOperationType type, |
| 95 const ConsumeCallback& callback) { | 95 const ConsumeCallback& callback) { |
| 96 if (origin.unique() || !content::IsOriginSecure(GURL(origin.Serialize()))) { | 96 if (origin.unique() || !content::IsOriginSecure(origin.GetURL())) { |
| 97 callback.Run(false /* success */); | 97 callback.Run(false /* success */); |
| 98 return; | 98 return; |
| 99 } | 99 } |
| 100 | 100 |
| 101 bool found_reservation = false; | 101 bool found_reservation = false; |
| 102 | 102 |
| 103 // First, see if there is a reservation already. | 103 // First, see if there is a reservation already. |
| 104 auto count = reservation_map_.find(origin); | 104 auto count = reservation_map_.find(origin); |
| 105 if (count != reservation_map_.end()) { | 105 if (count != reservation_map_.end()) { |
| 106 if (count->second == 1) | 106 if (count->second == 1) |
| (...skipping 30 matching lines...) Expand all Loading... |
| 137 void BudgetManager::DidReserve(const url::Origin& origin, | 137 void BudgetManager::DidReserve(const url::Origin& origin, |
| 138 const ReserveCallback& callback, | 138 const ReserveCallback& callback, |
| 139 blink::mojom::BudgetServiceErrorType error, | 139 blink::mojom::BudgetServiceErrorType error, |
| 140 bool success) { | 140 bool success) { |
| 141 // If the call succeeded, write the new reservation into the map. | 141 // If the call succeeded, write the new reservation into the map. |
| 142 if (success && error == blink::mojom::BudgetServiceErrorType::NONE) | 142 if (success && error == blink::mojom::BudgetServiceErrorType::NONE) |
| 143 reservation_map_[origin]++; | 143 reservation_map_[origin]++; |
| 144 | 144 |
| 145 callback.Run(error, success); | 145 callback.Run(error, success); |
| 146 } | 146 } |
| OLD | NEW |