yacy_search_server/source/de/anomic/plasma/plasmaCrawlLoader.java
2005-04-13 23:00:20 +00:00

260 lines
11 KiB
Java

// plasmaCrawlerLoader.java
// ------------------------
// part of YaCy
// (C) by Michael Peter Christen; mc@anomic.de
// first published on http://www.anomic.de
// Frankfurt, Germany, 2004
// last major change: 25.02.2004
//
// 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.*;
import java.util.*;
import java.net.*;
import de.anomic.net.*;
import de.anomic.http.*;
import de.anomic.server.*;
import de.anomic.tools.*;
import de.anomic.htmlFilter.*;
public class plasmaCrawlLoader {
private plasmaHTCache cacheManager;
private int socketTimeout;
private int loadTimeout;
private boolean remoteProxyUse;
private String remoteProxyHost;
private int remoteProxyPort;
private int maxSlots;
private List slots;
private serverLog log;
private HashSet acceptMimeTypes;
public plasmaCrawlLoader(plasmaHTCache cacheManager, serverLog log, int socketTimeout, int loadTimeout, int mslots, boolean proxyUse, String proxyHost, int proxyPort,
HashSet acceptMimeTypes) {
this.cacheManager = cacheManager;
this.log = log;
this.socketTimeout = socketTimeout;
this.loadTimeout = loadTimeout;
this.remoteProxyUse = proxyUse;
this.remoteProxyHost = proxyHost;
this.remoteProxyPort = proxyPort;
this.maxSlots = mslots;
this.slots = new LinkedList();
this.acceptMimeTypes = acceptMimeTypes;
}
private void killTimeouts() {
Exec thread;
for (int i = slots.size() - 1; i >= 0; i--) {
// check if thread is alive
thread = (Exec) slots.get(i);
if (thread.isAlive()) {
// check the age of the thread
if (System.currentTimeMillis() - thread.startdate > loadTimeout) {
// we kill that thread
thread.interrupt(); // hopefully this wakes him up.
slots.remove(i);
log.logDebug("IGNORING SLEEPING DOWNLOAD SLOT " + thread.url.toString());
}
} else {
// thread i is dead, remove it
slots.remove(i);
}
}
}
public synchronized void loadParallel(URL url, String referer, String initiator, int depth, plasmaCrawlProfile.entry profile) {
// wait until there is space in the download slots
Exec thread;
while (slots.size() >= maxSlots) {
killTimeouts();
// wait a while
try {
Thread.currentThread().sleep(1000);
} catch (InterruptedException e) {
break;
}
}
// we found space in the download slots
thread = new Exec(url, referer, initiator, depth, profile);
thread.start();
slots.add(thread);
}
public int size() {
killTimeouts();
return slots.size();
}
public Exec[] threadStatus() {
killTimeouts();
Exec[] result = new Exec[slots.size()];
for (int i = 0; i < slots.size(); i++) result[i] = (Exec) slots.get(i);
return result;
}
public class Exec extends Thread {
public URL url;
public String referer;
public String initiator;
public int depth;
public long startdate;
public plasmaCrawlProfile.entry profile;
public String error;
public Exec(URL url, String referer, String initiator, int depth, plasmaCrawlProfile.entry profile) {
this.url = url; // the url to crawl
this.referer = referer; // the url that contained this url as link
this.initiator = initiator;
this.depth = depth; // distance from start-url
this.startdate = System.currentTimeMillis();
this.profile = profile;
this.error = null;
}
public void run() {
try {
load(url, referer, initiator, depth, profile);
} catch (IOException e) {
}
}
private httpc newhttpc(String server, int port, boolean ssl) throws IOException {
// a new httpc connection, combined with possible remote proxy
if (remoteProxyUse)
return new httpc(server, port, socketTimeout, ssl, remoteProxyHost, remoteProxyPort);
else
return new httpc(server, port, socketTimeout, ssl);
}
private void load(URL url, String referer, String initiator, int depth, plasmaCrawlProfile.entry profile) throws IOException {
if (url == null) return;
Date requestDate = new Date(); // remember the time...
String host = url.getHost();
String path = url.getPath();
int port = url.getPort();
boolean ssl = url.getProtocol().equals("https");
if (port < 0) port = (ssl) ? 443 : 80;
// set referrer; in some case advertise a little bit:
referer = referer.trim();
if (referer.length() == 0) referer = "http://www.yacy.net/yacy/";
// take a file from the net
try {
// create a request header
httpHeader requestHeader = new httpHeader();
requestHeader.put("User-Agent", httpdProxyHandler.userAgent);
requestHeader.put("Referer", referer);
requestHeader.put("Accept-Encoding", "gzip,deflate");
//System.out.println("CRAWLER_REQUEST_HEADER=" + requestHeader.toString()); // DEBUG
// open the connection
httpc remote = newhttpc(host, port, ssl);
// send request
httpc.response res = remote.GET(path, requestHeader);
if (res.status.startsWith("200")) {
// the transfer is ok
long contentLength = res.responseHeader.contentLength();
// reserve cache entry
plasmaHTCache.Entry htCache = cacheManager.newEntry(requestDate, depth, url, requestHeader, res.status, res.responseHeader, initiator, profile);
// request has been placed and result has been returned. work off response
File cacheFile = cacheManager.getCachePath(url);
try {
if (!(httpd.isTextMime(res.responseHeader.mime().toLowerCase(), acceptMimeTypes))) {
// if the response has not the right file type then reject file
remote.close();
log.logInfo("REJECTED WRONG MIME TYPE " + res.responseHeader.mime() + " for url " + url.toString());
htCache.status = plasmaHTCache.CACHE_UNFILLED;
} else if ((profile.storeHTCache()) && ((error = htCache.shallStoreCache()) == null)) {
// we write the new cache entry to file system directly
cacheFile.getParentFile().mkdirs();
FileOutputStream fos = new FileOutputStream(cacheFile);
htCache.cacheArray = res.writeContent(fos); // writes in cacheArray and cache file
fos.close();
htCache.status = plasmaHTCache.CACHE_FILL;
} else {
if (error != null) log.logDebug("CRAWLER NOT STORED RESOURCE " + url.toString() + ": " + error);
// anyway, the content still lives in the content scraper
htCache.cacheArray = res.writeContent(null); // writes only into cacheArray
htCache.status = plasmaHTCache.CACHE_PASSING;
}
// enQueue new entry with response header
if ((initiator == null) || (initiator.length() == 0)) {
// enqueued for proxy writings
cacheManager.stackProcess(htCache);
} else {
// direct processing for crawling
cacheManager.process(htCache);
}
} catch (SocketException e) {
// this may happen if the client suddenly closes its connection
// maybe the user has stopped loading
// in that case, we are not responsible and just forget it
// but we clean the cache also, since it may be only partial
// and most possible corrupted
if (cacheFile.exists()) cacheFile.delete();
log.logError("CRAWLER LOADER ERROR1: with url=" + url.toString() + ": " + e.toString());
}
} else {
// if the response has not the right response type then reject file
log.logInfo("REJECTED WRONG STATUS TYPE '" + res.status + "' for url " + url.toString());
// not processed any further
}
remote.close();
} catch (Exception e) {
// this may happen if the targeted host does not exist or anything with the
// remote server was wrong.
log.logError("CRAWLER LOADER ERROR2 with url=" + url.toString() + ": " + e.toString());
e.printStackTrace();
}
}
}
}