服务器上传文件到oss,以及备份

扶醉桌前 提交于 2019-11-29 04:47:50

公司的前端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;
}

/**
* 还原
* */
@InterfacePermissions(permissionCode = "ossVersionPublishManage:btn_restore")
@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;
}

 

}

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!