org.eclipse.aether.internal.impl.DefaultRemoteRepositoryManager Maven / Gradle / Ivy
/*******************************************************************************
* Copyright (c) 2010, 2014 Sonatype, Inc.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Sonatype, Inc. - initial API and implementation
*******************************************************************************/
package org.eclipse.aether.internal.impl;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.ListIterator;
import javax.inject.Inject;
import javax.inject.Named;
import org.eclipse.aether.RepositoryCache;
import org.eclipse.aether.RepositorySystemSession;
import org.eclipse.aether.impl.RemoteRepositoryManager;
import org.eclipse.aether.impl.UpdatePolicyAnalyzer;
import org.eclipse.aether.repository.Authentication;
import org.eclipse.aether.repository.AuthenticationSelector;
import org.eclipse.aether.repository.MirrorSelector;
import org.eclipse.aether.repository.Proxy;
import org.eclipse.aether.repository.ProxySelector;
import org.eclipse.aether.repository.RemoteRepository;
import org.eclipse.aether.repository.RepositoryPolicy;
import org.eclipse.aether.spi.connector.checksum.ChecksumPolicyProvider;
import org.eclipse.aether.spi.locator.Service;
import org.eclipse.aether.spi.locator.ServiceLocator;
import org.eclipse.aether.spi.log.Logger;
import org.eclipse.aether.spi.log.LoggerFactory;
import org.eclipse.aether.spi.log.NullLoggerFactory;
import org.eclipse.aether.util.StringUtils;
/**
*/
@Named
public class DefaultRemoteRepositoryManager
implements RemoteRepositoryManager, Service
{
private static final class LoggedMirror
{
private final Object[] keys;
public LoggedMirror( RemoteRepository original, RemoteRepository mirror )
{
keys = new Object[] { mirror.getId(), mirror.getUrl(), original.getId(), original.getUrl() };
}
@Override
public boolean equals( Object obj )
{
if ( this == obj )
{
return true;
}
else if ( !( obj instanceof LoggedMirror ) )
{
return false;
}
LoggedMirror that = (LoggedMirror) obj;
return Arrays.equals( keys, that.keys );
}
@Override
public int hashCode()
{
return Arrays.hashCode( keys );
}
}
private Logger logger = NullLoggerFactory.LOGGER;
private UpdatePolicyAnalyzer updatePolicyAnalyzer;
private ChecksumPolicyProvider checksumPolicyProvider;
public DefaultRemoteRepositoryManager()
{
// enables default constructor
}
@Inject
DefaultRemoteRepositoryManager( UpdatePolicyAnalyzer updatePolicyAnalyzer,
ChecksumPolicyProvider checksumPolicyProvider, LoggerFactory loggerFactory )
{
setUpdatePolicyAnalyzer( updatePolicyAnalyzer );
setChecksumPolicyProvider( checksumPolicyProvider );
setLoggerFactory( loggerFactory );
}
public void initService( ServiceLocator locator )
{
setLoggerFactory( locator.getService( LoggerFactory.class ) );
setUpdatePolicyAnalyzer( locator.getService( UpdatePolicyAnalyzer.class ) );
setChecksumPolicyProvider( locator.getService( ChecksumPolicyProvider.class ) );
}
public DefaultRemoteRepositoryManager setLoggerFactory( LoggerFactory loggerFactory )
{
this.logger = NullLoggerFactory.getSafeLogger( loggerFactory, getClass() );
return this;
}
public DefaultRemoteRepositoryManager setUpdatePolicyAnalyzer( UpdatePolicyAnalyzer updatePolicyAnalyzer )
{
if ( updatePolicyAnalyzer == null )
{
throw new IllegalArgumentException( "update policy analyzer has not been specified" );
}
this.updatePolicyAnalyzer = updatePolicyAnalyzer;
return this;
}
public DefaultRemoteRepositoryManager setChecksumPolicyProvider( ChecksumPolicyProvider checksumPolicyProvider )
{
if ( checksumPolicyProvider == null )
{
throw new IllegalArgumentException( "checksum policy provider has not been specified" );
}
this.checksumPolicyProvider = checksumPolicyProvider;
return this;
}
public List aggregateRepositories( RepositorySystemSession session,
List dominantRepositories,
List recessiveRepositories,
boolean recessiveIsRaw )
{
if ( recessiveRepositories.isEmpty() )
{
return dominantRepositories;
}
MirrorSelector mirrorSelector = session.getMirrorSelector();
AuthenticationSelector authSelector = session.getAuthenticationSelector();
ProxySelector proxySelector = session.getProxySelector();
List result = new ArrayList( dominantRepositories );
next: for ( RemoteRepository recessiveRepository : recessiveRepositories )
{
RemoteRepository repository = recessiveRepository;
if ( recessiveIsRaw )
{
RemoteRepository mirrorRepository = mirrorSelector.getMirror( recessiveRepository );
if ( mirrorRepository != null )
{
logMirror( session, recessiveRepository, mirrorRepository );
repository = mirrorRepository;
}
}
String key = getKey( repository );
for ( ListIterator it = result.listIterator(); it.hasNext(); )
{
RemoteRepository dominantRepository = it.next();
if ( key.equals( getKey( dominantRepository ) ) )
{
if ( !dominantRepository.getMirroredRepositories().isEmpty()
&& !repository.getMirroredRepositories().isEmpty() )
{
RemoteRepository mergedRepository = mergeMirrors( session, dominantRepository, repository );
if ( mergedRepository != dominantRepository )
{
it.set( mergedRepository );
}
}
continue next;
}
}
if ( recessiveIsRaw )
{
RemoteRepository.Builder builder = null;
Authentication auth = authSelector.getAuthentication( repository );
if ( auth != null )
{
builder = new RemoteRepository.Builder( repository );
builder.setAuthentication( auth );
}
Proxy proxy = proxySelector.getProxy( repository );
if ( proxy != null )
{
if ( builder == null )
{
builder = new RemoteRepository.Builder( repository );
}
builder.setProxy( proxy );
}
if ( builder != null )
{
repository = builder.build();
}
}
result.add( repository );
}
return result;
}
private void logMirror( RepositorySystemSession session, RemoteRepository original, RemoteRepository mirror )
{
if ( !logger.isDebugEnabled() )
{
return;
}
RepositoryCache cache = session.getCache();
if ( cache != null )
{
Object key = new LoggedMirror( original, mirror );
if ( cache.get( session, key ) != null )
{
return;
}
cache.put( session, key, Boolean.TRUE );
}
logger.debug( "Using mirror " + mirror.getId() + " (" + mirror.getUrl() + ") for " + original.getId() + " ("
+ original.getUrl() + ")." );
}
private String getKey( RemoteRepository repository )
{
return repository.getId();
}
private RemoteRepository mergeMirrors( RepositorySystemSession session, RemoteRepository dominant,
RemoteRepository recessive )
{
RemoteRepository.Builder merged = null;
RepositoryPolicy releases = null, snapshots = null;
next: for ( RemoteRepository rec : recessive.getMirroredRepositories() )
{
String recKey = getKey( rec );
for ( RemoteRepository dom : dominant.getMirroredRepositories() )
{
if ( recKey.equals( getKey( dom ) ) )
{
continue next;
}
}
if ( merged == null )
{
merged = new RemoteRepository.Builder( dominant );
releases = dominant.getPolicy( false );
snapshots = dominant.getPolicy( true );
}
releases = merge( session, releases, rec.getPolicy( false ), false );
snapshots = merge( session, snapshots, rec.getPolicy( true ), false );
merged.addMirroredRepository( rec );
}
if ( merged == null )
{
return dominant;
}
return merged.setReleasePolicy( releases ).setSnapshotPolicy( snapshots ).build();
}
public RepositoryPolicy getPolicy( RepositorySystemSession session, RemoteRepository repository, boolean releases,
boolean snapshots )
{
RepositoryPolicy policy1 = releases ? repository.getPolicy( false ) : null;
RepositoryPolicy policy2 = snapshots ? repository.getPolicy( true ) : null;
RepositoryPolicy policy = merge( session, policy1, policy2, true );
return policy;
}
private RepositoryPolicy merge( RepositorySystemSession session, RepositoryPolicy policy1,
RepositoryPolicy policy2, boolean globalPolicy )
{
RepositoryPolicy policy;
if ( policy2 == null )
{
if ( globalPolicy )
{
policy = merge( policy1, session.getUpdatePolicy(), session.getChecksumPolicy() );
}
else
{
policy = policy1;
}
}
else if ( policy1 == null )
{
if ( globalPolicy )
{
policy = merge( policy2, session.getUpdatePolicy(), session.getChecksumPolicy() );
}
else
{
policy = policy2;
}
}
else if ( !policy2.isEnabled() )
{
if ( globalPolicy )
{
policy = merge( policy1, session.getUpdatePolicy(), session.getChecksumPolicy() );
}
else
{
policy = policy1;
}
}
else if ( !policy1.isEnabled() )
{
if ( globalPolicy )
{
policy = merge( policy2, session.getUpdatePolicy(), session.getChecksumPolicy() );
}
else
{
policy = policy2;
}
}
else
{
String checksums = session.getChecksumPolicy();
if ( globalPolicy && !StringUtils.isEmpty( checksums ) )
{
// use global override
}
else
{
checksums =
checksumPolicyProvider.getEffectiveChecksumPolicy( session, policy1.getChecksumPolicy(),
policy2.getChecksumPolicy() );
}
String updates = session.getUpdatePolicy();
if ( globalPolicy && !StringUtils.isEmpty( updates ) )
{
// use global override
}
else
{
updates =
updatePolicyAnalyzer.getEffectiveUpdatePolicy( session, policy1.getUpdatePolicy(),
policy2.getUpdatePolicy() );
}
policy = new RepositoryPolicy( true, updates, checksums );
}
return policy;
}
private RepositoryPolicy merge( RepositoryPolicy policy, String updates, String checksums )
{
if ( policy != null )
{
if ( StringUtils.isEmpty( updates ) )
{
updates = policy.getUpdatePolicy();
}
if ( StringUtils.isEmpty( checksums ) )
{
checksums = policy.getChecksumPolicy();
}
if ( !policy.getUpdatePolicy().equals( updates ) || !policy.getChecksumPolicy().equals( checksums ) )
{
policy = new RepositoryPolicy( policy.isEnabled(), updates, checksums );
}
}
return policy;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy