jio.test.junit.DatabaseBatchDebugger Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jio-test Show documentation
JIO test library based on Property Based Testing and Java Flight Recording Debuggers
There is a newer version: 3.0.0-RC2
package jio.test.junit;

import jdk.jfr.consumer.RecordedEvent;
import jio.test.Utils;
import jio.time.Fun;

import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.function.Consumer;

@SuppressWarnings("InlineFormatString")
final class DatabaseBatchDebugger implements Consumer {

  private static final String FORMAT_SUC = """
      ------ jdbc-client batch -----
      |  Label: %s
      |  Result: %s
      |  Duration: %s
      |  Rows Affected: %s
      |  Executed Batches: %s
      |  Batch Size: %s
      |  Statements Size: %s
      |  Operation Counter: %s
      |  Thread: %s
      |  Event Start Time: %s
      -------------------------------
      """;

  private static final String FORMAT_ERR = """
      ------ jdbc-client batch -----
      |  Label: %s
      |  Result: %s
      |  Duration: %s
      |  Rows Affected: %s
      |  Executed Batches: %s
      |  Batch Size: %s
      |  Statements Size: %s
      |  SQL: %s
      |  Exception: %s
      |  Operation Counter: %s
      |  Thread: %s
      |  Event Start Time: %s
      -------------------------------
      """;
  static final String EVENT_NAME = "jio.jdbc.BatchStm";


  @Override
  public void accept(RecordedEvent event) {
    assert event.getEventType()
                .getName()
                .equals(EVENT_NAME);
    var result = event.getValue(EventFields.RESULT);
    var label = event.getValue(EventFields.LABEL);
    boolean isSuccess = "SUCCESS".equals(result);
    var message = isSuccess ?
                  String.format(FORMAT_SUC,
                                label,
                                event.getValue(EventFields.RESULT),
                                Fun.formatTime(event.getDuration()),
                                event.getValue(EventFields.ROWS_AFFECTED),
                                event.getValue(EventFields.EXECUTED_BATCHES),
                                event.getValue(EventFields.BATCH_SIZE),
                                event.getValue(EventFields.TOTAL_STMS),
                                event.getValue(EventFields.OP_COUNTER),
                                Utils.getThreadName(event.getThread()),
                                event.getStartTime()
                                     .atZone(ZoneId.systemDefault())
                                     .format(DateTimeFormatter.ISO_OFFSET_DATE_TIME)
                               ) :
                  String.format(FORMAT_ERR,
                                label,
                                event.getValue(EventFields.RESULT),
                                Fun.formatTime(event.getDuration()),
                                event.getValue(EventFields.ROWS_AFFECTED),
                                event.getValue(EventFields.EXECUTED_BATCHES),
                                event.getValue(EventFields.BATCH_SIZE),
                                event.getValue(EventFields.TOTAL_STMS),
                                event.getValue(EventFields.SQL),
                                event.getValue(EventFields.EXCEPTION),
                                event.getValue(EventFields.OP_COUNTER),
                                Utils.getThreadName(event.getThread()),
                                event.getStartTime()
                                     .atZone(ZoneId.systemDefault())
                                     .format(DateTimeFormatter.ISO_OFFSET_DATE_TIME)
                               );
    synchronized (System.out) {
      System.out.println(message);
      System.out.flush();
    }
  }
}