APM - 零侵入监控JDBC服务
【摘要】
文章目录
Code
Code
public class JdbcCommonCollects extends AbstractByteTransformCollect implemen...
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 {
// connect
beginSrc = "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);
// 如果更換數據庫，需要替換改類 這裡目前僅適配了MySQL
if (!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 开始统计
jdbcStat = (JdbcStatistics) JdbcCommonCollects.this.begin(null, null);
jdbcStat.jdbcUrl = connection.getMetaData().getURL();
jdbcStat.sql = (String) args[0];
}
result = method.invoke(connection, args);
// 代理 PreparedStatement
if (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的參數 輸出 TODO 只能输出一个，后面的会把前面的参数覆盖掉
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;
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
- 133
- 134
- 135
- 136
- 137
- 138
- 139
- 140
- 141
- 142
- 143
- 144
- 145
- 146
- 147
- 148
- 149
- 150
- 151
- 152
- 153
- 154
- 155
- 156
- 157
- 158
- 159
- 160
- 161
- 162
- 163
- 164
- 165
- 166
- 167
- 168
- 169
- 170
- 171
- 172
- 173
- 174
- 175
- 176
- 177
- 178
- 179
- 180
- 181
- 182
- 183
- 184
- 185
- 186
- 187
- 188
- 189
- 190
- 191
- 192
- 193
- 194
文章来源: artisan.blog.csdn.net，作者：小小工匠，版权归原作者所有，如需转载，请联系作者。
原文链接：artisan.blog.csdn.net/article/details/108961689
【版权声明】本文为华为云社区用户转载文章，如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）