APM - 零侵入监控JDBC服务

亦凉 2022-12-12 15:22 159阅读 0赞

### 文章目录 ###

*  Code

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lhbmdzaGFuZ3dlaQ_size_16_color_FFFFFF_t_70_pic_center]

# 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;
        }
    
    	
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lhbmdzaGFuZ3dlaQ_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/4fb03f3aeafc49e9bc1b90085a7bea32.png