服务器上传文件到oss，以及备份
公司的前端js，css是发布到oss的，所以要求后台在发布的同时先把即将要覆盖的文件备份到服务器上
步骤是先上传文件到服务器，服务器上传到oss服务器之前先备份相同文件名的文件
public class OssVersionPublishController {
@Value("${oss.endpoint}")
public String endpoint;
@Value("${oss.accessKeyId}")
public String accessKeyId;
@Value("${oss.secretAccessKey}")
public String secretAccessKey;
@Value("${oss.bucketName}")
public String bucketName;
@Value("${oss.url}")
public String url;
public OSSClient getOSSClient(){
return new OSSClient(endpoint, accessKeyId, secretAccessKey);
}
/**
* 上传文件到oss
* */
private boolean publishFile(List<OssVersionPublishContent> contentList,String updName){
OSSClient ossClient = getOSSClient();
try {
// 文件的父父目录
String folder =contentList.get(0).getPublishBatch();
// 查找上传的文件是否存在oss？先备份后覆盖，先在本地服务器上创建文件夹（批次命名），先把上一次的删掉了
String deposeFilesDir="/home/admin/ossBackUp/"+folder;
File dest = new File(deposeFilesDir);
if(dest.exists()){
delFile(dest);
}
dest.mkdirs();//本地服务器创建好目录
for(OssVersionPublishContent content:contentList){
// 文件的父目录，判断输入的文件是否有父目录
boolean isExistSup= content.getFileName().contains("/");
boolean read = true;
boolean isDoesObjectExist=true;
try {
isDoesObjectExist=ossClient.doesObjectExist(bucketName,content.getFileName());
} catch (Exception e) {
isDoesObjectExist = false;
logger.warn("isDoesObjectExist异常：{}",e.getMessage());
}
if(isDoesObjectExist){//oss存在相同文件
//文件备份
String filePath = (deposeFilesDir +"/"+ content.getFileName()).replace("\\","/");
File targetFile= new File(filePath);
if(isExistSup){
if (!targetFile.getParentFile().exists()) {
targetFile.getParentFile().mkdirs();
}
}
OSSObject ossObject = ossClient.getObject(bucketName,content.getFileName());
FileOutputStream fos = null;
InputStream stream = null;
try {
fos = new FileOutputStream(targetFile);
stream = ossObject.getObjectContent();
byte[] buffer = new byte[1024*1024];
int byteRead = 0;
while((byteRead=stream.read(buffer))!=-1){
fos.write(buffer, 0, byteRead);
fos.flush();
}
} catch (Exception e) {
read = false;
logger.warn("oss文件备份到本地服务器失败{}",e.getMessage());
return false;
}finally{
try {
if(fos!=null){
fos.close();
}
if(stream!=null){
stream.close();
}
} catch (Exception e) {}
}
}
PutObjectResult result = ossClient.putObject(bucketName, content.getFileName(), new File(content.getFileUrl()));
logger.info("上传结果：{}",result.getETag());
if(read){
content.setOperateTime(new Date());
content.setOperator(updName);
content.setPublishStatus(true);
content.setPublishTime(new Date());
content.setFileStatus(true);
ossVersionPublishContentBiz.updateByPrimaryKey(content);
}
};
} catch (Exception e) {
logger.warn("发布失败={}",e.getMessage());
return false;
}finally{
ossClient.shutdown();
}
return true;
}
/**
* 还原
* */
@PostMapping(value = "/restore/{id}")
@ResponseBody
public ObjectRestResponse<OssVersionPublish> restore(@PathVariable Long id,HttpServletRequest request) {
String token = StringUtils.trimToEmpty(request.getHeader(userAuthConfig.getJwtTokenHeader()));//得到用户token
try {
OssVersionPublish ossVersionPublish = ossVersionPublishBiz.selectByPrimaryKey(id);
if(StringUtil.isBlank(ossVersionPublish.getPublishBatch())){
return new ObjectRestResponse<OssVersionPublish>(1, "获取参数异常");
}
// 修改发布版本批次的状态为未发布
String updName = userAuthUtil.getInfoFromToken(token).getUniqueName();
// 找到指定的目录下的所有文件，然后覆盖到oss相应的地方
String deposeFilesDir="/home/admin/ossBackUp/"+ossVersionPublish.getPublishBatch();
File dest = new File(deposeFilesDir);
if(!dest.exists()){
return new ObjectRestResponse<OssVersionPublish>(1, "备份文件丢失");
}
boolean flag = restoreToOss(dest,ossVersionPublish.getPublishBatch());
if(flag){
ossVersionPublish.setOperator(updName);
ossVersionPublish.setOperateTime(new Date());
ossVersionPublish.setPublishStatus(false);
int resCount = ossVersionPublishBiz.updateByPrimaryKey(ossVersionPublish);
if(resCount>0){
return new ObjectRestResponse<OssVersionPublish>(0, "还原成功");
}
}
} catch (Exception e) {
logger.warn("oss发布失败{}",e.getMessage());
return new ObjectRestResponse<OssVersionPublish>(1, "还原失败");
}
return new ObjectRestResponse<OssVersionPublish>(1, "还原失败");
}
/**
* 还原备份到oss
* @param publishBatch
* */
private boolean restoreToOss(File dest, String publishBatch){
OSSClient ossClient = getOSSClient();
if(dest.isDirectory()) {
File[] files = dest.listFiles();
logger.info("dest下的文件数量=={}",files.length);
if(files.length==0){
return true;
}
for (File f : files) {
if(f.isFile()){
logger.info("路径={},版本批次={}",f.getPath(),publishBatch);
String gt = f.getPath().substring(f.getPath().indexOf(publishBatch)+publishBatch.length()+1);
logger.info("文件路径：{}",gt);
// 上传文件
PutObjectResult result = ossClient.putObject(bucketName,gt, f);
logger.info("上传结果：{}",result.getETag());
}else{
logger.info("目录继续找文件=={0},{1}",f,publishBatch);
restoreToOss(f,publishBatch);
}
}
}
return true;
}
}
