智慧保安后台管理-外网
Administrator
2022-06-07 df054296327ebca617a9c19bb01139b9c40bf232
src/main/java/org/springblade/modules/quartz/task/Task.java
@@ -1,10 +1,14 @@
package org.springblade.modules.quartz.task;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.springblade.modules.FTP.*;
import org.springblade.modules.system.entity.User;
import org.springblade.modules.system.service.IUserService;
import org.springblade.modules.system.service.MyAsyncService;
import org.springblade.modules.training.service.TrainingRegistrationService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
@@ -15,6 +19,7 @@
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import static org.springblade.common.config.FtpConfig.*;
import static org.springblade.common.config.FtpConfig.ftpPath;
@@ -29,6 +34,13 @@
   @Autowired
   private TrainingRegistrationService trainingRegistrationService;
   @Autowired
   private IUserService userService;
   @Autowired
   private MyAsyncService myAsyncService;
   public void testTask(){
        System.out.println("测试定时任务执行-----------------");
@@ -62,9 +74,15 @@
//   @Scheduled(cron = "*/30 * * * * ?")
   public boolean readFile() {
      FTPClient ftp = new FTPClient();
      System.out.println("ftp 创建客户端");
      try {
         //设置传输超时时间为60秒
         ftp.setDataTimeout(10000);
         //连接超时为60秒
         ftp.setConnectTimeout(10000);
         //连接
         ftp.connect(ftpHost, ftpPort);
         System.out.println("ftp 连接成功!"+new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));
         // 登陆
         ftp.login(ftpUserNameIn, ftpPasswordIn);
         // 检验登陆操作的返回码是否正确
@@ -89,6 +107,7 @@
         FTPFile[] files = ftp.listFiles();
         if (files.length==0){
            ftp.logout();
            return false;
         }
         else {
@@ -96,10 +115,11 @@
               String fileName = file.getName();
               InputStream is = ftp.retrieveFileStream(new String(fileName.getBytes("GBK"), FTP.DEFAULT_CONTROL_ENCODING));
               if (null == is) {
                  ftp.logout();
                  return false;
               } else {
                  String substring1 = fileName.substring(0, 4);
                  if (substring1.equals("nsql")) {
                  String substring1 = fileName.substring(0, 2);
                  if (substring1.equals("nq")) {
                     //把文件下载到本地
                     FtpUtil.downloadFtpFile(ftpHost, ftpUserNameIn, ftpPasswordIn, ftpPort, ftpPath, localPath, fileName);
                     //
@@ -127,11 +147,13 @@
                     //删除本地服务器文件
                     MysqlCenlint.deletess(fileName);
                     //删除 ftp 服务器文件
                     FtpUtil.deleteFile(ftpHost, ftpPort, ftpUserNameIn, ftpPasswordIn, ftpPath, fileName);
                     ftp.dele(fileName);
//                     FtpUtil.deleteFile(ftpHost, ftpPort, ftpUserNameIn, ftpPasswordIn, ftpPath, fileName);
                  }
                  //关闭流
                  is.close();
                  ftp.completePendingCommand();
                  ftp.logout();
               }
               return true;
            }
@@ -140,7 +162,7 @@
         System.out.println("ftp连接失败");
         e.printStackTrace();
      } finally {
         if (ftp != null) {
         if (ftp.isConnected()) {
            try {
               ftp.disconnect();
            } catch (IOException e) {
@@ -151,102 +173,26 @@
      return false;
   }
//   /**
//    * 读取文件(用户对象)单用户新增
//    * @return
//    */
//   public Result readFile() {
//      String uuid = "123-456";
//      //创建 ftp 对象
//      FTPClient ftp = new FTPClient();
//      boolean flag = false;
//      try {
//         //连接
//         ftp.connect(ftpHost, ftpPort);
//
//         // 登陆
//         ftp.login(ftpUserNameIn, ftpPasswordIn);
//         // 检验登陆操作的返回码是否正确
//         if (!FTPReply.isPositiveCompletion(ftp.getReplyCode())) {
//            ftp.disconnect();
//            return new Result(400,null,"ftp 连接失败",null);
//         }
//
//         ftp.enterLocalActiveMode();
//
//         // 设置文件类型为二进制,与ASCII有区别
//         ftp.setFileType(FTP.BINARY_FILE_TYPE);
//
//         // 设置编码格式
//         ftp.setControlEncoding("GBK");
//         // 检验文件是否存在
//         boolean ftpFile = ftp.changeWorkingDirectory(ftpPath);
//         System.out.println("检验文件是否存在 = " + ftpFile);
//
//         // Use passive mode as default
//         ftp.enterLocalPassiveMode();
//
//         FTPFile[] files = ftp.listFiles();
//         System.out.println("files = " + files);
//         System.out.println("开始读取内网回传消息");
//         if (files.length==0){
//            System.out.println("未读取到文件");
//            return new Result(400,null,"未读取到文件",null);
//         } else {
//            System.out.println("-----有读取到文件--------");
//            for (FTPFile file : files){
//               String fileName = file.getName();
//               System.out.println("fileName = " + fileName);
//               InputStream is = ftp.retrieveFileStream(new String(fileName.getBytes("GBK"), FTP.DEFAULT_CONTROL_ENCODING));
//               String substring1 = fileName.substring(0, 1);
//               System.out.println("substring1 = " + substring1);
//               if (substring1.equals("n")){
//                  System.out.println("-------接收到内网回传的文件: " + substring1);
//                  //把文件下载到本地
//                  FtpUtil.downloadFtpFile(ftpHost, ftpUserNameIn, ftpPasswordIn, ftpPort, ftpPath, localPath, fileName);
//                  // 解析数据
//                  String s = OutJson.TestJson(fileName);
//                  //数据处理
//                  Result result = DataHandler.handler(s,uuid);
//                  //删除本地文件
//                  MysqlCenlint.deletess(fileName);
//                  FtpUtil.deleteFile(ftpHost, ftpPort, ftpUserNameIn, ftpPasswordIn, ftpPath, fileName);
//                  is.close();
//                  ftp.completePendingCommand();
//                  if (result.getCode()==200) {
//                     //返回
//                     return new Result(200,null,"新增成功",null);
//                  }
//                  if (result.getCode()==201) {
//                     //返回
//                     return new Result(201,null,"修改成功",null);
//                  }
//                  if (result.getCode()==202) {
//                     //返回
//                     return new Result(202,null,result.getMsg(),null);
//                  }
//                  if (result.getCode()==203) {
//                     //返回
//                     return new Result(203,null,result.getMsg(),null);
//                  }
//               }
//            }
//            System.out.println("未读取到对应的文件");
//            return new Result(400,null,"未读取到对应的文件",null);
//         }
//      } catch (Exception e) {
//         System.out.println("ftp连接失败");
//         e.printStackTrace();
//      } finally {
//         if (ftp != null) {
//            try {
//               ftp.disconnect();
//            } catch (IOException e) {
//               e.printStackTrace();
//            }
//         }
//      }
//      return new Result(400,null,"失败",null);
//   }
   /**
    *  用户审查
    * @return
    */
   public void userExamine() {
      //查询审查状态为未审查的保安员
      User user = new User();
      user.setStatus(1);
      user.setIsDeleted(0);
      user.setRoleId("1412226235153731586");
      //未审查
      user.setExaminationType("2");
      //查询
      List<User> users = userService.list(new QueryWrapper<>(user));
      //遍历
      if (users.size()>0){
         users.forEach(user1 -> {
            myAsyncService.userExamineFTP(user1);
         });
      }
   }
}