public void Download(byte[] data, int bufferSize, DownloadSegmentPositions[] segmentPositionInfos) { BeforeDownload(); DownloadStream inputStream = new DownloadStream(data); IList <ISegmentDownloadTask> segmentDownloadTasks = new List <ISegmentDownloadTask>(segmentPositionInfos.Length); IList <DownloadStream> downloadStreams = new List <DownloadStream>(segmentPositionInfos.Length); MemoryStream outputStream = new MemoryStream(); for (int i = 0; i < segmentPositionInfos.Length; i++) { DownloadSegmentPositions segmentPosition = segmentPositionInfos[i]; byte[] dataPart = data.Skip((int)segmentPosition.StartPosition).Take((int)(segmentPosition.EndPosition - segmentPosition.StartPosition + 1)).ToArray(); DownloadStream downloadStream = new DownloadStream(dataPart); segmentDownloadTasks.Add(CreateSegmentDownloadTask(bufferSize, CreateSegmentDownloader(downloadStream, segmentPosition), CreateSegmentWriter(outputStream))); downloadStreams.Add(downloadStream); } SegmentDownloadManager segmentDownloadManager = new SegmentDownloadManager(new SegmentDownloadTaskCollection(segmentDownloadTasks)); segmentDownloadManager.Start(); segmentDownloadManager.Finish(true); AfterDownload(); long totalDownloads = downloadStreams.Sum(x => x.TotalDownloads); Assert.AreEqual(data.Length, totalDownloads); Assert.AreEqual(inputStream.ToArray().Take(data.Length).ToArray(), outputStream.ToArray().Take(data.Length).ToArray()); Assert.AreEqual(inputStream.ToArray(), outputStream.ToArray()); }
/// <summary> /// Starts the download task. /// </summary> public void StartDownload() { RemoteFileInfo remoteFileInfo = null; try { ChangeState(DownloadTaskState.Preparing); m_StartDate = DateTime.Now; INetworkProtocolProvider networkProtocolProvider = m_NetworkProtocolProviderFactory.CreateProvider(m_File.Uri); remoteFileInfo = networkProtocolProvider.GetRemoteFileInfo(m_File); remoteFileInfo.FileName = m_File.FileName; m_FileSize = remoteFileInfo.FileSize; m_IsResumable = remoteFileInfo.AcceptRanges; DownloadSegmentPositions[] segmentPositionInfos; if (remoteFileInfo.AcceptRanges) { segmentPositionInfos = m_DownloadSegmentCalculator.Calculate(m_Settings.MinimumSegmentSize, m_Settings.MaximumSegmentCount, m_File.SegmentCount, remoteFileInfo.FileSize); } else { segmentPositionInfos = new[] { new DownloadSegmentPositions(0, remoteFileInfo.FileSize - 1) }; } using (Stream stream = m_DownloadStreamManager.CreateStream(remoteFileInfo)) { SegmentWriter segmentWriter = new SegmentWriter(stream); SegmentDownloadTasks = new SegmentDownloadTaskCollection(segmentPositionInfos.Length); for (int i = 0; i < segmentPositionInfos.Length; i++) { DownloadSegmentPositions segmentPosition = segmentPositionInfos[i]; SegmentDownloadTasks.Add(new DoubleBufferSegmentDownloadTask(m_Settings.DownloadBufferSize, new SegmentDownloader(m_File, networkProtocolProvider, segmentPosition, new SegmentDownloadRateCalculator(segmentPosition.StartPosition)), segmentWriter)); } SegmentDownloadManager segmentDownloadManager = new SegmentDownloadManager(SegmentDownloadTasks); segmentDownloadManager.Start(); ChangeState(DownloadTaskState.Working); segmentDownloadManager.Finish(true); ChangeState(DownloadTaskState.Ended); m_EventManager.EventPublisher.Publish(new DownloadTaskFinishedEventMessage(this, stream, remoteFileInfo)); } } catch (Exception ex) { if (m_RetryCount < m_Settings.MaximumRetries) { m_RetryCount++; StartDownload(); } else { m_LastException = ex; ChangeState(DownloadTaskState.EndedWithError); m_EventManager.EventPublisher.Publish(new DownloadTaskFinishedEventMessage(this, null, remoteFileInfo)); } } }