package org.apache.archiva.repository.scanner.functors;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *  http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import org.apache.archiva.common.utils.BaseFile;
  import org.apache.archiva.consumers.RepositoryContentConsumer;
  import org.apache.commons.collections4.Closure;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import java.util.Map;
  
  /**
   * ConsumerProcessFileClosure
   */
  public class ConsumerProcessFileClosure
      implements Closure<RepositoryContentConsumer>
  {
      private Logger log = LoggerFactory.getLogger( ConsumerProcessFileClosure.class );
  
      private BaseFile basefile;
  
      private boolean executeOnEntireRepo;
  
      private Map<String, Long> consumerTimings;
  
      private Map<String, Long> consumerCounts;
  
      @Override
      public void execute( RepositoryContentConsumer input )
      {
          RepositoryContentConsumer/apache/archiva/consumers/RepositoryContentConsumer.html#RepositoryContentConsumer">RepositoryContentConsumer consumer = (RepositoryContentConsumer) input;
  
          String id = consumer.getId( );
          try
          {
              log.debug( "Sending to consumer: {}", id );
  
              long startTime = System.currentTimeMillis( );
              consumer.processFile( basefile.getRelativePath( ), executeOnEntireRepo );
              long endTime = System.currentTimeMillis( );
  
              if ( consumerTimings != null )
              {
                  Long value = consumerTimings.get( id );
                  consumerTimings.put( id, ( value != null ? value : 0 ) + endTime - startTime );
              }
  
              if ( consumerCounts != null )
              {
                  Long value = consumerCounts.get( id );
                  consumerCounts.put( id, ( value != null ? value : 0 ) + 1 );
              }
          }
          catch ( Exception e )
          {
              /* Intentionally Catch all exceptions.
               * So that the discoverer processing can continue.
               */
              log.error( "Consumer [{}] had an error when processing file ["
                  + "{}]: {}", id, basefile.getAbsolutePath( ), e.getMessage( ), e );
          }
  
      }
  
      public BaseFile getBasefile( )
      {
          return basefile;
      }
  
      public void setBasefile( BaseFile basefile )
      {
          this.basefile = basefile;
      }
  
      public boolean isExecuteOnEntireRepo( )
      {
          return executeOnEntireRepo;
      }
  
      public void setExecuteOnEntireRepo( boolean executeOnEntireRepo )
      {
         this.executeOnEntireRepo = executeOnEntireRepo;
     }
 
     public void setConsumerTimings( Map<String, Long> consumerTimings )
     {
         this.consumerTimings = consumerTimings;
     }
 
     public void setConsumerCounts( Map<String, Long> consumerCounts )
     {
         this.consumerCounts = consumerCounts;
     }
 
     public Logger getLogger( )
     {
         return log;
     }
 
     public void setLogger( Logger logger )
     {
         this.log = logger;
     }
 }