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 org.apache.commons.lang.StringUtils;
  23 import org.apache.maven.archiva.configuration.ArchivaConfiguration;
  24 import org.apache.maven.archiva.configuration.ConfigurationNames;
  25 import org.apache.maven.archiva.configuration.RepositoryConfiguration;
  26 import org.apache.maven.archiva.consumers.AbstractMonitoredConsumer;
  27 import org.apache.maven.archiva.consumers.ConsumerException;
  28 import org.apache.maven.archiva.consumers.DatabaseUnprocessedArtifactConsumer;
  29 import org.apache.maven.archiva.indexer.RepositoryContentIndex;
  30 import org.apache.maven.archiva.indexer.RepositoryContentIndexFactory;
  31 import org.apache.maven.archiva.indexer.RepositoryIndexException;
  32 import org.apache.maven.archiva.indexer.hashcodes.HashcodesRecord;
  33 import org.apache.maven.archiva.model.ArchivaArtifact;
  34 import org.apache.maven.archiva.model.ArchivaRepository;
  35 import org.apache.maven.archiva.repository.ArchivaConfigurationAdaptor;
  36 import org.apache.maven.archiva.repository.layout.BidirectionalRepositoryLayout;
  37 import org.codehaus.plexus.personality.plexus.lifecycle.phase.Initializable;
  38 import org.codehaus.plexus.personality.plexus.lifecycle.phase.InitializationException;
  39 import org.codehaus.plexus.registry.Registry;
  40 import org.codehaus.plexus.registry.RegistryListener;
  41
  42 import java.util.HashMap;
  43 import java.util.Iterator;
  44 import java.util.List;
  45 import java.util.Map;
  46
  47 /**
  48  * IndexArtifactConsumer
  49  *
  50  * @author <a href="mailto:joakim@erdfelt.com">Joakim Erdfelt</a>
  51  * @version $Id$
  52  *
  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 static final String INDEX_ERROR = "indexing_error";
  62
  63     /**
  64      * @plexus.configuration default-value="index-artifact"
  65      */
  66     private String id;
  67
  68     /**
  69      * @plexus.configuration default-value="Index the artifact checksums for Find functionality."
  70      */
  71     private String description;
  72
  73     /**
  74      * @plexus.requirement
  75      */
  76     private ArchivaConfiguration configuration;
  77
  78     /**
  79      * @plexus.requirement role="org.apache.maven.archiva.repository.layout.BidirectionalRepositoryLayout"
  80      */
  81     private Map bidirectionalLayoutMap;
  82
  83     /**
  84      * @plexus.requirement role-hint="lucene"
  85      */
  86     private RepositoryContentIndexFactory indexFactory;
  87
  88     private Map repositoryMap = new HashMap();
  89
  90     public void beginScan()
  91     {
  92         /* nothing to do here */
  93     }
  94
  95     public void completeScan()
  96     {
  97         /* nothing to do here */
  98     }
  99
 100     public List getIncludedTypes()
 101     {
 102         return null; // TODO: define these as a list of artifacts.
 103     }
 104
 105     public void processArchivaArtifact( ArchivaArtifact artifact )
 106         throws ConsumerException
 107     {
 108         HashcodesRecord record = new HashcodesRecord();
 109         record.setRepositoryId( artifact.getModel().getRepositoryId() );
 110         record.setArtifact( artifact );
 111
 112         IndexedRepositoryDetails pnl = getIndexedRepositoryDetails( artifact );
 113         String artifactPath = pnl.layout.toPath( artifact );
 114         record.setFilename( artifactPath );
 115
 116         try
 117         {
 118             pnl.index.modifyRecord( record );
 119         }
 120         catch ( RepositoryIndexException e )
 121         {
 122             triggerConsumerError( INDEX_ERROR, "Unable to index hashcodes: " + e.getMessage() );
 123         }
 124     }
 125
 126     private IndexedRepositoryDetails getIndexedRepositoryDetails( ArchivaArtifact artifact )
 127     {
 128         String repoId = artifact.getModel().getRepositoryId();
 129         if ( StringUtils.isBlank( repoId ) )
 130         {
 131             throw new IllegalStateException( "Unable to process artifact [" + artifact
 132                 + "] as it has no repository id associated with it." );
 133         }
 134
 135         return getIndexedRepositoryDetails( repoId );
 136     }
 137
 138     private IndexedRepositoryDetails getIndexedRepositoryDetails( String id )
 139     {
 140         return (IndexedRepositoryDetails) this.repositoryMap.get( id );
 141     }
 142
 143     public String getDescription()
 144     {
 145         return description;
 146     }
 147
 148     public String getId()
 149     {
 150         return id;
 151     }
 152
 153     public boolean isPermanent()
 154     {
 155         return false;
 156     }
 157
 158     public void afterConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 159     {
 160         if ( ConfigurationNames.isRepositories( propertyName ) )
 161         {
 162             initRepositoryMap();
 163         }
 164     }
 165
 166     public void beforeConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 167     {
 168         /* do nothing */
 169     }
 170
 171     public void initialize()
 172         throws InitializationException
 173     {
 174         initRepositoryMap();
 175         configuration.addChangeListener( this );
 176     }
 177
 178     private void initRepositoryMap()
 179     {
 180         synchronized ( this.repositoryMap )
 181         {
 182             this.repositoryMap.clear();
 183
 184             Iterator it = configuration.getConfiguration().getRepositories().iterator();
 185             while ( it.hasNext() )
 186             {
 187                 RepositoryConfiguration repoconfig = (RepositoryConfiguration) it.next();
 188                 if ( !repoconfig.isManaged() )
 189                 {
 190                     continue;
 191                 }
 192
 193                 ArchivaRepository repository = ArchivaConfigurationAdaptor.toArchivaRepository( repoconfig );
 194                 IndexedRepositoryDetails pnl = new IndexedRepositoryDetails();
 195
 196                 pnl.path = repository.getUrl().getPath();
 197                 pnl.layout = (BidirectionalRepositoryLayout) this.bidirectionalLayoutMap.get( repoconfig.getLayout() );
 198
 199                 pnl.index = indexFactory.createHashcodeIndex( repository );
 200
 201                 this.repositoryMap.put( repoconfig.getId(), pnl );
 202             }
 203         }
 204     }
 205
 206     class IndexedRepositoryDetails
 207     {
 208         public String path;
 209
 210         public BidirectionalRepositoryLayout layout;
 211
 212         public RepositoryContentIndex index;
 213     }
 214 }