package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.whyc.constant.UserOperation;
|
import com.whyc.dto.FileDirPath;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.SoftcodeMapper;
|
import com.whyc.mapper.SoftwareMapper;
|
import com.whyc.pojo.Softcode;
|
import com.whyc.pojo.Software;
|
import com.whyc.util.ActionUtil;
|
import com.whyc.util.CommonUtil;
|
import com.whyc.util.DateUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLEncoder;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
@Service
|
public class SoftcodeService {
|
|
@Autowired(required = false)
|
private SoftcodeMapper mapper;
|
|
@Autowired
|
private DocLogService logService;
|
|
|
//源码上传
|
public Response uploadCode(MultipartFile file, String fileNames) throws IOException {
|
String[] nameList=fileNames.split(",");
|
String codeName = file.getOriginalFilename();
|
Date date = new Date();
|
String dateUnion = DateUtil.YYYY_MM_DD_HH_MM_SS_UNION.format(date);
|
//存储路径
|
String rootFile = CommonUtil.getRootFile();
|
String softcodeDir = rootFile + "software" + File.separator +"code"+ File.separator + dateUnion;
|
String softcodeHttpUrl = softcodeDir.substring(softcodeDir.lastIndexOf("doc_file"+ File.separator + "software"));
|
File softwareDirFile = new File(softcodeDir);
|
if(!softwareDirFile.exists()){
|
softwareDirFile.mkdirs();
|
}
|
String codeUrl=softcodeDir+File.separator+codeName;
|
file.transferTo(new File(codeUrl));
|
if(nameList.length>0){
|
for(String name:nameList){
|
Softcode softcode=new Softcode();
|
softcode.setFileName(name);
|
softcode.setCreateTime(new Date());
|
softcode.setCodeUrl(softcodeHttpUrl+File.separator+codeName);
|
softcode.setCodeName(codeName.substring(0,codeName.lastIndexOf(".")));
|
//判断是否存在源码，存在修改，不存在添加
|
if(mapper.selectOne(new QueryWrapper<Softcode>().eq("file_name",name))!=null){
|
mapper.update(softcode,new UpdateWrapper<Softcode>().eq("file_name",name));
|
}else{
|
mapper.insert(softcode);
|
}
|
}
|
}
|
return new Response().set(1,true,"源码上传");
|
}
|
|
|
//根据软件id实现源码下载
|
public void downLoadCode(HttpServletRequest req, HttpServletResponse resp, String fileName) {
|
String fileDirName = FileDirPath.getFileDirName();
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.eq("file_name",fileName);
|
wrapper.last("limit 1");
|
Softcode softcode=mapper.selectOne(wrapper);
|
|
String name=softcode.getCodeUrl().substring(softcode.getCodeUrl().lastIndexOf("\\")+1);
|
try {
|
// 转码防止乱码
|
//resp.addHeader("Content-Disposition", "attachment;filename=" + new String(softwareName.getBytes("UTF-8"), "ISO8859-1"));
|
resp.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode (name, "utf-8"));
|
OutputStream out = resp.getOutputStream();
|
FileInputStream in = new FileInputStream(fileDirName+ File.separator+softcode.getCodeUrl());
|
int len=0;
|
byte[] buffer =new byte[1024];
|
//7. 将缓冲区中的数据输出
|
while ((len=in.read(buffer))>0){
|
out.write(buffer,0,len);
|
}
|
in.close();
|
out.close();
|
} catch (FileNotFoundException | UnsupportedEncodingException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
//记录日志
|
logService.recordOperationLogDownLoad(ActionUtil.getUser().getId(),ActionUtil.getUser().getName(), UserOperation.TYPE_DOWNLOAD_SOFWARE.getType(),new Date(),req.getRemoteAddr()
|
,name,fileDirName+File.separator+softcode.getCodeUrl(),"",name,"");
|
}
|
}
|
