pass segmentStrategy

Author: rwv

src/get/manager/DownloadManger.ts

@@ -24,7 +24,7 @@ export class DownloadManger {
     requestConfig.logger.info(`Workers Count: ${this.workers.length}`)
 
     const contentLength = getContentLength(metadata.headers)
-    this.rangeProvider = new RangeProvider(contentLength)
+    this.rangeProvider = new RangeProvider(contentLength, this.requestConfig.segmentStrategy)
     requestConfig.logger.info(`Content length: ${contentLength}`)
 
     this.streamer = new DownloadStreamer(this.rangeProvider.maxRangeIndex, requestConfig?.init?.signal)

src/get/range/segmentStrategy.ts

@@ -6,14 +6,12 @@ const MB = 1024 * 1024
 function defaultSegmentSize (contentLength: number): number {
   if (contentLength <= 512 * KB) {
     return contentLength
-  } else if (contentLength <= 4 * MB) {
-    return 128 * KB
-  } else if (contentLength <= 16 * MB) {
-    return 256 * KB
+  } else if (contentLength <= 8 * MB) {
+    return Math.ceil(Math.max(512 * KB, (contentLength / 8)))
   } else if (contentLength <= 32 * MB) {
-    return 512 * KB
+    return Math.ceil((contentLength / 16))
   } else {
-    return 1 * MB
+    return 2 * MB
   }
 }
 

src/get/request/RequestConfig.ts

@@ -41,4 +41,8 @@ export class GETRequestConfig {
   get chunkCallback (): ChunkCallback {
     return this.fastFetchConfig?.chunkCallback ?? (() => undefined)
   }
+
+  get segmentStrategy (): ((contentLength: number) => Array<[number, number]>) | undefined {
+    return this.fastFetchConfig?.segmentStrategy
+  }
 }

src/get/request/types.ts

@@ -10,6 +10,7 @@ export interface FastFetchGetConfig {
   retryDelay?: number
   logger?: Partial<Logger>
   chunkCallback?: ChunkCallback
+  segmentStrategy?: (contentLength: number) => Array<[number, number]>
 }
 
 interface FastFetchGETInit {