From f21097421b67836100ff28c8c1ae163e3ec43917 Mon Sep 17 00:00:00 2001 From: "yihua.huang" Date: Sun, 11 Aug 2013 18:53:13 +0800 Subject: [PATCH] add new constructor to redisscheduler --- .../us/codecraft/webmagic/scheduler/RedisScheduler.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/webmagic-extension/src/main/java/us/codecraft/webmagic/scheduler/RedisScheduler.java b/webmagic-extension/src/main/java/us/codecraft/webmagic/scheduler/RedisScheduler.java index d6d6349..c377a12 100644 --- a/webmagic-extension/src/main/java/us/codecraft/webmagic/scheduler/RedisScheduler.java +++ b/webmagic-extension/src/main/java/us/codecraft/webmagic/scheduler/RedisScheduler.java @@ -29,6 +29,10 @@ public class RedisScheduler implements Scheduler { pool = new JedisPool(new JedisPoolConfig(), host); } + public RedisScheduler(JedisPool pool) { + this.pool = pool; + } + @Override public synchronized void push(Request request, Task task) { Jedis jedis = pool.getResource(); @@ -59,7 +63,7 @@ public class RedisScheduler implements Scheduler { } String key = ITEM_PREFIX + task.getUUID(); String field = DigestUtils.shaHex(url); - byte[] bytes = jedis.hget(key.getBytes(),field.getBytes()); + byte[] bytes = jedis.hget(key.getBytes(), field.getBytes()); if (bytes != null) { Request o = JSON.parseObject(new String(bytes), Request.class); return o;