ExtClasspathLoader

最新推荐文章于 2021-03-05 10:45:11 发布

deng_xintao

最新推荐文章于 2021-03-05 10:45:11 发布

阅读量105

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/deng_xintao/article/details/102795252

版权

java 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

/*
* Copyright (c) 2018,
*/
package com.rpc.jmeter.utils;

import java.io.File;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;

import com.alibaba.dubbo.common.json.JSON;
import com.alibaba.dubbo.config.ApplicationConfig;
import com.alibaba.dubbo.config.ReferenceConfig;
import com.alibaba.dubbo.config.RegistryConfig;

public final class ExtClasspathLoader {

private static Method addURL = initAddMethod();

private static ClassLoader systemClassLoader = ClassLoader.getSystemClassLoader();
private static ClassLoader webappClassLoader = ExtClasspathLoader.class.getClassLoader();

   /**
   * 初始化addUrl 方法.
   *
   * @return 可访问addUrl方法的Method对象
   */
   private static Method initAddMethod() {
       try {
           Method add = URLClassLoader.class.getDeclaredMethod("addURL", new Class[] { URL.class });
           add.setAccessible(true);
           return add;
       } catch (Exception e) {
           throw new RuntimeException(e);
       }
   }

   public static void loadClasspath(String filepath) {
       File file = new File(filepath);
       loopFiles(file);
   }

   /**
   * 循环遍历目录，找出所有的jar包。
   *
   * @param file
   * 当前遍历文件
   */
   private static void loopFiles(File file) {
       if (file.isDirectory()) {
           File[] tmps = file.listFiles();
           for (File tmp : tmps) {
               loopFiles(tmp);
           }
       } else {
           if (file.getAbsolutePath().endsWith(".jar") || file.getAbsolutePath().endsWith(".zip")) {
               addURL(file, false);
           }
       }
   }

   /**
   * 通过filepath加载文件到classpath。
   *
   * @param filePath
   * 文件路径
   * @return URL
   * @throws Exception
   * 异常
   */
   public static void addURL(File file, boolean system) {
       try {
           if (system) {
               addURL.invoke(systemClassLoader, new Object[] { file.toURI().toURL() });
           } else {
               addURL.invoke(webappClassLoader, new Object[] { file.toURI().toURL() });
           }
       } catch (Exception e) {
           e.printStackTrace();
       }
   }

   public static void main(String[] args) {
       String path = "D:\\*-resource-client-1.0.6-SNAPSHOT.jar";
       ExtClasspathLoader.loadClasspath(path);

       ApplicationConfig application = new ApplicationConfig();
       application.setName("username");
       RegistryConfig registry = new RegistryConfig();
       registry.setAddress("zookeeper://10.202.35.91:2181?backup=10.202.35.92:2181,10.202.35.96:2181");

       ReferenceConfig<?> reference = new ReferenceConfig<>();
       reference.setApplication(application);
       reference.setRegistry(registry);
       reference.setInterface("com.*.*.match.client.rpc.IMatchEmpService");
       reference.setGroup("ENV3_1");
       try {
           Object bean = reference.get();
           Class<?> paramClass = Class.forName("com.*.*.match.client.dto.MatchEmpReq");
           Object param = paramClass.newInstance();
           param = JSON.parse(
                   "{\"cityCode\":\"755\",\"intertype\":\"1\",\"orderTime\":null,\"orderType\":\"D\",\"parcelWeight\":10,\"unitAreaCode\":\"755Q001\"}",
                   paramClass);
           Method targetMethod = bean.getClass().getMethod("getEmpIdForDdsDelivery", paramClass);
           Object rpcResult = targetMethod.invoke(bean, param);
           //System.out.println("结果 = " + JSONUtils.toObject(rpcResult).toString());
       } catch (Exception e) {
           e.printStackTrace();
       }

       // File file = new File(path);
       // Map<String, List<String>> result = MyFileUtils.parseJar(file);
       // System.out.println("解析结果: ");
       // Iterator<Entry<String, List<String>>> iterator =
       // result.entrySet().iterator();
       // while (iterator.hasNext()) {
       // Entry<String, List<String>> entry = iterator.next();
       // System.out.print("接口名: " + entry.getKey() + "\t");
       // System.out.println("拥有方法: " + entry.getValue());
       // }
   }
}