智慧保安后台管理-外网
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("测试定时任务执行-----------------");
@@ -56,24 +68,27 @@
   }
   /**
    * 读取文件(用户对象)单用户新增
    * 读取内外推送的文件,每30秒读取一次
    * @return
    */
   public Result readFile() {
      String uuid = "123-456";
      //创建 ftp 对象
//   @Scheduled(cron = "*/30 * * * * ?")
   public boolean readFile() {
      FTPClient ftp = new FTPClient();
      boolean flag = false;
      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);
         // 检验登陆操作的返回码是否正确
         if (!FTPReply.isPositiveCompletion(ftp.getReplyCode())) {
            ftp.disconnect();
            return new Result(400,null,"ftp 连接失败",null);
            return false;
         }
         ftp.enterLocalActiveMode();
@@ -83,62 +98,71 @@
         // 设置编码格式
         ftp.setControlEncoding("GBK");
         // 检验文件是否存在
         boolean ftpFile = ftp.changeWorkingDirectory(ftpPath);
         System.out.println("检验文件是否存在 = " + ftpFile);
         ftp.changeWorkingDirectory(ftpPath);
         // 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){
            ftp.logout();
            return false;
         }
         else {
            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);
               if (null == is) {
                  ftp.logout();
                  return false;
               } else {
                  String substring1 = fileName.substring(0, 2);
                  if (substring1.equals("nq")) {
                     //把文件下载到本地
                     FtpUtil.downloadFtpFile(ftpHost, ftpUserNameIn, ftpPasswordIn, ftpPort, ftpPath, localPath, fileName);
                     //
                     String s = OutJson.TestJson(fileName);
                     //sql语句
                     String sql = OutJson.stringReplace(s);
                     String[] split = sql.split(";");//以逗号分割
                     for (String sqls : split) {
                        //判断是否是新增,删除,修改
                        String substring = sqls.substring(0, 2);
                        //新增
                        if (substring.equals("in")) {
                           //运行sql语句
                           MysqlCenlint.inster(sqls);
                        }
                        //修改
                        else if (substring.equals("up")) {
                           MysqlCenlint.update(sqls);
                        }
                        //删除
                        else {
                           MysqlCenlint.delete(sqls);
                        }
                     }
                     //删除本地服务器文件
                     MysqlCenlint.deletess(fileName);
                     //删除 ftp 服务器文件
                     ftp.dele(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);
                  }
                  ftp.logout();
               }
               return true;
            }
            System.out.println("未读取到对应的文件");
            return new Result(400,null,"未读取到对应的文件",null);
         }
      } catch (Exception e) {
         System.out.println("ftp连接失败");
         e.printStackTrace();
      } finally {
         if (ftp != null) {
         if (ftp.isConnected()) {
            try {
               ftp.disconnect();
            } catch (IOException e) {
@@ -146,7 +170,29 @@
            }
         }
      }
      return new Result(400,null,"失败",null);
      return false;
   }
   /**
    *  用户审查
    * @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);
         });
      }
   }
}