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.
189 lines
10 KiB
189 lines
10 KiB
|
|
import java.net.MalformedURLException;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.ArrayList;
|
|
import java.util.Date;
|
|
import java.util.HashSet;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.regex.Pattern;
|
|
import java.util.regex.PatternSyntaxException;
|
|
|
|
import net.yacy.cora.document.encoding.ASCII;
|
|
import net.yacy.cora.document.id.DigestURL;
|
|
import net.yacy.cora.protocol.RequestHeader;
|
|
import net.yacy.cora.util.ConcurrentLog;
|
|
import net.yacy.crawler.CrawlSwitchboard;
|
|
import net.yacy.crawler.data.CrawlProfile;
|
|
import net.yacy.crawler.data.NoticedURL.StackType;
|
|
import net.yacy.crawler.retrieval.Request;
|
|
import net.yacy.peers.Seed;
|
|
import net.yacy.search.Switchboard;
|
|
import net.yacy.server.serverObjects;
|
|
import net.yacy.server.serverSwitch;
|
|
|
|
public class IndexCreateQueues_p {
|
|
|
|
private static SimpleDateFormat dayFormatter = new SimpleDateFormat("yyyy/MM/dd", Locale.US);
|
|
private static String daydate(final Date date) {
|
|
if (date == null) return "";
|
|
return dayFormatter.format(date);
|
|
}
|
|
|
|
private static final int INVALID = 0;
|
|
private static final int URL = 1;
|
|
private static final int ANCHOR = 2;
|
|
private static final int PROFILE = 3;
|
|
private static final int DEPTH = 4;
|
|
private static final int INITIATOR = 5;
|
|
private static final int MODIFIED = 6;
|
|
|
|
public static serverObjects respond(@SuppressWarnings("unused") final RequestHeader header, final serverObjects post, final serverSwitch env) {
|
|
// return variable that accumulates replacements
|
|
final Switchboard sb = (Switchboard) env;
|
|
final serverObjects prop = new serverObjects();
|
|
StackType stackType = StackType.LOCAL;
|
|
int urlsPerHost = 5;
|
|
boolean embed = false;
|
|
String deletepattern = ".*";
|
|
|
|
if (post != null) {
|
|
stackType = StackType.valueOf(post.get("stack", stackType.name()).toUpperCase());
|
|
urlsPerHost = post.getInt("urlsPerHost", urlsPerHost);
|
|
if (post.containsKey("embed")) embed = true;
|
|
|
|
if (post.containsKey("delete")) {
|
|
deletepattern = post.get("pattern", deletepattern).trim();
|
|
final int option = post.getInt("option", INVALID);
|
|
if (".*".equals(deletepattern)) {
|
|
sb.crawlQueues.noticeURL.clear(stackType);
|
|
try { sb.cleanProfiles(); } catch (final InterruptedException e) {/* ignore this */}
|
|
} else if (option > INVALID) {
|
|
try {
|
|
// compiling the regular expression
|
|
final Pattern compiledPattern = Pattern.compile(deletepattern);
|
|
|
|
if (option == PROFILE) {
|
|
// search and delete the crawl profile (_much_ faster, independent of queue size)
|
|
CrawlProfile entry;
|
|
for (final byte[] handle: sb.crawler.getActive()) {
|
|
entry = sb.crawler.getActive(handle);
|
|
final String name = entry.name();
|
|
if (CrawlSwitchboard.DEFAULT_PROFILES.contains(name)) continue;
|
|
if (compiledPattern.matcher(name).find()) {
|
|
sb.crawler.removeActive(entry.handle().getBytes());
|
|
sb.crawler.removePassive(entry.handle().getBytes());
|
|
}
|
|
}
|
|
} else {
|
|
int removedByHosts = 0;
|
|
if (option == URL && deletepattern.startsWith(".*") && deletepattern.endsWith(".*")) {
|
|
// try to delete that using the host name
|
|
Set<String> hosthashes = new HashSet<String>();
|
|
String hn = deletepattern.substring(2, deletepattern.length() - 2);
|
|
try {
|
|
hosthashes.add(DigestURL.hosthash(hn, hn.startsWith("ftp") ? 21 : 80));
|
|
hosthashes.add(DigestURL.hosthash(hn, 443));
|
|
removedByHosts = sb.crawlQueues.removeHosts(hosthashes);
|
|
} catch (MalformedURLException e) {
|
|
}
|
|
}
|
|
|
|
if (removedByHosts == 0) {
|
|
// iterating through the list of URLs
|
|
final Iterator<Request> iter = sb.crawlQueues.noticeURL.iterator(stackType);
|
|
Request entry;
|
|
final List<byte[]> removehashes = new ArrayList<byte[]>();
|
|
while (iter.hasNext()) {
|
|
if ((entry = iter.next()) == null) continue;
|
|
String value = null;
|
|
|
|
location: switch (option) {
|
|
case URL: value = (entry.url() == null) ? null : entry.url().toString(); break location;
|
|
case ANCHOR: value = entry.name(); break location;
|
|
case DEPTH: value = Integer.toString(entry.depth()); break location;
|
|
case INITIATOR:
|
|
value = (entry.initiator() == null || entry.initiator().length == 0) ? "proxy" : ASCII.String(entry.initiator());
|
|
break location;
|
|
case MODIFIED: value = daydate(entry.appdate()); break location;
|
|
default: value = null; break location;
|
|
}
|
|
|
|
if (value != null && compiledPattern.matcher(value).matches()) removehashes.add(entry.url().hash());
|
|
}
|
|
ConcurrentLog.info("IndexCreateQueues_p", "created a remove list with " + removehashes.size() + " entries for pattern '" + deletepattern + "'");
|
|
for (final byte[] b: removehashes) {
|
|
sb.crawlQueues.noticeURL.removeByURLHash(b);
|
|
}
|
|
}
|
|
}
|
|
} catch (final PatternSyntaxException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int stackSize = sb.crawlQueues.noticeURL.stackSize(stackType);
|
|
if (stackSize == 0) {
|
|
prop.put("crawler", "0");
|
|
} else {
|
|
prop.put("crawler", "1");
|
|
prop.put("crawler_embed", embed ? 1 : 0);
|
|
prop.put("crawler_embed_deletepattern", deletepattern);
|
|
prop.put("crawler_embed_queuename", stackType.name());
|
|
|
|
final Map<String, Integer[]> hosts = sb.crawlQueues.noticeURL.getDomainStackHosts(stackType, sb.robots);
|
|
|
|
int hc = 0;
|
|
for (Map.Entry<String, Integer[]> host: hosts.entrySet()) {
|
|
String hostnameport = host.getKey();
|
|
int p = hostnameport.lastIndexOf(':');
|
|
String hostname = p < 0 ? hostnameport : hostnameport.substring(0, p);
|
|
prop.putHTML("crawler_host_" + hc + "_hostnameport", hostnameport);
|
|
prop.putHTML("crawler_host_" + hc + "_hostname", hostname);
|
|
prop.put("crawler_host_" + hc + "_embed", embed ? 1 : 0);
|
|
prop.put("crawler_host_" + hc + "_urlsPerHost", urlsPerHost);
|
|
prop.putHTML("crawler_host_" + hc + "_queuename", stackType.name());
|
|
prop.put("crawler_host_" + hc + "_hostcount", host.getValue()[0]);
|
|
prop.put("crawler_host_" + hc + "_hostdelta", host.getValue()[1] == Integer.MIN_VALUE ? "not accessed" : Integer.toString(host.getValue()[1]));
|
|
List<Request> domainStackReferences = sb.crawlQueues.noticeURL.getDomainStackReferences(stackType, hostname, urlsPerHost, 10000);
|
|
|
|
Seed initiator;
|
|
String profileHandle;
|
|
CrawlProfile profileEntry;
|
|
int count = 0;
|
|
for (Request request: domainStackReferences) {
|
|
if (request == null) continue;
|
|
initiator = sb.peers.getConnected(request.initiator() == null ? "" : ASCII.String(request.initiator()));
|
|
profileHandle = request.profileHandle();
|
|
profileEntry = profileHandle == null ? null : sb.crawler.getActive(profileHandle.getBytes());
|
|
String depthString = Integer.toString(request.depth());
|
|
while (depthString.length() < 4) depthString = "0" + depthString;
|
|
prop.putHTML("crawler_host_" + hc + "_list_" + count + "_initiator", ((initiator == null) ? "proxy" : initiator.getName()) );
|
|
prop.put("crawler_host_" + hc + "_list_" + count + "_profile", ((profileEntry == null) ? "unknown" : profileEntry.collectionName()));
|
|
prop.putHTML("crawler_host_" + hc + "_list_" + count + "_depth", depthString);
|
|
prop.put("crawler_host_" + hc + "_list_" + count + "_modified", daydate(request.appdate()) );
|
|
prop.putHTML("crawler_host_" + hc + "_list_" + count + "_anchor", request.name());
|
|
prop.putHTML("crawler_host_" + hc + "_list_" + count + "_url", request.url().toNormalform(true));
|
|
prop.put("crawler_host_" + hc + "_list_" + count + "_hash", request.url().hash());
|
|
count++;
|
|
}
|
|
prop.putNum("crawler_host_" + hc + "_list", count);
|
|
hc++;
|
|
}
|
|
prop.put("crawler_host", hc);
|
|
}
|
|
|
|
prop.put("embed", embed ? 1 : 0);
|
|
prop.put("queuename", stackType.name().charAt(0) + stackType.name().substring(1).toLowerCase());
|
|
prop.put("embed_queuename", stackType.name().charAt(0) + stackType.name().substring(1).toLowerCase());
|
|
|
|
// return rewrite properties
|
|
return prop;
|
|
}
|
|
}
|