adc6069eeda326de19e53b325e950a12827b057d
[idea/community.git] / plugins / groovy / groovy-psi / src / org / jetbrains / plugins / groovy / util / NotNullCachedComputableWrapper.java
1 /*
2  * Copyright 2000-2015 JetBrains s.r.o.
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  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.jetbrains.plugins.groovy.util;
17
18 import com.intellij.openapi.util.NotNullComputable;
19 import com.intellij.openapi.util.RecursionGuard;
20 import com.intellij.openapi.util.RecursionManager;
21 import org.jetbrains.annotations.NotNull;
22 import org.jetbrains.annotations.TestOnly;
23
24 public class NotNullCachedComputableWrapper<T> implements NotNullComputable<T> {
25
26   private static final RecursionGuard ourGuard = RecursionManager.createGuard(NotNullCachedComputableWrapper.class.getName());
27
28   private NotNullComputable<T> myComputable;
29   private volatile T myValue;
30
31   public NotNullCachedComputableWrapper(@NotNull NotNullComputable<T> computable) {
32     myComputable = computable;
33   }
34
35   @NotNull
36   @Override
37   public T compute() {
38     T result = myValue;
39     if (result != null) return result;
40
41     //noinspection SynchronizeOnThis
42     synchronized (this) {
43       result = myValue;
44       if (result == null) {
45         final RecursionGuard.StackStamp stamp = ourGuard.markStack();
46         result = myComputable.compute();
47         if (stamp.mayCacheNow()) {
48           myValue = result;
49           myComputable = null;  // allow gc to clean this up
50         }
51       }
52     }
53
54     return result;
55   }
56
57   @TestOnly
58   public boolean isComputed() {
59     return myValue != null;
60   }
61 }