wamblee.org Git - utils/blob

   1 /*
   2  * Copyright 2005 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.wamblee.crawler.impl;
  18
  19 import java.util.List;
  20
  21 import org.wamblee.crawler.Configuration;
  22 import org.wamblee.crawler.PageRequest;
  23 import org.wamblee.crawler.PageType;
  24
  25 /**
  26  * Implementation of the configuration for the crawler.
  27  */
  28 public class ConfigurationImpl implements Configuration {
  29
  30     private List<UrlConfig> _urlConfig;
  31
  32     private List<PageTypeConfig> _pageTypeConfig;
  33
  34     /**
  35      * Constructs the configuration.
  36      * @param aUrlConfig List of URL configuration elements.
  37      * @param aPageTypeConfig List of page type configuration elements.
  38      */
  39     public ConfigurationImpl(List<UrlConfig> aUrlConfig,
  40             List<PageTypeConfig> aPageTypeConfig) {
  41         _urlConfig = aUrlConfig;
  42         _pageTypeConfig = aPageTypeConfig;
  43     }
  44
  45     /*
  46      * (non-Javadoc)
  47      *
  48      * @see org.wamblee.crawler.Configuration#getRequest(java.lang.String)
  49      */
  50     public PageRequest getRequest(String aUrl) {
  51
  52         for (UrlConfig config : _urlConfig) {
  53             PageRequest request = config.getRequest(aUrl);
  54             if (request != null) {
  55                 return request;
  56             }
  57         }
  58         throw new RuntimeException("No configuration matched the URL '" + aUrl
  59                 + "'");
  60     }
  61
  62     /*
  63      * (non-Javadoc)
  64      *
  65      * @see org.wamblee.crawler.Configuration#getRequest(org.wamblee.crawler.PageType)
  66      */
  67     public PageRequest getRequest(PageType aType) {
  68         for (PageTypeConfig config : _pageTypeConfig) {
  69             PageRequest request = config.getRequest(aType.getType());
  70             if (request != null) {
  71                 return request;
  72             }
  73         }
  74         throw new RuntimeException("No configuration matched type '" + aType
  75                 + "'");
  76     }
  77 }