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 "base/threading/sequenced_worker_pool.h" | 5 #include "base/threading/sequenced_worker_pool.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
(...skipping 886 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
897 // Should be fine to call on an idle instance with all threads created, and | 897 // Should be fine to call on an idle instance with all threads created, and |
898 // spamming the method shouldn't deadlock or confuse the class. | 898 // spamming the method shouldn't deadlock or confuse the class. |
899 pool()->FlushForTesting(); | 899 pool()->FlushForTesting(); |
900 pool()->FlushForTesting(); | 900 pool()->FlushForTesting(); |
901 | 901 |
902 // Should be fine to call after shutdown too. | 902 // Should be fine to call after shutdown too. |
903 pool()->Shutdown(); | 903 pool()->Shutdown(); |
904 pool()->FlushForTesting(); | 904 pool()->FlushForTesting(); |
905 } | 905 } |
906 | 906 |
907 namespace { | 907 void VerifySequencedTaskRunner( |
gab
2015/11/10 19:54:19
IsSameSequence() or something? (+ a commment?) --
Bernhard Bauer
2015/11/10 20:31:47
Done -- tried to make the name as descriptive as p
| |
908 | 908 const scoped_refptr<SequencedTaskRunner>& task_runner) { |
909 void CheckWorkerPoolAndSequenceToken( | 909 EXPECT_TRUE(task_runner->RunsTasksOnCurrentThread()); |
910 const scoped_refptr<SequencedWorkerPool>& expected_pool, | |
911 SequencedWorkerPool::SequenceToken expected_token) { | |
912 SequencedWorkerPool::SequenceToken token = | |
913 SequencedWorkerPool::GetSequenceTokenForCurrentThread(); | |
914 EXPECT_EQ(expected_token.ToString(), token.ToString()); | |
915 | |
916 scoped_refptr<SequencedWorkerPool> pool = | |
917 SequencedWorkerPool::GetWorkerPoolForCurrentThread(); | |
918 EXPECT_EQ(expected_pool, pool); | |
919 } | 910 } |
920 | 911 |
921 } // namespace | 912 void GetCurrentSequencedTaskRunner( |
gab
2015/11/10 19:54:19
Again, a "Get" method that returns void feels weir
Bernhard Bauer
2015/11/10 20:31:47
Done.
| |
913 scoped_refptr<SequencedTaskRunner> expected_task_runner) { | |
914 scoped_refptr<SequencedTaskRunner> task_runner = | |
915 SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread(); | |
922 | 916 |
923 TEST_F(SequencedWorkerPoolTest, GetWorkerPoolAndSequenceTokenForCurrentThread) { | 917 // If the expected task runner is null, get another one for the current |
918 // thread. They should be the same. | |
919 if (!expected_task_runner) { | |
920 expected_task_runner = | |
921 SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread(); | |
922 } | |
923 | |
924 // SequencedTaskRunner does not allow directly checking for equality, but we | |
925 // can post a task to one task runner and verify that the other task runner | |
926 // is on the same sequence. | |
927 task_runner->PostTask( | |
928 FROM_HERE, base::Bind(&VerifySequencedTaskRunner, expected_task_runner)); | |
929 } | |
930 | |
931 TEST_F(SequencedWorkerPoolTest, GetSequencedTaskRunnerForCurrentThread) { | |
924 EnsureAllWorkersCreated(); | 932 EnsureAllWorkersCreated(); |
925 | 933 |
926 // The current thread should have neither a worker pool nor a sequence token. | 934 // The current thread should not have a sequenced task runner from a |
927 SequencedWorkerPool::SequenceToken local_token = | 935 // worker pool. |
928 SequencedWorkerPool::GetSequenceTokenForCurrentThread(); | 936 scoped_refptr<SequencedTaskRunner> local_task_runner = |
929 scoped_refptr<SequencedWorkerPool> local_pool = | 937 SequencedWorkerPool::GetSequencedTaskRunnerForCurrentThread(); |
930 SequencedWorkerPool::GetWorkerPoolForCurrentThread(); | 938 EXPECT_FALSE(local_task_runner); |
931 EXPECT_FALSE(local_token.IsValid()) << local_token.ToString(); | |
932 EXPECT_FALSE(local_pool); | |
933 | 939 |
934 SequencedWorkerPool::SequenceToken token1 = pool()->GetSequenceToken(); | 940 scoped_refptr<SequencedTaskRunner> task_runner_1 = |
935 SequencedWorkerPool::SequenceToken token2 = pool()->GetSequenceToken(); | 941 pool()->GetSequencedTaskRunner(SequencedWorkerPool::GetSequenceToken()); |
936 pool()->PostSequencedWorkerTask( | 942 scoped_refptr<SequencedTaskRunner> task_runner_2 = |
937 token1, FROM_HERE, | 943 pool()->GetSequencedTaskRunner(SequencedWorkerPool::GetSequenceToken()); |
938 base::Bind(&CheckWorkerPoolAndSequenceToken, pool(), token1)); | 944 task_runner_1->PostTask( |
939 pool()->PostSequencedWorkerTask( | 945 FROM_HERE, base::Bind(&GetCurrentSequencedTaskRunner, task_runner_1)); |
940 token2, FROM_HERE, | 946 task_runner_2->PostTask( |
941 base::Bind(&CheckWorkerPoolAndSequenceToken, pool(), token2)); | 947 FROM_HERE, base::Bind(&GetCurrentSequencedTaskRunner, task_runner_2)); |
942 | 948 |
943 pool()->PostWorkerTask(FROM_HERE, | 949 pool()->PostWorkerTask(FROM_HERE, |
944 base::Bind(&CheckWorkerPoolAndSequenceToken, pool(), | 950 base::Bind(&GetCurrentSequencedTaskRunner, nullptr)); |
945 SequencedWorkerPool::SequenceToken())); | |
946 | 951 |
947 pool()->FlushForTesting(); | 952 pool()->FlushForTesting(); |
953 pool()->Shutdown(); | |
948 } | 954 } |
949 | 955 |
950 TEST(SequencedWorkerPoolRefPtrTest, ShutsDownCleanWithContinueOnShutdown) { | 956 TEST(SequencedWorkerPoolRefPtrTest, ShutsDownCleanWithContinueOnShutdown) { |
951 MessageLoop loop; | 957 MessageLoop loop; |
952 scoped_refptr<SequencedWorkerPool> pool(new SequencedWorkerPool(3, "Pool")); | 958 scoped_refptr<SequencedWorkerPool> pool(new SequencedWorkerPool(3, "Pool")); |
953 scoped_refptr<SequencedTaskRunner> task_runner = | 959 scoped_refptr<SequencedTaskRunner> task_runner = |
954 pool->GetSequencedTaskRunnerWithShutdownBehavior( | 960 pool->GetSequencedTaskRunnerWithShutdownBehavior( |
955 pool->GetSequenceToken(), | 961 pool->GetSequenceToken(), |
956 base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN); | 962 base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN); |
957 | 963 |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
1078 SequencedWorkerPoolSequencedTaskRunner, SequencedTaskRunnerTest, | 1084 SequencedWorkerPoolSequencedTaskRunner, SequencedTaskRunnerTest, |
1079 SequencedWorkerPoolSequencedTaskRunnerTestDelegate); | 1085 SequencedWorkerPoolSequencedTaskRunnerTestDelegate); |
1080 INSTANTIATE_TYPED_TEST_CASE_P( | 1086 INSTANTIATE_TYPED_TEST_CASE_P( |
1081 SequencedWorkerPoolSequencedTaskRunner, | 1087 SequencedWorkerPoolSequencedTaskRunner, |
1082 SequencedTaskRunnerDelayedTest, | 1088 SequencedTaskRunnerDelayedTest, |
1083 SequencedWorkerPoolSequencedTaskRunnerTestDelegate); | 1089 SequencedWorkerPoolSequencedTaskRunnerTestDelegate); |
1084 | 1090 |
1085 } // namespace | 1091 } // namespace |
1086 | 1092 |
1087 } // namespace base | 1093 } // namespace base |
OLD | NEW |