1 /*
2  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License").
5  * You may not use this file except in compliance with the License.
6  * A copy of the License is located at
7  *
8  *  http://aws.amazon.com/apache2.0
9  *
10  * or in the "license" file accompanying this file. This file is distributed
11  * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12  * express or implied. See the License for the specific language governing
13  * permissions and limitations under the License.
14  */
15
16 package software.amazon.awssdk.utils.cache;
17
18 import java.util.concurrent.atomic.AtomicBoolean;
19 import software.amazon.awssdk.annotations.SdkProtectedApi;
20
21 /**
22  * A {@link CachedSupplier.PrefetchStrategy} that will have one caller at a time block to update the value.
23  *
24  * Multiple calls to {@link #prefetch(Runnable)} will result in only one caller actually performing the update, with the others
25  * immediately returning.
26  */
27 @SdkProtectedApi
28 public class OneCallerBlocks implements CachedSupplier.PrefetchStrategy {
29     /**
30      * Whether we are currently refreshing the supplier. This is used to make sure only one caller is blocking at a time.
31      */
32     private final AtomicBoolean currentlyRefreshing = new AtomicBoolean(false);
33
34     @Override
35     public void prefetch(Runnable valueUpdater) {
36         if (currentlyRefreshing.compareAndSet(false, true)) {
37             try {
38                 valueUpdater.run();
39             } finally {
40                 currentlyRefreshing.set(false);
41             }
42         }
43     }
44 }
45