private static Consumer newConsumer() throws DTraceException { Consumer cons = new LocalConsumer(); cons.addConsumerListener( new ConsumerAdapter() { @Override public void consumerStarted(final ConsumerEvent ce) { rt.send( DTraceStartCommand.class, new AbstractCommand.Initializer<DTraceStartCommand>() { public void init(DTraceStartCommand cmd) { cmd.setConsumerEvent(ce); } }); } @Override public void consumerStopped(final ConsumerEvent ce) { Consumer cons = (Consumer) ce.getSource(); Aggregate ag = null; try { ag = cons.getAggregate(); } catch (final DTraceException dexp) { rt.send( ErrorCommand.class, new AbstractCommand.Initializer<ErrorCommand>() { public void init(ErrorCommand cmd) { cmd.setCause(dexp); } }); } StringBuilder buf = new StringBuilder(); if (ag != null) { for (Aggregation agg : ag.asMap().values()) { String name = agg.getName(); if (name != null && name.length() > 0) { buf.append(name); buf.append('\n'); } for (AggregationRecord rec : agg.asMap().values()) { buf.append('\t'); buf.append(rec.getTuple()); buf.append(" "); buf.append(rec.getValue()); buf.append('\n'); } } } final String msg = buf.toString(); if (msg.length() > 0) { rt.send( MessageCommand.class, new AbstractCommand.Initializer<MessageCommand>() { public void init(MessageCommand cmd) { cmd.setMessage(msg); } }); } rt.send( DTraceStopCommand.class, new AbstractCommand.Initializer<DTraceStopCommand>() { public void init(DTraceStopCommand cmd) { cmd.setConsumerEvent(ce); } }); cons.close(); } @Override public void dataReceived(final DataEvent de) { rt.send( DTraceDataCommand.class, new AbstractCommand.Initializer<DTraceDataCommand>() { public void init(DTraceDataCommand cmd) { cmd.setDataEvent(de); } }); } @Override public void dataDropped(final DropEvent de) { rt.send( DTraceDropCommand.class, new AbstractCommand.Initializer<DTraceDropCommand>() { public void init(DTraceDropCommand cmd) { cmd.setDropEvent(de); } }); } @Override public void errorEncountered(final ErrorEvent ee) throws ConsumerException { try { super.errorEncountered(ee); } catch (final ConsumerException ce) { rt.send( DTraceErrorCommand.class, new AbstractCommand.Initializer<DTraceErrorCommand>() { public void init(DTraceErrorCommand cmd) { cmd.setErrorEvent(ee); cmd.setException(ce); } }); throw ce; } } }); // open DTrace Consumer cons.open(); // unused macro arguments are fine cons.setOption(Option.argref, ""); // if no macro arg passed use "" or NULL cons.setOption(Option.defaultargs, ""); // allow empty D-scripts cons.setOption(Option.empty, ""); // be quiet! equivalent to DTrace's -q cons.setOption(Option.quiet, ""); // undefined user land symbols are fine cons.setOption(Option.unodefs, ""); // allow zero matching of probes (needed for late loading) cons.setOption(Option.zdefs, ""); try { int pid = Integer.parseInt(rt.$(0)); cons.grabProcess(pid); } catch (Exception ignored) { } return cons; }
private static Consumer newConsumer(String[] args, final CommandListener listener) throws DTraceException { Consumer cons = new LocalConsumer(); cons.addConsumerListener( new ConsumerAdapter() { private void notify(Command cmd) { try { listener.onCommand(cmd); } catch (IOException ioexp) { ioexp.printStackTrace(); } } @Override public void consumerStarted(ConsumerEvent ce) { notify(new DTraceStartCommand(ce)); } @Override public void consumerStopped(ConsumerEvent ce) { Consumer cons = (Consumer) ce.getSource(); Aggregate ag = null; try { ag = cons.getAggregate(); } catch (DTraceException dexp) { notify(new ErrorCommand(dexp)); } StringBuilder buf = new StringBuilder(); if (ag != null) { for (Aggregation agg : ag.asMap().values()) { String name = agg.getName(); if (name != null && name.length() > 0) { buf.append(name); buf.append('\n'); } for (AggregationRecord rec : agg.asMap().values()) { buf.append('\t'); buf.append(rec.getTuple()); buf.append(" "); buf.append(rec.getValue()); buf.append('\n'); } } } String msg = buf.toString(); if (msg.length() > 0) { notify(new MessageCommand(msg)); } notify(new DTraceStopCommand(ce)); cons.close(); } @Override public void dataReceived(DataEvent de) { notify(new DTraceDataCommand(de)); } @Override public void dataDropped(DropEvent de) { notify(new DTraceDropCommand(de)); } @Override public void errorEncountered(ErrorEvent ee) throws ConsumerException { try { super.errorEncountered(ee); } catch (ConsumerException ce) { notify(new DTraceErrorCommand(ce, ee)); throw ce; } } }); // open DTrace Consumer cons.open(); // unused macro arguments are fine cons.setOption(Option.argref, ""); // if no macro arg passed use "" or NULL cons.setOption(Option.defaultargs, ""); // allow empty D-scripts cons.setOption(Option.empty, ""); // be quiet! equivalent to DTrace's -q cons.setOption(Option.quiet, ""); // undefined user land symbols are fine cons.setOption(Option.unodefs, ""); // allow zero matching of probes (needed for late loading) cons.setOption(Option.zdefs, ""); try { int pid = Integer.parseInt(args[0]); cons.grabProcess(pid); } catch (Exception ignored) { } return cons; }