Share project "druid" into "http://code.alibabatech.com/svn/druid"

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@2 b9813039-fb51-4c41-a8b9-e21c2acb5095

src/main/java/com/alibaba/druid/util/DruidLoaderUtils.java

@@ -0,0 +1,148 @@
+/**
+ * Copyright 1999-2011 Alibaba.com All right reserved. This software is the confidential and proprietary information of
+ * Alibaba.com ("Confidential Information"). You shall not disclose such Confidential Information and shall use it only
+ * in accordance with the terms of the license agreement you entered into with Alibaba.com.
+ */
+package com.alibaba.druid.util;
+
+import java.sql.SQLException;
+import java.util.Iterator;
+import java.util.List;
+
+import com.alibaba.druid.filter.Filter;
+import com.alibaba.druid.filter.FilterManager;
+import com.alibaba.druid.proxy.config.AbstractDruidFilterConfig;
+
+/**
+ * druid loader utils
+ * 
+ * @author gang.su
+ */
+public class DruidLoaderUtils {
+
+    public static void loadFilter(List<Filter> filterList, List<AbstractDruidFilterConfig> druidFilterConfigList)
+                                                                                                                 throws SQLException {
+        for (Iterator<AbstractDruidFilterConfig> iterator = druidFilterConfigList.iterator(); iterator.hasNext();) {
+            AbstractDruidFilterConfig druidFilterConfig = iterator.next();
+            String filterItem = druidFilterConfig.getClazz();
+
+            Class<?> filterClass = loadClass(druidFilterConfig.getClazz());
+
+            if (filterClass == null) {
+                return;
+            }
+
+            Filter jdbcFilter;
+            try {
+                jdbcFilter = (Filter) filterClass.newInstance();
+                jdbcFilter.loadConfig(druidFilterConfig);
+            } catch (InstantiationException e) {
+                throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+            } catch (IllegalAccessException e) {
+                throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+            }
+            filterList.add(jdbcFilter);
+        }
+
+    }
+
+    public static void loadFilter(List<Filter> filterConfigList, String filterItem) throws SQLException {
+        if (filterItem.length() == 0) {
+            return;
+        }
+
+        String filterClassNames = FilterManager.getFilter(filterItem);
+        if (filterClassNames != null) {
+            filterClassNames = filterClassNames.trim();
+        }
+
+        if (filterClassNames != null) {
+            for (String filterClassName : filterClassNames.split(",")) {
+
+                if (!isExist(filterConfigList, filterClassName)) {
+                    Class<?> filterClass = loadClass(filterClassName);
+
+                    if (filterClass != null) {
+                        Filter filter;
+
+                        try {
+                            filter = (Filter) filterClass.newInstance();
+                        } catch (InstantiationException e) {
+                            throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+                        } catch (IllegalAccessException e) {
+                            throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+                        }
+
+                        filterConfigList.add(filter);
+                    }
+                }
+            }
+        } else {
+            if (!isExist(filterConfigList, filterItem)) {
+                Class<?> filterClass = loadClass(filterItem);
+
+                if (filterClass != null) {
+                    try {
+                        Filter filter = (Filter) filterClass.newInstance();
+                        filterConfigList.add(filter);
+                    } catch (InstantiationException e) {
+                        throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+                    } catch (IllegalAccessException e) {
+                        throw new SQLException("load managed jdbc driver event listener error. " + filterItem, e);
+                    }
+                }
+            }
+        }
+    }
+
+    /**
+     * 是否存在load
+     * @param filterConfigList
+     * @param filterClassName
+     * @return
+     */
+    private static boolean isExist(List<Filter> filterConfigList, String filterClassName) {
+        for (Iterator<Filter> iterator = filterConfigList.iterator(); iterator.hasNext();) {
+            Filter filter = (Filter) iterator.next();
+            if (filter.getClass().getName().equalsIgnoreCase(filterClassName)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public static Class<?> loadClass(String className) {
+        Class<?> clazz = null;
+
+        if (className == null) {
+            return null;
+        }
+
+        try {
+            clazz = Thread.currentThread().getContextClassLoader().loadClass(className);
+        } catch (ClassNotFoundException e) {
+            e.printStackTrace();
+        }
+
+        if (clazz == null) {
+            loadClass2(className);
+        }
+
+        return clazz;
+    }
+
+    public static Class<?> loadClass2(String className) {
+        if (className == null) {
+            return null;
+        }
+
+        try {
+            return Class.forName(className);
+        } catch (ClassNotFoundException e) {
+            e.printStackTrace();
+        }
+
+        return null;
+    }
+
+}