From a1272d78844574b86645149a0d8ca257c2b8fe91 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Thu, 26 Dec 2024 09:43:14 +0800
Subject: [PATCH] 风险源统计等级空的设置为其他

---
 src/main/java/org/springblade/modules/yw/service/impl/YwAttachServiceImpl.java |   46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 46 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/yw/service/impl/YwAttachServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/YwAttachServiceImpl.java
index d457ae0..7b96dfc 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/YwAttachServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/YwAttachServiceImpl.java
@@ -1,12 +1,20 @@
 package org.springblade.modules.yw.service.impl;
 
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springblade.core.oss.model.BladeFile;
+import org.springblade.core.tool.api.R;
+import org.springblade.modules.resource.builder.OssBuilder;
 import org.springblade.modules.yw.entity.YwAttachEntity;
 import org.springblade.modules.yw.vo.YwAttachVO;
 import org.springblade.modules.yw.mapper.YwAttachMapper;
 import org.springblade.modules.yw.service.IYwAttachService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.util.List;
 
 /**
  * 业务附件表 服务实现类
@@ -17,9 +25,47 @@
 @Service
 public class YwAttachServiceImpl extends ServiceImpl<YwAttachMapper, YwAttachEntity> implements IYwAttachService {
 
+
+	/**
+	 * 对象存储构建类
+	 */
+	@Autowired
+	private OssBuilder ossBuilder;
+
 	@Override
 	public IPage<YwAttachVO> selectAttachPage(IPage<YwAttachVO> page, YwAttachVO attach) {
 		return page.setRecords(baseMapper.selectAttachPage(page, attach));
 	}
 
+	/**
+	 * 导入单位附件信息
+	 * @param files
+	 * @param firmId
+	 * @param type
+	 * @return
+	 */
+	@Override
+	public R importAttach(List<MultipartFile> files, Long firmId,Integer type) {
+		// 遍历文件导入
+		for (MultipartFile file : files) {
+			String originalFilename = file.getOriginalFilename();
+			String[] split = originalFilename.split("\\.");
+			// 设置
+			try {
+				// 文件上传
+				BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
+				// 设置信息
+				YwAttachEntity attachEntity = new YwAttachEntity();
+				attachEntity.setName(split[0]);
+				attachEntity.setLink(bladeFile.getLink());
+				attachEntity.setFirmId(firmId);
+				attachEntity.setType(type);
+				// 保存
+				save(attachEntity);
+			} catch (IOException e) {
+				e.printStackTrace();
+			}
+		}
+		return null;
+	}
 }

--
Gitblit v1.9.3