support h2 connection pool (7b05bede) · Commits · SUMMER2020 / students / proj-2005254

apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/JdbcXAConnectionConstructorInterceptor.java

0 → 100644

+36 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright 2017, OpenSkywalking Organization All rights reserved.
		*
		* Licensed under the Apache License, Version 2.0 (the "License");
		* you may not use this file except in compliance with the License.
		* You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*
		* Project repository: https://github.com/OpenSkywalking/skywalking
		*/

		package org.skywalking.apm.plugin.jdbc.h2;

		import org.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;
		import org.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceConstructorInterceptor;

		/**
		* {@link JdbcXAConnectionConstructorInterceptor } store {@link org.skywalking.apm.plugin.jdbc.trace.ConnectionInfo}
		* when the client new instance of {@link org.h2.jdbcx.JdbcXAConnection}.
		*
		* @author zhangxin
		*/
		public class JdbcXAConnectionConstructorInterceptor implements InstanceConstructorInterceptor {

		@Override
		public void onConstruct(EnhancedInstance objInst, Object[] allArguments) {
		objInst.setSkyWalkingDynamicField(((EnhancedInstance)allArguments[2]).getSkyWalkingDynamicField());
		}
		}

apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/PooledJdbcConnectionConstructorInterceptor.java

0 → 100644

+35 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright 2017, OpenSkywalking Organization All rights reserved.
		*
		* Licensed under the Apache License, Version 2.0 (the "License");
		* you may not use this file except in compliance with the License.
		* You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*
		* Project repository: https://github.com/OpenSkywalking/skywalking
		*/

		package org.skywalking.apm.plugin.jdbc.h2;

		import org.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;
		import org.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceConstructorInterceptor;

		/**
		* {@link PooledJdbcConnectionConstructorInterceptor } store {@link org.skywalking.apm.plugin.jdbc.trace.ConnectionInfo}
		* when the client new instance of {@link org.h2.jdbcx.JdbcXAConnection$PooledJdbcConnection}.
		*
		* @author zhangxin
		*/
		public class PooledJdbcConnectionConstructorInterceptor implements InstanceConstructorInterceptor {
		@Override
		public void onConstruct(EnhancedInstance objInst, Object[] allArguments) {
		objInst.setSkyWalkingDynamicField(((EnhancedInstance)allArguments[0]).getSkyWalkingDynamicField());
		}
		}

apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/define/ConnectionInstrumentation.java→apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/define/AbstractConnectionInstrumentation.java

+6 −6

Original line number	Diff line number	Diff line
		@@ -41,8 +41,8 @@ import static org.skywalking.apm.plugin.jdbc.define.Constants.ROLLBACK_METHOD_NA
		import static org.skywalking.apm.plugin.jdbc.define.Constants.SERVICE_METHOD_INTERCEPT_CLASS;

		/**
		* {@link ConnectionInstrumentation} intercept the following methods that the class which extend {@link
		* org.h2.jdbc.JdbcConnection}. <br/>
		* {@link AbstractConnectionInstrumentation} define how to enhance the following methods that the class which extend
		* {@link java.sql.Connection}. <br/>
		*
		* 1. Enhance <code>prepareStatement</code> by <code>org.skywalking.apm.plugin.jdbc.define.JDBCPrepareStatementInterceptor</code>
		* 3. Enhance <code>prepareCall</code> by <code>org.skywalking.apm.plugin.jdbc.define.JDBCPrepareCallInterceptor</code>
		@@ -51,9 +51,7 @@ import static org.skywalking.apm.plugin.jdbc.define.Constants.SERVICE_METHOD_INT
		*
		* @author zhangxin
		*/
		public class ConnectionInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

		public static final String ENHANCE_CLASS = "org.h2.jdbc.JdbcConnection";
		public abstract class AbstractConnectionInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

		@Override protected ConstructorInterceptPoint[] getConstructorsInterceptPoints() {
		return new ConstructorInterceptPoint[0];
		@@ -143,6 +141,8 @@ public class ConnectionInstrumentation extends ClassInstanceMethodsEnhancePlugin
		}

		@Override protected ClassMatch enhanceClass() {
		return byName(ENHANCE_CLASS);
		return byName(getEnhanceClass());
		}

		public abstract String getEnhanceClass();
		}

apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/define/JdbcConnectionInstrumentation.java

0 → 100644

+32 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright 2017, OpenSkywalking Organization All rights reserved.
		*
		* Licensed under the Apache License, Version 2.0 (the "License");
		* you may not use this file except in compliance with the License.
		* You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*
		* Project repository: https://github.com/OpenSkywalking/skywalking
		*/

		package org.skywalking.apm.plugin.jdbc.h2.define;

		/**
		* {@link JdbcConnectionInstrumentation} presents that skywalking intercepts {@link org.h2.jdbc.JdbcConnection}.
		*
		* @author zhangxin
		*/
		public class JdbcConnectionInstrumentation extends AbstractConnectionInstrumentation {
		public static final String ENHANCE_CLASS = "org.h2.jdbc.JdbcConnection";

		@Override public String getEnhanceClass() {
		return ENHANCE_CLASS;
		}
		}

apm-sniffer/apm-sdk-plugin/h2-1.x-plugin/src/main/java/org/skywalking/apm/plugin/jdbc/h2/define/JdbcXAConnectionInstrumentation.java

0 → 100644

+54 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright 2017, OpenSkywalking Organization All rights reserved.
		*
		* Licensed under the Apache License, Version 2.0 (the "License");
		* you may not use this file except in compliance with the License.
		* You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*
		* Project repository: https://github.com/OpenSkywalking/skywalking
		*/

		package org.skywalking.apm.plugin.jdbc.h2.define;

		import net.bytebuddy.description.method.MethodDescription;
		import net.bytebuddy.matcher.ElementMatcher;
		import org.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint;

		import static net.bytebuddy.matcher.ElementMatchers.any;

		/**
		* {@link JdbcXAConnectionInstrumentation} presents that skywalking intercepts {@link org.h2.jdbcx.JdbcXAConnection}.
		*
		* @author zhangxin
		*/
		public class JdbcXAConnectionInstrumentation extends AbstractConnectionInstrumentation {

		public static final String ENHANCE_CLASS = "org.h2.jdbcx.JdbcXAConnection";
		public static final String CONSTRUCTOR_INTERCEPT_CLASS = "org.skywalking.apm.plugin.jdbc.h2.JdbcXAConnectionConstructorInterceptor";

		@Override protected ConstructorInterceptPoint[] getConstructorsInterceptPoints() {
		return new ConstructorInterceptPoint[] {
		new ConstructorInterceptPoint() {
		@Override public ElementMatcher<MethodDescription> getConstructorMatcher() {
		return any();
		}

		@Override public String getConstructorInterceptor() {
		return CONSTRUCTOR_INTERCEPT_CLASS;
		}
		}
		};
		}

		@Override public String getEnhanceClass() {
		return ENHANCE_CLASS;
		}
		}