[duplicates] enable duplicates analysis in PyCharm/WebStorm/PhpStorm/RubyMine
[idea/community.git] / platform / core-api / src / com / intellij / openapi / util / TimedOutCallback.java
1 /*
2  * Copyright 2000-2017 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 com.intellij.openapi.util;
17
18 import com.intellij.openapi.diagnostic.Logger;
19
20 import javax.swing.*;
21
22 /**
23  * @author Konstantin Bulenkov
24 */
25 @SuppressWarnings({"UnusedDeclaration", "SSBasedInspection"})
26 public class TimedOutCallback extends ActionCallback implements Runnable {
27   private static final Logger LOG = Logger.getInstance("#com.intellij.openapi.util.ActionCallback.TimedOutCallback");
28
29   private Throwable myAllocation;
30   private String myMessage;
31   private SimpleTimerTask myTask;
32   private boolean myShouldDumpError;
33
34   public TimedOutCallback(final long timeOut, String message, Throwable allocation, boolean isEdt) {
35     scheduleCheck(timeOut, message, allocation, isEdt);
36   }
37
38   public TimedOutCallback(int countToDone, long timeOut, String message, Throwable allocation, boolean isEdt) {
39     super(countToDone);
40     scheduleCheck(timeOut, message, allocation, isEdt);
41   }
42
43   private void scheduleCheck(final long timeOut, final String message, Throwable allocation, final boolean isEdt) {
44     myMessage = message;
45     myAllocation = allocation;
46     final long current = System.currentTimeMillis();
47     myTask = SimpleTimer.getInstance().setUp(() -> {
48       myShouldDumpError = System.currentTimeMillis() - current > timeOut; //double check is necessary :-(
49       if (isEdt) {
50         SwingUtilities.invokeLater(this);
51       } else {
52         this.run();
53       }
54     }, timeOut);
55   }
56
57   @Override
58   public final void run() {
59     if (!isProcessed()) {
60       setRejected();
61
62       if (myShouldDumpError) {
63         dumpError();
64       }
65
66       onTimeout();
67     }
68   }
69
70   protected void dumpError() {
71     if (myAllocation != null) {
72       LOG.error(myMessage, myAllocation);
73     } else {
74       LOG.error(myMessage);
75     }
76   }
77
78   public String getMessage() {
79     return myMessage;
80   }
81
82   public Throwable getAllocation() {
83     return myAllocation;
84   }
85
86   @Override
87   public void dispose() {
88     super.dispose();
89     myTask.cancel();
90   }
91
92   protected void onTimeout() {
93   }
94 }