You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
871 lines
38 KiB
871 lines
38 KiB
// plasmaCrawlStacker.java
|
|
// -----------------------
|
|
// part of YaCy
|
|
// (C) by Michael Peter Christen; mc@anomic.de
|
|
// first published on http://www.anomic.de
|
|
// Frankfurt, Germany, 2005
|
|
//
|
|
// This file was contributed by Martin Thelian
|
|
//
|
|
// $LastChangedDate$
|
|
// $LastChangedRevision$
|
|
// $LastChangedBy$
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
//
|
|
// Using this software in any meaning (reading, learning, copying, compiling,
|
|
// running) means that you agree that the Author(s) is (are) not responsible
|
|
// for cost, loss of data or any harm that may be caused directly or indirectly
|
|
// by usage of this softare or this documentation. The usage of this software
|
|
// is on your own risk. The installation and usage (starting/running) of this
|
|
// software may allow other people or application to access your computer and
|
|
// any attached devices and is highly dependent on the configuration of the
|
|
// software which must be done by the user of the software; the author(s) is
|
|
// (are) also not responsible for proper configuration and usage of the
|
|
// software, even if provoked by documentation provided together with
|
|
// the software.
|
|
//
|
|
// Any changes to this file according to the GPL as documented in the file
|
|
// gpl.txt aside this file in the shipment you received can be done to the
|
|
// lines that follows this copyright notice here, but changes must not be
|
|
// done inside the copyright notive above. A re-distribution must contain
|
|
// the intact and unchanged copyright notice.
|
|
// Contributions and changes to the program code must be marked as such.
|
|
|
|
package de.anomic.plasma;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.net.InetAddress;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.util.Date;
|
|
import java.util.Iterator;
|
|
import java.util.LinkedList;
|
|
|
|
import org.apache.commons.pool.impl.GenericObjectPool;
|
|
|
|
import de.anomic.data.robotsParser;
|
|
import de.anomic.http.httpc;
|
|
import de.anomic.kelondro.kelondroException;
|
|
import de.anomic.kelondro.kelondroTree;
|
|
import de.anomic.kelondro.kelondroRecords.Node;
|
|
import de.anomic.server.serverCodings;
|
|
import de.anomic.server.serverSemaphore;
|
|
import de.anomic.server.logging.serverLog;
|
|
import de.anomic.tools.bitfield;
|
|
import de.anomic.yacy.yacyCore;
|
|
|
|
public final class plasmaCrawlStacker {
|
|
|
|
final WorkerPool theWorkerPool;
|
|
private GenericObjectPool.Config theWorkerPoolConfig = null;
|
|
final ThreadGroup theWorkerThreadGroup = new ThreadGroup("stackCrawlThreadGroup");
|
|
final serverLog log = new serverLog("STACKCRAWL");
|
|
final plasmaSwitchboard sb;
|
|
//private boolean stopped = false;
|
|
private stackCrawlQueue queue;
|
|
|
|
public plasmaCrawlStacker(plasmaSwitchboard sb, File dbPath, int dbCacheSize) {
|
|
this.sb = sb;
|
|
|
|
this.queue = new stackCrawlQueue(dbPath,dbCacheSize);
|
|
this.log.logInfo(this.queue.size() + " entries in the stackCrawl queue.");
|
|
this.log.logInfo("STACKCRAWL thread initialized.");
|
|
|
|
// configuring the thread pool
|
|
// implementation of session thread pool
|
|
this.theWorkerPoolConfig = new GenericObjectPool.Config();
|
|
|
|
// The maximum number of active connections that can be allocated from pool at the same time,
|
|
// 0 for no limit
|
|
this.theWorkerPoolConfig.maxActive = Integer.parseInt(sb.getConfig("stacker.MaxActiveThreads","50"));
|
|
|
|
// The maximum number of idle connections connections in the pool
|
|
// 0 = no limit.
|
|
this.theWorkerPoolConfig.maxIdle = Integer.parseInt(sb.getConfig("stacker.MaxIdleThreads","10"));
|
|
this.theWorkerPoolConfig.minIdle = Integer.parseInt(sb.getConfig("stacker.MinIdleThreads","5"));
|
|
|
|
// block undefinitely
|
|
this.theWorkerPoolConfig.maxWait = -1;
|
|
|
|
// Action to take in case of an exhausted DBCP statement pool
|
|
// 0 = fail, 1 = block, 2= grow
|
|
this.theWorkerPoolConfig.whenExhaustedAction = GenericObjectPool.WHEN_EXHAUSTED_BLOCK;
|
|
this.theWorkerPoolConfig.minEvictableIdleTimeMillis = 30000;
|
|
|
|
// creating worker pool
|
|
this.theWorkerPool = new WorkerPool(new WorkterFactory(this.theWorkerThreadGroup),this.theWorkerPoolConfig);
|
|
|
|
}
|
|
|
|
public GenericObjectPool.Config getPoolConfig() {
|
|
return this.theWorkerPoolConfig;
|
|
}
|
|
|
|
public void setPoolConfig(GenericObjectPool.Config newConfig) {
|
|
this.theWorkerPool.setConfig(newConfig);
|
|
}
|
|
|
|
public void close() {
|
|
try {
|
|
this.log.logFine("Shutdown. Terminating worker threads.");
|
|
if (this.theWorkerPool != null) this.theWorkerPool.close();
|
|
} catch (Exception e1) {
|
|
this.log.logSevere("Unable to shutdown all remaining stackCrawl threads", e1);
|
|
}
|
|
|
|
try {
|
|
this.log.logFine("Shutdown. Closing stackCrawl queue.");
|
|
if (this.queue != null) this.queue.close();
|
|
this.queue = null;
|
|
} catch (IOException e) {
|
|
this.log.logSevere("DB could not be closed properly.", e);
|
|
}
|
|
}
|
|
|
|
public int size() {
|
|
return this.queue.size();
|
|
}
|
|
|
|
public void job() {
|
|
try {
|
|
// getting a new message from the crawler queue
|
|
stackCrawlMessage theMsg = this.queue.waitForMessage();
|
|
|
|
// getting a free session thread from the pool
|
|
Worker worker = (Worker) this.theWorkerPool.borrowObject();
|
|
|
|
// processing the new request
|
|
worker.execute(theMsg);
|
|
} catch (InterruptedException e) {
|
|
Thread.interrupted();
|
|
//this.stopped = true;
|
|
}
|
|
catch (Exception e) {
|
|
this.log.logSevere("plasmaStackCrawlThread.run/loop", e);
|
|
}
|
|
}
|
|
|
|
public void enqueue(
|
|
String nexturlString,
|
|
String referrerString,
|
|
String initiatorHash,
|
|
String name,
|
|
Date loadDate,
|
|
int currentdepth,
|
|
plasmaCrawlProfile.entry profile) {
|
|
try {
|
|
this.queue.addMessage(new stackCrawlMessage(
|
|
initiatorHash,
|
|
nexturlString,
|
|
referrerString,
|
|
name,
|
|
loadDate,
|
|
profile.handle(),
|
|
currentdepth,
|
|
0,
|
|
0
|
|
));
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public String dequeue(stackCrawlMessage theMsg) {
|
|
|
|
plasmaCrawlProfile.entry profile = this.sb.profiles.getEntry(theMsg.profileHandle());
|
|
if (profile == null) {
|
|
String errorMsg = "LOST PROFILE HANDLE '" + theMsg.profileHandle() + "' (must be internal error) for URL " + theMsg.url();
|
|
this.log.logSevere(errorMsg);
|
|
throw new IllegalStateException(errorMsg);
|
|
}
|
|
|
|
return stackCrawl(
|
|
theMsg.url().toString(),
|
|
theMsg.referrerHash(),
|
|
theMsg.initiatorHash(),
|
|
theMsg.name(),
|
|
theMsg.loaddate(),
|
|
theMsg.depth(),
|
|
profile);
|
|
}
|
|
|
|
public String stackCrawl(String nexturlString, String referrerString, String initiatorHash, String name, Date loadDate, int currentdepth, plasmaCrawlProfile.entry profile) {
|
|
// stacks a crawl item. The position can also be remote
|
|
// returns null if successful, a reason string if not successful
|
|
this.log.logFinest("stackCrawl: nexturlString='" + nexturlString + "'");
|
|
|
|
long startTime = System.currentTimeMillis();
|
|
String reason = null; // failure reason
|
|
|
|
// strange errors
|
|
if (nexturlString == null) {
|
|
reason = "denied_(url_null)";
|
|
this.log.logSevere("Wrong URL in stackCrawl: url=null");
|
|
return reason;
|
|
}
|
|
/*
|
|
if (profile == null) {
|
|
reason = "denied_(profile_null)";
|
|
log.logError("Wrong Profile for stackCrawl: profile=null");
|
|
return reason;
|
|
}
|
|
*/
|
|
URL nexturl = null;
|
|
if ((initiatorHash == null) || (initiatorHash.length() == 0)) initiatorHash = plasmaURL.dummyHash;
|
|
String referrerHash = (referrerString==null)?null:plasmaURL.urlHash(referrerString);
|
|
try {
|
|
nexturl = new URL(nexturlString);
|
|
} catch (MalformedURLException e) {
|
|
reason = "denied_(url_'" + nexturlString + "'_wrong)";
|
|
this.log.logSevere("Wrong URL in stackCrawl: " + nexturlString +
|
|
". Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// check if ip is local ip address
|
|
InetAddress hostAddress = httpc.dnsResolve(nexturl.getHost());
|
|
if (hostAddress == null) {
|
|
reason = "denied_(unknown_host)";
|
|
this.log.logFine("Unknown host in URL '" + nexturlString + "'." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
} else if (hostAddress.isSiteLocalAddress()) {
|
|
reason = "denied_(private_ip_address)";
|
|
this.log.logFine("Host in URL '" + nexturlString + "' has private ip address." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
} else if (hostAddress.isLoopbackAddress()) {
|
|
reason = "denied_(loopback_ip_address)";
|
|
this.log.logFine("Host in URL '" + nexturlString + "' has loopback ip address." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// check blacklist
|
|
String hostlow = nexturl.getHost().toLowerCase();
|
|
if (plasmaSwitchboard.urlBlacklist.isListed(hostlow, nexturl.getPath())) {
|
|
reason = "denied_(url_in_blacklist)";
|
|
this.log.logFine("URL '" + nexturlString + "' is in blacklist." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// filter deny
|
|
if ((currentdepth > 0) && (profile != null) && (!(nexturlString.matches(profile.generalFilter())))) {
|
|
reason = "denied_(does_not_match_filter)";
|
|
/*
|
|
urlPool.errorURL.newEntry(nexturl, referrerHash, initiatorHash, yacyCore.seedDB.mySeed.hash,
|
|
name, reason, new bitfield(plasmaURL.urlFlagLength), false);*/
|
|
this.log.logFine("URL '" + nexturlString + "' does not match crawling filter '" + profile.generalFilter() + "'." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// deny cgi
|
|
if (plasmaHTCache.isCGI(nexturlString)) {
|
|
reason = "denied_(cgi_url)";
|
|
/*
|
|
urlPool.errorURL.newEntry(nexturl, referrerHash, initiatorHash, yacyCore.seedDB.mySeed.hash,
|
|
name, reason, new bitfield(plasmaURL.urlFlagLength), false);*/
|
|
this.log.logFine("URL '" + nexturlString + "' is cgi URL." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// deny post properties
|
|
if ((plasmaHTCache.isPOST(nexturlString)) && (profile != null) && (!(profile.crawlingQ()))) {
|
|
reason = "denied_(post_url)";
|
|
/*
|
|
urlPool.errorURL.newEntry(nexturl, referrerHash, initiatorHash, yacyCore.seedDB.mySeed.hash,
|
|
name, reason, new bitfield(plasmaURL.urlFlagLength), false);*/
|
|
this.log.logFine("URL '" + nexturlString + "' is post URL. " +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
String nexturlhash = plasmaURL.urlHash(nexturl);
|
|
String dbocc = "";
|
|
if ((dbocc = this.sb.urlPool.exists(nexturlhash)) != null) {
|
|
// DISTIGUISH OLD/RE-SEARCH CASES HERE!
|
|
reason = "double_(registered_in_" + dbocc + ")";
|
|
/*
|
|
urlPool.errorURL.newEntry(nexturl, referrerHash, initiatorHash, yacyCore.seedDB.mySeed.hash,
|
|
name, reason, new bitfield(plasmaURL.urlFlagLength), false);*/
|
|
this.log.logFine("URL '" + nexturlString + "' is double registered in '" + dbocc + "'." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// checking robots.txt
|
|
if (robotsParser.isDisallowed(nexturl)) {
|
|
reason = "denied_(robots.txt)";
|
|
/*
|
|
urlPool.errorURL.newEntry(nexturl, referrerHash, initiatorHash, yacyCore.seedDB.mySeed.hash,
|
|
name, reason, new bitfield(plasmaURL.urlFlagLength), false);*/
|
|
this.log.logFine("Crawling of URL '" + nexturlString + "' disallowed by robots.txt." +
|
|
"Stack processing time: " + (System.currentTimeMillis()-startTime));
|
|
return reason;
|
|
}
|
|
|
|
// store information
|
|
boolean local = ((initiatorHash.equals(plasmaURL.dummyHash)) || (initiatorHash.equals(yacyCore.seedDB.mySeed.hash)));
|
|
boolean global =
|
|
(profile != null) &&
|
|
(profile.remoteIndexing()) /* granted */ &&
|
|
(currentdepth == profile.generalDepth()) /* leaf node */ &&
|
|
(initiatorHash.equals(yacyCore.seedDB.mySeed.hash)) /* not proxy */ &&
|
|
((yacyCore.seedDB.mySeed.isSenior()) ||
|
|
(yacyCore.seedDB.mySeed.isPrincipal())) /* qualified */;
|
|
|
|
if ((!local)&&(!global)) {
|
|
this.log.logSevere("URL '" + nexturlString + "' can neither be crawled local nor global.");
|
|
}
|
|
|
|
this.sb.urlPool.noticeURL.newEntry(initiatorHash, /* initiator, needed for p2p-feedback */
|
|
nexturl, /* url clear text string */
|
|
loadDate, /* load date */
|
|
referrerHash, /* last url in crawling queue */
|
|
name, /* the anchor name */
|
|
(profile == null) ? null : profile.handle(), // profile must not be null!
|
|
currentdepth, /*depth so far*/
|
|
0, /*anchors, default value */
|
|
0, /*forkfactor, default value */
|
|
((global) ? plasmaCrawlNURL.STACK_TYPE_LIMIT :
|
|
((local) ? plasmaCrawlNURL.STACK_TYPE_CORE : plasmaCrawlNURL.STACK_TYPE_REMOTE)) /*local/remote stack*/
|
|
);
|
|
|
|
return null;
|
|
}
|
|
|
|
public final class stackCrawlMessage {
|
|
private String initiator; // the initiator hash, is NULL or "" if it is the own proxy;
|
|
String urlHash; // the url's hash
|
|
private String referrerHash; // the url's referrer hash
|
|
private String url; // the url as string
|
|
String name; // the name of the url, from anchor tag <a>name</a>
|
|
private Date loaddate; // the time when the url was first time appeared
|
|
private String profileHandle; // the name of the prefetch profile
|
|
private int depth; // the prefetch depth so far, starts at 0
|
|
private int anchors; // number of anchors of the parent
|
|
private int forkfactor; // sum of anchors of all ancestors
|
|
private bitfield flags;
|
|
private int handle;
|
|
|
|
// loadParallel(URL url, String referer, String initiator, int depth, plasmaCrawlProfile.entry profile) {
|
|
public stackCrawlMessage(
|
|
String initiator,
|
|
String urlString,
|
|
String referrerUrlString,
|
|
String name,
|
|
Date loaddate,
|
|
String profileHandle,
|
|
int depth,
|
|
int anchors,
|
|
int forkfactor) {
|
|
try {
|
|
// create new entry and store it into database
|
|
this.urlHash = plasmaURL.urlHash(urlString);
|
|
this.initiator = initiator;
|
|
this.url = urlString;
|
|
this.referrerHash = (referrerUrlString == null) ? plasmaURL.dummyHash : plasmaURL.urlHash(referrerUrlString);
|
|
this.name = (name == null) ? "" : name;
|
|
this.loaddate = (loaddate == null) ? new Date() : loaddate;
|
|
this.profileHandle = profileHandle; // must not be null
|
|
this.depth = depth;
|
|
this.anchors = anchors;
|
|
this.forkfactor = forkfactor;
|
|
this.flags = new bitfield(plasmaURL.urlFlagLength);
|
|
this.handle = 0;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public stackCrawlMessage(String urlHash, byte[][] entryBytes) {
|
|
if (urlHash == null) throw new NullPointerException();
|
|
if (entryBytes == null) throw new NullPointerException();
|
|
|
|
try {
|
|
this.urlHash = urlHash;
|
|
this.initiator = new String(entryBytes[1]);
|
|
this.url = new String(entryBytes[2]).trim();
|
|
this.referrerHash = (entryBytes[3]==null) ? plasmaURL.dummyHash : new String(entryBytes[3]);
|
|
this.name = (entryBytes[4] == null) ? "" : new String(entryBytes[4]).trim();
|
|
this.loaddate = new Date(86400000 * serverCodings.enhancedCoder.decodeBase64Long(new String(entryBytes[5])));
|
|
this.profileHandle = (entryBytes[6] == null) ? null : new String(entryBytes[6]).trim();
|
|
this.depth = (int) serverCodings.enhancedCoder.decodeBase64Long(new String(entryBytes[7]));
|
|
this.anchors = (int) serverCodings.enhancedCoder.decodeBase64Long(new String(entryBytes[8]));
|
|
this.forkfactor = (int) serverCodings.enhancedCoder.decodeBase64Long(new String(entryBytes[9]));
|
|
this.flags = new bitfield(entryBytes[10]);
|
|
this.handle = Integer.parseInt(new String(entryBytes[11]));
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
throw new IllegalStateException();
|
|
}
|
|
}
|
|
|
|
public String url() {
|
|
return this.url;
|
|
}
|
|
|
|
public String referrerHash() {
|
|
return this.referrerHash;
|
|
}
|
|
|
|
public String initiatorHash() {
|
|
if (this.initiator == null) return null;
|
|
if (this.initiator.length() == 0) return null;
|
|
return this.initiator;
|
|
}
|
|
|
|
public Date loaddate() {
|
|
return this.loaddate;
|
|
}
|
|
|
|
public String name() {
|
|
return this.name;
|
|
}
|
|
|
|
public int depth() {
|
|
return this.depth;
|
|
}
|
|
|
|
public String profileHandle() {
|
|
return this.profileHandle;
|
|
}
|
|
|
|
public String toString() {
|
|
StringBuffer str = new StringBuffer();
|
|
str.append("urlHash: ").append(urlHash==null ? "null" : urlHash).append(" | ")
|
|
.append("initiator: ").append(initiator==null?"null":initiator).append(" | ")
|
|
.append("url: ").append(url==null?"null":url).append(" | ")
|
|
.append("referrer: ").append((referrerHash == null) ? plasmaURL.dummyHash : referrerHash).append(" | ")
|
|
.append("name: ").append((name == null) ? "null" : name).append(" | ")
|
|
.append("loaddate: ").append((loaddate == null) ? new Date() : loaddate).append(" | ")
|
|
.append("profile: ").append(profileHandle==null?"null":profileHandle).append(" | ")
|
|
.append("depth: ").append(Integer.toString(depth)).append(" | ")
|
|
.append("forkfactor: ").append(Integer.toString(forkfactor)).append(" | ")
|
|
.append("flags: ").append((flags==null) ? "null" : flags.toString());
|
|
return str.toString();
|
|
}
|
|
|
|
public byte[][] getBytes() {
|
|
// stores the values from the object variables into the database
|
|
String loaddatestr = serverCodings.enhancedCoder.encodeBase64Long(loaddate.getTime() / 86400000, plasmaURL.urlDateLength);
|
|
// store the hash in the hash cache
|
|
|
|
// even if the entry exists, we simply overwrite it
|
|
byte[][] entry = new byte[][] {
|
|
this.urlHash.getBytes(),
|
|
(this.initiator == null) ? "".getBytes() : this.initiator.getBytes(),
|
|
this.url.getBytes(),
|
|
this.referrerHash.getBytes(),
|
|
this.name.getBytes(),
|
|
loaddatestr.getBytes(),
|
|
(this.profileHandle == null) ? null : this.profileHandle.getBytes(),
|
|
serverCodings.enhancedCoder.encodeBase64Long(this.depth, plasmaURL.urlCrawlDepthLength).getBytes(),
|
|
serverCodings.enhancedCoder.encodeBase64Long(this.anchors, plasmaURL.urlParentBranchesLength).getBytes(),
|
|
serverCodings.enhancedCoder.encodeBase64Long(this.forkfactor, plasmaURL.urlForkFactorLength).getBytes(),
|
|
this.flags.getBytes(),
|
|
normalizeHandle(this.handle).getBytes()
|
|
};
|
|
return entry;
|
|
}
|
|
|
|
private String normalizeHandle(int h) {
|
|
String d = Integer.toHexString(h);
|
|
while (d.length() < plasmaURL.urlHandleLength) d = "0" + d;
|
|
return d;
|
|
}
|
|
}
|
|
|
|
final class stackCrawlQueue {
|
|
|
|
private final serverSemaphore readSync;
|
|
private final serverSemaphore writeSync;
|
|
private final LinkedList urlEntryHashCache;
|
|
private kelondroTree urlEntryCache;
|
|
|
|
public stackCrawlQueue(File cacheStacksPath, int bufferkb) {
|
|
// init the read semaphore
|
|
this.readSync = new serverSemaphore (0);
|
|
|
|
// init the write semaphore
|
|
this.writeSync = new serverSemaphore (1);
|
|
|
|
// init the message list
|
|
this.urlEntryHashCache = new LinkedList();
|
|
|
|
// create a stack for newly entered entries
|
|
if (!(cacheStacksPath.exists())) cacheStacksPath.mkdir(); // make the path
|
|
|
|
File cacheFile = new File(cacheStacksPath, "urlPreNotice.db");
|
|
if (cacheFile.exists()) {
|
|
// open existing cache
|
|
try {
|
|
this.urlEntryCache = new kelondroTree(cacheFile, bufferkb * 0x400);
|
|
} catch (IOException e) {
|
|
cacheFile.delete();
|
|
this.urlEntryCache = new kelondroTree(cacheFile, bufferkb * 0x400, plasmaCrawlNURL.ce, true);
|
|
}
|
|
try {
|
|
// loop through the list and fill the messageList with url hashs
|
|
Iterator iter = this.urlEntryCache.nodeIterator(true,false);
|
|
Node n;
|
|
while (iter.hasNext()) {
|
|
n = (Node) iter.next();
|
|
if (n == null) {
|
|
System.out.println("ERROR! null element found");
|
|
continue;
|
|
}
|
|
String urlHash = new String(n.getKey());
|
|
this.urlEntryHashCache.add(urlHash);
|
|
this.readSync.V();
|
|
}
|
|
} catch (kelondroException e) {
|
|
/* if we have an error, we start with a fresh database */
|
|
plasmaCrawlStacker.this.log.logSevere("Unable to initialize crawl stacker queue. Reseting DB.\n",e);
|
|
|
|
// deleting old db and creating a new db
|
|
try {this.urlEntryCache.close();}catch(Exception ex){}
|
|
cacheFile.delete();
|
|
this.urlEntryCache = new kelondroTree(cacheFile, bufferkb * 0x400, plasmaCrawlNURL.ce, true);
|
|
}
|
|
} else {
|
|
// create new cache
|
|
cacheFile.getParentFile().mkdirs();
|
|
this.urlEntryCache = new kelondroTree(cacheFile, bufferkb * 0x400, plasmaCrawlNURL.ce, true);
|
|
}
|
|
}
|
|
|
|
public void close() throws IOException {
|
|
// closing the db
|
|
this.urlEntryCache.close();
|
|
|
|
// clearing the hash list
|
|
this.urlEntryHashCache.clear();
|
|
}
|
|
|
|
public void addMessage(stackCrawlMessage newMessage)
|
|
throws InterruptedException, IOException {
|
|
if (newMessage == null) throw new NullPointerException();
|
|
|
|
this.writeSync.P();
|
|
try {
|
|
|
|
boolean insertionDoneSuccessfully = false;
|
|
synchronized(this.urlEntryHashCache) {
|
|
byte[][] oldValue = this.urlEntryCache.put(newMessage.getBytes());
|
|
if (oldValue == null) {
|
|
insertionDoneSuccessfully = this.urlEntryHashCache.add(newMessage.urlHash);
|
|
}
|
|
}
|
|
|
|
if (insertionDoneSuccessfully) {
|
|
this.readSync.V();
|
|
}
|
|
} finally {
|
|
this.writeSync.V();
|
|
}
|
|
}
|
|
|
|
public int size() {
|
|
synchronized(this.urlEntryHashCache) {
|
|
return this.urlEntryHashCache.size();
|
|
}
|
|
}
|
|
|
|
public stackCrawlMessage waitForMessage() throws InterruptedException, IOException {
|
|
this.readSync.P();
|
|
this.writeSync.P();
|
|
|
|
String urlHash = null;
|
|
byte[][] entryBytes = null;
|
|
stackCrawlMessage newMessage = null;
|
|
synchronized(this.urlEntryHashCache) {
|
|
urlHash = (String) this.urlEntryHashCache.removeFirst();
|
|
entryBytes = this.urlEntryCache.remove(urlHash.getBytes());
|
|
}
|
|
|
|
this.writeSync.V();
|
|
|
|
newMessage = new stackCrawlMessage(urlHash,entryBytes);
|
|
return newMessage;
|
|
}
|
|
}
|
|
|
|
public final class WorkterFactory implements org.apache.commons.pool.PoolableObjectFactory {
|
|
|
|
final ThreadGroup workerThreadGroup;
|
|
public WorkterFactory(ThreadGroup theWorkerThreadGroup) {
|
|
super();
|
|
|
|
if (theWorkerThreadGroup == null)
|
|
throw new IllegalArgumentException("The threadgroup object must not be null.");
|
|
|
|
this.workerThreadGroup = theWorkerThreadGroup;
|
|
}
|
|
|
|
public Object makeObject() {
|
|
Worker newWorker = new Worker(this.workerThreadGroup);
|
|
newWorker.setPriority(Thread.MAX_PRIORITY);
|
|
return newWorker;
|
|
}
|
|
|
|
/**
|
|
* @see org.apache.commons.pool.PoolableObjectFactory#destroyObject(java.lang.Object)
|
|
*/
|
|
public void destroyObject(Object obj) {
|
|
if (obj instanceof Worker) {
|
|
Worker theWorker = (Worker) obj;
|
|
theWorker.setStopped(true);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @see org.apache.commons.pool.PoolableObjectFactory#validateObject(java.lang.Object)
|
|
*/
|
|
public boolean validateObject(Object obj) {
|
|
if (obj instanceof Worker)
|
|
{
|
|
Worker theWorker = (Worker) obj;
|
|
if (!theWorker.isAlive() || theWorker.isInterrupted()) return false;
|
|
if (theWorker.isRunning()) return true;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* @param obj
|
|
*
|
|
*/
|
|
public void activateObject(Object obj) {
|
|
//log.debug(" activateObject...");
|
|
}
|
|
|
|
/**
|
|
* @param obj
|
|
*
|
|
*/
|
|
public void passivateObject(Object obj) {
|
|
//log.debug(" passivateObject..." + obj);
|
|
// if (obj instanceof Session) {
|
|
// Session theSession = (Session) obj;
|
|
// }
|
|
}
|
|
}
|
|
|
|
public final class WorkerPool extends GenericObjectPool {
|
|
public boolean isClosed = false;
|
|
|
|
/**
|
|
* First constructor.
|
|
* @param objFactory
|
|
*/
|
|
public WorkerPool(WorkterFactory objFactory) {
|
|
super(objFactory);
|
|
this.setMaxIdle(10); // Maximum idle threads.
|
|
this.setMaxActive(50); // Maximum active threads.
|
|
this.setMinEvictableIdleTimeMillis(30000); //Evictor runs every 30 secs.
|
|
//this.setMaxWait(1000); // Wait 1 second till a thread is available
|
|
}
|
|
|
|
public WorkerPool(plasmaCrawlStacker.WorkterFactory objFactory,
|
|
GenericObjectPool.Config config) {
|
|
super(objFactory, config);
|
|
}
|
|
|
|
public Object borrowObject() throws Exception {
|
|
return super.borrowObject();
|
|
}
|
|
|
|
public void returnObject(Object obj) throws Exception {
|
|
super.returnObject(obj);
|
|
}
|
|
|
|
public synchronized void close() throws Exception {
|
|
|
|
/*
|
|
* shutdown all still running session threads ...
|
|
*/
|
|
this.isClosed = true;
|
|
|
|
/* waiting for all threads to finish */
|
|
int threadCount = theWorkerThreadGroup.activeCount();
|
|
Thread[] threadList = new Thread[threadCount];
|
|
threadCount = theWorkerThreadGroup.enumerate(threadList);
|
|
|
|
try {
|
|
// trying to gracefull stop all still running sessions ...
|
|
log.logInfo("Signaling shutdown to " + threadCount + " remaining stackCrawl threads ...");
|
|
for ( int currentThreadIdx = 0; currentThreadIdx < threadCount; currentThreadIdx++ ) {
|
|
Thread currentThread = threadList[currentThreadIdx];
|
|
if (currentThread.isAlive()) {
|
|
((Worker)currentThread).setStopped(true);
|
|
}
|
|
}
|
|
|
|
// waiting a frew ms for the session objects to continue processing
|
|
try { Thread.sleep(500); } catch (InterruptedException ex) {}
|
|
|
|
// interrupting all still running or pooled threads ...
|
|
log.logInfo("Sending interruption signal to " + theWorkerThreadGroup.activeCount() + " remaining stackCrawl threads ...");
|
|
theWorkerThreadGroup.interrupt();
|
|
|
|
// if there are some sessions that are blocking in IO, we simply close the socket
|
|
log.logFine("Trying to abort " + theWorkerThreadGroup.activeCount() + " remaining stackCrawl threads ...");
|
|
for ( int currentThreadIdx = 0; currentThreadIdx < threadCount; currentThreadIdx++ ) {
|
|
Thread currentThread = threadList[currentThreadIdx];
|
|
if (currentThread.isAlive()) {
|
|
log.logInfo("Trying to shutdown stackCrawl thread '" + currentThread.getName() + "' [" + currentThreadIdx + "].");
|
|
((Worker)currentThread).close();
|
|
}
|
|
}
|
|
|
|
// we need to use a timeout here because of missing interruptable session threads ...
|
|
log.logFine("Waiting for " + theWorkerThreadGroup.activeCount() + " remaining stackCrawl threads to finish shutdown ...");
|
|
for ( int currentThreadIdx = 0; currentThreadIdx < threadCount; currentThreadIdx++ ) {
|
|
Thread currentThread = threadList[currentThreadIdx];
|
|
if (currentThread.isAlive()) {
|
|
log.logFine("Waiting for stackCrawl thread '" + currentThread.getName() + "' [" + currentThreadIdx + "] to finish shutdown.");
|
|
try { currentThread.join(500); } catch (InterruptedException ex) {}
|
|
}
|
|
}
|
|
|
|
log.logInfo("Shutdown of remaining stackCrawl threads finished.");
|
|
} catch (Exception e) {
|
|
log.logSevere("Unexpected error while trying to shutdown all remaining stackCrawl threads.",e);
|
|
}
|
|
|
|
super.close();
|
|
}
|
|
|
|
}
|
|
|
|
public final class Worker extends Thread {
|
|
private boolean running = false;
|
|
private boolean stopped = false;
|
|
private boolean done = false;
|
|
private stackCrawlMessage theMsg;
|
|
|
|
public Worker(ThreadGroup theThreadGroup) {
|
|
super(theThreadGroup,"stackCrawlThread");
|
|
}
|
|
|
|
public void setStopped(boolean stopped) {
|
|
this.stopped = stopped;
|
|
}
|
|
|
|
public void close() {
|
|
if (this.isAlive()) {
|
|
try {
|
|
// trying to close all still open httpc-Sockets first
|
|
int closedSockets = httpc.closeOpenSockets(this);
|
|
if (closedSockets > 0) {
|
|
log.logInfo(closedSockets + " HTTP-client sockets of thread '" + this.getName() + "' closed.");
|
|
}
|
|
} catch (Exception e) {}
|
|
}
|
|
}
|
|
|
|
public synchronized void execute(stackCrawlMessage newMsg) {
|
|
this.theMsg = newMsg;
|
|
this.done = false;
|
|
|
|
if (!this.running) {
|
|
// this.setDaemon(true);
|
|
this.start();
|
|
} else {
|
|
this.notifyAll();
|
|
}
|
|
}
|
|
|
|
public void reset() {
|
|
this.done = true;
|
|
this.theMsg = null;
|
|
}
|
|
|
|
public boolean isRunning() {
|
|
return this.running;
|
|
}
|
|
|
|
public void run() {
|
|
this.running = true;
|
|
|
|
// The thread keeps running.
|
|
while (!this.stopped && !Thread.interrupted()) {
|
|
if (this.done) {
|
|
// We are waiting for a task now.
|
|
synchronized (this) {
|
|
try {
|
|
this.wait(); //Wait until we get a request to process.
|
|
} catch (InterruptedException e) {
|
|
this.stopped = true;
|
|
// log.error("", e);
|
|
}
|
|
}
|
|
} else {
|
|
//There is a task....let us execute it.
|
|
try {
|
|
execute();
|
|
} catch (Exception e) {
|
|
// log.error("", e);
|
|
} finally {
|
|
reset();
|
|
|
|
if (!this.stopped && !this.isInterrupted() && !plasmaCrawlStacker.this.theWorkerPool.isClosed) {
|
|
try {
|
|
this.setName("stackCrawlThread_inPool");
|
|
plasmaCrawlStacker.this.theWorkerPool.returnObject(this);
|
|
} catch (Exception e1) {
|
|
// e1.printStackTrace();
|
|
this.stopped = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private void execute() {
|
|
try {
|
|
String rejectReason = dequeue(this.theMsg);
|
|
|
|
if (rejectReason != null) {
|
|
plasmaCrawlStacker.this.sb.urlPool.errorURL.newEntry(
|
|
new URL(this.theMsg.url()),
|
|
this.theMsg.referrerHash(),
|
|
this.theMsg.initiatorHash(),
|
|
yacyCore.seedDB.mySeed.hash,
|
|
this.theMsg.name,
|
|
rejectReason,
|
|
new bitfield(plasmaURL.urlFlagLength),
|
|
false
|
|
);
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
} finally {
|
|
this.done = true;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
}
|