SpringBoot實現(xiàn)文件下載的限速功能
前言
在文件下載過程中,如果不加以控制,可能會導致服務器帶寬被單個或少數(shù)用戶占用,影響其他用戶的訪問體驗。通過實現(xiàn)文件下載的限速,可以平衡帶寬資源的使用,確保所有用戶都有良好的下載體驗。
實現(xiàn)思路
為了實現(xiàn)文件下載的限速,我們需要以下幾個關鍵步驟:
- 創(chuàng)建一個工具類,用于限制下載速率。
- 在控制器中使用該工具類處理文件下載請求。
- 使用
StreamingResponseBody
實現(xiàn)流式響應,確保大文件可以逐步傳輸。
代碼實現(xiàn)
步驟1:創(chuàng)建限速工具類
首先,我們創(chuàng)建一個限速工具類RateLimiter
,該類包含一個方法limitDownloadSpeed
,用于限制下載速率。
import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; /** * @desc: 文件工具類 * @author: shy * @date: 2024/06/28 11:27 */ public class FileUtil { private static final int BUFFER_SIZE = 1024; /** * 文件下載限速 * * @param in 輸入流 * @param out 輸出流 * @param bytesPerSecond 每秒允許下載的字節(jié)數(shù) * @throws IOException */ public static void limitDownloadSpeed(InputStream in, OutputStream out, int bytesPerSecond) throws IOException { byte[] buffer = new byte[BUFFER_SIZE]; int bytesRead; long bytesSent = 0; long startTime = System.currentTimeMillis(); try { while ((bytesRead = in.read(buffer)) != -1) { // 將數(shù)據(jù)寫入輸出流 out.write(buffer, 0, bytesRead); bytesSent += bytesRead; if (bytesSent >= bytesPerSecond) { long elapsedTime = System.currentTimeMillis() - startTime; if (elapsedTime < 1000) { // 如果時間少于1秒,則休眠剩余時間 Thread.sleep(1000 - elapsedTime); } // 重置已發(fā)送字節(jié)計數(shù)和開始時間 bytesSent = 0; startTime = System.currentTimeMillis(); } } } catch (InterruptedException e) { // 恢復線程的中斷狀態(tài) Thread.currentThread().interrupt(); throw new IOException("Thread was interrupted", e); } } }
解釋:
BUFFER_SIZE
:定義緩沖區(qū)大小。limitDownloadSpeed
:通過try-with-resources
管理InputStream
,根據(jù)設定的速率讀取數(shù)據(jù)并寫入輸出流,控制傳輸速率。
步驟2:修改文件下載控制器
接下來,我們在控制器中使用StreamingResponseBody
來實現(xiàn)文件下載,并調用限速工具類的方法。
import java.io.File; import java.io.InputStream; import java.nio.file.Files; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody; import com.shy.admin.common.annotation.WithoutLogin; import com.shy.common.utils.FileUtil; /** * @desc: 文件下載Controller * @author: shy * @date: 2024/06/28 10:48 */ @RestController @RequestMapping("/file") public class FileController { // 每秒允許下載的字節(jié)數(shù)(例如100KB/s) private static final int BYTES_PER_SECOND = 1024 * 100; @WithoutLogin @GetMapping("/download/{filename}") public ResponseEntity<StreamingResponseBody> downloadFile(@PathVariable String filename) { // 獲取要下載的文件 File file = new File("D:\\tools\\" + filename); // 使用 StreamingResponseBody 實現(xiàn)流式響應體 StreamingResponseBody responseBody = outputStream -> { try (InputStream inputStream = Files.newInputStream(file.toPath())) { // 調用限速方法 FileUtil.limitDownloadSpeed(inputStream, outputStream, BYTES_PER_SECOND); } }; // 返回 ResponseEntity,包含響應頭和流式響應體 return ResponseEntity.ok() .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + file.getName()) .contentType(MediaType.APPLICATION_OCTET_STREAM) .contentLength(file.length()) .body(responseBody); } }
解釋:
StreamingResponseBody
:實現(xiàn)流式響應體,用于處理大文件的逐步傳輸。responseBody
:通過lambda表達式實現(xiàn)StreamingResponseBody
的writeTo
方法,在方法中使用try-with-resources
管理InputStream
,并調用RateLimiter
的方法實現(xiàn)限速。
工作流程
- 請求處理:當客戶端發(fā)送下載請求時,Spring 調用控制器方法
downloadFile
。 - 創(chuàng)建
StreamingResponseBody
:控制器方法創(chuàng)建StreamingResponseBody
實例。 - 返回
ResponseEntity
:控制器方法返回包含StreamingResponseBody
的ResponseEntity
,并設置適當?shù)捻憫^(如Content-Disposition
和Content-Type
)。 - 調用
writeTo
方法:Spring 在準備向客戶端發(fā)送響應時,調用StreamingResponseBody
的writeTo
方法,并傳入與客戶端連接的OutputStream
。 - 寫入數(shù)據(jù):
writeTo
方法中,從文件輸入流讀取數(shù)據(jù),并通過RateLimiter
方法將數(shù)據(jù)寫入OutputStream
,同時控制傳輸速率。
總結
通過以上步驟,我們成功在SpringBoot項目中實現(xiàn)了文件下載的限速功能。核心思路是通過一個限速工具類控制數(shù)據(jù)傳輸速率,并使用StreamingResponseBody
實現(xiàn)流式響應,確保大文件可以逐步傳輸。這種設計既能有效控制帶寬資源的使用,又能提供良好的用戶下載體驗。
以上就是SpringBoot實現(xiàn)文件下載的限速功能的詳細內容,更多關于SpringBoot文件下載限速的資料請關注腳本之家其它相關文章!
相關文章
SpringBoot項目的配置文件中設置server.port不生效問題
這篇文章主要介紹了SpringBoot項目的配置文件中設置server.port不生效問題,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-11-11JavaWeb實現(xiàn)學生信息管理系統(tǒng)(2)
這篇文章主要介紹了JavaWeb實現(xiàn)學生信息管理系統(tǒng)的第二篇,實現(xiàn)學生管理系統(tǒng)的查找和添加功能,文中示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下2021-08-08springboot時間格式化的五種方法總結(解決后端傳給前端的時間顯示不一致)
這篇文章主要給大家介紹了關于springboot時間格式化的五種方法,文中介紹的方法解決了后端傳給前端的時間顯示不一致,文中通過圖文以及代碼介紹的非常詳細,需要的朋友可以參考下2024-01-01Java生成日期時間存入Mysql數(shù)據(jù)庫的實現(xiàn)方法
本文主要介紹了Java生成日期時間存入Mysql數(shù)據(jù)庫的實現(xiàn)方法,文中通過示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下2022-03-03