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;
|
|
/**
|
* 业务附件表 服务实现类
|
*
|
* @author BladeX
|
* @since 2024-11-11
|
*/
|
@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;
|
}
|
}
|
