1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.omid.tso;
19
20 import org.apache.omid.metrics.MetricsRegistry;
21 import org.apache.omid.timestamp.storage.TimestampStorage;
22 import org.slf4j.Logger;
23 import org.slf4j.LoggerFactory;
24
25 import javax.inject.Inject;
26 import java.io.IOException;
27
28 public class PausableTimestampOracle extends TimestampOracleImpl {
29
30 private static final Logger LOG = LoggerFactory.getLogger(PausableTimestampOracle.class);
31
32 private volatile boolean tsoPaused = false;
33
34 @Inject
35 public PausableTimestampOracle(MetricsRegistry metrics,
36 TimestampStorage tsStorage,
37 Panicker panicker) throws IOException {
38 super(metrics, tsStorage, panicker);
39 }
40
41 @Override
42 public long next() {
43 while (tsoPaused) {
44 synchronized (this) {
45 try {
46 this.wait();
47 } catch (InterruptedException e) {
48 LOG.error("Interrupted whilst paused");
49 Thread.currentThread().interrupt();
50 }
51 }
52 }
53 return super.next();
54 }
55
56 public synchronized void pause() {
57 tsoPaused = true;
58 this.notifyAll();
59 }
60
61 public synchronized void resume() {
62 tsoPaused = false;
63 this.notifyAll();
64 }
65
66 public boolean isTSOPaused() {
67 return tsoPaused;
68 }
69
70 }