目錄前言數(shù)據(jù)庫(kù)持久化支持Swagger 文檔支持參數(shù)校驗(yàn) JSR303跨域解決整合MongoDB實(shí)現(xiàn)文件上傳下載刪除前言

如果根據(jù)之前做的 Nest.js 后端項(xiàng)目功能為標(biāo)準(zhǔn)的話(huà)，那么 Spring Boot 項(xiàng)目需要幾種功能進(jìn)行整合，好在生態(tài)豐富，集成也不算困難。所以打算根據(jù)之前的項(xiàng)目使用 Spring Boot 重寫(xiě)個(gè)新的項(xiàng)目：

Restful API CRUD 功能實(shí)現(xiàn) 數(shù)據(jù)庫(kù)對(duì)象關(guān)系映射功能持久化支持 OpenAPI 文檔支持 參數(shù)校驗(yàn)判斷業(yè)務(wù) redis 緩存 ... 數(shù)據(jù)庫(kù)持久化支持

目前數(shù)據(jù)庫(kù)持久化主要是 Spring Boot Jpa 和 Spring Boot Mybatis 。如果看過(guò) JPA 的業(yè)務(wù)過(guò)程會(huì)發(fā)現(xiàn)和 Nodejs 中的 TypeORM 及其相似。Mybatis 主要可以靈活調(diào)試動(dòng)態(tài) Sql 。不管怎么說(shuō)根據(jù)自己項(xiàng)目業(yè)務(wù)需求選定其中功能吧。

安裝 MyBatis 教程可以官方文檔查閱：mybatis-spring-boot-autoconfigure

Swagger 文檔支持

集成 Swagger UI 文檔支持也非常簡(jiǎn)單，生態(tài)中的 springfox 做的不錯(cuò)，添加依賴(lài)：

<dependency> <groupId>io.springfox</groupId> <artifactId>springfox-boot-starter</artifactId> <version>3.0.0</version></dependency>

這里需要指定版本，不指定拉取依賴(lài)會(huì)報(bào)錯(cuò)。

然后在啟動(dòng)方法添加注解：

@EnableOpenApipublic class YasuoApplication {public static void main(String[] args) { // ...}}

然后在 Controller 類(lèi)上添加標(biāo)識(shí)：

@Api(value = 'global', tags = '全局接口')@RestController@RequestMapping('/')public class AppController {}

在然后在方法里添加詳細(xì)信息：

@Api(value = 'global', tags = '全局接口')@RestController@RequestMapping('/')public class AppController { UserService userService; @ApiOperation(value = '用戶(hù)登錄', notes = '系統(tǒng)用戶(hù)登錄') @PostMapping('login') public JSONObject login(@RequestParam('username') String username, @RequestParam('password') String password) {System.out.println(username);System.out.println(password);JSONObject info = new JSONObject();return info; }}

啟動(dòng)項(xiàng)目訪(fǎng)問(wèn)：http://localhost:8080/swagger-ui 即可訪(fǎng)問(wèn)。值得注意是如果你在 application 添加 server.servlet.contextPath 選項(xiàng)的時(shí)候記得添加對(duì)應(yīng)的字段。

參數(shù)校驗(yàn) JSR303

從 springboot-2.3 開(kāi)始，校驗(yàn)包被獨(dú)立成了一個(gè) starter 組件：

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-validation</artifactId></dependency>

比如在 DTO 類(lèi)里：

package com.iiong.yasuo.dto;import lombok.Data;import javax.validation.constraints.NotEmpty;/** * Author: Jaxson * Description: 請(qǐng)求用戶(hù)登錄參數(shù) * Date: 2021-05-26 */@Datapublic class UserLoginRequestDTO { @NotEmpty(message = '登錄名稱(chēng)不得為空！') private String username; @NotEmpty(message = '登錄密碼不得為空！') private String password;}

內(nèi)置的校驗(yàn)注解可以查看官方文檔，然后進(jìn)行參數(shù)校驗(yàn)：

@ApiOperation(value = '用戶(hù)登錄', notes = '系統(tǒng)用戶(hù)登錄')@PostMapping('login')public RestfulModel<UserLoginResponseDTO> login(@RequestBody @Validated UserLoginRequestDTO userLoginRequestDTO) { System.out.println(userLoginRequestDTO); UserLoginResponseDTO userLoginResponseDTO = new UserLoginResponseDTO(); userLoginResponseDTO.setId(1013401346173L); userLoginResponseDTO.setLoginName('112233'); userLoginResponseDTO.setName('系統(tǒng)管理員'); userLoginResponseDTO.setToken('test'); return new RestfulModel<>(0, '登錄成功！', userLoginResponseDTO);}

不過(guò)默認(rèn)返回的異常信息并不是很友好，需要再次簡(jiǎn)化，所以需要做個(gè)全局異常處理。如果需要可以使用 @RestControllerAdvice 注解來(lái)表示全局處理類(lèi)：

/** * Author: Jaxson * Description: 全局異常處理類(lèi) * Date: 2021-05-26 */@ControllerAdvicepublic class ExceptionHandlerConfig { /** * 統(tǒng)一處理參數(shù)校驗(yàn)異常 * @param bindException 捕捉到的異常 * @return 返回?cái)?shù)據(jù) */ @ExceptionHandler(value = BindException.class) @ResponseBody public RestfulModel<Object> validExceptionHandler(BindException bindException) {String exceptionMsg = bindException.getBindingResult().getAllErrors().get(0).getDefaultMessage();return new RestfulModel<>(1000, exceptionMsg, null); }}

當(dāng)然這里面還可以處理一些系統(tǒng)級(jí)別的異常，自己拋出即可。

跨域解決

解決跨域問(wèn)題也很簡(jiǎn)單，只需要實(shí)現(xiàn)接口 WebMvcConfigurer 重寫(xiě)方法即可：

/** * Author: Jaxson * Description: 運(yùn)行跨域 * Date: 2021-05-26 */@Configurationpublic class WebMvcConfig implements WebMvcConfigurer { @Override public void addCorsMappings(CorsRegistry corsRegistry) {corsRegistry.addMapping('/**').allowedOriginPatterns('*').allowedHeaders(CorsConfiguration.ALL).allowedMethods(CorsConfiguration.ALL).allowCredentials(true).maxAge(3600); // 1小時(shí)內(nèi)不需要再預(yù)檢（發(fā)OPTIONS請(qǐng)求） }}整合MongoDB實(shí)現(xiàn)文件上傳下載刪除

引入pom依賴(lài)

<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-mongodb</artifactId></dependency>

配置yml

spring: data: mongodb: host: *.*.*.* username: *** password: *** database: *** port: 27017 # 設(shè)置文件上傳的大小限制 servlet: multipart: max-file-size: 10MB max-request-size: 50MB

上傳下載刪除

/** * @author Mr.Horse * @version 1.0 * @description: MongoDB的文件上傳、下載、刪除等基本操作(集合HuTool工具庫(kù)) * @date 2021/4/29 9:53 */@Validated@Controller@RequestMapping('/mongo')public class MongoUploadController { private static Logger logger = LoggerFactory.getLogger(MongoUploadController.class); @Autowired private GridFsTemplate gridFsTemplate; @Autowired private MongoTemplate mongoTemplate; private static final List<String> CONTENT_TYPES = Arrays.asList('image/gif', 'image/jpeg', 'image/jpg', 'image/png'); /** * MongoDB文件上傳(圖片上傳) * * @param file * @return */ @PostMapping('/upload') public ResponseEntity<String> fileUpload(@RequestParam('file') MultipartFile file) {try { // 校驗(yàn)文件信息(文件類(lèi)型,文件內(nèi)容) String originalFilename = file.getOriginalFilename(); if (StrUtil.isBlank(originalFilename)) {return ResponseEntity.badRequest().body('參數(shù)錯(cuò)誤'); } String contentType = file.getContentType(); if (!CONTENT_TYPES.contains(contentType)) {return ResponseEntity.badRequest().body('文件類(lèi)型錯(cuò)誤'); } InputStream inputStream = file.getInputStream(); BufferedImage bufferedImage = ImageIO.read(inputStream); if (ObjectUtil.isEmpty(bufferedImage)) {return ResponseEntity.badRequest().body('文件內(nèi)容錯(cuò)誤'); } // 文件重命名 String suffix = FileNameUtil.getSuffix(originalFilename); String fileName = IdUtil.simpleUUID().concat('.').concat(suffix); // 文件上傳,返回ObjectId ObjectId objectId = gridFsTemplate.store(inputStream, fileName, contentType); return StrUtil.isBlank(String.valueOf(objectId)) ? ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body('文件上傳失敗') : ResponseEntity.ok(String.valueOf(objectId));} catch (IOException e) { return ResponseEntity.badRequest().body('文件上傳異常');} } /** * 根據(jù)ObjectId讀取文件并寫(xiě)入響應(yīng)流,頁(yè)面進(jìn)行進(jìn)行相關(guān)操作,可以進(jìn)行文件的下載和展示 * * @param objectId */ @GetMapping('/read') public void queryFileByObjectId(@RequestParam('objectId') @NotBlank(message = 'ObjectId不能為空') String objectId, HttpServletResponse response) {// 根據(jù)objectId查詢(xún)文件GridFSFile file = gridFsTemplate.findOne(new Query(Criteria.where('_id').is(objectId)));// 創(chuàng)建一個(gè)文件桶GridFSBucket gridFsBucket = GridFSBuckets.create(mongoTemplate.getDb());InputStream inputStream = null;OutputStream outputStream = null;try { if (ObjectUtil.isNotNull(file)) {// 打開(kāi)下載流對(duì)象GridFSDownloadStream fileStream = gridFsBucket.openDownloadStream(file.getObjectId());// 創(chuàng)建girdFsResource，傳入下載流對(duì)象，獲取流對(duì)象GridFsResource gridFsResource = new GridFsResource(file, fileStream);// 寫(xiě)入輸出流inputStream = gridFsResource.getInputStream();outputStream = response.getOutputStream();byte[] bytes = new byte[1024];if (inputStream.read(bytes) != -1) { outputStream.write(bytes);} }} catch (IOException e) { logger.error('文件讀取異常: {}', e.getMessage());} finally { IoUtil.close(outputStream); IoUtil.close(inputStream);} } /** * 根據(jù)ObjectId刪除文件 * * @param objectId * @return */ @DeleteMapping('/remove') public ResponseEntity<String> removeFileByObjectId(@RequestParam('objectId') @NotBlank(message = 'ObjectId不能為空') String objectId) {gridFsTemplate.delete(new Query(Criteria.where('_id').is(objectId)));return ResponseEntity.ok('刪除成功'); }}

如果需要實(shí)現(xiàn)在瀏覽器頁(yè)面下載此資源的功能，可結(jié)合js進(jìn)行操作（文件類(lèi)型根據(jù)具體業(yè)務(wù)需求而定）。主要實(shí)現(xiàn)代碼如下所示：

downloadNotes(noteId) { axios({url: this.BASE_API + ’/admin/mongo/file/query/’ + noteId,method: ’get’,responseType: ’arraybuffer’,params: { type: ’download’ } }).then(res => {// type類(lèi)型可以設(shè)置為文本類(lèi)型，這里是pdf類(lèi)型const pdfUrl = window.URL.createObjectURL(new Blob([res.data], { type: `application/pdf` }))const fname = noteId // 下載文件的名字const link = document.createElement(’a’)link.href = pdfUrllink.setAttribute(’download’, fname)document.body.appendChild(link)link.click()URL.revokeObjectURL(pdfUrl) // 釋放URL 對(duì)象 }) }

以上就是Spring Boot 功能整合的實(shí)現(xiàn)的詳細(xì)內(nèi)容，更多關(guān)于Spring Boot 功能整合的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！