source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.consumers.lucene;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import java.util.HashMap;
  23 import java.util.Iterator;
  24 import java.util.List;
  25 import java.util.Map;
  26
  27 import org.apache.commons.lang.StringUtils;
  28 import org.apache.maven.archiva.configuration.ArchivaConfiguration;
  29 import org.apache.maven.archiva.configuration.ConfigurationNames;
  30 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  31 import org.apache.maven.archiva.consumers.AbstractMonitoredConsumer;
  32 import org.apache.maven.archiva.consumers.ConsumerException;
  33 import org.apache.maven.archiva.consumers.DatabaseUnprocessedArtifactConsumer;
  34 import org.apache.maven.archiva.indexer.RepositoryContentIndex;
  35 import org.apache.maven.archiva.indexer.RepositoryContentIndexFactory;
  36 import org.apache.maven.archiva.indexer.RepositoryIndexException;
  37 import org.apache.maven.archiva.indexer.hashcodes.HashcodesRecord;
  38 import org.apache.maven.archiva.model.ArchivaArtifact;
  39 import org.apache.maven.archiva.repository.ManagedRepositoryContent;
  40 import org.apache.maven.archiva.repository.RepositoryContentFactory;
  41 import org.apache.maven.archiva.repository.RepositoryException;
  42 import org.codehaus.plexus.personality.plexus.lifecycle.phase.Initializable;
  43 import org.codehaus.plexus.personality.plexus.lifecycle.phase.InitializationException;
  44 import org.codehaus.plexus.registry.Registry;
  45 import org.codehaus.plexus.registry.RegistryListener;
  46 import org.slf4j.Logger;
  47 import org.slf4j.LoggerFactory;
  48
  49 /**
  50  * IndexArtifactConsumer
  51  *
  52  * @version $Id$
  53  * @plexus.component role="org.apache.maven.archiva.consumers.DatabaseUnprocessedArtifactConsumer"
  54  * role-hint="index-artifact"
  55  * instantiation-strategy="per-lookup"
  56  */
  57 public class IndexArtifactConsumer
  58     extends AbstractMonitoredConsumer
  59     implements DatabaseUnprocessedArtifactConsumer, RegistryListener, Initializable
  60 {
  61     private Logger log = LoggerFactory.getLogger( IndexArtifactConsumer.class );
  62
  63     private static final String INDEX_ERROR = "indexing_error";
  64
  65     /**
  66      * @plexus.configuration default-value="index-artifact"
  67      */
  68     private String id;
  69
  70     /**
  71      * @plexus.configuration default-value="Index the artifact checksums for Find functionality."
  72      */
  73     private String description;
  74
  75     /**
  76      * @plexus.requirement
  77      */
  78     private ArchivaConfiguration configuration;
  79
  80     /**
  81      * @plexus.requirement
  82      */
  83     private RepositoryContentFactory repositoryFactory;
  84
  85     /**
  86      * @plexus.requirement role-hint="lucene"
  87      */
  88     private RepositoryContentIndexFactory indexFactory;
  89
  90     private Map<String, IndexedRepositoryDetails> repositoryMap = new HashMap<String, IndexedRepositoryDetails>();
  91
  92     public void beginScan()
  93     {
  94         /* nothing to do here */
  95     }
  96
  97     public void completeScan()
  98     {
  99         /* nothing to do here */
 100     }
 101
 102     public List<String> getIncludedTypes()
 103     {
 104         return null; // TODO: define these as a list of artifacts.
 105     }
 106
 107     public void processArchivaArtifact( ArchivaArtifact artifact )
 108         throws ConsumerException
 109     {
 110         HashcodesRecord record = new HashcodesRecord();
 111         record.setRepositoryId( artifact.getModel().getRepositoryId() );
 112         record.setArtifact( artifact );
 113
 114         IndexedRepositoryDetails pnl = getIndexedRepositoryDetails( artifact );
 115
 116         String artifactPath = pnl.repository.toPath( artifact );
 117         record.setFilename( artifactPath );
 118
 119         try
 120         {
 121             pnl.index.modifyRecord( record );
 122         }
 123         catch ( RepositoryIndexException e )
 124         {
 125             triggerConsumerError( INDEX_ERROR, "Unable to index hashcodes: " + e.getMessage() );
 126         }
 127     }
 128
 129     private IndexedRepositoryDetails getIndexedRepositoryDetails( ArchivaArtifact artifact )
 130     {
 131         String repoId = artifact.getModel().getRepositoryId();
 132         if ( StringUtils.isBlank( repoId ) )
 133         {
 134             throw new IllegalStateException(
 135                 "Unable to process artifact [" + artifact + "] as it has no repository id associated with it." );
 136         }
 137
 138         return getIndexedRepositoryDetails( repoId );
 139     }
 140
 141     private IndexedRepositoryDetails getIndexedRepositoryDetails( String id )
 142     {
 143         return this.repositoryMap.get( id );
 144     }
 145
 146     public String getDescription()
 147     {
 148         return description;
 149     }
 150
 151     public String getId()
 152     {
 153         return id;
 154     }
 155
 156     public boolean isPermanent()
 157     {
 158         return false;
 159     }
 160
 161     public void afterConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 162     {
 163         if ( ConfigurationNames.isManagedRepositories( propertyName ) )
 164         {
 165             initRepositoryMap();
 166         }
 167     }
 168
 169     public void beforeConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 170     {
 171         /* do nothing */
 172     }
 173
 174     public void initialize()
 175         throws InitializationException
 176     {
 177         initRepositoryMap();
 178         configuration.addChangeListener( this );
 179     }
 180
 181     private void initRepositoryMap()
 182     {
 183         synchronized ( this.repositoryMap )
 184         {
 185             this.repositoryMap.clear();
 186
 187             Iterator<ManagedRepositoryConfiguration> it = configuration.getConfiguration().getManagedRepositories().iterator();
 188             while ( it.hasNext() )
 189             {
 190                 ManagedRepositoryConfiguration repository = it.next();
 191
 192                 try
 193                 {
 194                     IndexedRepositoryDetails pnl = new IndexedRepositoryDetails();
 195
 196                     pnl.repository = repositoryFactory.getManagedRepositoryContent( repository.getId() );
 197
 198                     pnl.index = indexFactory.createHashcodeIndex( repository );
 199
 200                     this.repositoryMap.put( repository.getId(), pnl );
 201                 }
 202                 catch ( RepositoryException e )
 203                 {
 204                     log.error( "Unable to load repository content object: " + e.getMessage(), e );
 205                 }
 206             }
 207         }
 208     }
 209
 210     class IndexedRepositoryDetails
 211     {
 212         public ManagedRepositoryContent repository;
 213
 214         public RepositoryContentIndex index;
 215     }
 216 }