/*
|
* 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.amazonaws.services.s3.model.Bucket;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.chinaztt.mes.basic.entity.Company;
|
import com.chinaztt.mes.basic.entity.CompanyAddress;
|
import com.chinaztt.mes.basic.entity.CompanyContact;
|
import com.chinaztt.mes.basic.mapper.CompanyAddressMapper;
|
import com.chinaztt.mes.basic.mapper.CompanyContactMapper;
|
import com.chinaztt.mes.basic.mapper.CompanyMapper;
|
import com.chinaztt.mes.basic.service.CompanyService;
|
import com.chinaztt.mes.common.numgen.NumberGenerator;
|
import com.chinaztt.ztt.common.oss.OssProperties;
|
import com.chinaztt.ztt.common.oss.service.OssTemplate;
|
import lombok.AllArgsConstructor;
|
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.HashSet;
|
import java.util.List;
|
import java.util.Optional;
|
import java.util.Set;
|
|
/**
|
* 公司
|
*
|
* @author liuth
|
* @date 2020-09-18 15:02:11
|
*/
|
@Service
|
@AllArgsConstructor
|
public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> implements CompanyService {
|
|
private final OssProperties ossProperties;
|
private final OssTemplate minioTemplate;
|
private CompanyAddressMapper companyAddressMapper;
|
private CompanyContactMapper companyContactMapper;
|
private CompanyMapper companyMapper;
|
private NumberGenerator<Company> numberGenerator;
|
|
/**
|
* 关联删除
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean fullDelete(Long id) {
|
companyAddressMapper.delete(Wrappers.<CompanyAddress>lambdaQuery().eq(CompanyAddress::getCompanyId, id));
|
companyContactMapper.delete(Wrappers.<CompanyContact>lambdaQuery().eq(CompanyContact::getCompanyId, id));
|
baseMapper.deleteById(id);
|
return true;
|
}
|
|
@Override
|
public Company getFullById(Long id) {
|
return companyMapper.getFullById(id);
|
}
|
|
|
@Override
|
public Company getByStaffId(Long staffId) {
|
return companyMapper.getByStaffId(staffId);
|
}
|
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean fullSave(Company company) {
|
companyMapper.insert(company);
|
if (CollectionUtil.isNotEmpty(company.getAddresses())) {
|
company.getAddresses().forEach(a -> {
|
a.setCompanyId(company.getId());
|
companyAddressMapper.insert(a);
|
});
|
}
|
if (CollectionUtil.isNotEmpty(company.getContacts())) {
|
company.getContacts().forEach(c -> {
|
c.setCompanyId(company.getId());
|
companyContactMapper.insert(c);
|
});
|
}
|
return true;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean fullUpdate(Company company) {
|
Set<Long> addressIds = new HashSet();
|
Set<Long> contactIds = new HashSet();
|
companyMapper.updateById(company);
|
if (CollectionUtil.isNotEmpty(company.getAddresses())) {
|
company.getAddresses().forEach(a -> {
|
if (a.getId() == null || a.getId() == 0) {
|
a.setCompanyId(company.getId());
|
companyAddressMapper.insert(a);
|
} else {
|
companyAddressMapper.updateById(a);
|
}
|
addressIds.add(a.getId());
|
});
|
}
|
if (CollectionUtil.isNotEmpty(company.getContacts())) {
|
company.getContacts().forEach(c -> {
|
if (c.getId() == null || c.getId() == 0) {
|
c.setCompanyId(company.getId());
|
companyContactMapper.insert(c);
|
} else {
|
companyContactMapper.updateById(c);
|
}
|
contactIds.add(c.getId());
|
});
|
}
|
LambdaQueryWrapper<CompanyAddress> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(CompanyAddress::getCompanyId, company.getId());
|
if (CollectionUtil.isNotEmpty(addressIds)) {
|
wrapper.notIn(CompanyAddress::getId, addressIds);
|
}
|
LambdaQueryWrapper<CompanyContact> wrapper2 = new LambdaQueryWrapper<>();
|
wrapper2.eq(CompanyContact::getCompanyId, company.getId());
|
if (CollectionUtil.isNotEmpty(contactIds)) {
|
wrapper2.notIn(CompanyContact::getId, contactIds);
|
}
|
companyAddressMapper.delete(wrapper);
|
companyContactMapper.delete(wrapper2);
|
return true;
|
}
|
|
@Override
|
public int deleteContactById(Long id) {
|
return companyContactMapper.deleteById(id);
|
}
|
|
@Override
|
public int deleteAddressById(Long id) {
|
return companyAddressMapper.deleteById(id);
|
}
|
|
@Override
|
public void clone(Company[] companies) {
|
for (Company data : companies) {
|
Company company = this.getById(data.getId());
|
List<CompanyAddress> addresses = companyAddressMapper.selectList(Wrappers.<CompanyAddress>lambdaQuery().eq(CompanyAddress::getCompanyId, data.getId()));
|
List<CompanyContact> contacts = companyContactMapper.selectList(Wrappers.<CompanyContact>lambdaQuery().eq(CompanyContact::getCompanyId, data.getId()));
|
|
Company copy = (Company) data.clone();
|
copy.setCompanyNo(numberGenerator.getCopyValueOfUniqueField(company.getCompanyNo(), Company::getCompanyNo));
|
copy.setAddresses(addresses);
|
copy.setContacts(contacts);
|
|
this.fullSave(copy);
|
}
|
}
|
|
/**
|
* 上传文件
|
*
|
* @param file 附件
|
* @param id id
|
* @return
|
*/
|
@Override
|
public boolean uploadFile(MultipartFile file, Long id) {
|
//图片存在mes的桶里面如果没有桶 创建一个
|
Optional<Bucket> optionalBucket = minioTemplate.getBucket("mes");
|
if (optionalBucket.isPresent()) {
|
minioTemplate.createBucket("mes");
|
}
|
Company company = baseMapper.selectById(id);
|
String fileName = IdUtil.simpleUUID() + StrUtil.DOT + FileUtil.extName(file.getOriginalFilename());
|
try {
|
//如果原先存在图片就将原先的图片删掉
|
if (StringUtils.isNotBlank(company.getIconName())) {
|
minioTemplate.removeObject(ossProperties.getBucketName() + "/company", company.getIconName());
|
}
|
//上传新的图片
|
minioTemplate.putObject(ossProperties.getBucketName() + "/company", fileName, file.getInputStream());
|
} catch (Exception e) {
|
throw new RuntimeException("上传失败：" + e.getMessage());
|
}
|
company.setIconName(fileName);
|
baseMapper.updateById(company);
|
return true;
|
}
|
|
/**
|
* 附件删除
|
*
|
* @param id id
|
* @return
|
*/
|
@Override
|
public boolean removeFile(Long id) {
|
//图片存在mes的桶里面如果没有桶 创建一个
|
Company company = baseMapper.selectById(id);
|
if (StringUtils.isBlank(company.getIconName())) {
|
throw new RuntimeException("不存在图片");
|
}
|
try {
|
//如果原先存在图片就将原先的图片删掉
|
minioTemplate.removeObject(ossProperties.getBucketName() + "/company", company.getIconName());
|
} catch (Exception e) {
|
throw new RuntimeException("文件移除失败：" + e.getMessage());
|
}
|
baseMapper.clearIconName(id);
|
return true;
|
}
|
|
/**
|
* 读取文件
|
*
|
* @param id
|
* @param response
|
*/
|
@Override
|
public void getFile(Long id, HttpServletResponse response) {
|
Company company = baseMapper.selectById(id);
|
try (InputStream inputStream = minioTemplate.getObject(ossProperties.getBucketName() + "/company", company.getIconName())) {
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + company.getIconName());
|
IoUtil.copy(inputStream, response.getOutputStream());
|
} catch (Exception e) {
|
throw new RuntimeException("文件读取失败" + e.getMessage());
|
}
|
}
|
}
|
