webMagic-Mobai
/
magic-Dependency
mirror of https://gitea.qinmian.online/Mobai/magic-Dependency.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add new constructor to redisscheduler

master
yihua.huang 2013-08-11 18:53:13 +08:00
parent 0f2c5b5723
commit f21097421b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
webmagic-extension/src/main/java/us/codecraft/webmagic/scheduler/RedisScheduler.java
View File

@ -29,6 +29,10 @@ public class RedisScheduler implements Scheduler {
pool = new JedisPool(new JedisPoolConfig(), host); pool = new JedisPool(new JedisPoolConfig(), host);
} }
public RedisScheduler(JedisPool pool) {
this.pool = pool;
}
@Override @Override
public synchronized void push(Request request, Task task) { public synchronized void push(Request request, Task task) {
Jedis jedis = pool.getResource(); Jedis jedis = pool.getResource();
@ -59,7 +63,7 @@ public class RedisScheduler implements Scheduler {
} }
String key = ITEM_PREFIX + task.getUUID(); String key = ITEM_PREFIX + task.getUUID();
String field = DigestUtils.shaHex(url); String field = DigestUtils.shaHex(url);
byte[] bytes = jedis.hget(key.getBytes(),field.getBytes()); byte[] bytes = jedis.hget(key.getBytes(), field.getBytes());
if (bytes != null) { if (bytes != null) {
Request o = JSON.parseObject(new String(bytes), Request.class); Request o = JSON.parseObject(new String(bytes), Request.class);
return o; return o;