文章目錄

• Code

Code

public class JdbcCommonCollects extends AbstractByteTransformCollect implements ICollect {public static JdbcCommonCollects INSTANCE;private ApmContext context;public JdbcCommonCollects(ApmContext context, Instrumentation instrumentation) {super(instrumentation);INSTANCE=this;this.context=context;}private final static String[] connection_agent_methods = new String[]{"prepareStatement"};private final static String[] prepared_statement_methods = new String[]{"execute", "executeUpdate", "executeQuery"};private static final String beginSrc;private static final String endSrc;private static final String errorSrc;static {// connectbeginSrc = "com.artisan.collects.JdbcCommonCollects inst=com.artisan.collects.JdbcCommonCollects.INSTANCE;";errorSrc = "inst.error(null,e);";endSrc = "result=inst.proxyConnection((java.sql.Connection)result);";}public JdbcStatistics begin(String className, String method) {JdbcStatistics jdbcStat = new JdbcStatistics();jdbcStat.begin=System.currentTimeMillis();jdbcStat.setModelType("jdbc");return jdbcStat;}public void end(JdbcStatistics stat) {JdbcStatistics jdbcStat= (JdbcStatistics) stat;jdbcStat.end=System.currentTimeMillis();jdbcStat.useTime =jdbcStat.end-jdbcStat.begin;if (jdbcStat.jdbcUrl != null) {jdbcStat.databaseName = getDbName(jdbcStat.jdbcUrl);}this.context.submitCollectResult(stat);}public void error(JdbcStatistics stat, Throwable throwable) {if (stat != null) {stat.error = throwable.getMessage();stat.errorType = throwable.getClass().getName();if (throwable instanceof InvocationTargetException) {stat.errorType = ((InvocationTargetException) throwable).getTargetException().getClass().getName();stat.error = ((InvocationTargetException) throwable).getTargetException().getMessage();}}}public void sendStatistics(JdbcStatistics stat) {}public Connection proxyConnection(final Connection connection) {Object c = Proxy.newProxyInstance(JdbcCommonCollects.class.getClassLoader(), new Class[]{Connection.class}, new ConnectionHandler(connection));return (Connection) c;}public PreparedStatement proxyPreparedStatement(final PreparedStatement statement, JdbcStatistics jdbcStat) {Object c = Proxy.newProxyInstance(JdbcCommonCollects.class.getClassLoader(), new Class[]{PreparedStatement.class}, new PreparedStatementHandler(statement, jdbcStat));return (PreparedStatement) c;}public byte[] transform(ClassLoader loader, String className) throws Exception {// System.out.println("JDBC className:" + className);// 如果更換數(shù)據(jù)庫(kù)，需要替換改類(lèi) 這裡目前僅適配了MySQLif (!className.equals("com.mysql.cj.jdbc.NonRegisteringDriver")) {return null;}CtClass ctclass = super.toCtClass(loader, className);AgentByteBuild byteLoade = new AgentByteBuild(className, loader, ctclass);CtMethod connectMethod = ctclass.getMethod("connect", "(Ljava/lang/String;Ljava/util/Properties;)Ljava/sql/Connection;"); // connectMethod.getMethodInfo().getDescriptor();AgentByteBuild.MethodSrcBuild build = new AgentByteBuild.MethodSrcBuild();build.setBeginSrc(beginSrc);build.setErrorSrc(errorSrc);build.setEndSrc(endSrc);byteLoade.updateMethod(connectMethod, build);return byteLoade.toBytecote();}/*** connection 代理處理*/public class ConnectionHandler implements InvocationHandler {private final Connection connection;private ConnectionHandler(Connection connection) {this.connection = connection;}public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {boolean isTargetMethod = false;for (String agentm : connection_agent_methods) {if (agentm.equals(method.getName())) {isTargetMethod = true;}}Object result = null;JdbcStatistics jdbcStat = null;try {if (isTargetMethod) { // 獲取PreparedStatement 開(kāi)始統(tǒng)計(jì)jdbcStat = (JdbcStatistics) JdbcCommonCollects.this.begin(null, null);jdbcStat.jdbcUrl = connection.getMetaData().getURL();jdbcStat.sql = (String) args[0];}result = method.invoke(connection, args);// 代理 PreparedStatementif (isTargetMethod && result instanceof PreparedStatement) {PreparedStatement ps = (PreparedStatement) result;result = proxyPreparedStatement(ps, jdbcStat);}} catch (Throwable e) {JdbcCommonCollects.this.error(jdbcStat, e);JdbcCommonCollects.this.end(jdbcStat);throw e;}return result;}}/*** PreparedStatement 代理處理*/public class PreparedStatementHandler implements InvocationHandler {private final PreparedStatement statement;private final JdbcStatistics jdbcStat;public PreparedStatementHandler(PreparedStatement statement, JdbcStatistics jdbcStat) {this.statement = statement;this.jdbcStat = jdbcStat;}public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {System.out.println("++++++++++++++++++++++++++++++++++++++++++++");boolean isTargetMethod = false;for (String agentm : prepared_statement_methods) {if (agentm.equals(method.getName())) {isTargetMethod = true;break;}}Object result = null;try {// SQL的參數(shù) 輸出 TODO 只能輸出一個(gè)，后面的會(huì)把前面的參數(shù)覆蓋掉if (args != null ){List<JdbcStatistics.ParamValues> paramValues = SplitUtils.split2Parts(args, 2);jdbcStat.setParams(paramValues);//System.out.println("KKKKKKK:" + JsonWriter.objectToJson(jdbcStat));}result = method.invoke(statement, args);} catch (Throwable e) {if (isTargetMethod) {JdbcCommonCollects.this.error(jdbcStat, e);}throw e;} finally {if (isTargetMethod) {JdbcCommonCollects.this.end(jdbcStat);}}return result;}}private static String getDbName(String url) {int index = url.indexOf("?"); //$NON-NLS-1$if (index != -1) {String paramString = url.substring(index + 1, url.length());url = url.substring(0, index);}String dbName = url.substring(url.lastIndexOf("/") + 1);return dbName;}}

總結(jié)

以上是生活随笔為你收集整理的APM - 零侵入监控JDBC服务的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。