fetchstatusmatchesregexpdeciderule.java

这是个爬虫和lucece相结合最好了

/* $Id: FetchStatusMatchesRegExpDecideRule.java 4649 2006-09-25 17:16:55Z paul_jack $
*
* Created on Sep 4, 2006
*
* Copyright (C) 2006 Olaf Freyer.
*
* This file is part of the Heritrix web crawler (crawler.archive.org).
*
* Heritrix is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* any later version.
*
* Heritrix 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 Lesser Public License for more details.
*
* You should have received a copy of the GNU Lesser Public License
* along with Heritrix; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
package org.archive.crawler.deciderules;

import java.util.logging.Level;
import java.util.logging.Logger;
import javax.management.AttributeNotFoundException;

import org.archive.crawler.datamodel.CrawlURI;
import org.archive.crawler.settings.SimpleType;
import org.archive.util.TextUtils;


public class FetchStatusMatchesRegExpDecideRule extends PredicatedDecideRule {

    private static final long serialVersionUID = -3088156729860241312L;

    private  final Logger logger = Logger.getLogger(this.getClass().getName());
    
    public static final String ATTR_REGEXP = "regexp";
    
    /**
     * Usual constructor. 
     * @param name Name of this DecideRule.
     */
    public FetchStatusMatchesRegExpDecideRule(String name) {
        super(name);
        setDescription("FetchStatusMatchesRegExpDecideRule. Applies " +
        	"configured decision to any URI that has a fetch status matching " +
        	"the given regular expression.");
        addElementToDefinition(new SimpleType(ATTR_REGEXP, "Java regular" +
                "expression to match.", ""));
    }

    protected boolean evaluate(Object object) {
        try {
            String regexp = getRegexp(object);
            CrawlURI curi = (CrawlURI)object;
            String str = String.valueOf(curi.getFetchStatus());
            boolean result = (regexp == null)?
                    false: TextUtils.matches(regexp, str);
            if (logger.isLoggable(Level.FINE)) {
                logger.fine("Tested '" + str + "' match with regex '" +
                        regexp + " and result was " + result);
            }
            return result;
        } catch (ClassCastException e) {
            // if not CrawlURI, always disregard
            return false; 
        }
    }
    
    /** 
     * Get the regular expression string to match the URI against.
     *
     * @param o the object for which the regular expression should be
     *          matched against.
     * @return the regular expression to match against.
     */
    protected String getRegexp(Object o) {
        try {
            return (String) getAttribute(o, ATTR_REGEXP);
        } catch (AttributeNotFoundException e) {
            logger.severe(e.getMessage());
            return null;  // Basically the filter is inactive if this occurs.
        }
    }
}