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

服务器上传文件到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;
　　}

 

}

posted on 2019-09-03 18:20  幽魂~~ 阅读( ...) 评论( ...) 编辑 收藏

转载于:https://www.cnblogs.com/you-hun/p/11454687.html