mirror of
https://github.com/yacy/yacy_search_server.git
synced 2024-09-19 00:01:41 +02:00
866 lines
41 KiB
Java
866 lines
41 KiB
Java
// Segment.java
|
|
// (C) 2005-2009 by Michael Peter Christen; mc@yacy.net, Frankfurt a. M., Germany
|
|
// first published 2005 on http://yacy.net; full redesign for segments 28.5.2009
|
|
//
|
|
// This is a part of YaCy, a peer-to-peer based web search engine
|
|
//
|
|
// $LastChangedDate$
|
|
// $LastChangedRevision$
|
|
// $LastChangedBy$
|
|
//
|
|
// LICENSE
|
|
//
|
|
// 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.search.index;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.net.MalformedURLException;
|
|
import java.util.Arrays;
|
|
import java.util.Collection;
|
|
import java.util.Date;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.concurrent.ArrayBlockingQueue;
|
|
import java.util.concurrent.BlockingQueue;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
import java.util.regex.Pattern;
|
|
|
|
import org.apache.solr.common.SolrDocument;
|
|
import org.apache.solr.common.SolrInputDocument;
|
|
|
|
import net.yacy.cora.document.encoding.ASCII;
|
|
import net.yacy.cora.document.encoding.UTF8;
|
|
import net.yacy.cora.document.id.DigestURL;
|
|
import net.yacy.cora.document.id.MultiProtocolURL;
|
|
import net.yacy.cora.federate.solr.connector.AbstractSolrConnector;
|
|
import net.yacy.cora.federate.solr.connector.SolrConnector;
|
|
import net.yacy.cora.federate.yacy.CacheStrategy;
|
|
import net.yacy.cora.order.Base64Order;
|
|
import net.yacy.cora.order.ByteOrder;
|
|
import net.yacy.cora.protocol.ClientIdentification;
|
|
import net.yacy.cora.protocol.ResponseHeader;
|
|
import net.yacy.cora.storage.HandleSet;
|
|
import net.yacy.cora.util.ByteBuffer;
|
|
import net.yacy.cora.util.CommonPattern;
|
|
import net.yacy.cora.util.ConcurrentLog;
|
|
import net.yacy.cora.util.LookAheadIterator;
|
|
import net.yacy.cora.util.SpaceExceededException;
|
|
import net.yacy.crawler.data.CrawlProfile;
|
|
import net.yacy.crawler.data.Transactions;
|
|
import net.yacy.crawler.retrieval.Response;
|
|
import net.yacy.document.Condenser;
|
|
import net.yacy.document.Document;
|
|
import net.yacy.document.Parser;
|
|
import net.yacy.document.parser.htmlParser;
|
|
import net.yacy.kelondro.data.citation.CitationReference;
|
|
import net.yacy.kelondro.data.citation.CitationReferenceFactory;
|
|
import net.yacy.kelondro.data.word.Word;
|
|
import net.yacy.kelondro.data.word.WordReference;
|
|
import net.yacy.kelondro.data.word.WordReferenceFactory;
|
|
import net.yacy.kelondro.data.word.WordReferenceRow;
|
|
import net.yacy.kelondro.index.RowHandleSet;
|
|
import net.yacy.kelondro.rwi.IODispatcher;
|
|
import net.yacy.kelondro.rwi.IndexCell;
|
|
import net.yacy.kelondro.rwi.ReferenceContainer;
|
|
import net.yacy.kelondro.rwi.ReferenceFactory;
|
|
import net.yacy.kelondro.table.IndexTable;
|
|
import net.yacy.kelondro.util.Bitfield;
|
|
import net.yacy.kelondro.util.ISO639;
|
|
import net.yacy.kelondro.util.MemoryControl;
|
|
import net.yacy.repository.LoaderDispatcher;
|
|
import net.yacy.search.query.SearchEvent;
|
|
import net.yacy.search.schema.CollectionConfiguration;
|
|
import net.yacy.search.schema.CollectionSchema;
|
|
import net.yacy.search.schema.WebgraphConfiguration;
|
|
import net.yacy.search.schema.WebgraphSchema;
|
|
|
|
public class Segment {
|
|
|
|
// catchall word
|
|
public final static String catchallString = "yacyall"; // a word that is always in all indexes; can be used for zero-word searches to find ALL documents
|
|
public final static byte[] catchallHash;
|
|
final static Word catchallWord = new Word(0, 0, 0);
|
|
static {
|
|
catchallHash = Word.word2hash(catchallString); // "KZzU-Vf6h5k-"
|
|
catchallWord.flags = new Bitfield(4);
|
|
for (int i = 0; i < catchallWord.flags.length(); i++) catchallWord.flags.set(i, true);
|
|
}
|
|
|
|
// environment constants
|
|
public static final long wCacheMaxAge = 1000 * 60 * 30; // milliseconds; 30 minutes
|
|
public static final int wCacheMaxChunk = 800; // maximum number of references for each urlhash
|
|
public static final int lowcachedivisor = 900;
|
|
public static final long targetFileSize = 64 * 1024 * 1024; // 256 MB
|
|
public static final int writeBufferSize = 4 * 1024 * 1024;
|
|
public static final String termIndexName = "text.index";
|
|
public static final String citationIndexName = "citation.index";
|
|
public static final String firstseenIndexName = "firstseen.index";
|
|
public static final String loadtimeIndexName = "loadtime.index";
|
|
|
|
// the reference factory
|
|
public static final ReferenceFactory<WordReference> wordReferenceFactory = new WordReferenceFactory();
|
|
public static final ReferenceFactory<CitationReference> citationReferenceFactory = new CitationReferenceFactory();
|
|
public static final ByteOrder wordOrder = Base64Order.enhancedCoder;
|
|
|
|
private final ConcurrentLog log;
|
|
private final File segmentPath;
|
|
protected final Fulltext fulltext;
|
|
protected IndexCell<WordReference> termIndex;
|
|
private IndexCell<CitationReference> urlCitationIndex;
|
|
private IndexTable firstSeenIndex;
|
|
private IndexTable loadTimeIndex;
|
|
private IODispatcher merger = null; // shared iodispatcher for kelondro indexes
|
|
|
|
/**
|
|
* create a new Segment
|
|
* @param log logger instance
|
|
* @param segmentPath that should be the path pointing to the directory "SEGMENT"
|
|
* @throws IOException when an error occurs
|
|
*/
|
|
public Segment(final ConcurrentLog log, final File segmentPath, final File archivePath,
|
|
final CollectionConfiguration collectionConfiguration, final WebgraphConfiguration webgraphConfiguration) throws IOException {
|
|
log.info("Initializing Segment '" + segmentPath + ".");
|
|
this.log = log;
|
|
this.segmentPath = segmentPath;
|
|
archivePath.mkdirs();
|
|
|
|
this.fulltext = new Fulltext(segmentPath, archivePath, collectionConfiguration, webgraphConfiguration);
|
|
this.termIndex = null;
|
|
this.urlCitationIndex = null;
|
|
this.firstSeenIndex = new IndexTable(new File(segmentPath, firstseenIndexName), 12, 8, false, false);
|
|
this.loadTimeIndex = new IndexTable(new File(segmentPath, loadtimeIndexName), 12, 8, false, false);
|
|
}
|
|
|
|
public boolean connectedRWI() {
|
|
return this.termIndex != null;
|
|
}
|
|
|
|
public void connectRWI(final int entityCacheMaxSize, final long maxFileSize) throws IOException {
|
|
if (this.termIndex != null) return;
|
|
|
|
if (this.merger == null) { // init shared iodispatcher if none running
|
|
this.merger = new IODispatcher(2, 2, writeBufferSize);
|
|
this.merger.start();
|
|
}
|
|
this.termIndex = new IndexCell<WordReference>(
|
|
new File(this.segmentPath, "default"),
|
|
termIndexName,
|
|
wordReferenceFactory,
|
|
wordOrder,
|
|
Word.commonHashLength,
|
|
entityCacheMaxSize,
|
|
targetFileSize,
|
|
maxFileSize,
|
|
writeBufferSize,
|
|
this.merger);
|
|
}
|
|
|
|
public void disconnectRWI() {
|
|
if (this.termIndex == null) return;
|
|
this.termIndex.close();
|
|
this.termIndex = null;
|
|
}
|
|
|
|
public boolean connectedCitation() {
|
|
return this.urlCitationIndex != null;
|
|
}
|
|
|
|
public void connectCitation(final int entityCacheMaxSize, final long maxFileSize) throws IOException {
|
|
if (this.urlCitationIndex != null) return;
|
|
|
|
if (this.merger == null) { // init shared iodispatcher if none running
|
|
this.merger = new IODispatcher(2,2,writeBufferSize);
|
|
this.merger.start();
|
|
}
|
|
this.urlCitationIndex = new IndexCell<CitationReference>(
|
|
new File(this.segmentPath, "default"),
|
|
citationIndexName,
|
|
citationReferenceFactory,
|
|
wordOrder,
|
|
Word.commonHashLength,
|
|
entityCacheMaxSize,
|
|
targetFileSize,
|
|
maxFileSize,
|
|
writeBufferSize,
|
|
this.merger);
|
|
}
|
|
|
|
public void disconnectCitation() {
|
|
if (this.urlCitationIndex == null) return;
|
|
this.urlCitationIndex.close();
|
|
this.urlCitationIndex = null;
|
|
}
|
|
|
|
public int citationCount() {
|
|
return this.urlCitationIndex == null ? 0 : this.urlCitationIndex.sizesMax();
|
|
}
|
|
|
|
public long citationSegmentCount() {
|
|
return this.urlCitationIndex == null ? 0 : this.urlCitationIndex.getSegmentCount();
|
|
}
|
|
|
|
public Fulltext fulltext() {
|
|
return this.fulltext;
|
|
}
|
|
|
|
public IndexCell<WordReference> termIndex() {
|
|
return this.termIndex;
|
|
}
|
|
|
|
public IndexCell<CitationReference> urlCitation() {
|
|
return this.urlCitationIndex;
|
|
}
|
|
|
|
public IndexTable firstSeenIndex() {
|
|
return this.firstSeenIndex;
|
|
}
|
|
|
|
public IndexTable loadTimeIndex() {
|
|
return this.loadTimeIndex;
|
|
}
|
|
|
|
public ReferenceReportCache getReferenceReportCache() {
|
|
return new ReferenceReportCache();
|
|
}
|
|
|
|
public class ReferenceReportCache {
|
|
private final Map<String, ReferenceReport> cache;
|
|
public ReferenceReportCache() {
|
|
this.cache = new ConcurrentHashMap<String, ReferenceReport>();
|
|
}
|
|
public ReferenceReport getReferenceReport(final String id, final boolean acceptSelfReference) throws IOException {
|
|
ReferenceReport rr = this.cache.get(id);
|
|
if (MemoryControl.shortStatus()) this.cache.clear();
|
|
if (rr != null) return rr;
|
|
try {
|
|
rr = new ReferenceReport(ASCII.getBytes(id), acceptSelfReference);
|
|
this.cache.put(id, rr);
|
|
return rr;
|
|
} catch (final SpaceExceededException e) {
|
|
ConcurrentLog.logException(e);
|
|
throw new IOException(e.getMessage());
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* A ReferenceReport object is a container for all references to a specific url.
|
|
* The class stores the number of links from domain-internal and domain-external backlinks,
|
|
* and the host hashes of all externally linking documents,
|
|
* all IDs from external hosts and all IDs from the same domain.
|
|
*/
|
|
public final class ReferenceReport {
|
|
private int internal, external;
|
|
private HandleSet externalHosts, externalIDs, internalIDs;
|
|
public ReferenceReport(final byte[] id, final boolean acceptSelfReference) throws IOException, SpaceExceededException {
|
|
this.internal = 0;
|
|
this.external = 0;
|
|
this.externalHosts = new RowHandleSet(6, Base64Order.enhancedCoder, 0);
|
|
this.internalIDs = new RowHandleSet(Word.commonHashLength, Base64Order.enhancedCoder, 0);
|
|
this.externalIDs = new RowHandleSet(Word.commonHashLength, Base64Order.enhancedCoder, 0);
|
|
if (connectedCitation()) try {
|
|
// read the references from the citation index
|
|
ReferenceContainer<CitationReference> references;
|
|
references = urlCitation().get(id, null);
|
|
if (references == null) return; // no references at all
|
|
Iterator<CitationReference> ri = references.entries();
|
|
while (ri.hasNext()) {
|
|
CitationReference ref = ri.next();
|
|
byte[] hh = ref.hosthash(); // host hash
|
|
if (ByteBuffer.equals(hh, 0, id, 6, 6)) {
|
|
this.internalIDs.put(ref.urlhash());
|
|
this.internal++;
|
|
} else {
|
|
this.externalHosts.put(hh);
|
|
this.externalIDs.put(ref.urlhash());
|
|
this.external++;
|
|
}
|
|
}
|
|
} catch (SpaceExceededException e) {
|
|
// the Citation Index got too large, we ignore the problem and hope that a second solr index is attached which will take over now
|
|
if (Segment.this.fulltext.useWebgraph()) this.internalIDs.clear();
|
|
}
|
|
if ((this.internalIDs.size() == 0 || !connectedCitation()) && Segment.this.fulltext.useWebgraph()) {
|
|
// reqd the references from the webgraph
|
|
SolrConnector webgraph = Segment.this.fulltext.getWebgraphConnector();
|
|
BlockingQueue<SolrDocument> docs = webgraph.concurrentDocumentsByQuery("{!cache=false raw f=" + WebgraphSchema.target_id_s.getSolrFieldName() + "}" + ASCII.String(id), WebgraphSchema.source_chars_i.getSolrFieldName() + " asc", 0, 10000000, Long.MAX_VALUE, 100, 1, false, WebgraphSchema.source_id_s.getSolrFieldName());
|
|
SolrDocument doc;
|
|
try {
|
|
while ((doc = docs.take()) != AbstractSolrConnector.POISON_DOCUMENT) {
|
|
if (MemoryControl.shortStatus()) break;
|
|
String refid = (String) doc.getFieldValue(WebgraphSchema.source_id_s.getSolrFieldName());
|
|
if (refid == null) continue;
|
|
byte[] refidh = ASCII.getBytes(refid);
|
|
byte[] hh = new byte[6]; // host hash
|
|
System.arraycopy(refidh, 6, hh, 0, 6);
|
|
if (ByteBuffer.equals(hh, 0, id, 6, 6)) {
|
|
if (acceptSelfReference || !Arrays.equals(refidh, id)) {
|
|
this.internalIDs.put(refidh);
|
|
this.internal++;
|
|
}
|
|
} else {
|
|
this.externalHosts.put(hh);
|
|
this.externalIDs.put(refidh);
|
|
this.external++;
|
|
}
|
|
}
|
|
} catch (final InterruptedException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
this.externalHosts.optimize();
|
|
this.internalIDs.optimize();
|
|
this.externalIDs.optimize();
|
|
}
|
|
public int getInternalCount() {
|
|
return this.internal;
|
|
}
|
|
public int getExternalCount() {
|
|
return this.external;
|
|
}
|
|
public HandleSet getExternalHostIDs() {
|
|
return this.externalHosts;
|
|
}
|
|
public HandleSet getExternalIDs() {
|
|
return this.externalIDs;
|
|
}
|
|
public HandleSet getInternallIDs() {
|
|
return this.internalIDs;
|
|
}
|
|
}
|
|
|
|
public long RWICount() {
|
|
if (this.termIndex == null) return 0;
|
|
return this.termIndex.sizesMax();
|
|
}
|
|
|
|
public long RWISegmentCount() {
|
|
if (this.termIndex == null) return 0;
|
|
return this.termIndex.getSegmentCount();
|
|
}
|
|
|
|
public int RWIBufferCount() {
|
|
if (this.termIndex == null) return 0;
|
|
return this.termIndex.getBufferSize();
|
|
}
|
|
|
|
/**
|
|
* get a guess about the word count. This is only a guess because it uses the term index if present and this index may be
|
|
* influenced by index transmission processes in its statistic word distribution. However, it can be a hint for heuristics
|
|
* which use the word count. Please do NOT use this if the termIndex is not present because it otherwise uses the solr index
|
|
* which makes it painfully slow.
|
|
* @param word
|
|
* @return the number of references for this word.
|
|
*/
|
|
public int getWordCountGuess(String word) {
|
|
if (word == null || word.indexOf(':') >= 0 || word.indexOf(' ') >= 0 || word.indexOf('/') >= 0 || word.indexOf('\"') >= 0) return 0;
|
|
if (this.termIndex != null) {
|
|
int count = this.termIndex.count(Word.word2hash(word));
|
|
return count;
|
|
}
|
|
if (this.fulltext.getDefaultConnector() == null) return 0;
|
|
try {
|
|
return (int) this.fulltext.getDefaultConnector().getCountByQuery(CollectionSchema.text_t.getSolrFieldName() + ":\"" + word + "\"");
|
|
} catch (final Throwable e) {
|
|
ConcurrentLog.warn("Segment", "problem with word guess for word: " + word);
|
|
ConcurrentLog.logException(e);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
public void setFirstSeenTime(final byte[] urlhash, long time) {
|
|
if (urlhash == null || time <= 0) return;
|
|
try {
|
|
if (this.firstSeenIndex.has(urlhash)) return; // NEVER overwrite, that is the purpose of this index.
|
|
this.firstSeenIndex.put(urlhash, time);
|
|
} catch (IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
|
|
public long getFirstSeenTime(final byte[] urlhash) {
|
|
if (urlhash == null) return -1;
|
|
try {
|
|
return this.firstSeenIndex.get(urlhash);
|
|
} catch (IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
public void setLoadTime(final byte[] urlhash, long time) {
|
|
if (urlhash == null || time <= 0) return;
|
|
try {
|
|
this.loadTimeIndex.put(urlhash, time); // ALWAYS overwrite!
|
|
} catch (IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
|
|
public long getLoadTime(final byte[] urlhash) {
|
|
if (urlhash == null) return -1;
|
|
try {
|
|
return this.loadTimeIndex.get(urlhash);
|
|
} catch (IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* check if a given document, identified by url hash as document id exists
|
|
* @param id the url hash and document id
|
|
* @return whether the documents exists
|
|
*/
|
|
public boolean exists(final String id) {
|
|
return this.fulltext.exists(id);
|
|
}
|
|
|
|
/**
|
|
* discover all urls that start with a given url stub
|
|
* @param stub
|
|
* @return an iterator for all matching urls
|
|
*/
|
|
public Iterator<DigestURL> urlSelector(final MultiProtocolURL stub, final long maxtime, final int maxcount) {
|
|
final BlockingQueue<SolrDocument> docQueue;
|
|
final String urlstub;
|
|
if (stub == null) {
|
|
docQueue = this.fulltext.getDefaultConnector().concurrentDocumentsByQuery(AbstractSolrConnector.CATCHALL_QUERY, CollectionSchema.url_chars_i.getSolrFieldName() + " asc", 0, Integer.MAX_VALUE, maxtime, maxcount, 1, false, CollectionSchema.id.getSolrFieldName(), CollectionSchema.sku.getSolrFieldName());
|
|
urlstub = null;
|
|
} else {
|
|
final String host = stub.getHost();
|
|
String hh = null;
|
|
try {
|
|
hh = DigestURL.hosthash(host, stub.getPort());
|
|
} catch (MalformedURLException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
docQueue = hh == null ? new ArrayBlockingQueue<SolrDocument>(0) : this.fulltext.getDefaultConnector().concurrentDocumentsByQuery(CollectionSchema.host_id_s + ":\"" + hh + "\"", CollectionSchema.url_chars_i.getSolrFieldName() + " asc", 0, Integer.MAX_VALUE, maxtime, maxcount, 1, false, CollectionSchema.id.getSolrFieldName(), CollectionSchema.sku.getSolrFieldName());
|
|
urlstub = stub.toNormalform(true);
|
|
}
|
|
|
|
// now filter the stub from the iterated urls
|
|
return new LookAheadIterator<DigestURL>() {
|
|
@Override
|
|
protected DigestURL next0() {
|
|
while (true) {
|
|
SolrDocument doc;
|
|
try {
|
|
doc = docQueue.take();
|
|
} catch (final InterruptedException e) {
|
|
ConcurrentLog.logException(e);
|
|
return null;
|
|
}
|
|
if (doc == null || doc == AbstractSolrConnector.POISON_DOCUMENT) return null;
|
|
String u = (String) doc.getFieldValue(CollectionSchema.sku.getSolrFieldName());
|
|
String id = (String) doc.getFieldValue(CollectionSchema.id.getSolrFieldName());
|
|
DigestURL url;
|
|
try {
|
|
url = new DigestURL(u, ASCII.getBytes(id));
|
|
} catch (final MalformedURLException e) {
|
|
continue;
|
|
}
|
|
if (urlstub == null || u.startsWith(urlstub)) return url;
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
public void clear() {
|
|
try {
|
|
if (this.termIndex != null) this.termIndex.clear();
|
|
if (this.fulltext != null) this.fulltext.clearLocalSolr();
|
|
if (this.fulltext != null) this.fulltext.clearRemoteSolr();
|
|
if (this.urlCitationIndex != null) this.urlCitationIndex.clear();
|
|
} catch (final IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
|
|
public void clearCaches() {
|
|
if (this.urlCitationIndex != null) this.urlCitationIndex.clearCache();
|
|
if (this.termIndex != null) this.termIndex.clearCache();
|
|
this.fulltext.clearCaches();
|
|
}
|
|
|
|
public File getLocation() {
|
|
return this.segmentPath;
|
|
}
|
|
|
|
public synchronized void close() {
|
|
if (this.termIndex != null) this.termIndex.close();
|
|
if (this.fulltext != null) this.fulltext.close();
|
|
if (this.urlCitationIndex != null) this.urlCitationIndex.close();
|
|
if (this.firstSeenIndex != null) this.firstSeenIndex.close();
|
|
if (this.loadTimeIndex != null) this.loadTimeIndex.close();
|
|
if (this.merger != null) {
|
|
this.merger.terminate();
|
|
this.merger = null;
|
|
}
|
|
}
|
|
|
|
public static String votedLanguage(
|
|
final DigestURL url,
|
|
final String urlNormalform,
|
|
final Document document,
|
|
final Condenser condenser) {
|
|
// do a identification of the language
|
|
String language = condenser.language(); // this is a statistical analysation of the content: will be compared with other attributes
|
|
final String bymetadata = document.dc_language(); // the languageByMetadata may return null if there was no declaration
|
|
if (language == null) {
|
|
// no statistics available, we take either the metadata (if given) or the TLD
|
|
language = (bymetadata == null) ? url.language() : bymetadata;
|
|
} else {
|
|
if (bymetadata == null) {
|
|
if (condenser.languageProbability() < 0.9) { // if probability of statistic is not very high, examine url
|
|
// two possible results: compare and report conflicts
|
|
if (!language.equals(url.language())) {
|
|
// see if we have a hint in the url that the statistic was right
|
|
final String u = urlNormalform.toLowerCase(Locale.ROOT);
|
|
String ISO639_country = ISO639.country(language);
|
|
if (u.contains("/" + language + "/") ||
|
|
(ISO639_country != null && u.contains("/" + ISO639.country(language).toLowerCase(Locale.ROOT) + "/"))) {
|
|
// this is a strong hint that the statistics was in fact correct
|
|
} else {
|
|
// no confirmation using the url, use the TLD
|
|
language = url.language();
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
// here we have three results: we can do a voting
|
|
if (language.equals(bymetadata)) {
|
|
//if (log.isFine()) log.logFine("LANGUAGE-BY-STATISTICS: " + entry.url() + " CONFIRMED - METADATA IDENTICAL: " + language);
|
|
} else if (language.equals(url.language())) {
|
|
//if (log.isFine()) log.logFine("LANGUAGE-BY-STATISTICS: " + entry.url() + " CONFIRMED - TLD IS IDENTICAL: " + language);
|
|
} else if (bymetadata.equals(url.language())) {
|
|
//if (log.isFine()) log.logFine("LANGUAGE-BY-STATISTICS: " + entry.url() + " CONFLICTING: " + language + " BUT METADATA AND TLD ARE IDENTICAL: " + bymetadata + ")");
|
|
language = bymetadata;
|
|
} else {
|
|
//if (log.isFine()) log.logFine("LANGUAGE-BY-STATISTICS: " + entry.url() + " CONFLICTING: ALL DIFFERENT! statistic: " + language + ", metadata: " + bymetadata + ", TLD: + " + entry.url().language() + ". taking metadata.");
|
|
language = bymetadata;
|
|
}
|
|
}
|
|
}
|
|
return language;
|
|
}
|
|
|
|
public void storeRWI(final ReferenceContainer<WordReference> wordContainer) throws IOException, SpaceExceededException {
|
|
if (this.termIndex != null) this.termIndex.add(wordContainer);
|
|
}
|
|
|
|
public void storeRWI(final byte[] termHash, final WordReference entry) throws IOException, SpaceExceededException {
|
|
if (this.termIndex != null) this.termIndex.add(termHash, entry);
|
|
}
|
|
|
|
/**
|
|
* putDocument should not be called directly; instead, put queueEntries to
|
|
* indexingPutDocumentProcessor
|
|
* (this must be public, otherwise the WorkflowProcessor - calling by reflection - does not work)
|
|
* ATTENTION: do not remove! profiling tools will show that this is not called, which is not true (using reflection)
|
|
* @param queueEntry
|
|
* @throws IOException
|
|
*/
|
|
public void putDocument(final SolrInputDocument queueEntry) {
|
|
try {
|
|
this.fulltext().putDocument(queueEntry);
|
|
} catch (final IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
|
|
public SolrInputDocument storeDocument(
|
|
final DigestURL url,
|
|
final DigestURL referrerURL,
|
|
final Map<String, Pattern> collections,
|
|
final CrawlProfile crawlProfile,
|
|
final ResponseHeader responseHeader,
|
|
final Document document,
|
|
final Condenser condenser,
|
|
final SearchEvent searchEvent,
|
|
final String sourceName, // contains the crawl profile hash if this comes from a web crawl
|
|
final boolean storeToRWI,
|
|
final String proxy,
|
|
final String acceptLanguage
|
|
) {
|
|
final CollectionConfiguration collectionConfig = this.fulltext.getDefaultConfiguration();
|
|
final String language = votedLanguage(url, url.toNormalform(true), document, condenser); // identification of the language
|
|
|
|
final CollectionConfiguration.SolrVector vector = collectionConfig.yacy2solr(this, collections, responseHeader,
|
|
document, condenser, referrerURL, language, crawlProfile.isPushCrawlProfile(),
|
|
this.fulltext().useWebgraph() ? this.fulltext.getWebgraphConfiguration() : null, sourceName);
|
|
|
|
return storeDocument(url, crawlProfile, responseHeader, document, vector, language, condenser,
|
|
searchEvent, sourceName, storeToRWI, proxy, acceptLanguage);
|
|
}
|
|
|
|
public SolrInputDocument storeDocument(
|
|
final DigestURL url,
|
|
final CrawlProfile crawlProfile,
|
|
final ResponseHeader responseHeader,
|
|
final Document document,
|
|
final CollectionConfiguration.SolrVector vector,
|
|
final String language,
|
|
final Condenser condenser,
|
|
final SearchEvent searchEvent,
|
|
final String sourceName, // contains the crawl profile hash if this comes from a web crawl
|
|
final boolean storeToRWI,
|
|
final String proxy,
|
|
final String acceptLanguage
|
|
) {
|
|
final long startTime = System.currentTimeMillis();
|
|
|
|
final CollectionConfiguration collectionConfig = this.fulltext.getDefaultConfiguration();
|
|
final String urlNormalform = url.toNormalform(true);
|
|
|
|
// CREATE INDEX
|
|
// load some document metadata
|
|
final Date loadDate = new Date();
|
|
final String id = ASCII.String(url.hash());
|
|
final String dc_title = document.dc_title();
|
|
|
|
// get last modified date of the document to be used for the rwi index
|
|
// (the lastmodified document propery should be the same in rwi and fulltext (calculated in yacy2solr))
|
|
Date modDate = responseHeader == null ? document.getLastModified() : responseHeader.lastModified();
|
|
if (modDate == null) modDate = new Date();
|
|
if (document.getLastModified().before(modDate)) modDate = document.getLastModified();
|
|
if (modDate.getTime() > loadDate.getTime()) modDate = loadDate;
|
|
char docType = Response.docType(document.dc_format());
|
|
|
|
// ENRICH DOCUMENT WITH RANKING INFORMATION
|
|
this.fulltext.getDefaultConfiguration().postprocessing_references(this.getReferenceReportCache(), vector, url, null);
|
|
|
|
// CREATE SNAPSHOT
|
|
if ((url.getProtocol().equals("http") || url.getProtocol().equals("https")) &&
|
|
crawlProfile != null && document.getDepth() <= crawlProfile.snapshotMaxdepth() &&
|
|
!crawlProfile.snapshotsMustnotmatch().matcher(urlNormalform).matches()) {
|
|
// load pdf in case that is wanted. This can later be used to compute a web page preview in the search results
|
|
Parser p = document.getParserObject();
|
|
boolean mimesupported = false;
|
|
if (p instanceof htmlParser)
|
|
mimesupported = ((htmlParser)p).supportedMimeTypes().contains(document.dc_format());
|
|
|
|
if (mimesupported)
|
|
// STORE IMAGE AND METADATA
|
|
Transactions.store(vector, true, crawlProfile.snapshotLoadImage(), crawlProfile.snapshotReplaceold(), proxy, acceptLanguage);
|
|
}
|
|
|
|
// STORE TO SOLR
|
|
this.putDocument(vector);
|
|
List<SolrInputDocument> webgraph = vector.getWebgraphDocuments();
|
|
String error = null;
|
|
if (webgraph != null && webgraph.size() > 0) {
|
|
|
|
// write the edges to the webgraph solr index
|
|
if (this.fulltext.useWebgraph()) {
|
|
tryloop: for (int i = 0; i < 20; i++) {
|
|
try {
|
|
error = null;
|
|
this.fulltext.putEdges(webgraph);
|
|
break tryloop;
|
|
} catch (final IOException e ) {
|
|
error = "failed to send " + urlNormalform + " to solr: " + e.getMessage();
|
|
ConcurrentLog.warn("SOLR", error);
|
|
if (i == 10) this.fulltext.commit(true);
|
|
try {Thread.sleep(1000);} catch (final InterruptedException e1) {}
|
|
continue tryloop;
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// REMEMBER FIRST SEEN
|
|
long now = System.currentTimeMillis();
|
|
setFirstSeenTime(url.hash(), Math.min(document.getLastModified().getTime(), now)); // should exist already in the index at this time, but just to make sure
|
|
setLoadTime(url.hash(), now); // always overwrites index entry
|
|
|
|
// write the edges to the citation reference index
|
|
if (this.connectedCitation()) try {
|
|
// we use the subgraph to write the citation index, that shall cause that the webgraph and the citation index is identical
|
|
|
|
if (collectionConfig.contains(CollectionSchema.inboundlinks_protocol_sxt) || collectionConfig.contains(CollectionSchema.inboundlinks_urlstub_sxt)) {
|
|
Collection<Object> inboundlinks_urlstub = vector.getFieldValues(CollectionSchema.inboundlinks_urlstub_sxt.getSolrFieldName());
|
|
List<String> inboundlinks_protocol = inboundlinks_urlstub == null ? null : CollectionConfiguration.indexedList2protocolList(vector.getFieldValues(CollectionSchema.inboundlinks_protocol_sxt.getSolrFieldName()), inboundlinks_urlstub.size());
|
|
if (inboundlinks_protocol != null && inboundlinks_urlstub != null && inboundlinks_protocol.size() == inboundlinks_urlstub.size() && inboundlinks_urlstub instanceof List<?>) {
|
|
for (int i = 0; i < inboundlinks_protocol.size(); i++) try {
|
|
String targetURL = inboundlinks_protocol.get(i) + "://" + ((String) ((List<?>) inboundlinks_urlstub).get(i));
|
|
String referrerhash = id;
|
|
String anchorhash = ASCII.String(new DigestURL(targetURL).hash());
|
|
if (referrerhash != null && anchorhash != null) {
|
|
this.urlCitationIndex.add(ASCII.getBytes(anchorhash), new CitationReference(ASCII.getBytes(referrerhash), loadDate.getTime()));
|
|
}
|
|
} catch (Throwable e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
}
|
|
if (collectionConfig.contains(CollectionSchema.outboundlinks_protocol_sxt) || collectionConfig.contains(CollectionSchema.outboundlinks_urlstub_sxt)) {
|
|
Collection<Object> outboundlinks_urlstub = vector.getFieldValues(CollectionSchema.outboundlinks_urlstub_sxt.getSolrFieldName());
|
|
List<String> outboundlinks_protocol = outboundlinks_urlstub == null ? null : CollectionConfiguration.indexedList2protocolList(vector.getFieldValues(CollectionSchema.outboundlinks_protocol_sxt.getSolrFieldName()), outboundlinks_urlstub.size());
|
|
if (outboundlinks_protocol != null && outboundlinks_urlstub != null && outboundlinks_protocol.size() == outboundlinks_urlstub.size() && outboundlinks_urlstub instanceof List<?>) {
|
|
for (int i = 0; i < outboundlinks_protocol.size(); i++) try {
|
|
String targetURL = outboundlinks_protocol.get(i) + "://" + ((String) ((List<?>) outboundlinks_urlstub).get(i));
|
|
String referrerhash = id;
|
|
String anchorhash = ASCII.String(new DigestURL(targetURL).hash());
|
|
if (referrerhash != null && anchorhash != null) {
|
|
this.urlCitationIndex.add(ASCII.getBytes(anchorhash), new CitationReference(ASCII.getBytes(referrerhash), loadDate.getTime()));
|
|
}
|
|
} catch (Throwable e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
}
|
|
}
|
|
} catch (Throwable e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
|
|
if (error != null) {
|
|
ConcurrentLog.severe("SOLR", error + ", PLEASE REPORT TO https://github.com/yacy/yacy_search_server/issues");
|
|
//Switchboard.getSwitchboard().pauseCrawlJob(SwitchboardConstants.CRAWLJOB_LOCAL_CRAWL, error);
|
|
//Switchboard.getSwitchboard().pauseCrawlJob(SwitchboardConstants.CRAWLJOB_REMOTE_TRIGGERED_CRAWL, error);
|
|
}
|
|
final long storageEndTime = System.currentTimeMillis();
|
|
|
|
// STORE PAGE INDEX INTO WORD INDEX DB
|
|
// create a word prototype which is re-used for all entries
|
|
if ((this.termIndex != null && storeToRWI) || searchEvent != null) {
|
|
final int outlinksSame = document.inboundLinks().size();
|
|
final int outlinksOther = document.outboundLinks().size();
|
|
final int urlLength = urlNormalform.length();
|
|
final int urlComps = MultiProtocolURL.urlComps(url.toNormalform(false)).length;
|
|
final int wordsintitle = CommonPattern.SPACES.split(dc_title).length; // same calculation as for CollectionSchema.title_words_val
|
|
final WordReferenceRow ientry = new WordReferenceRow(
|
|
url.hash(),
|
|
urlLength, urlComps, wordsintitle,
|
|
condenser.RESULT_NUMB_WORDS,
|
|
condenser.RESULT_NUMB_SENTENCES,
|
|
modDate.getTime(),
|
|
System.currentTimeMillis(),
|
|
UTF8.getBytes(language),
|
|
docType,
|
|
outlinksSame, outlinksOther);
|
|
|
|
// iterate over all words of content text
|
|
Word wprop = null;
|
|
byte[] wordhash;
|
|
String word;
|
|
for (Map.Entry<String, Word> wentry: condenser.words().entrySet()) {
|
|
word = wentry.getKey();
|
|
wprop = wentry.getValue();
|
|
assert (wprop.flags != null);
|
|
ientry.setWord(wprop);
|
|
wordhash = Word.word2hash(word);
|
|
if (this.termIndex != null && storeToRWI) try {
|
|
this.termIndex.add(wordhash, ientry);
|
|
} catch (final Exception e) {
|
|
ConcurrentLog.logException(e);
|
|
}
|
|
|
|
// during a search event it is possible that a heuristic is used which aquires index
|
|
// data during search-time. To transfer indexed data directly to the search process
|
|
// the following lines push the index data additionally to the search process
|
|
// this is done only for searched words
|
|
if (searchEvent != null && !searchEvent.query.getQueryGoal().getExcludeHashes().has(wordhash) && searchEvent.query.getQueryGoal().getIncludeHashes().has(wordhash)) {
|
|
// if the page was added in the context of a heuristic this shall ensure that findings will fire directly into the search result
|
|
ReferenceContainer<WordReference> container;
|
|
try {
|
|
container = ReferenceContainer.emptyContainer(Segment.wordReferenceFactory, wordhash, 1);
|
|
container.add(ientry);
|
|
searchEvent.addRWIs(container, true, sourceName, 1, 5000);
|
|
} catch (final SpaceExceededException e) {
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
if (searchEvent != null) searchEvent.addFinalize();
|
|
|
|
// assign the catchall word
|
|
ientry.setWord(wprop == null ? catchallWord : wprop); // we use one of the word properties as template to get the document characteristics
|
|
if (this.termIndex != null) try {this.termIndex.add(catchallHash, ientry);} catch (final Throwable e) {ConcurrentLog.logException(e);}
|
|
}
|
|
|
|
// finish index time
|
|
final long indexingEndTime = System.currentTimeMillis();
|
|
|
|
if (this.log.isInfo()) {
|
|
this.log.info("*Indexed " + condenser.words().size() + " words in URL " + url.toNormalform(true) +
|
|
" [" + id + "]" +
|
|
"\n\tDescription: " + dc_title +
|
|
"\n\tMimeType: " + document.dc_format() + " | Charset: " + document.getCharset() + " | " +
|
|
"Size: " + document.getTextLength() + " bytes | " +
|
|
//"Anchors: " + refs +
|
|
"\n\tLinkStorageTime: " + (storageEndTime - startTime) + " ms | " +
|
|
"indexStorageTime: " + (indexingEndTime - storageEndTime) + " ms");
|
|
}
|
|
|
|
// finished
|
|
return vector;
|
|
}
|
|
|
|
public void removeAllUrlReferences(final HandleSet urls, final LoaderDispatcher loader, final ClientIdentification.Agent agent, final CacheStrategy cacheStrategy) {
|
|
for (final byte[] urlhash: urls) removeAllUrlReferences(urlhash, loader, agent, cacheStrategy);
|
|
}
|
|
|
|
/**
|
|
* find all the words in a specific resource and remove the url reference from every word index
|
|
* finally, delete the url entry
|
|
* @param urlhash the hash of the url that shall be removed
|
|
* @param loader
|
|
* @param cacheStrategy
|
|
* @return number of removed words
|
|
*/
|
|
public int removeAllUrlReferences(final byte[] urlhash, final LoaderDispatcher loader, final ClientIdentification.Agent agent, final CacheStrategy cacheStrategy) {
|
|
|
|
if (urlhash == null) return 0;
|
|
// determine the url string
|
|
try {
|
|
final String u = fulltext().getURL(ASCII.String(urlhash));
|
|
final DigestURL url = u == null ? null : new DigestURL(u);
|
|
if (url == null) return 0;
|
|
|
|
// parse the resource
|
|
final Document document = Document.mergeDocuments(url, null, loader.loadDocuments(loader.request(url, true, false), cacheStrategy, Integer.MAX_VALUE, null, agent));
|
|
if (document == null) {
|
|
// delete just the url entry
|
|
fulltext().remove(urlhash);
|
|
return 0;
|
|
}
|
|
// get the word set
|
|
Set<String> words = null;
|
|
words = new Condenser(document, null, true, true, null, false, false, 0).words().keySet();
|
|
|
|
// delete all word references
|
|
int count = 0;
|
|
if (words != null && termIndex() != null) count = termIndex().remove(Word.words2hashesHandles(words), urlhash);
|
|
|
|
// finally delete the url entry itself
|
|
fulltext().remove(urlhash);
|
|
return count;
|
|
} catch (final Parser.Failure e) {
|
|
return 0;
|
|
} catch (final IOException e) {
|
|
ConcurrentLog.logException(e);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
}
|