DB2 Profiler Plugin added

plugins/db2/.gitignore

@@ -0,0 +1,5 @@
+/target/
+/.settings/
+/.classpath
+/.project
+/*.iml

plugins/db2/pom.xml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+ xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
+
+ <modelVersion>4.0.0</modelVersion>
+
+ <parent>
+ <groupId>com.navercorp.pinpoint</groupId>
+ <artifactId>pinpoint-plugins</artifactId>
+ <version>3.0.0-SNAPSHOT</version>
+ </parent>
+
+ <artifactId>pinpoint-db2-plugin</artifactId>
+ <name>pinpoint-db2-plugin</name>
+ <packaging>jar</packaging>
+
+ <dependencies>
+ <dependency>
+ <groupId>com.navercorp.pinpoint</groupId>
+ <artifactId>pinpoint-bootstrap-core</artifactId>
+ <scope>provided</scope>
+ </dependency>
+
+ </dependencies>
+</project>

plugins/db2/src/main/java/com/navercorp/pinpoint/plugin/db2/DB2Config.java

@@ -0,0 +1,40 @@
+
+package com.navercorp.pinpoint.plugin.db2;
+
+import com.navercorp.pinpoint.bootstrap.config.ProfilerConfig;
+import com.navercorp.pinpoint.bootstrap.plugin.jdbc.JdbcConfig;
+
+
+public class DB2Config extends JdbcConfig {
+ private final boolean profileSetAutoCommit;
+ private final boolean profileCommit;
+ private final boolean profileRollback;
+
+ public DB2Config(ProfilerConfig config) {
+ super(config.readBoolean("profiler.jdbc.DB2", false),
+ config.readBoolean("profiler.jdbc.DB2.tracesqlbindvalue", config.isTraceSqlBindValue()),
+ config.getMaxSqlBindValueSize());
+ this.profileSetAutoCommit = config.readBoolean("profiler.jdbc.DB2.setautocommit", false);
+ this.profileCommit = config.readBoolean("profiler.jdbc.DB2.commit", false);
+ this.profileRollback = config.readBoolean("profiler.jdbc.DB2.rollback", false);
+ }
+
+ public boolean isProfileSetAutoCommit() {
+ return profileSetAutoCommit;
+ }
+
+ public boolean isProfileCommit() {
+ return profileCommit;
+ }
+
+ public boolean isProfileRollback() {
+ return profileRollback;
+ }
+
+
+ @Override
+ public String toString() {
+ return "DB2Config [" + super.toString() + ", profileSetAutoCommit=" + profileSetAutoCommit + ", profileCommit=" + profileCommit + ", profileRollback=" + profileRollback + "]";
+ }
+
+}

plugins/db2/src/main/java/com/navercorp/pinpoint/plugin/db2/DB2JdbcUrlParser.java

@@ -0,0 +1,110 @@
+
+package com.navercorp.pinpoint.plugin.db2;
+
+import com.navercorp.pinpoint.bootstrap.context.DatabaseInfo;
+import com.navercorp.pinpoint.bootstrap.logging.PLogger;
+import com.navercorp.pinpoint.bootstrap.logging.PLoggerFactory;
+import com.navercorp.pinpoint.bootstrap.plugin.jdbc.DefaultDatabaseInfo;
+import com.navercorp.pinpoint.bootstrap.plugin.jdbc.JdbcUrlParserV2;
+import com.navercorp.pinpoint.bootstrap.plugin.jdbc.StringMaker;
+import com.navercorp.pinpoint.bootstrap.plugin.jdbc.UnKnownDatabaseInfo;
+import com.navercorp.pinpoint.common.trace.ServiceType;
+
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.EnumSet;
+import java.util.List;
+import java.util.Set;
+
+
+public class DB2JdbcUrlParser implements JdbcUrlParserV2 {
+
+
+ static final String DB2_URL_PREFIX = "jdbc:db2:";
+ private static final Set<Type> TYPES = EnumSet.allOf(Type.class);
+
+ private final PLogger logger = PLoggerFactory.getLogger(this.getClass());
+
+ @Override
+ public DatabaseInfo parse(String jdbcUrl) {
+ if (jdbcUrl == null) {
+ logger.info("jdbcUrl");
+ return UnKnownDatabaseInfo.INSTANCE;
+ }
+
+ Type type = getType(jdbcUrl);
+ if (type == null) {
+ logger.info("jdbcUrl has invalid prefix.(url:{}, valid prefixes:{}, {})", jdbcUrl, DB2_URL_PREFIX, DB2_URL_PREFIX);
+ return UnKnownDatabaseInfo.INSTANCE;
+ }
+
+ try {
+ return parse0(jdbcUrl, type);
+ } catch (Exception e) {
+ logger.info("DB2JdbcUrl parse error. url: {}, Caused: {}", jdbcUrl, e.getMessage(), e);
+ return UnKnownDatabaseInfo.createUnknownDataBase(DB2PluginConstants.DB2,DB2PluginConstants.DB2_EXECUTE_QUERY, jdbcUrl);
+ }
+ }
+
+ private DatabaseInfo parse0(String url, Type type) {
+ return parseNormal(url, type);
+ }
+
+ private DatabaseInfo parseNormal(String url, Type type) {
+
+ StringMaker maker = new StringMaker(url);
+ maker.after(type.getUrlPrefix());
+
+ String host = maker.after("//").before('/').value();
+ List<String> hostList = parseHost(host);
+
+ String databaseId = maker.next().after('/').before('?').value();
+ String normalizedUrl = maker.clear().before('?').value();
+ return new DefaultDatabaseInfo(DB2PluginConstants.DB2, DB2PluginConstants.DB2_EXECUTE_QUERY, url,
+ normalizedUrl, hostList, databaseId);
+ }
+
+ private List<String> parseHost(String host) {
+ final int multiHost = host.indexOf(",");
+ if (multiHost == -1) {
+ return Collections.singletonList(host);
+ }
+
+ String[] parsedHost = host.split(",");
+ return Arrays.asList(parsedHost);
+ }
+
+ @Override
+ public ServiceType getServiceType() {
+ return DB2PluginConstants.DB2;
+ }
+
+ private static Type getType(String jdbcUrl) {
+ for (Type type : TYPES) {
+ if (jdbcUrl.startsWith(type.getUrlPrefix())) {
+ return type;
+ }
+ }
+ return null;
+ }
+
+ private enum Type {
+ DB2(DB2_URL_PREFIX);
+
+ private final String urlPrefix;
+ private final String loadbalanceUrlPrefix;
+
+ Type(String urlPrefix) {
+ this.urlPrefix = urlPrefix;
+ this.loadbalanceUrlPrefix = urlPrefix + "loadbalance:";
+ }
+
+ private String getUrlPrefix() {
+ return urlPrefix;
+ }
+
+ private String getLoadbalanceUrlPrefix() {
+ return loadbalanceUrlPrefix;
+ }
+ }
+}