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

   1 package org.apache.maven.archiva.consumers.database;
   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.maven.archiva.configuration.ArchivaConfiguration;
  23 import org.apache.maven.archiva.configuration.ConfigurationNames;
  24 import org.apache.maven.archiva.configuration.FileTypes;
  25 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  26 import org.apache.maven.archiva.consumers.AbstractMonitoredConsumer;
  27 import org.apache.maven.archiva.consumers.ConsumerException;
  28 import org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer;
  29 import org.apache.maven.archiva.database.ArchivaDAO;
  30 import org.apache.maven.archiva.database.ArchivaDatabaseException;
  31 import org.apache.maven.archiva.model.ArchivaArtifact;
  32 import org.apache.maven.archiva.model.ArtifactReference;
  33 import org.apache.maven.archiva.repository.ManagedRepositoryContent;
  34 import org.apache.maven.archiva.repository.RepositoryContentFactory;
  35 import org.apache.maven.archiva.repository.RepositoryException;
  36 import org.apache.maven.archiva.repository.layout.LayoutException;
  37 import org.codehaus.plexus.digest.Digester;
  38 import org.codehaus.plexus.digest.DigesterException;
  39 import org.codehaus.plexus.personality.plexus.lifecycle.phase.Initializable;
  40 import org.codehaus.plexus.personality.plexus.lifecycle.phase.InitializationException;
  41 import org.codehaus.plexus.registry.Registry;
  42 import org.codehaus.plexus.registry.RegistryListener;
  43
  44 import java.io.File;
  45 import java.util.ArrayList;
  46 import java.util.Date;
  47 import java.util.List;
  48
  49 /**
  50  * ArtifactUpdateDatabaseConsumer - Take an artifact off of disk and put it into the repository.
  51  *
  52  * @author <a href="mailto:joakime@apache.org">Joakim Erdfelt</a>
  53  * @version $Id$
  54  * @plexus.component role="org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer"
  55  *                   role-hint="update-db-artifact"
  56  *                   instantiation-strategy="per-lookup"
  57  */
  58 public class ArtifactUpdateDatabaseConsumer
  59     extends AbstractMonitoredConsumer
  60     implements KnownRepositoryContentConsumer, RegistryListener, Initializable
  61 {
  62     private static final String TYPE_NOT_ARTIFACT = "file-not-artifact";
  63
  64     private static final String DB_ERROR = "db-error";
  65
  66     private static final String CHECKSUM_CALCULATION = "checksum-calc";
  67
  68     /**
  69      * @plexus.configuration default-value="update-db-artifact"
  70      */
  71     private String id;
  72
  73     /**
  74      * @plexus.configuration default-value="Update the Artifact in the Database"
  75      */
  76     private String description;
  77
  78     /**
  79      * @plexus.requirement role-hint="jdo"
  80      */
  81     private ArchivaDAO dao;
  82
  83     /**
  84      * @plexus.requirement
  85      */
  86     private ArchivaConfiguration configuration;
  87
  88     /**
  89      * @plexus.requirement
  90      */
  91     private FileTypes filetypes;
  92
  93     /**
  94      * @plexus.requirement
  95      */
  96     private RepositoryContentFactory repositoryFactory;
  97
  98     /**
  99      * @plexus.requirement role-hint="sha1"
 100      */
 101     private Digester digestSha1;
 102
 103     /**
 104      * @plexus.requirement role-hint="md5";
 105      */
 106     private Digester digestMd5;
 107
 108     private ManagedRepositoryContent repository;
 109
 110     private File repositoryDir;
 111
 112     private List<String> includes = new ArrayList<String>();
 113
 114     public String getId()
 115     {
 116         return this.id;
 117     }
 118
 119     public String getDescription()
 120     {
 121         return this.description;
 122     }
 123
 124     public boolean isPermanent()
 125     {
 126         return true;
 127     }
 128
 129     public List<String> getExcludes()
 130     {
 131         return null;
 132     }
 133
 134     public List<String> getIncludes()
 135     {
 136         return this.includes;
 137     }
 138
 139     public void beginScan( ManagedRepositoryConfiguration repo )
 140         throws ConsumerException
 141     {
 142         try
 143         {
 144             this.repository = repositoryFactory.getManagedRepositoryContent( repo.getId() );
 145             this.repositoryDir = new File( repository.getRepoRoot() );
 146         }
 147         catch(RepositoryException e)
 148         {
 149             throw new ConsumerException( "Unable to start ArtifactUpdateDatabaseConsumer: " + e.getMessage(), e );
 150         }
 151     }
 152
 153     public void processFile( String path )
 154         throws ConsumerException
 155     {
 156         ArchivaArtifact artifact = getLiveArtifact( path );
 157
 158         if ( artifact == null )
 159         {
 160             return;
 161         }
 162
 163         try
 164         {
 165             artifact.getModel().setRepositoryId( this.repository.getId() );
 166
 167             // Calculate the hashcodes.
 168             File artifactFile = new File( this.repositoryDir, path );
 169             try
 170             {
 171                 artifact.getModel().setChecksumMD5( digestMd5.calc( artifactFile ) );
 172             }
 173             catch ( DigesterException e )
 174             {
 175                 triggerConsumerWarning( CHECKSUM_CALCULATION,
 176                                         "Unable to calculate the MD5 checksum: " + e.getMessage() );
 177             }
 178
 179             try
 180             {
 181                 artifact.getModel().setChecksumSHA1( digestSha1.calc( artifactFile ) );
 182             }
 183             catch ( DigesterException e )
 184             {
 185                 triggerConsumerWarning( CHECKSUM_CALCULATION,
 186                                         "Unable to calculate the SHA1 checksum: " + e.getMessage() );
 187             }
 188
 189             artifact.getModel().setLastModified( new Date( artifactFile.lastModified() ) );
 190             artifact.getModel().setSize( artifactFile.length() );
 191             artifact.getModel().setOrigin( "FileSystem" );
 192
 193             dao.getArtifactDAO().saveArtifact( artifact );
 194         }
 195         catch ( ArchivaDatabaseException e )
 196         {
 197             triggerConsumerError( DB_ERROR, "Unable to save artifact to database: " + e.getMessage() );
 198         }
 199     }
 200
 201     /**
 202      * Get a Live Artifact from a Path.
 203      * <p/>
 204      * Will resolve the artifact details from the path, and then return a database live version
 205      * of that artifact.  Suitable for modification and saving (without the need to check for
 206      * existance in database prior to save.)
 207      *
 208      * @param path the path to work from.
 209      * @return the artifact that is suitable for database saving.
 210      */
 211     public ArchivaArtifact getLiveArtifact( String path )
 212     {
 213         try
 214         {
 215             ArtifactReference artifact = repository.toArtifactReference( path );
 216
 217             ArchivaArtifact liveArtifact = dao.getArtifactDAO().createArtifact( artifact.getGroupId(),
 218                                                                                 artifact.getArtifactId(),
 219                                                                                 artifact.getVersion(),
 220                                                                                 artifact.getClassifier(),
 221                                                                                 artifact.getType() );
 222
 223             return liveArtifact;
 224         }
 225         catch ( LayoutException e )
 226         {
 227             triggerConsumerError( TYPE_NOT_ARTIFACT,
 228                                   "Path " + path + " cannot be converted to artifact: " + e.getMessage() );
 229             return null;
 230         }
 231     }
 232
 233     public void completeScan()
 234     {
 235         /* do nothing */
 236     }
 237
 238     public void afterConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 239     {
 240         if ( ConfigurationNames.isRepositoryScanning( propertyName ) )
 241         {
 242             initIncludes();
 243         }
 244     }
 245
 246     public void beforeConfigurationChange( Registry registry, String propertyName, Object propertyValue )
 247     {
 248         /* do nothing */
 249     }
 250
 251     private void initIncludes()
 252     {
 253         includes.clear();
 254
 255         includes.addAll( filetypes.getFileTypePatterns( FileTypes.ARTIFACTS ) );
 256     }
 257
 258     public void initialize()
 259         throws InitializationException
 260     {
 261         configuration.addChangeListener( this );
 262
 263         initIncludes();
 264     }
 265 }