/* * Copyright (c) 2018-2025, ztt All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the pig4cloud.com developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: ztt */ package com.chinaztt.mes.basic.service.impl; import cn.hutool.core.collection.CollectionUtil; import cn.hutool.core.io.FileUtil; import cn.hutool.core.io.IoUtil; import cn.hutool.core.util.IdUtil; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.chinaztt.mes.basic.entity.Warehouse; import com.chinaztt.mes.basic.excel.WarehouseData; import com.chinaztt.mes.basic.mapper.FactoryMapper; import com.chinaztt.mes.basic.mapper.WarehouseMapper; import com.chinaztt.mes.basic.service.WarehouseService; import com.chinaztt.mes.basic.util.DictUtils; import com.chinaztt.ztt.admin.api.entity.SysDictItem; import com.chinaztt.ztt.admin.api.feign.RemoteDictService; import com.chinaztt.ztt.common.core.util.R; import com.chinaztt.ztt.common.oss.OssProperties; import com.chinaztt.ztt.common.oss.service.OssTemplate; import lombok.AllArgsConstructor; import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.InputStream; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Objects; /** * 仓库基础数据 * * @author sunxl * @date 2020-09-18 09:32:01 */ @Service @Slf4j @AllArgsConstructor public class WarehouseServiceImpl extends ServiceImpl implements WarehouseService { private final OssProperties ossProperties; private final OssTemplate minioTemplate; private FactoryMapper factoryMapper; private DictUtils dictUtils; private RemoteDictService remoteDictService; /** * Description: 分页 * * @param page * @param warehouse * @return IPage> * @author: fenglang * @date: 2020/8/18 11:06 */ @Override public IPage> getWarehouse(Page page, QueryWrapper warehouse) { return baseMapper.getWarehouse(page,warehouse); } @Override public void deleteWarehouse(Long id) { // 2.删除选中的仓库 baseMapper.deleteById(id); } @Override public R uploadFile(MultipartFile file) { String fileName = IdUtil.simpleUUID() + StrUtil.DOT + FileUtil.extName(file.getOriginalFilename()); Map resultMap = new HashMap<>(4); resultMap.put("bucketName", ossProperties.getBucketName()); resultMap.put("fileName", fileName); resultMap.put("url", String.format("/admin/sys-file/%s/%s", ossProperties.getBucketName(), fileName)); try { minioTemplate.putObject(ossProperties.getBucketName(), fileName, file.getInputStream()); // 文件管理数据记录,收集管理追踪文件 } catch (Exception e) { log.error("上传失败", e); return R.failed(e.getLocalizedMessage()); } return R.ok(resultMap); } @Override @SneakyThrows public R deleteFile(String fileName) { minioTemplate.removeObject(ossProperties.getBucketName(), fileName); return R.ok(); } @Override public void getFile(String bucket, String url, HttpServletResponse response) { try (InputStream inputStream = minioTemplate.getObject(bucket, url)) { response.setContentType("application/octet-stream; charset=UTF-8"); IoUtil.copy(inputStream, response.getOutputStream()); } catch (Exception e) { log.error("文件读取异常: {}", e.getLocalizedMessage()); } } @Override public R deleteByFile(Long id) { baseMapper.deleteWarehouse(id); return R.ok(); } @Override @Transactional(rollbackFor = Exception.class) public void importWarehouseExcel(List list) { if (CollectionUtil.isEmpty(list)) { return; } List dict = dictUtils.getDict("storehouse_type"); for (WarehouseData data :list){ QueryWrapper wrapper = new QueryWrapper<>(); wrapper.lambda().eq(Warehouse::getWarehouseName,data.getWarehouseName()); Warehouse warehouse1 = baseMapper.selectOne(wrapper); if(!Objects.isNull(warehouse1)){ continue; } Warehouse warehouse = new Warehouse(); dict.forEach(a -> { if (a.getLabel().equals(data.getType())){ warehouse.setWarehouseType(a.getValue()); } }); if("启用".equals(data.getState())){ warehouse.setWarehouseStatus(1); }else{ warehouse.setWarehouseStatus(0); } warehouse.setWarehouseNo(data.getWarehouseNo()); warehouse.setWarehouseName(data.getWarehouseName()); //warehouse.setWarehouseStatus(1); baseMapper.insert(warehouse); } } }