CrawlFile.java

/**
 * License Agreement for OpenSearchServer
 * <p>
 * Copyright (C) 2008-2014 Emmanuel Keller / Jaeksoft
 * <p>
 * http://www.open-search-server.com
 * <p>
 * This file is part of OpenSearchServer.
 * <p>
 * OpenSearchServer 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 3 of the License, or
 * (at your option) any later version.
 * <p>
 * OpenSearchServer 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.
 * <p>
 * You should have received a copy of the GNU General Public License
 * along with OpenSearchServer.
 * If not, see <http://www.gnu.org/licenses/>.
 **/

package com.jaeksoft.searchlib.crawler.file.spider;

import com.jaeksoft.searchlib.Client;
import com.jaeksoft.searchlib.Logging;
import com.jaeksoft.searchlib.SearchLibException;
import com.jaeksoft.searchlib.config.Config;
import com.jaeksoft.searchlib.crawler.FieldMap;
import com.jaeksoft.searchlib.crawler.common.database.FetchStatus;
import com.jaeksoft.searchlib.crawler.common.database.IndexStatus;
import com.jaeksoft.searchlib.crawler.common.database.ParserStatus;
import com.jaeksoft.searchlib.crawler.common.process.CrawlStatistics;
import com.jaeksoft.searchlib.crawler.file.database.FileItem;
import com.jaeksoft.searchlib.crawler.file.process.FileInstanceAbstract;
import com.jaeksoft.searchlib.index.IndexDocument;
import com.jaeksoft.searchlib.parser.Parser;
import com.jaeksoft.searchlib.parser.ParserIndexDocumentIterator;
import com.jaeksoft.searchlib.parser.ParserSelector;
import com.jaeksoft.searchlib.plugin.IndexPluginList;
import com.jaeksoft.searchlib.streamlimiter.LimitException;
import com.jaeksoft.searchlib.streamlimiter.StreamLimiter;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;

public class CrawlFile {

	private final FileInstanceAbstract fileInstance;
	private final FileItem fileItem;
	private Parser parser;
	private String error;
	private final FieldMap fileFieldMap;
	private final Config config;

	public CrawlFile(FileInstanceAbstract fileInstance, FileItem fileItem, Config config, CrawlStatistics currentStats)
			throws SearchLibException {
		this.fileFieldMap = config.getFileCrawlerFieldMap();
		this.fileInstance = fileInstance;
		this.fileItem = fileItem;
		this.fileItem.setCrawlDate(System.currentTimeMillis());
		this.parser = null;
		this.error = null;
		this.config = config;
	}

	/**
	 * Télécharge le fichier et extrait les informations
	 *
	 */
	public void download() {
		synchronized (this) {
			try {
				fileItem.setFetchStatus(FetchStatus.FETCHED);

				ParserSelector parserSelector = config.getParserSelector();

				fileItem.setParserStatus(ParserStatus.PARSED);

				IndexDocument sourceDocument = fileItem.getIndexDocument();

				parser = config.getParserSelector()
						.parseFileInstance(sourceDocument, fileItem.getFileName(), null, null, fileInstance, null,
								parserSelector.getFileCrawlerDefaultParser(),
								parserSelector.getFileCrawlerFailOverParser());

				if (parser == null)
					fileItem.setParserStatus(ParserStatus.NOPARSER);
				else {
					if (parser.getError() != null)
						fileItem.setParserStatus(ParserStatus.PARSER_ERROR);
					else {
						fileItem.setLang(parser.getFirstLang());
						fileItem.setParser(parser.getParserName());
					}
				}

			} catch (MalformedURLException e) {
				Logging.warn(e.toString() + " (" + fileItem.getUri() + ")", e);
				fileItem.setFetchStatus(FetchStatus.ERROR);
			} catch (FileNotFoundException e) {
				fileItem.setFetchStatus(FetchStatus.GONE);
			} catch (LimitException e) {
				Logging.warn(e.toString() + " (" + fileItem.getUri() + ")", e);
				fileItem.setFetchStatus(FetchStatus.SIZE_EXCEED);
				setError(e.getMessage());
			} catch (IOException e) {
				Logging.warn(e.getMessage(), e);
				fileItem.setFetchStatus(FetchStatus.ERROR);
			} catch (Exception e) {
				Logging.warn(e.getMessage(), e);
				fileItem.setFetchStatus(FetchStatus.ERROR);
				fileItem.setParserStatus(ParserStatus.PARSER_ERROR);
			}
		}
	}

	public String getError() {
		return error;
	}

	public FileItem getFileItem() {
		return fileItem;
	}

	public Parser getParser() {
		return parser;
	}

	public class FileIndexDocumentIterator extends ParserIndexDocumentIterator {

		private FileIndexDocumentIterator() {
			super(parser, fileFieldMap);
		}

		@Override
		protected IndexDocument getCrawlItemIndexDocument() throws UnsupportedEncodingException {
			return fileItem.getIndexDocument();
		}

		@Override
		protected boolean checkPlugins(IndexDocument crawlItemIndexDocument, IndexDocument targetIndexDocument)
				throws SearchLibException, IOException {
			IndexPluginList indexPluginList = config.getFileCrawlMaster().getIndexPluginList();
			if (indexPluginList == null)
				return true;
			StreamLimiter streamLimiter = parser.getStreamLimiter();
			if (indexPluginList.run((Client) config, "octet/stream", streamLimiter, targetIndexDocument))
				return true;
			fileItem.setIndexStatus(IndexStatus.PLUGIN_REJECTED);
			fileItem.populate(targetIndexDocument);
			return false;
		}
	}

	public FileIndexDocumentIterator getTargetIndexDocumentIterator() throws SearchLibException, IOException {
		synchronized (this) {
			return new FileIndexDocumentIterator();
		}
	}

	public void setError(String error) {
		this.error = error;
	}
}