forked from btraceio/btrace
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' of github.com:btraceio/btrace
- Loading branch information
Showing
4 changed files
with
164 additions
and
110 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
89 changes: 89 additions & 0 deletions
89
src/share/classes/com/sun/btrace/api/impl/BTraceTaskDispatcher.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,89 @@ | ||
package com.sun.btrace.api.impl; | ||
|
||
import com.sun.btrace.api.BTraceTask; | ||
import com.sun.btrace.comm.*; | ||
|
||
import java.util.HashSet; | ||
import java.util.Set; | ||
import java.util.concurrent.ExecutorService; | ||
import java.util.concurrent.Executors; | ||
|
||
class BTraceTaskDispatcher { | ||
private final static ExecutorService dispatcher = Executors.newSingleThreadExecutor(); | ||
private final Set<BTraceTask.MessageDispatcher> messageDispatchers = new HashSet<>(); | ||
|
||
BTraceTaskDispatcher() { | ||
} | ||
|
||
/** | ||
* Dispatcher management | ||
* | ||
* @param dispatcher {@linkplain BTraceTask.MessageDispatcher} instance to add | ||
*/ | ||
void addMessageDispatcher(BTraceTask.MessageDispatcher dispatcher) { | ||
synchronized (messageDispatchers) { | ||
messageDispatchers.add(dispatcher); | ||
} | ||
} | ||
|
||
/** | ||
* Dispatcher management | ||
* | ||
* @param dispatcher {@linkplain BTraceTask.MessageDispatcher} instance to remove | ||
*/ | ||
void removeMessageDispatcher(BTraceTask.MessageDispatcher dispatcher) { | ||
synchronized (messageDispatchers) { | ||
messageDispatchers.remove(dispatcher); | ||
} | ||
} | ||
|
||
@SuppressWarnings("FutureReturnValueIgnored") | ||
void dispatchCommand(final Command cmd) { | ||
final Set<BTraceTask.MessageDispatcher> dispatchingSet = new HashSet<BTraceTask.MessageDispatcher>(); | ||
synchronized (messageDispatchers) { | ||
dispatchingSet.addAll(messageDispatchers); | ||
} | ||
dispatcher.submit(new Runnable() { | ||
@Override | ||
public void run() { | ||
for (BTraceTask.MessageDispatcher listener : dispatchingSet) { | ||
switch (cmd.getType()) { | ||
case Command.MESSAGE: { | ||
listener.onPrintMessage(((MessageCommand) cmd).getMessage()); | ||
break; | ||
} | ||
case Command.RETRANSFORM_CLASS: { | ||
listener.onClassInstrumented(((RetransformClassNotification) cmd).getClassName()); | ||
break; | ||
} | ||
case Command.NUMBER: { | ||
NumberDataCommand ndc = (NumberDataCommand) cmd; | ||
listener.onNumberMessage(ndc.getName(), ndc.getValue()); | ||
break; | ||
} | ||
case Command.NUMBER_MAP: { | ||
NumberMapDataCommand nmdc = (NumberMapDataCommand) cmd; | ||
listener.onNumberMap(nmdc.getName(), nmdc.getData()); | ||
break; | ||
} | ||
case Command.STRING_MAP: { | ||
StringMapDataCommand smdc = (StringMapDataCommand) cmd; | ||
listener.onStringMap(smdc.getName(), smdc.getData()); | ||
break; | ||
} | ||
case Command.GRID_DATA: { | ||
GridDataCommand gdc = (GridDataCommand) cmd; | ||
listener.onGrid(gdc.getName(), gdc.getData()); | ||
break; | ||
} | ||
case Command.ERROR: { | ||
ErrorCommand ec = (ErrorCommand) cmd; | ||
listener.onError(ec.getCause()); | ||
break; | ||
} | ||
} | ||
} | ||
} | ||
}); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,55 @@ | ||
package com.sun.btrace; | ||
|
||
import org.junit.Before; | ||
import org.junit.Test; | ||
import static org.junit.Assert.*; | ||
|
||
public class ArgsMapTest { | ||
private static final String KEY1 = "key1"; | ||
private static final String KEY2 = "key2"; | ||
private static final String VALUE1 = "value1"; | ||
private static final String VALUE2 = "value2"; | ||
|
||
private ArgsMap instance; | ||
|
||
@Before | ||
public void setUp() { | ||
instance = new ArgsMap(); | ||
instance.put(KEY1, VALUE1); | ||
instance.put(KEY2, VALUE2); | ||
} | ||
|
||
@Test | ||
public void templateExisting() { | ||
String value = instance.template(KEY1 + "=${" + KEY1 + "}"); | ||
assertEquals(KEY1 + "=" + VALUE1, value); | ||
} | ||
|
||
@Test | ||
public void templateNonExisting() { | ||
String orig = KEY1 + "=${key3}"; | ||
String value = instance.template(orig); | ||
assertEquals(orig, value); | ||
} | ||
|
||
@Test | ||
public void templateTrailing$() { | ||
String orig = KEY1 + "$"; | ||
String value = instance.template(orig); | ||
assertEquals(orig, value); | ||
} | ||
|
||
@Test | ||
public void templateUnclosedPlaceholder() { | ||
String orig = KEY1 + "${"; | ||
String value = instance.template(orig); | ||
assertEquals(orig, value); | ||
} | ||
|
||
@Test | ||
public void templateSingle$() { | ||
String orig = KEY1 + "$" + KEY2; | ||
String value = instance.template(orig); | ||
assertEquals(orig, value); | ||
} | ||
} |