• Home
• com.github.marschall
• jfr-jdbctemplate
• 0.1.0
• source code
• JfrJdbcOperations.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.marschall.jfrjdbctemplate.JfrJdbcOperations Maven / Gradle / Ivy

package com.github.marschall.jfrjdbctemplate;

import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import org.springframework.jdbc.core.BatchPreparedStatementSetter;
import org.springframework.jdbc.core.CallableStatementCallback;
import org.springframework.jdbc.core.CallableStatementCreator;
import org.springframework.jdbc.core.ConnectionCallback;
import org.springframework.jdbc.core.JdbcOperations;
import org.springframework.jdbc.core.ParameterizedPreparedStatementSetter;
import org.springframework.jdbc.core.PreparedStatementCallback;
import org.springframework.jdbc.core.PreparedStatementCreator;
import org.springframework.jdbc.core.PreparedStatementSetter;
import org.springframework.jdbc.core.ResultSetExtractor;
import org.springframework.jdbc.core.RowCallbackHandler;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.SqlParameter;
import org.springframework.jdbc.core.SqlProvider;
import org.springframework.jdbc.core.StatementCallback;
import org.springframework.jdbc.support.KeyHolder;
import org.springframework.jdbc.support.rowset.SqlRowSet;

import jdk.jfr.Category;
import jdk.jfr.Description;
import jdk.jfr.Event;
import jdk.jfr.Label;

/**
 * An implementation of {@link JdbcOperations} that generates JFR events.
 */
public final class JfrJdbcOperations implements JdbcOperations {

  // http://hirt.se/blog/?p=870
  // https://www.oracle.com/technetwork/oem/soa-mgmt/con10912-javaflightrecorder-2342054.pdf

  private final JdbcOperations delegate;

  /**
   * Constructs a new {@link JfrJdbcOperations}.
   * 
   * @param delegate the actual {@link JdbcOperations} implementation
   */
  public JfrJdbcOperations(JdbcOperations delegate) {
    Objects.requireNonNull(delegate, "delegate");
    this.delegate = delegate;
  }

  @Override
  public  T execute(ConnectionCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(getSql(action));
    event.begin();
    try {
      return this.delegate.execute(action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T execute(StatementCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(getSql(action));
    event.begin();
    try {
      return this.delegate.execute(action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void execute(String sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.execute(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(String sql, ResultSetExtractor rse) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, rse);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(String sql, RowCallbackHandler rch) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.query(sql, rch);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(String sql, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Class requiredType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, requiredType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public Map queryForMap(String sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForMap");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForMap(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List queryForList(String sql, Class elementType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, elementType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public List> queryForList(String sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public SqlRowSet queryForRowSet(String sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForRowSet");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForRowSet(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(String sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.update(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int[] batchUpdate(String... sql) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    if (sql != null && sql.length > 0) {
      event.setQuery(sql[0]);
    }
    event.begin();
    try {
      return this.delegate.batchUpdate(sql);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T execute(PreparedStatementCreator psc, PreparedStatementCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(getSql(psc));
    event.begin();
    try {
      return this.delegate.execute(psc, action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T execute(String sql, PreparedStatementCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.execute(sql, action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(PreparedStatementCreator psc, ResultSetExtractor rse) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(getSql(psc));
    event.begin();
    try {
      return this.delegate.query(psc, rse);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(String sql, PreparedStatementSetter pss, ResultSetExtractor rse) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, pss, rse);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(String sql, Object[] args, int[] argTypes, ResultSetExtractor rse) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, args, argTypes, rse);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(String sql, Object[] args, ResultSetExtractor rse) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, args, rse);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T query(String sql, ResultSetExtractor rse, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, rse, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(PreparedStatementCreator psc, RowCallbackHandler rch) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(getSql(psc));
    event.begin();
    try {
      this.delegate.query(psc, rch);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(String sql, PreparedStatementSetter pss, RowCallbackHandler rch) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.query(sql, pss, rch);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(String sql, Object[] args, int[] argTypes, RowCallbackHandler rch) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.query(sql, args, argTypes, rch);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(String sql, Object[] args, RowCallbackHandler rch) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.query(sql, args, rch);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public void query(String sql, RowCallbackHandler rch, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      this.delegate.query(sql, rch, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(PreparedStatementCreator psc, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(getSql(psc));
    event.begin();
    try {
      return this.delegate.query(psc, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(String sql, PreparedStatementSetter pss, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, pss, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(String sql, Object[] args, int[] argTypes, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, args, argTypes, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(String sql, Object[] args, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, args, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List query(String sql, RowMapper rowMapper, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("query");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.query(sql, rowMapper, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Object[] args, int[] argTypes, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, args, argTypes, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Object[] args, RowMapper rowMapper) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, args, rowMapper);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, RowMapper rowMapper, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, rowMapper, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Object[] args, int[] argTypes, Class requiredType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, args, argTypes, requiredType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Object[] args, Class requiredType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, args, requiredType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T queryForObject(String sql, Class requiredType, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForObject");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForObject(sql, requiredType, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public Map queryForMap(String sql, Object[] args, int[] argTypes) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForMap");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForMap(sql, args, argTypes);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public Map queryForMap(String sql, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForMap");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForMap(sql, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List queryForList(String sql, Object[] args, int[] argTypes, Class elementType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, args, argTypes, elementType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List queryForList(String sql, Object[] args, Class elementType) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, args, elementType);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  List queryForList(String sql, Class elementType, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, elementType, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public List> queryForList(String sql, Object[] args, int[] argTypes) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, args, argTypes);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public List> queryForList(String sql, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForList");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForList(sql, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public SqlRowSet queryForRowSet(String sql, Object[] args, int[] argTypes) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForRowSet");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForRowSet(sql, args, argTypes);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public SqlRowSet queryForRowSet(String sql, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("queryForRowSet");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.queryForRowSet(sql, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(PreparedStatementCreator psc) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(getSql(psc));
    event.begin();
    try {
      return this.delegate.update(psc);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(PreparedStatementCreator psc, KeyHolder generatedKeyHolder) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(getSql(generatedKeyHolder));
    event.begin();
    try {
      return this.delegate.update(psc, generatedKeyHolder);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(String sql, PreparedStatementSetter pss) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.update(sql, pss);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(String sql, Object[] args, int[] argTypes) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.update(sql, args, argTypes);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int update(String sql, Object... args) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("update");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.update(sql, args);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int[] batchUpdate(String sql, BatchPreparedStatementSetter pss) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.batchUpdate(sql, pss);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int[] batchUpdate(String sql, List batchArgs) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.batchUpdate(sql, batchArgs);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public int[] batchUpdate(String sql, List batchArgs, int[] argTypes) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.batchUpdate(sql, batchArgs, argTypes);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  int[][] batchUpdate(String sql, Collection batchArgs, int batchSize, ParameterizedPreparedStatementSetter pss) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    event.setQuery(sql);
    event.begin();
    try {
      return this.delegate.batchUpdate(sql, batchArgs, batchSize, pss);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T execute(CallableStatementCreator csc, CallableStatementCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(getSql(csc));
    event.begin();
    try {
      return this.delegate.execute(csc, action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public  T execute(String callString, CallableStatementCallback action) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("execute");
    event.setQuery(callString);
    event.begin();
    try {
      return this.delegate.execute(callString, action);
    } finally {
      event.end();
      event.commit();
    }
  }

  @Override
  public Map call(CallableStatementCreator csc, List declaredParameters) {
    JdbcEvent event = new JdbcEvent();
    event.setOperationName("batchUpdate");
    event.setQuery(getSql(csc));
    event.begin();
    try {
      return this.delegate.call(csc, declaredParameters);
    } finally {
      event.end();
      event.commit();
    }
  }

  private static String getSql(Object o) {
    if (o instanceof SqlProvider) {
      return ((SqlProvider) o).getSql();
    }
    return null;
  }

  @Label("JDBC Operation")
  @Description("A JDBC Operation")
  @Category("JDBC")
  static class JdbcEvent extends Event {

    @Label("Operation Name")
    @Description("The name of the JDBC operation")
    private String operationName;

    @Label("Query")
    @Description("The SQL query string")
    private String query;

    String getOperationName() {
      return this.operationName;
    }

    void setOperationName(String operationName) {
      this.operationName = operationName;
    }

    String getQuery() {
      return this.query;
    }

    void setQuery(String query) {
      this.query = query;
    }

  }

}