1 | package com.framsticks.util.dispatching; |
---|
2 | |
---|
3 | |
---|
4 | import org.apache.logging.log4j.Logger; |
---|
5 | import org.apache.logging.log4j.LogManager; |
---|
6 | |
---|
7 | import com.framsticks.util.FramsticksException; |
---|
8 | import com.framsticks.util.lang.Strings; |
---|
9 | // import static com.framsticks.util.dispatching.AbstractJoinable.wrap; |
---|
10 | |
---|
11 | public class BufferedDispatcher<C> extends AbstractJoinable implements Dispatcher<C>, JoinableParent { |
---|
12 | private static final Logger log = LogManager.getLogger(BufferedDispatcher.class); |
---|
13 | |
---|
14 | protected final RunnableQueue<C> queue = new RunnableQueue<>(); |
---|
15 | |
---|
16 | protected Dispatcher<C> targetDispatcher; |
---|
17 | |
---|
18 | protected boolean buffer = true; |
---|
19 | protected Dispatcher<C> parent; |
---|
20 | |
---|
21 | /** |
---|
22 | * @param parent |
---|
23 | */ |
---|
24 | public BufferedDispatcher(Dispatcher<C> parent) { |
---|
25 | this.parent = parent; |
---|
26 | } |
---|
27 | |
---|
28 | /** |
---|
29 | * @return the targetDispatcher |
---|
30 | */ |
---|
31 | public synchronized Dispatcher<C> getTargetDispatcher() { |
---|
32 | return targetDispatcher; |
---|
33 | } |
---|
34 | |
---|
35 | /** |
---|
36 | * @param targetDispatcher the targetDispatcher to set |
---|
37 | */ |
---|
38 | public synchronized void setTargetDispatcher(Dispatcher<C> targetDispatcher) { |
---|
39 | if (this.targetDispatcher != null) { |
---|
40 | throw new FramsticksException().msg("dispatcher is already set").arg("dispatcher", this.targetDispatcher).arg("in", this); |
---|
41 | } |
---|
42 | if (targetDispatcher == null) { |
---|
43 | throw new FramsticksException().msg("trying to set empty target dispatcher").arg("in", this); |
---|
44 | } |
---|
45 | log.debug("setting {} to {}", this, targetDispatcher); |
---|
46 | this.targetDispatcher = targetDispatcher; |
---|
47 | flushQueue(); |
---|
48 | } |
---|
49 | |
---|
50 | public synchronized boolean isActive() { |
---|
51 | if (targetDispatcher == null || buffer) { |
---|
52 | return false; |
---|
53 | // throw new FramsticksException().msg("no dispatcher is set for tree yet").arg("tree", this); |
---|
54 | } |
---|
55 | return targetDispatcher.isActive(); |
---|
56 | } |
---|
57 | |
---|
58 | public synchronized void dispatch(final RunAt<? extends C> runnable) { |
---|
59 | if (targetDispatcher != null && !buffer) { |
---|
60 | targetDispatcher.dispatch(runnable); |
---|
61 | return; |
---|
62 | } |
---|
63 | queue.push(runnable); |
---|
64 | } |
---|
65 | |
---|
66 | protected void flushQueue() { |
---|
67 | if (this.buffer || targetDispatcher == null) { |
---|
68 | return; |
---|
69 | } |
---|
70 | log.debug("flushing {} tasks in {}", queue.size(), this); |
---|
71 | while (!queue.isEmpty()) { |
---|
72 | targetDispatcher.dispatch(queue.pollFirst()); |
---|
73 | } |
---|
74 | } |
---|
75 | |
---|
76 | public synchronized void setBuffer(final boolean buffer) { |
---|
77 | if (this.buffer == buffer) { |
---|
78 | return; |
---|
79 | } |
---|
80 | if (buffer) { |
---|
81 | this.buffer = true; |
---|
82 | return; |
---|
83 | } |
---|
84 | this.buffer = false; |
---|
85 | flushQueue(); |
---|
86 | } |
---|
87 | |
---|
88 | @Override |
---|
89 | public String toString() { |
---|
90 | return parent + " -> " + Strings.toStringNullProof(targetDispatcher, "<null>"); |
---|
91 | } |
---|
92 | |
---|
93 | public void createThreadIfNeeded() { |
---|
94 | synchronized (this) { |
---|
95 | if (targetDispatcher != null) { |
---|
96 | return; |
---|
97 | } |
---|
98 | } |
---|
99 | this.setTargetDispatcher(new Thread<C>().setName(parent.getName())); |
---|
100 | } |
---|
101 | |
---|
102 | @Override |
---|
103 | public String getName() { |
---|
104 | return parent + " buffered dispatcher"; |
---|
105 | } |
---|
106 | |
---|
107 | @Override |
---|
108 | protected synchronized void joinableStart() { |
---|
109 | Dispatching.use(targetDispatcher, this); |
---|
110 | } |
---|
111 | |
---|
112 | @Override |
---|
113 | protected synchronized void joinableInterrupt() { |
---|
114 | Dispatching.drop(targetDispatcher, this); |
---|
115 | |
---|
116 | finishJoinable(); |
---|
117 | } |
---|
118 | |
---|
119 | @Override |
---|
120 | protected void joinableFinish() { |
---|
121 | |
---|
122 | } |
---|
123 | |
---|
124 | @Override |
---|
125 | protected synchronized void joinableJoin() throws InterruptedException { |
---|
126 | Dispatching.join(targetDispatcher); |
---|
127 | } |
---|
128 | |
---|
129 | @Override |
---|
130 | public void childChangedState(Joinable joinable, JoinableState state) { |
---|
131 | proceedToState(state); |
---|
132 | } |
---|
133 | } |
---|