View Javadoc
1   package org.eclipse.aether.transport.http;
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.io.ByteArrayOutputStream;
23  import java.nio.Buffer;
24  import java.nio.ByteBuffer;
25  
26  import org.eclipse.aether.spi.connector.transport.TransportListener;
27  import org.eclipse.aether.transfer.TransferCancelledException;
28  
29  class RecordingTransportListener
30      extends TransportListener
31  {
32  
33      public final ByteArrayOutputStream baos = new ByteArrayOutputStream( 1024 );
34  
35      public long dataOffset;
36  
37      public long dataLength;
38  
39      public int startedCount;
40  
41      public int progressedCount;
42  
43      public boolean cancelStart;
44  
45      public boolean cancelProgress;
46  
47      @Override
48      public void transportStarted( long dataOffset, long dataLength )
49          throws TransferCancelledException
50      {
51          startedCount++;
52          progressedCount = 0;
53          this.dataLength = dataLength;
54          this.dataOffset = dataOffset;
55          baos.reset();
56          if ( cancelStart )
57          {
58              throw new TransferCancelledException();
59          }
60      }
61  
62      @Override
63      public void transportProgressed( ByteBuffer data )
64          throws TransferCancelledException
65      {
66          progressedCount++;
67          baos.write( data.array(), data.arrayOffset() + ( (Buffer) data ).position(), data.remaining() );
68          if ( cancelProgress )
69          {
70              throw new TransferCancelledException();
71          }
72      }
73  
74  }