OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "net/http/http_cache.h" | 5 #include "net/http/http_cache.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 true, | 89 true, |
90 thread_, | 90 thread_, |
91 net_log, | 91 net_log, |
92 backend, | 92 backend, |
93 callback); | 93 callback); |
94 } | 94 } |
95 | 95 |
96 //----------------------------------------------------------------------------- | 96 //----------------------------------------------------------------------------- |
97 | 97 |
98 HttpCache::ActiveEntry::ActiveEntry(disk_cache::Entry* entry) | 98 HttpCache::ActiveEntry::ActiveEntry(disk_cache::Entry* entry) |
99 : disk_entry(entry), | 99 : disk_entry(entry) {} |
100 writer(NULL), | |
101 will_process_pending_queue(false), | |
102 doomed(false) { | |
103 } | |
104 | 100 |
105 HttpCache::ActiveEntry::~ActiveEntry() { | 101 HttpCache::ActiveEntry::~ActiveEntry() { |
106 if (disk_entry) { | 102 if (disk_entry) { |
107 disk_entry->Close(); | 103 disk_entry->Close(); |
108 disk_entry = NULL; | 104 disk_entry = nullptr; |
109 } | 105 } |
110 } | 106 } |
111 | 107 |
112 size_t HttpCache::ActiveEntry::EstimateMemoryUsage() const { | 108 size_t HttpCache::ActiveEntry::EstimateMemoryUsage() const { |
113 // Skip |disk_entry| which is tracked in simple_backend_impl; Skip |readers| | 109 // Skip |disk_entry| which is tracked in simple_backend_impl; Skip |readers| |
114 // and |pending_queue| because the Transactions are owned by their respective | 110 // and |add_to_entry_queue| because the Transactions are owned by their |
115 // URLRequestHttpJobs. | 111 // respective URLRequestHttpJobs. |
116 return 0; | 112 return 0; |
117 } | 113 } |
118 | 114 |
119 bool HttpCache::ActiveEntry::HasNoTransactions() { | 115 bool HttpCache::ActiveEntry::HasNoTransactions() { |
120 return !writer && readers.empty() && pending_queue.empty(); | 116 return !writer && readers.empty() && add_to_entry_queue.empty() && |
| 117 done_headers_queue.empty() && !headers_transaction; |
| 118 } |
| 119 |
| 120 bool HttpCache::ActiveEntry::HasNoActiveTransactions() { |
| 121 return !writer && readers.empty() && !headers_transaction; |
121 } | 122 } |
122 | 123 |
123 //----------------------------------------------------------------------------- | 124 //----------------------------------------------------------------------------- |
124 | 125 |
125 // This structure keeps track of work items that are attempting to create or | 126 // This structure keeps track of work items that are attempting to create or |
126 // open cache entries or the backend itself. | 127 // open cache entries or the backend itself. |
127 struct HttpCache::PendingOp { | 128 struct HttpCache::PendingOp { |
128 PendingOp() : disk_entry(NULL) {} | 129 PendingOp() : disk_entry(NULL) {} |
129 ~PendingOp() {} | 130 ~PendingOp() {} |
130 | 131 |
(...skipping 222 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
353 } | 354 } |
354 } | 355 } |
355 } | 356 } |
356 | 357 |
357 HttpCache::~HttpCache() { | 358 HttpCache::~HttpCache() { |
358 // Transactions should see an invalid cache after this point; otherwise they | 359 // Transactions should see an invalid cache after this point; otherwise they |
359 // could see an inconsistent object (half destroyed). | 360 // could see an inconsistent object (half destroyed). |
360 weak_factory_.InvalidateWeakPtrs(); | 361 weak_factory_.InvalidateWeakPtrs(); |
361 | 362 |
362 // If we have any active entries remaining, then we need to deactivate them. | 363 // If we have any active entries remaining, then we need to deactivate them. |
363 // We may have some pending calls to OnProcessPendingQueue, but since those | 364 // We may have some pending tasks to process queued transactions ,but since |
364 // won't run (due to our destruction), we can simply ignore the corresponding | 365 // those won't run (due to our destruction), we can simply ignore the |
365 // will_process_pending_queue flag. | 366 // corresponding flags. |
366 while (!active_entries_.empty()) { | 367 while (!active_entries_.empty()) { |
367 ActiveEntry* entry = active_entries_.begin()->second.get(); | 368 ActiveEntry* entry = active_entries_.begin()->second.get(); |
368 entry->will_process_pending_queue = false; | 369 entry->will_process_queued_transactions = false; |
369 entry->pending_queue.clear(); | 370 entry->add_to_entry_queue.clear(); |
370 entry->readers.clear(); | 371 entry->readers.clear(); |
371 entry->writer = NULL; | 372 entry->done_headers_queue.clear(); |
| 373 entry->headers_transaction = nullptr; |
| 374 entry->writer = nullptr; |
372 DeactivateEntry(entry); | 375 DeactivateEntry(entry); |
373 } | 376 } |
374 | 377 |
375 doomed_entries_.clear(); | 378 doomed_entries_.clear(); |
376 | 379 |
377 // Before deleting pending_ops_, we have to make sure that the disk cache is | 380 // Before deleting pending_ops_, we have to make sure that the disk cache is |
378 // done with said operations, or it will attempt to use deleted data. | 381 // done with said operations, or it will attempt to use deleted data. |
379 disk_cache_.reset(); | 382 disk_cache_.reset(); |
380 | 383 |
381 for (auto pending_it = pending_ops_.begin(); pending_it != pending_ops_.end(); | 384 for (auto pending_it = pending_ops_.begin(); pending_it != pending_ops_.end(); |
(...skipping 239 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
621 // We keep track of doomed entries so that we can ensure that they are | 624 // We keep track of doomed entries so that we can ensure that they are |
622 // cleaned up properly when the cache is destroyed. | 625 // cleaned up properly when the cache is destroyed. |
623 ActiveEntry* entry_ptr = entry.get(); | 626 ActiveEntry* entry_ptr = entry.get(); |
624 DCHECK_EQ(0u, doomed_entries_.count(entry_ptr)); | 627 DCHECK_EQ(0u, doomed_entries_.count(entry_ptr)); |
625 doomed_entries_[entry_ptr] = std::move(entry); | 628 doomed_entries_[entry_ptr] = std::move(entry); |
626 | 629 |
627 entry_ptr->disk_entry->Doom(); | 630 entry_ptr->disk_entry->Doom(); |
628 entry_ptr->doomed = true; | 631 entry_ptr->doomed = true; |
629 | 632 |
630 DCHECK(entry_ptr->writer || !entry_ptr->readers.empty() || | 633 DCHECK(entry_ptr->writer || !entry_ptr->readers.empty() || |
631 entry_ptr->will_process_pending_queue); | 634 entry_ptr->headers_transaction || |
| 635 entry_ptr->will_process_queued_transactions); |
632 return OK; | 636 return OK; |
633 } | 637 } |
634 | 638 |
635 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { | 639 int HttpCache::AsyncDoomEntry(const std::string& key, Transaction* trans) { |
636 std::unique_ptr<WorkItem> item = | 640 std::unique_ptr<WorkItem> item = |
637 base::MakeUnique<WorkItem>(WI_DOOM_ENTRY, trans, nullptr); | 641 base::MakeUnique<WorkItem>(WI_DOOM_ENTRY, trans, nullptr); |
638 PendingOp* pending_op = GetPendingOp(key); | 642 PendingOp* pending_op = GetPendingOp(key); |
639 if (pending_op->writer) { | 643 if (pending_op->writer) { |
640 pending_op->pending_queue.push_back(std::move(item)); | 644 pending_op->pending_queue.push_back(std::move(item)); |
641 return ERR_IO_PENDING; | 645 return ERR_IO_PENDING; |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
677 DCHECK(entry->doomed); | 681 DCHECK(entry->doomed); |
678 DCHECK(entry->HasNoTransactions()); | 682 DCHECK(entry->HasNoTransactions()); |
679 | 683 |
680 auto it = doomed_entries_.find(entry); | 684 auto it = doomed_entries_.find(entry); |
681 DCHECK(it != doomed_entries_.end()); | 685 DCHECK(it != doomed_entries_.end()); |
682 doomed_entries_.erase(it); | 686 doomed_entries_.erase(it); |
683 } | 687 } |
684 | 688 |
685 HttpCache::ActiveEntry* HttpCache::FindActiveEntry(const std::string& key) { | 689 HttpCache::ActiveEntry* HttpCache::FindActiveEntry(const std::string& key) { |
686 auto it = active_entries_.find(key); | 690 auto it = active_entries_.find(key); |
687 return it != active_entries_.end() ? it->second.get() : NULL; | 691 return it != active_entries_.end() ? it->second.get() : nullptr; |
688 } | 692 } |
689 | 693 |
690 HttpCache::ActiveEntry* HttpCache::ActivateEntry( | 694 HttpCache::ActiveEntry* HttpCache::ActivateEntry( |
691 disk_cache::Entry* disk_entry) { | 695 disk_cache::Entry* disk_entry) { |
692 DCHECK(!FindActiveEntry(disk_entry->GetKey())); | 696 DCHECK(!FindActiveEntry(disk_entry->GetKey())); |
693 ActiveEntry* entry = new ActiveEntry(disk_entry); | 697 ActiveEntry* entry = new ActiveEntry(disk_entry); |
694 active_entries_[disk_entry->GetKey()] = base::WrapUnique(entry); | 698 active_entries_[disk_entry->GetKey()] = base::WrapUnique(entry); |
695 return entry; | 699 return entry; |
696 } | 700 } |
697 | 701 |
698 void HttpCache::DeactivateEntry(ActiveEntry* entry) { | 702 void HttpCache::DeactivateEntry(ActiveEntry* entry) { |
699 DCHECK(!entry->will_process_pending_queue); | 703 DCHECK(!entry->will_process_queued_transactions); |
700 DCHECK(!entry->doomed); | 704 DCHECK(!entry->doomed); |
701 DCHECK(entry->disk_entry); | 705 DCHECK(entry->disk_entry); |
702 DCHECK(entry->HasNoTransactions()); | 706 DCHECK(entry->HasNoTransactions()); |
703 | 707 |
704 std::string key = entry->disk_entry->GetKey(); | 708 std::string key = entry->disk_entry->GetKey(); |
705 if (key.empty()) | 709 if (key.empty()) |
706 return SlowDeactivateEntry(entry); | 710 return SlowDeactivateEntry(entry); |
707 | 711 |
708 auto it = active_entries_.find(key); | 712 auto it = active_entries_.find(key); |
709 DCHECK(it != active_entries_.end()); | 713 DCHECK(it != active_entries_.end()); |
(...skipping 109 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
819 } | 823 } |
820 | 824 |
821 void HttpCache::DestroyEntry(ActiveEntry* entry) { | 825 void HttpCache::DestroyEntry(ActiveEntry* entry) { |
822 if (entry->doomed) { | 826 if (entry->doomed) { |
823 FinalizeDoomedEntry(entry); | 827 FinalizeDoomedEntry(entry); |
824 } else { | 828 } else { |
825 DeactivateEntry(entry); | 829 DeactivateEntry(entry); |
826 } | 830 } |
827 } | 831 } |
828 | 832 |
829 int HttpCache::AddTransactionToEntry(ActiveEntry* entry, Transaction* trans) { | 833 int HttpCache::AddTransactionToEntry(ActiveEntry* entry, |
| 834 Transaction* transaction) { |
830 DCHECK(entry); | 835 DCHECK(entry); |
831 DCHECK(entry->disk_entry); | 836 DCHECK(entry->disk_entry); |
832 | 837 // Always add a new transaction to the queue to maintain FIFO order. |
833 // We implement a basic reader/writer lock for the disk cache entry. If | 838 entry->add_to_entry_queue.push_back(transaction); |
834 // there is already a writer, then everyone has to wait for the writer to | 839 ProcessQueuedTransactions(entry); |
835 // finish before they can access the cache entry. There can be multiple | 840 return ERR_IO_PENDING; |
836 // readers. | 841 } |
837 // | 842 |
838 // NOTE: If the transaction can only write, then the entry should not be in | 843 int HttpCache::DoneWithResponseHeaders(ActiveEntry* entry, |
839 // use (since any existing entry should have already been doomed). | 844 Transaction* transaction) { |
840 | 845 // If |transaction| is the current writer, do nothing. This can happen for |
841 if (entry->writer || entry->will_process_pending_queue) { | 846 // range requests since they can go back to headers phase after starting to |
842 entry->pending_queue.push_back(trans); | 847 // write. |
843 return ERR_IO_PENDING; | 848 if (entry->writer == transaction) |
844 } | 849 return OK; |
845 | 850 |
846 if (trans->mode() & Transaction::WRITE) { | 851 // Proceed only if |transaction| is a headers_transaction and not already a |
847 // transaction needs exclusive access to the entry | 852 // reader. |
848 if (entry->readers.empty()) { | 853 if (entry->headers_transaction != transaction) |
849 entry->writer = trans; | 854 return OK; |
850 } else { | 855 |
851 entry->pending_queue.push_back(trans); | 856 entry->headers_transaction = nullptr; |
852 return ERR_IO_PENDING; | 857 |
853 } | 858 entry->done_headers_queue.push_back(transaction); |
| 859 ProcessQueuedTransactions(entry); |
| 860 return ERR_IO_PENDING; |
| 861 } |
| 862 |
| 863 void HttpCache::RemoveAllQueuedTransactions(ActiveEntry* entry, |
| 864 TransactionList* list) { |
| 865 // Process done_headers_queue before add_to_entry_queue to maintain FIFO |
| 866 // order. |
| 867 for (auto* transaction : entry->done_headers_queue) |
| 868 list->push_back(transaction); |
| 869 entry->done_headers_queue.clear(); |
| 870 |
| 871 for (auto* transaction : entry->add_to_entry_queue) |
| 872 list->push_back(transaction); |
| 873 entry->add_to_entry_queue.clear(); |
| 874 } |
| 875 |
| 876 void HttpCache::ProcessEntryFailure(ActiveEntry* entry) { |
| 877 // Failure case is either writer failing to completely write the response to |
| 878 // the cache or validating transaction received a non-304 response. |
| 879 TransactionList list; |
| 880 if (entry->HasNoActiveTransactions() && |
| 881 !entry->will_process_queued_transactions) { |
| 882 entry->disk_entry->Doom(); |
| 883 RemoveAllQueuedTransactions(entry, &list); |
| 884 DestroyEntry(entry); |
854 } else { | 885 } else { |
855 // transaction needs read access to the entry | 886 DoomActiveEntry(entry->disk_entry->GetKey()); |
856 entry->readers.insert(trans); | 887 RemoveAllQueuedTransactions(entry, &list); |
857 } | 888 } |
858 | 889 // ERR_CACHE_RACE causes the transaction to restart the whole process. |
859 // We do this before calling EntryAvailable to force any further calls to | 890 for (auto* transaction : list) |
860 // AddTransactionToEntry to add their transaction to the pending queue, which | 891 transaction->io_callback().Run(net::ERR_CACHE_RACE); |
861 // ensures FIFO ordering. | 892 } |
862 if (!entry->writer && !entry->pending_queue.empty()) | 893 |
863 ProcessPendingQueue(entry); | 894 void HttpCache::ProcessQueuedTransactions(ActiveEntry* entry) { |
864 | 895 // Multiple readers may finish with an entry at once, so we want to batch up |
865 return OK; | 896 // calls to OnProcessQueuedTransactions. This flag also tells us that we |
866 } | 897 // should not delete the entry before OnProcessQueuedTransactions runs. |
867 | 898 if (entry->will_process_queued_transactions) |
868 void HttpCache::DoneWithEntry(ActiveEntry* entry, Transaction* trans, | 899 return; |
| 900 |
| 901 entry->will_process_queued_transactions = true; |
| 902 |
| 903 // Post a task instead of invoking the io callback of another transaction here |
| 904 // to avoid re-entrancy. |
| 905 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 906 FROM_HERE, |
| 907 base::Bind(&HttpCache::OnProcessQueuedTransactions, GetWeakPtr(), entry)); |
| 908 } |
| 909 |
| 910 void HttpCache::OnProcessQueuedTransactions(ActiveEntry* entry) { |
| 911 entry->will_process_queued_transactions = false; |
| 912 |
| 913 // If no one is interested in this entry, then we can deactivate it. |
| 914 if (entry->HasNoTransactions()) { |
| 915 DestroyEntry(entry); |
| 916 return; |
| 917 } |
| 918 |
| 919 if (entry->done_headers_queue.empty() && entry->add_to_entry_queue.empty()) |
| 920 return; |
| 921 |
| 922 // To maintain FIFO order of transactions, done_headers_queue should be |
| 923 // processed first, and then add_to_entry_queue. |
| 924 // If another transaction is writing the response, let validated transactions |
| 925 // wait till the response is complete. If the response is not yet started, the |
| 926 // done_headers_queue transaction should start writing it. |
| 927 if (!entry->writer && !entry->done_headers_queue.empty()) { |
| 928 ProcessDoneHeadersQueue(entry); |
| 929 return; |
| 930 } |
| 931 |
| 932 if (!entry->add_to_entry_queue.empty()) |
| 933 ProcessAddToEntryQueue(entry); |
| 934 } |
| 935 |
| 936 void HttpCache::ProcessAddToEntryQueue(ActiveEntry* entry) { |
| 937 if (entry->headers_transaction) { |
| 938 // Note the entry may be new or may already have a response body written to |
| 939 // it. In both cases, a transaction with write mode set, needs to wait |
| 940 // because only one transaction is allowed to be validating at a time. |
| 941 // If transaction is read-only and entry is not yet complete, then it |
| 942 // needs to wait. If entry is complete, it needs to wait because its |
| 943 // possible that the headers_transaction dooms the entry if its not a match. |
| 944 return; |
| 945 } |
| 946 |
| 947 Transaction* transaction = entry->add_to_entry_queue.front(); |
| 948 |
| 949 // If a writer is present, read-only transactions cannot proceed. |
| 950 if (entry->writer && !(transaction->mode() & Transaction::WRITE)) |
| 951 return; |
| 952 |
| 953 entry->add_to_entry_queue.erase(entry->add_to_entry_queue.begin()); |
| 954 |
| 955 // This state may be reached either during the start of an entry or when the |
| 956 // response is written or being written. |
| 957 if (transaction->mode() & Transaction::WRITE) { |
| 958 entry->headers_transaction = transaction; |
| 959 } else { |
| 960 // A read-only transaction can start reading the response. |
| 961 // If a read transaction is here, then it has to be an already created |
| 962 // entry as the other case is already handled in the Transaction state |
| 963 // machine. |
| 964 entry->readers.insert(transaction); |
| 965 |
| 966 // More readers can join too. |
| 967 // TODO (shivanisha@) Consider a loop instead of posting a task. |
| 968 ProcessQueuedTransactions(entry); |
| 969 } |
| 970 |
| 971 transaction->io_callback().Run(OK); |
| 972 } |
| 973 |
| 974 void HttpCache::ProcessDoneHeadersQueue(ActiveEntry* entry) { |
| 975 DCHECK(!entry->writer && !entry->done_headers_queue.empty()); |
| 976 Transaction* transaction = entry->done_headers_queue.front(); |
| 977 // If this transaction is responsible for writing the response body. |
| 978 if (transaction->mode() & Transaction::WRITE) { |
| 979 entry->writer = transaction; |
| 980 } else { |
| 981 // Response body is already written, convert to a reader. |
| 982 auto return_val = entry->readers.insert(transaction); |
| 983 DCHECK_EQ(return_val.second, true); |
| 984 } |
| 985 // Post another task to give a chance to more transactions to either join |
| 986 // readers or another transaction to start parallel validation. |
| 987 // TODO (shivanisha@) Consider a loop instead of posting a task. |
| 988 ProcessQueuedTransactions(entry); |
| 989 entry->done_headers_queue.erase(entry->done_headers_queue.begin()); |
| 990 transaction->io_callback().Run(OK); |
| 991 } |
| 992 |
| 993 bool HttpCache::CanTransactionWriteResponseHeaders(ActiveEntry* entry, |
| 994 Transaction* transaction, |
| 995 int response_code) const { |
| 996 if (transaction != entry->headers_transaction) |
| 997 return false; |
| 998 |
| 999 // If its not a match then it should be the transaction responsible for |
| 1000 // writing the response body. |
| 1001 if (response_code != 304) { |
| 1002 return !entry->writer && entry->done_headers_queue.empty(); |
| 1003 } |
| 1004 |
| 1005 return true; |
| 1006 } |
| 1007 |
| 1008 bool HttpCache::CanTransactionWriteResponseBody( |
| 1009 ActiveEntry* entry, |
| 1010 Transaction* transaction, |
| 1011 const std::string& method) const { |
| 1012 if (transaction == entry->writer) |
| 1013 return true; |
| 1014 |
| 1015 if (method == "HEAD" || method == "DELETE") |
| 1016 return false; |
| 1017 |
| 1018 // Check if transaction is about to start writing to the cache. |
| 1019 |
| 1020 // Transaction's mode may have been set to NONE if StopCaching was invoked. |
| 1021 if (!(transaction->mode() & Transaction::WRITE || |
| 1022 transaction->mode() == Transaction::NONE)) |
| 1023 return false; |
| 1024 |
| 1025 if (transaction == entry->headers_transaction) |
| 1026 return !entry->writer && entry->done_headers_queue.empty(); |
| 1027 |
| 1028 // Check the first of done_headers_queue, since the rest should anyways be |
| 1029 // READ mode transactions as they would be a result of validation match. |
| 1030 if (transaction == entry->done_headers_queue.front()) |
| 1031 return !entry->writer; |
| 1032 |
| 1033 return false; |
| 1034 } |
| 1035 |
| 1036 void HttpCache::DoneWithEntry(ActiveEntry* entry, |
| 1037 Transaction* transaction, |
869 bool cancel) { | 1038 bool cancel) { |
870 // If we already posted a task to move on to the next transaction and this was | 1039 // Transaction is waiting in the done_headers_queue. |
871 // the writer, there is nothing to cancel. | 1040 auto i = std::find(entry->done_headers_queue.begin(), |
872 if (entry->will_process_pending_queue && entry->readers.empty()) | 1041 entry->done_headers_queue.end(), transaction); |
873 return; | 1042 if (i != entry->done_headers_queue.end()) { |
874 | 1043 entry->done_headers_queue.erase(i); |
875 if (entry->writer) { | 1044 if (cancel) |
876 DCHECK(trans == entry->writer); | 1045 ProcessEntryFailure(entry); |
877 | 1046 return; |
| 1047 } |
| 1048 |
| 1049 // Transaction is removed in the headers phase. |
| 1050 if (transaction == entry->headers_transaction) { |
| 1051 // If the response is not written (cancel is true), consider it a failure. |
| 1052 DoneWritingToEntry(entry, !cancel, transaction); |
| 1053 return; |
| 1054 } |
| 1055 |
| 1056 // Transaction is removed in the writing phase. |
| 1057 if (transaction == entry->writer) { |
878 // Assume there was a failure. | 1058 // Assume there was a failure. |
879 bool success = false; | 1059 bool success = false; |
880 if (cancel) { | 1060 if (cancel) { |
881 DCHECK(entry->disk_entry); | 1061 DCHECK(entry->disk_entry); |
882 // This is a successful operation in the sense that we want to keep the | 1062 // This is a successful operation in the sense that we want to keep the |
883 // entry. | 1063 // entry. |
884 success = trans->AddTruncatedFlag(); | 1064 success = transaction->AddTruncatedFlag(); |
885 // The previous operation may have deleted the entry. | 1065 // The previous operation may have deleted the entry. |
886 if (!trans->entry()) | 1066 if (!transaction->entry()) |
887 return; | 1067 return; |
888 } | 1068 } |
889 DoneWritingToEntry(entry, success); | 1069 DoneWritingToEntry(entry, success, transaction); |
890 } else { | 1070 return; |
891 DoneReadingFromEntry(entry, trans); | 1071 } |
892 } | 1072 |
893 } | 1073 // Transaction is reading from the entry. |
894 | 1074 DoneReadingFromEntry(entry, transaction); |
895 void HttpCache::DoneWritingToEntry(ActiveEntry* entry, bool success) { | 1075 } |
896 DCHECK(entry->readers.empty()); | 1076 |
897 | 1077 void HttpCache::DoneWritingToEntry(ActiveEntry* entry, |
898 entry->writer = NULL; | 1078 bool success, |
899 | 1079 Transaction* transaction) { |
900 if (success) { | 1080 DCHECK(transaction == entry->writer || |
901 ProcessPendingQueue(entry); | 1081 transaction == entry->headers_transaction); |
902 } else { | 1082 |
903 DCHECK(!entry->will_process_pending_queue); | 1083 if (transaction == entry->writer) |
904 | 1084 entry->writer = nullptr; |
905 // We failed to create this entry. | 1085 else |
906 TransactionList pending_queue; | 1086 entry->headers_transaction = nullptr; |
907 pending_queue.swap(entry->pending_queue); | 1087 |
908 | 1088 // If writer fails, all transactions restart the headers_transaction. |
909 entry->disk_entry->Doom(); | 1089 if (!success && entry->headers_transaction) { |
910 DestroyEntry(entry); | 1090 entry->headers_transaction->SetValidatingCannotProceed(); |
911 | 1091 entry->headers_transaction = nullptr; |
912 // We need to do something about these pending entries, which now need to | 1092 DCHECK(entry->HasNoActiveTransactions()); |
913 // be added to a new entry. | 1093 } |
914 while (!pending_queue.empty()) { | 1094 if (!success) |
915 // ERR_CACHE_RACE causes the transaction to restart the whole process. | 1095 ProcessEntryFailure(entry); |
916 pending_queue.front()->io_callback().Run(ERR_CACHE_RACE); | 1096 else |
917 pending_queue.pop_front(); | 1097 ProcessQueuedTransactions(entry); |
918 } | 1098 } |
919 } | 1099 |
920 } | 1100 void HttpCache::DoneReadingFromEntry(ActiveEntry* entry, |
921 | 1101 Transaction* transaction) { |
922 void HttpCache::DoneReadingFromEntry(ActiveEntry* entry, Transaction* trans) { | |
923 DCHECK(!entry->writer); | 1102 DCHECK(!entry->writer); |
924 | 1103 auto it = entry->readers.find(transaction); |
925 auto it = entry->readers.find(trans); | |
926 DCHECK(it != entry->readers.end()); | 1104 DCHECK(it != entry->readers.end()); |
927 | |
928 entry->readers.erase(it); | 1105 entry->readers.erase(it); |
929 | 1106 |
930 ProcessPendingQueue(entry); | 1107 ProcessQueuedTransactions(entry); |
931 } | |
932 | |
933 void HttpCache::ConvertWriterToReader(ActiveEntry* entry) { | |
934 DCHECK(entry->writer); | |
935 DCHECK(entry->writer->mode() == Transaction::READ_WRITE); | |
936 DCHECK(entry->readers.empty()); | |
937 | |
938 Transaction* trans = entry->writer; | |
939 | |
940 entry->writer = NULL; | |
941 entry->readers.insert(trans); | |
942 | |
943 ProcessPendingQueue(entry); | |
944 } | 1108 } |
945 | 1109 |
946 LoadState HttpCache::GetLoadStateForPendingTransaction( | 1110 LoadState HttpCache::GetLoadStateForPendingTransaction( |
947 const Transaction* trans) { | 1111 const Transaction* trans) { |
948 auto i = active_entries_.find(trans->key()); | 1112 auto i = active_entries_.find(trans->key()); |
949 if (i == active_entries_.end()) { | 1113 if (i == active_entries_.end()) { |
950 // If this is really a pending transaction, and it is not part of | 1114 // If this is really a pending transaction, and it is not part of |
951 // active_entries_, we should be creating the backend or the entry. | 1115 // active_entries_, we should be creating the backend or the entry. |
952 return LOAD_STATE_WAITING_FOR_CACHE; | 1116 return LOAD_STATE_WAITING_FOR_CACHE; |
953 } | 1117 } |
(...skipping 29 matching lines...) Expand all Loading... |
983 | 1147 |
984 for (auto k = doomed_entries_.begin(); k != doomed_entries_.end() && !found; | 1148 for (auto k = doomed_entries_.begin(); k != doomed_entries_.end() && !found; |
985 ++k) { | 1149 ++k) { |
986 found = RemovePendingTransactionFromEntry(k->first, trans); | 1150 found = RemovePendingTransactionFromEntry(k->first, trans); |
987 } | 1151 } |
988 | 1152 |
989 DCHECK(found) << "Pending transaction not found"; | 1153 DCHECK(found) << "Pending transaction not found"; |
990 } | 1154 } |
991 | 1155 |
992 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, | 1156 bool HttpCache::RemovePendingTransactionFromEntry(ActiveEntry* entry, |
993 Transaction* trans) { | 1157 Transaction* transaction) { |
994 TransactionList& pending_queue = entry->pending_queue; | 1158 TransactionList& add_to_entry_queue = entry->add_to_entry_queue; |
995 | 1159 |
996 auto j = find(pending_queue.begin(), pending_queue.end(), trans); | 1160 auto j = |
997 if (j == pending_queue.end()) | 1161 find(add_to_entry_queue.begin(), add_to_entry_queue.end(), transaction); |
| 1162 if (j == add_to_entry_queue.end()) |
998 return false; | 1163 return false; |
999 | 1164 |
1000 pending_queue.erase(j); | 1165 add_to_entry_queue.erase(j); |
1001 return true; | 1166 return true; |
1002 } | 1167 } |
1003 | 1168 |
1004 bool HttpCache::RemovePendingTransactionFromPendingOp(PendingOp* pending_op, | 1169 bool HttpCache::RemovePendingTransactionFromPendingOp(PendingOp* pending_op, |
1005 Transaction* trans) { | 1170 Transaction* trans) { |
1006 if (pending_op->writer->Matches(trans)) { | 1171 if (pending_op->writer->Matches(trans)) { |
1007 pending_op->writer->ClearTransaction(); | 1172 pending_op->writer->ClearTransaction(); |
1008 pending_op->writer->ClearEntry(); | 1173 pending_op->writer->ClearEntry(); |
1009 return true; | 1174 return true; |
1010 } | 1175 } |
1011 WorkItemList& pending_queue = pending_op->pending_queue; | 1176 WorkItemList& pending_queue = pending_op->pending_queue; |
1012 | 1177 |
1013 for (auto it = pending_queue.begin(); it != pending_queue.end(); ++it) { | 1178 for (auto it = pending_queue.begin(); it != pending_queue.end(); ++it) { |
1014 if ((*it)->Matches(trans)) { | 1179 if ((*it)->Matches(trans)) { |
1015 pending_queue.erase(it); | 1180 pending_queue.erase(it); |
1016 return true; | 1181 return true; |
1017 } | 1182 } |
1018 } | 1183 } |
1019 return false; | 1184 return false; |
1020 } | 1185 } |
1021 | 1186 |
1022 void HttpCache::ProcessPendingQueue(ActiveEntry* entry) { | |
1023 // Multiple readers may finish with an entry at once, so we want to batch up | |
1024 // calls to OnProcessPendingQueue. This flag also tells us that we should | |
1025 // not delete the entry before OnProcessPendingQueue runs. | |
1026 if (entry->will_process_pending_queue) | |
1027 return; | |
1028 entry->will_process_pending_queue = true; | |
1029 | |
1030 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
1031 FROM_HERE, | |
1032 base::Bind(&HttpCache::OnProcessPendingQueue, GetWeakPtr(), entry)); | |
1033 } | |
1034 | |
1035 void HttpCache::OnProcessPendingQueue(ActiveEntry* entry) { | |
1036 entry->will_process_pending_queue = false; | |
1037 DCHECK(!entry->writer); | |
1038 | |
1039 // If no one is interested in this entry, then we can deactivate it. | |
1040 if (entry->HasNoTransactions()) { | |
1041 DestroyEntry(entry); | |
1042 return; | |
1043 } | |
1044 | |
1045 if (entry->pending_queue.empty()) | |
1046 return; | |
1047 | |
1048 // Promote next transaction from the pending queue. | |
1049 Transaction* next = entry->pending_queue.front(); | |
1050 if ((next->mode() & Transaction::WRITE) && !entry->readers.empty()) | |
1051 return; // Have to wait. | |
1052 | |
1053 entry->pending_queue.erase(entry->pending_queue.begin()); | |
1054 | |
1055 int rv = AddTransactionToEntry(entry, next); | |
1056 if (rv != ERR_IO_PENDING) { | |
1057 next->io_callback().Run(rv); | |
1058 } | |
1059 } | |
1060 | |
1061 void HttpCache::OnIOComplete(int result, PendingOp* pending_op) { | 1187 void HttpCache::OnIOComplete(int result, PendingOp* pending_op) { |
1062 WorkItemOperation op = pending_op->writer->operation(); | 1188 WorkItemOperation op = pending_op->writer->operation(); |
1063 | 1189 |
1064 // Completing the creation of the backend is simpler than the other cases. | 1190 // Completing the creation of the backend is simpler than the other cases. |
1065 if (op == WI_CREATE_BACKEND) | 1191 if (op == WI_CREATE_BACKEND) |
1066 return OnBackendCreated(result, pending_op); | 1192 return OnBackendCreated(result, pending_op); |
1067 | 1193 |
1068 std::unique_ptr<WorkItem> item = std::move(pending_op->writer); | 1194 std::unique_ptr<WorkItem> item = std::move(pending_op->writer); |
1069 bool fail_requests = false; | 1195 bool fail_requests = false; |
1070 | 1196 |
(...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1195 building_backend_ = false; | 1321 building_backend_ = false; |
1196 DeletePendingOp(pending_op); | 1322 DeletePendingOp(pending_op); |
1197 } | 1323 } |
1198 | 1324 |
1199 // The cache may be gone when we return from the callback. | 1325 // The cache may be gone when we return from the callback. |
1200 if (!item->DoCallback(result, disk_cache_.get())) | 1326 if (!item->DoCallback(result, disk_cache_.get())) |
1201 item->NotifyTransaction(result, NULL); | 1327 item->NotifyTransaction(result, NULL); |
1202 } | 1328 } |
1203 | 1329 |
1204 } // namespace net | 1330 } // namespace net |
OLD | NEW |