package org.apache.archiva.rest.services;
/*
* 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.redback.components.cache.Cache;
import org.apache.archiva.redback.components.cache.CacheStatistics;
import org.apache.archiva.redback.components.taskqueue.TaskQueue;
import org.apache.archiva.redback.components.taskqueue.TaskQueueException;
import org.apache.archiva.repository.scanner.RepositoryScanner;
import org.apache.archiva.repository.scanner.RepositoryScannerInstance;
import org.apache.archiva.rest.api.model.CacheEntry;
import org.apache.archiva.rest.api.model.ConsumerScanningStatistics;
import org.apache.archiva.rest.api.model.QueueEntry;
import org.apache.archiva.rest.api.model.RepositoryScannerStatistics;
import org.apache.archiva.rest.api.services.ArchivaRestServiceException;
import org.apache.archiva.rest.api.services.SystemStatusService;
import org.apache.archiva.rest.services.utils.ConsumerScanningStatisticsComparator;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Service;
import javax.inject.Inject;
import javax.ws.rs.core.Response;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
/**
* @author Olivier Lamy
* @since 1.4-M3
*/
@Service( "systemStatusService#rest" )
public class DefaultSystemStatusService
extends AbstractRestService
implements SystemStatusService
{
private Map<String, TaskQueue> queues = null;
private Map<String, Cache> caches = null;
private RepositoryScanner scanner;
// display spring scheduled
//@Inject @Named (value="springScheduler");
@Inject
public DefaultSystemStatusService( ApplicationContext applicationContext, RepositoryScanner scanner )
{
this.scanner = scanner;
queues = getBeansOfType( applicationContext, TaskQueue.class );
caches = getBeansOfType( applicationContext, Cache.class );
}
@Override
public String getMemoryStatus()
throws ArchivaRestServiceException
{
Runtime runtime = Runtime.getRuntime();
long total = runtime.totalMemory();
long used = total - runtime.freeMemory();
long max = runtime.maxMemory();
return formatMemory( used ) + "/" + formatMemory( total ) + " (Max: " + formatMemory( max ) + ")";
}
private static String formatMemory( long l )
{
return l / ( 1024 * 1024 ) + "M";
}
@Override
public String getCurrentServerTime( String locale )
throws ArchivaRestServiceException
{
SimpleDateFormat sdf = new SimpleDateFormat( "EEE, d MMM yyyy HH:mm:ss Z", new Locale( locale ) );
return sdf.format( new Date() );
}
@Override
public List<QueueEntry> getQueueEntries()
throws ArchivaRestServiceException
{
try
{
List<QueueEntry> queueEntries = new ArrayList<QueueEntry>( queues.size() );
for ( Map.Entry<String, TaskQueue> entry : queues.entrySet() )
{
queueEntries.add( new QueueEntry( entry.getKey(), entry.getValue().getQueueSnapshot().size() ) );
}
return queueEntries;
}
catch ( TaskQueueException e )
{
log.error( e.getMessage(), e );
throw new ArchivaRestServiceException( e.getMessage(),
Response.Status.INTERNAL_SERVER_ERROR.getStatusCode(), e );
}
}
@Override
public List<CacheEntry> getCacheEntries()
throws ArchivaRestServiceException
{
List<CacheEntry> cacheEntries = new ArrayList<CacheEntry>( caches.size() );
DecimalFormat decimalFormat = new DecimalFormat( "#%" );
for ( Map.Entry<String, Cache> entry : caches.entrySet() )
{
CacheStatistics cacheStatistics = entry.getValue().getStatistics();
cacheEntries.add( new CacheEntry( entry.getKey(), cacheStatistics.getSize(), cacheStatistics.getCacheHits(),
cacheStatistics.getCacheMiss(),
decimalFormat.format( cacheStatistics.getCacheHitRate() ).toString(),
cacheStatistics.getInMemorySize() ) );
}
Collections.sort( cacheEntries );
return cacheEntries;
}
@Override
public Boolean clearCache( String cacheKey )
throws ArchivaRestServiceException
{
Cache cache = caches.get( cacheKey );
if ( cache == null )
{
throw new ArchivaRestServiceException( "no cache for key: " + cacheKey,
Response.Status.BAD_REQUEST.getStatusCode(), null );
}
cache.clear();
return Boolean.TRUE;
}
@Override
public Boolean clearAllCaches()
throws ArchivaRestServiceException
{
for ( Cache cache : caches.values() )
{
cache.clear();
}
return Boolean.TRUE;
}
@Override
public List<RepositoryScannerStatistics> getRepositoryScannerStatistics()
throws ArchivaRestServiceException
{
Set<RepositoryScannerInstance> repositoryScannerInstances = scanner.getInProgressScans();
if ( repositoryScannerInstances.isEmpty() )
{
return Collections.emptyList();
}
List<RepositoryScannerStatistics> repositoryScannerStatisticsList =
new ArrayList<RepositoryScannerStatistics>( repositoryScannerInstances.size() );
for ( RepositoryScannerInstance instance : repositoryScannerInstances )
{
RepositoryScannerStatistics repositoryScannerStatistics = new RepositoryScannerStatistics();
repositoryScannerStatisticsList.add( repositoryScannerStatistics );
repositoryScannerStatistics.setManagedRepository( instance.getRepository() );
repositoryScannerStatistics.setNewFileCount( instance.getStats().getNewFileCount() );
repositoryScannerStatistics.setTotalFileCount( instance.getStats().getTotalFileCount() );
repositoryScannerStatistics.setConsumerScanningStatistics( mapConsumerScanningStatistics( instance ) );
}
return repositoryScannerStatisticsList;
}
private List<ConsumerScanningStatistics> mapConsumerScanningStatistics( RepositoryScannerInstance instance )
{
DecimalFormat decimalFormat = new DecimalFormat( "###.##" );
if ( instance.getConsumerCounts() == null )
{
return Collections.emptyList();
}
List<ConsumerScanningStatistics> ret =
new ArrayList<ConsumerScanningStatistics>( instance.getConsumerCounts().size() );
for ( Map.Entry<String, Long> entry : instance.getConsumerCounts().entrySet() )
{
ConsumerScanningStatistics consumerScanningStatistics = new ConsumerScanningStatistics();
consumerScanningStatistics.setConsumerKey( entry.getKey() );
consumerScanningStatistics.setCount( entry.getValue() );
consumerScanningStatistics.setTime( instance.getConsumerTimings().get( entry.getKey() ) );
if ( consumerScanningStatistics.getCount() > 0 )
{
consumerScanningStatistics.setAverage( decimalFormat.format(
consumerScanningStatistics.getTime() / consumerScanningStatistics.getCount() ) );
}
ret.add( consumerScanningStatistics );
}
Collections.sort( ret, ConsumerScanningStatisticsComparator.INSTANCE );
return ret;
}
}