mirror of
https://github.com/yacy/yacy_search_server.git
synced 2024-09-19 00:01:41 +02:00
329 lines
13 KiB
Java
329 lines
13 KiB
Java
// yacySearch.java
|
|
// -------------------------------------
|
|
// (C) by Michael Peter Christen; mc@yacy.net
|
|
// first published on http://www.anomic.de
|
|
// Frankfurt, Germany, 2004
|
|
//
|
|
// $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
|
|
|
|
package net.yacy.peers;
|
|
|
|
import java.util.HashSet;
|
|
import java.util.Iterator;
|
|
import java.util.Set;
|
|
import java.util.SortedMap;
|
|
|
|
import net.yacy.cora.document.ASCII;
|
|
import net.yacy.cora.storage.HandleSet;
|
|
import net.yacy.kelondro.logging.Log;
|
|
import net.yacy.peers.dht.PeerSelection;
|
|
import net.yacy.repository.Blacklist;
|
|
import net.yacy.search.query.QueryParams;
|
|
import net.yacy.search.query.SearchEvent;
|
|
|
|
|
|
public class RemoteSearch extends Thread {
|
|
|
|
private static final ThreadGroup ysThreadGroup = new ThreadGroup("yacySearchThreadGroup");
|
|
|
|
final private SearchEvent event;
|
|
final private String wordhashes, excludehashes, sitehash, authorhash, contentdom;
|
|
final private int partitions;
|
|
final private SearchEvent.SecondarySearchSuperviser secondarySearchSuperviser;
|
|
final private Blacklist blacklist;
|
|
final private Seed targetPeer;
|
|
private int urls;
|
|
private final int count, maxDistance;
|
|
private final long time;
|
|
final private QueryParams.Modifier modifier;
|
|
final private String language;
|
|
|
|
public RemoteSearch(
|
|
final SearchEvent event,
|
|
final String wordhashes,
|
|
final String excludehashes,
|
|
final QueryParams.Modifier modifier,
|
|
final String language,
|
|
final String sitehash,
|
|
final String authorhash,
|
|
final String contentdom,
|
|
final int count,
|
|
final long time,
|
|
final int maxDistance,
|
|
final int partitions,
|
|
final Seed targetPeer,
|
|
final SearchEvent.SecondarySearchSuperviser secondarySearchSuperviser,
|
|
final Blacklist blacklist) {
|
|
super(ysThreadGroup, "yacySearch_" + targetPeer.getName());
|
|
this.event = event;
|
|
this.wordhashes = wordhashes;
|
|
this.excludehashes = excludehashes;
|
|
this.modifier = modifier;
|
|
this.language = language;
|
|
this.sitehash = sitehash;
|
|
this.authorhash = authorhash;
|
|
this.contentdom = contentdom;
|
|
this.partitions = partitions;
|
|
this.secondarySearchSuperviser = secondarySearchSuperviser;
|
|
this.blacklist = blacklist;
|
|
this.targetPeer = targetPeer;
|
|
this.urls = -1;
|
|
this.count = count;
|
|
this.time = time;
|
|
this.maxDistance = maxDistance;
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
this.event.rankingProcess.oneFeederStarted();
|
|
try {
|
|
this.urls = Protocol.primarySearch(
|
|
this.event,
|
|
this.wordhashes,
|
|
this.excludehashes,
|
|
this.modifier.getModifier(),
|
|
this.language,
|
|
this.sitehash,
|
|
this.authorhash,
|
|
this.contentdom,
|
|
this.count,
|
|
this.time,
|
|
this.maxDistance,
|
|
this.partitions,
|
|
this.targetPeer,
|
|
this.secondarySearchSuperviser,
|
|
this.blacklist);
|
|
if (this.urls >= 0) {
|
|
// urls is an array of url hashes. this is only used for log output
|
|
this.event.peers.mySeed().incRI(this.urls);
|
|
this.event.peers.mySeed().incRU(this.urls);
|
|
} else {
|
|
Network.log.logInfo("REMOTE SEARCH - no answer from remote peer " + this.targetPeer.hash + ":" + this.targetPeer.getName());
|
|
}
|
|
} catch (final Exception e) {
|
|
Log.logException(e);
|
|
} finally {
|
|
this.event.rankingProcess.oneFeederTerminated();
|
|
}
|
|
}
|
|
|
|
public static String set2string(final HandleSet hashes) {
|
|
final StringBuilder wh = new StringBuilder(hashes.size() * 12);
|
|
final Iterator<byte[]> iter = hashes.iterator();
|
|
while (iter.hasNext()) { wh.append(ASCII.String(iter.next())); }
|
|
return wh.toString();
|
|
}
|
|
|
|
public int links() {
|
|
return this.urls;
|
|
}
|
|
|
|
public int count() {
|
|
return this.count;
|
|
}
|
|
|
|
public Seed target() {
|
|
return this.targetPeer;
|
|
}
|
|
|
|
public static void primaryRemoteSearches(
|
|
final SearchEvent event,
|
|
final int count, final long time,
|
|
final Blacklist blacklist,
|
|
final SortedMap<byte[], String> clusterselection,
|
|
final int burstRobinsonPercent,
|
|
final int burstMultiwordPercent) {
|
|
// check own peer status
|
|
//if (wordIndex.seedDB.mySeed() == null || wordIndex.seedDB.mySeed().getPublicAddress() == null) { return null; }
|
|
|
|
// prepare seed targets and threads
|
|
final Seed[] targetPeers =
|
|
(clusterselection == null) ?
|
|
PeerSelection.selectSearchTargets(
|
|
event.peers,
|
|
event.getQuery().query_include_hashes,
|
|
event.peers.redundancy(),
|
|
burstRobinsonPercent,
|
|
burstMultiwordPercent)
|
|
: PeerSelection.selectClusterPeers(event.peers, clusterselection);
|
|
if (targetPeers == null) return;
|
|
|
|
// start solr searches
|
|
Set<Seed> omit = new HashSet<Seed>();
|
|
for (Seed s: targetPeers) omit.add(s);
|
|
Seed[] nodes = PeerSelection.selectNodeSearchTargets(event.peers, 20, omit);
|
|
for (Seed s: nodes) {
|
|
solrRemoteSearch(event, count, time, s, blacklist);
|
|
}
|
|
|
|
// start search to YaCy peers
|
|
final int targets = targetPeers.length;
|
|
if (targets == 0) return;
|
|
for (int i = 0; i < targets; i++) {
|
|
if (targetPeers[i] == null || targetPeers[i].hash == null) continue;
|
|
try {
|
|
RemoteSearch rs = new RemoteSearch(
|
|
event,
|
|
QueryParams.hashSet2hashString(event.getQuery().query_include_hashes),
|
|
QueryParams.hashSet2hashString(event.getQuery().query_exclude_hashes),
|
|
event.getQuery().modifier,
|
|
event.getQuery().targetlang == null ? "" : event.getQuery().targetlang,
|
|
event.getQuery().sitehash == null ? "" : event.getQuery().sitehash,
|
|
event.getQuery().authorhash == null ? "" : event.getQuery().authorhash,
|
|
event.getQuery().contentdom == null ? "all" : event.getQuery().contentdom.toString(),
|
|
count,
|
|
time,
|
|
event.getQuery().maxDistance,
|
|
targets,
|
|
targetPeers[i],
|
|
event.secondarySearchSuperviser,
|
|
blacklist);
|
|
rs.start();
|
|
event.primarySearchThreadsL.add(rs);
|
|
} catch (final OutOfMemoryError e) {
|
|
Log.logException(e);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static Thread secondaryRemoteSearch(
|
|
final SearchEvent event,
|
|
final Set<String> wordhashes,
|
|
final String urlhashes,
|
|
final long time,
|
|
final String targethash,
|
|
final Blacklist blacklist) {
|
|
|
|
// check own peer status
|
|
if (event.peers.mySeed() == null || event.peers.mySeed().getPublicAddress() == null) { return null; }
|
|
assert urlhashes != null;
|
|
assert urlhashes.length() > 0;
|
|
|
|
// prepare seed targets and threads
|
|
final Seed targetPeer = event.peers.getConnected(targethash);
|
|
if (targetPeer == null || targetPeer.hash == null) return null;
|
|
if (event.preselectedPeerHashes != null) targetPeer.setAlternativeAddress(event.preselectedPeerHashes.get(ASCII.getBytes(targetPeer.hash)));
|
|
Thread secondary = new Thread() {
|
|
@Override
|
|
public void run() {
|
|
event.rankingProcess.oneFeederStarted();
|
|
try {
|
|
int urls = Protocol.secondarySearch(
|
|
event,
|
|
QueryParams.hashSet2hashString(wordhashes),
|
|
urlhashes,
|
|
"all",
|
|
20,
|
|
time,
|
|
999,
|
|
0,
|
|
targetPeer,
|
|
blacklist);
|
|
if (urls >= 0) {
|
|
// urls is an array of url hashes. this is only used for log output
|
|
if (urlhashes != null && urlhashes.length() > 0) Network.log.logInfo("SECONDARY REMOTE SEARCH - remote peer " + targetPeer.hash + ":" + targetPeer.getName() + " contributed " + urls + " links for word hash " + wordhashes);
|
|
event.peers.mySeed().incRI(urls);
|
|
event.peers.mySeed().incRU(urls);
|
|
} else {
|
|
Network.log.logInfo("REMOTE SEARCH - no answer from remote peer " + targetPeer.hash + ":" + targetPeer.getName());
|
|
}
|
|
} catch (final Exception e) {
|
|
Log.logException(e);
|
|
} finally {
|
|
event.rankingProcess.oneFeederTerminated();
|
|
}
|
|
}
|
|
};
|
|
secondary.start();
|
|
return secondary;
|
|
}
|
|
|
|
public static Thread solrRemoteSearch(
|
|
final SearchEvent event,
|
|
final int count,
|
|
final long time,
|
|
final Seed targetPeer,
|
|
final Blacklist blacklist) {
|
|
|
|
|
|
// check own peer status
|
|
if (event.peers.mySeed() == null || event.peers.mySeed().getPublicAddress() == null) { return null; }
|
|
|
|
// prepare seed targets and threads
|
|
if (targetPeer != null && targetPeer.hash != null && event.preselectedPeerHashes != null) targetPeer.setAlternativeAddress(event.preselectedPeerHashes.get(ASCII.getBytes(targetPeer.hash)));
|
|
Thread solr = new Thread() {
|
|
@Override
|
|
public void run() {
|
|
event.rankingProcess.oneFeederStarted();
|
|
try {
|
|
int urls = Protocol.solrQuery(
|
|
event,
|
|
0,
|
|
count,
|
|
time,
|
|
targetPeer,
|
|
blacklist);
|
|
if (urls >= 0) {
|
|
// urls is an array of url hashes. this is only used for log output
|
|
event.peers.mySeed().incRI(urls);
|
|
event.peers.mySeed().incRU(urls);
|
|
} else {
|
|
if (targetPeer != null) {
|
|
Network.log.logInfo("REMOTE SEARCH - no answer from remote peer " + targetPeer.hash + ":" + targetPeer.getName());
|
|
}
|
|
}
|
|
} catch (final Exception e) {
|
|
Log.logException(e);
|
|
} finally {
|
|
event.rankingProcess.oneFeederTerminated();
|
|
}
|
|
}
|
|
};
|
|
solr.start();
|
|
return solr;
|
|
}
|
|
|
|
public static int remainingWaiting(final RemoteSearch[] searchThreads) {
|
|
if (searchThreads == null) return 0;
|
|
int alive = 0;
|
|
for (final RemoteSearch searchThread : searchThreads) {
|
|
if (searchThread.isAlive()) alive++;
|
|
}
|
|
return alive;
|
|
}
|
|
|
|
public static int collectedLinks(final RemoteSearch[] searchThreads) {
|
|
int links = 0;
|
|
for (final RemoteSearch searchThread : searchThreads) {
|
|
if (!(searchThread.isAlive()) && searchThread.urls > 0) {
|
|
links += searchThread.urls;
|
|
}
|
|
}
|
|
return links;
|
|
}
|
|
|
|
public static void interruptAlive(final RemoteSearch[] searchThreads) {
|
|
for (final RemoteSearch searchThread : searchThreads) {
|
|
if (searchThread.isAlive()) searchThread.interrupt();
|
|
}
|
|
}
|
|
|
|
}
|