新增附件信息导入，新增应急物资查询接口

zhongrj

2024-11-12 52a29d9515a4e9cec9f5f975cb7566adca12540e

 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;
   }
}