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

   1 package org.apache.archiva.repository.scanner;
   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.collections.CollectionUtils;
  23 import org.apache.maven.archiva.configuration.FileTypes;
  24 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  25 import org.apache.maven.archiva.consumers.InvalidRepositoryContentConsumer;
  26 import org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer;
  27 import org.apache.maven.archiva.consumers.RepositoryContentConsumer;
  28 import org.codehaus.plexus.util.DirectoryWalker;
  29 import org.springframework.stereotype.Service;
  30
  31 import java.io.File;
  32 import java.util.ArrayList;
  33 import java.util.LinkedHashSet;
  34 import java.util.List;
  35 import java.util.Set;
  36 import javax.inject.Inject;
  37
  38 /**
  39  * DefaultRepositoryScanner
  40  *
  41  * @version $Id$
  42  */
  43 @Service("repositoryScanner#default")
  44 public class DefaultRepositoryScanner
  45     implements RepositoryScanner
  46 {
  47     /**
  48      *
  49      */
  50     @Inject
  51     private FileTypes filetypes;
  52
  53     /**
  54      *
  55      */
  56     @Inject
  57     private RepositoryContentConsumers consumerUtil;
  58
  59     private Set<RepositoryScannerInstance> inProgressScans = new LinkedHashSet<RepositoryScannerInstance>();
  60
  61     public RepositoryScanStatistics scan( ManagedRepositoryConfiguration repository, long changesSince )
  62         throws RepositoryScannerException
  63     {
  64         List<KnownRepositoryContentConsumer> knownContentConsumers = consumerUtil.getSelectedKnownConsumers();
  65         List<InvalidRepositoryContentConsumer> invalidContentConsumers = consumerUtil.getSelectedInvalidConsumers();
  66         List<String> ignoredPatterns = filetypes.getFileTypePatterns( FileTypes.IGNORED );
  67
  68         return scan( repository, knownContentConsumers, invalidContentConsumers, ignoredPatterns, changesSince );
  69     }
  70
  71     public RepositoryScanStatistics scan( ManagedRepositoryConfiguration repository,
  72                                           List<KnownRepositoryContentConsumer> knownContentConsumers,
  73                                           List<InvalidRepositoryContentConsumer> invalidContentConsumers,
  74                                           List<String> ignoredContentPatterns, long changesSince )
  75         throws RepositoryScannerException
  76     {
  77         if ( repository == null )
  78         {
  79             throw new IllegalArgumentException( "Unable to operate on a null repository." );
  80         }
  81
  82         File repositoryBase = new File( repository.getLocation() );
  83
  84         //MRM-1342 Repository statistics report doesn't appear to be working correctly
  85         //create the repo if not existing to have an empty stats
  86         if ( !repositoryBase.exists() && !repositoryBase.mkdirs() )
  87         {
  88             throw new UnsupportedOperationException( "Unable to scan a repository, directory "
  89                 + repositoryBase.getPath() + " does not exist." );
  90         }
  91
  92         if ( !repositoryBase.isDirectory() )
  93         {
  94             throw new UnsupportedOperationException( "Unable to scan a repository, path "
  95                 + repositoryBase.getPath() + " is not a directory." );
  96         }
  97
  98         // Setup Includes / Excludes.
  99
 100         List<String> allExcludes = new ArrayList<String>();
 101         List<String> allIncludes = new ArrayList<String>();
 102
 103         if ( CollectionUtils.isNotEmpty( ignoredContentPatterns ) )
 104         {
 105             allExcludes.addAll( ignoredContentPatterns );
 106         }
 107
 108         // Scan All Content. (intentional)
 109         allIncludes.add( "**/*" );
 110
 111         // Setup Directory Walker
 112         DirectoryWalker dirWalker = new DirectoryWalker();
 113
 114         dirWalker.setBaseDir( repositoryBase );
 115
 116         dirWalker.setIncludes( allIncludes );
 117         dirWalker.setExcludes( allExcludes );
 118
 119         // Setup the Scan Instance
 120         RepositoryScannerInstance scannerInstance = new RepositoryScannerInstance( repository, knownContentConsumers,
 121                                                                                    invalidContentConsumers, changesSince );
 122
 123         inProgressScans.add( scannerInstance );
 124
 125         dirWalker.addDirectoryWalkListener( scannerInstance );
 126
 127         // Execute scan.
 128         dirWalker.scan();
 129
 130         RepositoryScanStatistics stats = scannerInstance.getStatistics();
 131
 132         stats.setKnownConsumers( gatherIds( knownContentConsumers ) );
 133         stats.setInvalidConsumers( gatherIds( invalidContentConsumers ) );
 134
 135         inProgressScans.remove( scannerInstance );
 136
 137         return stats;
 138     }
 139
 140     private List<String> gatherIds( List<? extends RepositoryContentConsumer> consumers )
 141     {
 142         List<String> ids = new ArrayList<String>();
 143         for ( RepositoryContentConsumer consumer : consumers )
 144         {
 145             ids.add( consumer.getId() );
 146         }
 147         return ids;
 148     }
 149
 150     public Set<RepositoryScannerInstance> getInProgressScans()
 151     {
 152         return inProgressScans;
 153     }
 154 }