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

   1 package org.apache.archiva.scheduler.repository;
   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.archiva.consumers.AbstractMonitoredConsumer;
  23 import org.apache.archiva.consumers.ConsumerException;
  24 import org.apache.archiva.consumers.KnownRepositoryContentConsumer;
  25 import org.apache.archiva.model.ArtifactReference;
  26 import org.apache.archiva.repository.ManagedRepository;
  27 import org.apache.archiva.repository.ManagedRepositoryContent;
  28 import org.apache.archiva.repository.RepositoryContentFactory;
  29 import org.apache.archiva.repository.layout.LayoutException;
  30 import org.springframework.stereotype.Service;
  31
  32 import javax.inject.Inject;
  33 import java.util.Collection;
  34 import java.util.Collections;
  35 import java.util.Date;
  36 import java.util.HashSet;
  37 import java.util.List;
  38 import java.util.Set;
  39
  40 @Service( "knownRepositoryContentConsumer#test-consumer" )
  41 public class TestConsumer
  42     extends AbstractMonitoredConsumer
  43     implements KnownRepositoryContentConsumer
  44 {
  45     private Set<ArtifactReference> consumed = new HashSet<ArtifactReference>();
  46
  47     @Inject
  48     private RepositoryContentFactory factory;
  49
  50     private ManagedRepositoryContent repository;
  51
  52     @Override
  53     public String getId()
  54     {
  55         return "test-consumer";
  56     }
  57
  58     @Override
  59     public String getDescription()
  60     {
  61         return null;
  62     }
  63
  64     @Override
  65     public List<String> getIncludes()
  66     {
  67         return Collections.singletonList( "**/**" );
  68     }
  69
  70     @Override
  71     public List<String> getExcludes()
  72     {
  73         return null;
  74     }
  75
  76     @Override
  77     public void beginScan( ManagedRepository repository, Date whenGathered )
  78         throws ConsumerException
  79     {
  80         consumed.clear();
  81
  82         this.repository = repository.getContent();
  83     }
  84
  85     @Override
  86     public void beginScan( ManagedRepository repository, Date whenGathered, boolean executeOnEntireRepo )
  87         throws ConsumerException
  88     {
  89         beginScan( repository, whenGathered );
  90     }
  91
  92     @Override
  93     public void processFile( String path )
  94         throws ConsumerException
  95     {
  96         if ( !path.endsWith( ".sha1" ) && !path.endsWith( ".md5" ) )
  97         {
  98             try
  99             {
 100                 consumed.add( repository.toArtifactReference( path ) );
 101             }
 102             catch ( LayoutException e )
 103             {
 104                 throw new ConsumerException( e.getMessage(), e );
 105             }
 106         }
 107     }
 108
 109     @Override
 110     public void processFile( String path, boolean executeOnEntireRepo )
 111         throws Exception
 112     {
 113         processFile( path );
 114     }
 115
 116     @Override
 117     public void completeScan()
 118     {
 119     }
 120
 121     @Override
 122     public void completeScan( boolean executeOnEntireRepo )
 123     {
 124         completeScan();
 125     }
 126
 127     public Collection<ArtifactReference> getConsumed()
 128     {
 129         return consumed;
 130     }
 131 }