Search in sources :

Example 11 with MetadataUpload

use of org.sonatype.aether.spi.connector.MetadataUpload in project sonatype-aether by sonatype.

the class FileRepositoryConnector method put.

public void put(Collection<? extends ArtifactUpload> artifactUploads, Collection<? extends MetadataUpload> metadataUploads) {
    checkClosed();
    artifactUploads = notNull(artifactUploads);
    metadataUploads = notNull(metadataUploads);
    RunnableErrorForwarder errorForwarder = new RunnableErrorForwarder();
    for (ArtifactUpload artifactUpload : artifactUploads) {
        FileRepositoryWorker worker = new FileRepositoryWorker(artifactUpload, repository, session);
        worker.setLogger(logger);
        worker.setFileProcessor(fileProcessor);
        executor.execute(errorForwarder.wrap(worker));
    }
    for (MetadataUpload metadataUpload : metadataUploads) {
        FileRepositoryWorker worker = new FileRepositoryWorker(metadataUpload, repository, session);
        worker.setLogger(logger);
        worker.setFileProcessor(fileProcessor);
        executor.execute(errorForwarder.wrap(worker));
    }
    errorForwarder.await();
}
Also used : ArtifactUpload(org.sonatype.aether.spi.connector.ArtifactUpload) MetadataUpload(org.sonatype.aether.spi.connector.MetadataUpload) RunnableErrorForwarder(org.sonatype.aether.util.concurrency.RunnableErrorForwarder)

Example 12 with MetadataUpload

use of org.sonatype.aether.spi.connector.MetadataUpload in project sonatype-aether by sonatype.

the class WagonRepositoryConnector method put.

public void put(Collection<? extends ArtifactUpload> artifactUploads, Collection<? extends MetadataUpload> metadataUploads) {
    if (closed) {
        throw new IllegalStateException("connector closed");
    }
    artifactUploads = safe(artifactUploads);
    metadataUploads = safe(metadataUploads);
    for (ArtifactUpload upload : artifactUploads) {
        String path = layout.getPath(upload.getArtifact()).getPath();
        PutTask<?> task = new PutTask<ArtifactTransfer>(path, upload.getFile(), upload, ARTIFACT);
        task.run();
    }
    for (MetadataUpload upload : metadataUploads) {
        String path = layout.getPath(upload.getMetadata()).getPath();
        PutTask<?> task = new PutTask<MetadataTransfer>(path, upload.getFile(), upload, METADATA);
        task.run();
    }
}
Also used : ArtifactUpload(org.sonatype.aether.spi.connector.ArtifactUpload) MetadataUpload(org.sonatype.aether.spi.connector.MetadataUpload)

Example 13 with MetadataUpload

use of org.sonatype.aether.spi.connector.MetadataUpload in project sonatype-aether by sonatype.

the class RecordingRepositoryConnector method put.

public void put(Collection<? extends ArtifactUpload> artifactUploads, Collection<? extends MetadataUpload> metadataUploads) {
    if (artifactUploads != null) {
        for (ArtifactUpload artifactUpload : artifactUploads) {
            // mimic "real" connector
            artifactUpload.setState(State.ACTIVE);
            if (artifactUpload.getFile() == null) {
                artifactUpload.setException(new ArtifactTransferException(artifactUpload.getArtifact(), null, "no file"));
            }
            this.actualPut.add(artifactUpload.getArtifact());
            artifactUpload.setState(State.DONE);
        }
    }
    if (metadataUploads != null) {
        for (MetadataUpload metadataUpload : metadataUploads) {
            // mimic "real" connector
            metadataUpload.setState(State.ACTIVE);
            if (metadataUpload.getFile() == null) {
                metadataUpload.setException(new MetadataTransferException(metadataUpload.getMetadata(), null, "no file"));
            }
            this.actualPutMD.add(metadataUpload.getMetadata());
            metadataUpload.setState(State.DONE);
        }
    }
}
Also used : ArtifactTransferException(org.sonatype.aether.transfer.ArtifactTransferException) ArtifactUpload(org.sonatype.aether.spi.connector.ArtifactUpload) MetadataUpload(org.sonatype.aether.spi.connector.MetadataUpload) MetadataTransferException(org.sonatype.aether.transfer.MetadataTransferException)

Example 14 with MetadataUpload

use of org.sonatype.aether.spi.connector.MetadataUpload in project sonatype-aether by sonatype.

the class ConnectorTestSuite method testBlocking.

@Test
public void testBlocking() throws NoRepositoryConnectorException, IOException {
    RepositoryConnector connector = factory().newInstance(session, repository);
    int count = 10;
    byte[] pattern = "tmpFile".getBytes("UTF-8");
    File tmpFile = TestFileUtils.createTempFile(pattern, 100000);
    List<ArtifactUpload> artUps = ConnectorTestUtils.createTransfers(ArtifactUpload.class, count, tmpFile);
    List<MetadataUpload> metaUps = ConnectorTestUtils.createTransfers(MetadataUpload.class, count, tmpFile);
    List<ArtifactDownload> artDowns = ConnectorTestUtils.createTransfers(ArtifactDownload.class, count, null);
    List<MetadataDownload> metaDowns = ConnectorTestUtils.createTransfers(MetadataDownload.class, count, null);
    // this should block until all transfers are done - racing condition, better way to test this?
    connector.put(artUps, metaUps);
    connector.get(artDowns, metaDowns);
    for (int i = 0; i < count; i++) {
        ArtifactUpload artUp = artUps.get(i);
        MetadataUpload metaUp = metaUps.get(i);
        ArtifactDownload artDown = artDowns.get(i);
        MetadataDownload metaDown = metaDowns.get(i);
        assertTrue(Transfer.State.DONE.equals(artUp.getState()));
        assertTrue(Transfer.State.DONE.equals(artDown.getState()));
        assertTrue(Transfer.State.DONE.equals(metaUp.getState()));
        assertTrue(Transfer.State.DONE.equals(metaDown.getState()));
    }
    connector.close();
}
Also used : ArtifactDownload(org.sonatype.aether.spi.connector.ArtifactDownload) ArtifactUpload(org.sonatype.aether.spi.connector.ArtifactUpload) RepositoryConnector(org.sonatype.aether.spi.connector.RepositoryConnector) MetadataUpload(org.sonatype.aether.spi.connector.MetadataUpload) MetadataDownload(org.sonatype.aether.spi.connector.MetadataDownload) File(java.io.File) Test(org.junit.Test)

Example 15 with MetadataUpload

use of org.sonatype.aether.spi.connector.MetadataUpload in project sonatype-aether by sonatype.

the class ConnectorTestSuite method testProgressEventsDataBuffer.

@Test
public void testProgressEventsDataBuffer() throws UnsupportedEncodingException, IOException, NoSuchAlgorithmException, NoRepositoryConnectorException {
    byte[] bytes = "These are the test contents.\n".getBytes("UTF-8");
    int count = 120000;
    MessageDigest digest = MessageDigest.getInstance("SHA-1");
    for (int i = 0; i < count; i++) {
        digest.update(bytes);
    }
    byte[] hash = digest.digest();
    File file = TestFileUtils.createTempFile(bytes, count);
    Artifact artifact = new StubArtifact("gid:aid:ext:ver");
    ArtifactUpload upA = new ArtifactUpload(artifact, file);
    File dir = TestFileUtils.createTempDir("con-test");
    File downAFile = new File(dir, "downA.file");
    downAFile.deleteOnExit();
    ArtifactDownload downA = new ArtifactDownload(artifact, "", downAFile, RepositoryPolicy.CHECKSUM_POLICY_FAIL);
    Metadata metadata = new StubMetadata("gid", "aid", "ver", "maven-metadata.xml", Metadata.Nature.RELEASE_OR_SNAPSHOT);
    MetadataUpload upM = new MetadataUpload(metadata, file);
    File downMFile = new File(dir, "downM.file");
    downMFile.deleteOnExit();
    MetadataDownload downM = new MetadataDownload(metadata, "", downMFile, RepositoryPolicy.CHECKSUM_POLICY_FAIL);
    DigestingTransferListener listener = new DigestingTransferListener();
    session.setTransferListener(listener);
    RepositoryConnector connector = factory().newInstance(session, repository);
    connector.put(Arrays.asList(upA), null);
    assertArrayEquals(hash, listener.getHash());
    listener.rewind();
    connector.put(null, Arrays.asList(upM));
    assertArrayEquals(hash, listener.getHash());
    listener.rewind();
    connector.get(Arrays.asList(downA), null);
    assertArrayEquals(hash, listener.getHash());
    listener.rewind();
    connector.get(null, Arrays.asList(downM));
    assertArrayEquals(hash, listener.getHash());
    listener.rewind();
    connector.close();
}
Also used : ArtifactUpload(org.sonatype.aether.spi.connector.ArtifactUpload) StubMetadata(org.sonatype.aether.test.util.impl.StubMetadata) Metadata(org.sonatype.aether.metadata.Metadata) MetadataUpload(org.sonatype.aether.spi.connector.MetadataUpload) StubArtifact(org.sonatype.aether.test.util.impl.StubArtifact) Artifact(org.sonatype.aether.artifact.Artifact) ArtifactDownload(org.sonatype.aether.spi.connector.ArtifactDownload) StubArtifact(org.sonatype.aether.test.util.impl.StubArtifact) StubMetadata(org.sonatype.aether.test.util.impl.StubMetadata) RepositoryConnector(org.sonatype.aether.spi.connector.RepositoryConnector) MetadataDownload(org.sonatype.aether.spi.connector.MetadataDownload) MessageDigest(java.security.MessageDigest) File(java.io.File) Test(org.junit.Test)

Aggregations

MetadataUpload (org.sonatype.aether.spi.connector.MetadataUpload)18 ArtifactUpload (org.sonatype.aether.spi.connector.ArtifactUpload)13 Test (org.junit.Test)10 MetadataDownload (org.sonatype.aether.spi.connector.MetadataDownload)9 File (java.io.File)8 Metadata (org.sonatype.aether.metadata.Metadata)8 ArtifactDownload (org.sonatype.aether.spi.connector.ArtifactDownload)8 RepositoryConnector (org.sonatype.aether.spi.connector.RepositoryConnector)8 StubArtifact (org.sonatype.aether.test.util.impl.StubArtifact)5 StubMetadata (org.sonatype.aether.test.util.impl.StubMetadata)5 Artifact (org.sonatype.aether.artifact.Artifact)4 ArrayList (java.util.ArrayList)3 LinkedList (java.util.LinkedList)2 RecordingTransferListener (org.sonatype.aether.test.impl.RecordingTransferListener)2 TransferEvent (org.sonatype.aether.transfer.TransferEvent)2 IOException (java.io.IOException)1 MessageDigest (java.security.MessageDigest)1 IdentityHashMap (java.util.IdentityHashMap)1 CountDownLatch (java.util.concurrent.CountDownLatch)1 Ignore (org.junit.Ignore)1