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 java.util.ArrayList;
  23 import java.util.Date;
  24 import java.util.HashMap;
  25 import java.util.List;
  26 import java.util.Map;
  27
  28 import org.apache.archiva.metadata.repository.stats.RepositoryStatistics;
  29 import org.apache.archiva.metadata.repository.stats.RepositoryStatisticsManager;
  30
  31 public class TestRepositoryStatisticsManager
  32     implements RepositoryStatisticsManager
  33 {
  34     private Map<String, List<RepositoryStatistics>> repoStats = new HashMap<String, List<RepositoryStatistics>>();
  35
  36     public RepositoryStatistics getLastStatistics( String repositoryId )
  37     {
  38         List<RepositoryStatistics> repositoryStatisticsList = getStatsList( repositoryId );
  39         return !repositoryStatisticsList.isEmpty()
  40             ? repositoryStatisticsList.get( repositoryStatisticsList.size() - 1 )
  41             : null;
  42     }
  43
  44     public void addStatisticsAfterScan( String repositoryId, Date startTime, Date endTime, long totalFiles,
  45                                         long newFiles )
  46     {
  47         List<RepositoryStatistics> stats = getStatsList( repositoryId );
  48
  49         RepositoryStatistics repositoryStatistics = new RepositoryStatistics();
  50         repositoryStatistics.setScanStartTime( startTime );
  51         repositoryStatistics.setScanEndTime( endTime );
  52         repositoryStatistics.setNewFileCount( newFiles );
  53         repositoryStatistics.setTotalFileCount( totalFiles );
  54
  55         stats.add( repositoryStatistics );
  56     }
  57
  58     public void deleteStatistics( String repositoryId )
  59     {
  60         repoStats.remove( repositoryId );
  61     }
  62
  63     public List<RepositoryStatistics> getStatisticsInRange( String repositoryId, Date startDate, Date endDate )
  64     {
  65         throw new UnsupportedOperationException();
  66     }
  67
  68     private List<RepositoryStatistics> getStatsList( String repositoryId )
  69     {
  70         List<RepositoryStatistics> stats = repoStats.get( repositoryId );
  71         if ( stats == null )
  72         {
  73             stats = new ArrayList<RepositoryStatistics>();
  74             repoStats.put( repositoryId, stats );
  75         }
  76         return stats;
  77     }
  78 }