|
|
|
@ -4,6 +4,10 @@ import com.xxl.job.core.glue.impl.SpringGlueFactory; |
|
|
|
|
import com.xxl.job.core.handler.IJobHandler; |
|
|
|
|
import groovy.lang.GroovyClassLoader; |
|
|
|
|
|
|
|
|
|
import java.math.BigInteger; |
|
|
|
|
import java.security.MessageDigest; |
|
|
|
|
import java.util.concurrent.ConcurrentHashMap; |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* glue factory, product class/object by name |
|
|
|
|
* |
|
|
|
@ -30,6 +34,8 @@ public class GlueFactory { |
|
|
|
|
*/ |
|
|
|
|
private GroovyClassLoader groovyClassLoader = new GroovyClassLoader(); |
|
|
|
|
|
|
|
|
|
private static final ConcurrentHashMap<String, Class<?>> CLASS_CACHE = new ConcurrentHashMap<>(); |
|
|
|
|
private static final ConcurrentHashMap<Long, String> JOBID_MD5KEY_CACHE = new ConcurrentHashMap<>(); |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* load new instance, prototype |
|
|
|
@ -38,9 +44,9 @@ public class GlueFactory { |
|
|
|
|
* @return |
|
|
|
|
* @throws Exception |
|
|
|
|
*/ |
|
|
|
|
public IJobHandler loadNewInstance(String codeSource) throws Exception{ |
|
|
|
|
public IJobHandler loadNewInstance(long jobId, String codeSource) throws Exception{ |
|
|
|
|
if (codeSource!=null && codeSource.trim().length()>0) { |
|
|
|
|
Class<?> clazz = groovyClassLoader.parseClass(codeSource); |
|
|
|
|
Class<?> clazz = getCodeSourceClass(jobId, codeSource); |
|
|
|
|
if (clazz != null) { |
|
|
|
|
Object instance = clazz.newInstance(); |
|
|
|
|
if (instance!=null) { |
|
|
|
@ -66,4 +72,28 @@ public class GlueFactory { |
|
|
|
|
// do something
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private Class<?> getCodeSourceClass(long jobId, String codeSource){ |
|
|
|
|
try { |
|
|
|
|
MessageDigest md = MessageDigest.getInstance("MD5"); |
|
|
|
|
byte[] md5 = md.digest(codeSource.getBytes()); |
|
|
|
|
BigInteger no = new BigInteger(1, md5); |
|
|
|
|
String md5Str = no.toString(16); |
|
|
|
|
Class<?> clazz = CLASS_CACHE.get(md5Str); |
|
|
|
|
if(clazz == null){ |
|
|
|
|
clazz = groovyClassLoader.parseClass(codeSource); |
|
|
|
|
Class<?> preClazz = CLASS_CACHE.putIfAbsent(md5Str, clazz); |
|
|
|
|
|
|
|
|
|
// 如果代碼有變化則刪除之前class緩存
|
|
|
|
|
if(preClazz == null){ |
|
|
|
|
String preMd5 = JOBID_MD5KEY_CACHE.put(jobId, md5Str); |
|
|
|
|
if(preMd5 != null){ |
|
|
|
|
CLASS_CACHE.remove(preMd5); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return clazz; |
|
|
|
|
} catch (Exception e) { |
|
|
|
|
return groovyClassLoader.parseClass(codeSource); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|