【Mybatis-Plus租户id结合定时任务 @Scheduled】

---

前言

提示：这里可以添加本文要记录的大概内容：

例如：随着人工智能的不断发展，机器学习这门技术也越来越重要，很多人都开启了学习机器学习，本文就介绍了机器学习的基础内容。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、启动类加上@EnableScheduling、@EnableAsync

二、创建定时任务类

package com.jsh.erp.task;

import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.jsh.erp.datasource.entities.project.ProTask;
import com.jsh.erp.datasource.mappers.project.ProTaskMapper;
import com.jsh.erp.service.project.ProTaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.List;
import java.util.concurrent.Executor;

@Component
@Slf4j
@Async
public class TaskOutTime {
    
    

   @Autowired
   private ProTaskService proTaskService;

   /*@Autowired
   private Executor taskTimeOutExecutor;

   @Autowired
   private ProTaskMapper proTaskMapper;*/

//   @Scheduled(cron = "0 0 0 * * ?")
   @Scheduled(cron = "0/2 * * * * ?")
   public void proTaskTimeOut(){
    
    

      Date date = new Date();
      SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
      String date1 = sdf.format(date);
      log.info("date1"+date1);
      LambdaQueryWrapper<ProTask> proTaskLambdaQueryWrapper = new LambdaQueryWrapper<>();
      proTaskLambdaQueryWrapper.eq(ProTask::getEndDate,date1);
      List<ProTask> proTaskList = proTaskService.list(proTaskLambdaQueryWrapper);
      for (ProTask proTask : proTaskList) {
    
    
         proTask.setState(4);
         proTaskService.updateById(proTask);
      }


      /*taskTimeOutExecutor.execute(() -> {
         // 在新的线程中执行的任务
         Date date = new Date();
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
         String date1 = sdf.format(date);
         log.info("date1"+date1);
         LambdaQueryWrapper<ProTask> proTaskLambdaQueryWrapper = new LambdaQueryWrapper<>();
         proTaskLambdaQueryWrapper.eq(ProTask::getEndDate,date1);
         List<ProTask> proTaskList = proTaskService.list(proTaskLambdaQueryWrapper);
         for (ProTask proTask : proTaskList) {
            proTask.setState(4);
            proTaskService.updateById(proTask);
         }
      });*/
   }
}

3、多线程环境下配置多线程配置

package com.jsh.erp.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
@Configuration
public class TimerTaskConfig {
    
    
    @Bean
    public Executor taskTimeOutExecutor() {
    
    
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(20);
        executor.setQueueCapacity(30);
        executor.setThreadNamePrefix("MyTaskExecutor-");
        executor.initialize();
        return executor;
    }
}

4、以下是一个示例代码，展示了如何在Mybatis-Plus中配置租户解析器，以便在定时任务中使用：

public PaginationInterceptor paginationInterceptor(HttpServletRequest request) {
    
    
    PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
    List<ISqlParser> sqlParserList = new ArrayList<>();
    TenantSqlParser tenantSqlParser = new TenantSqlParser();
    tenantSqlParser.setTenantHandler(new TenantHandler() {
    
    
        @Override
        public Expression getTenantId() {
    
    
            String token = request.getHeader("X-Access-Token");
            Long tenantId = Tools.getTenantIdByToken(token);
            if (tenantId != 0L) {
    
    
                return new LongValue(tenantId);
            } else {
    
    
                //超管
                return null;
            }
        }
        @Override
        public String getTenantIdColumn() {
    
    
            return "tenant_id";
        }
        @Override
        public boolean doTableFilter(String tableName) {
    
    
            if ("t_xxxx".equals(tableName)) {
    
    //你要过滤的表名
                return true;
            }
        });
}

其中，getTenantId()方法从请求中获取租户ID，并返回一个LongValue对象；getTenantIdColumn()方法返回租户ID所在的列名；doTableFilter()方法用于过滤不需要使用租户ID的表。

或者

package com.jsh.erp.config;

import com.baomidou.mybatisplus.core.parser.ISqlParser;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.extension.plugins.tenant.TenantHandler;
import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser;
import com.jsh.erp.utils.Tools;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Executor;
@Configuration
public class TimerTaskConfig {
    
    
    @Bean
    public Executor taskTimeOutExecutor() {
    
    
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(20);
        executor.setQueueCapacity(30);
        executor.setThreadNamePrefix("MyTaskExecutor-");
        executor.initialize();
        return executor;
    }

    public PaginationInterceptor paginationInterceptor(HttpServletRequest request) {
    
    
        PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
        List<ISqlParser> sqlParserList = new ArrayList<>();
        TenantSqlParser tenantSqlParser = new TenantSqlParser();
        tenantSqlParser.setTenantHandler(new TenantHandler() {
    
    
            @Override
            public Expression getTenantId() {
    
    
                String token = request.getHeader("X-Access-Token");
                Long tenantId = Tools.getTenantIdByToken(token);
                if (tenantId != 0L) {
    
    
                    return new LongValue(tenantId);
                } else {
    
    
                    //超管
                    return null;
                }
            }
            @Override
            public String getTenantIdColumn() {
    
    
                return "tenant_id";
            }
            @Override
            public boolean doTableFilter(String tableName) {
    
    
                if ("t_xxxx".equals(tableName)) {
    
    //你要过滤的表名
                    return true;
                }
                return false;
            }
            });
        return null;
        }

    }

5、有可能会报没有线程绑定错误，添加一个线程绑定就可以了。

package com.jsh.erp.task;

import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.jsh.erp.datasource.entities.project.ProTask;
import com.jsh.erp.datasource.mappers.project.ProTaskMapper;
import com.jsh.erp.service.project.ProTaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.List;
import java.util.concurrent.Executor;

/**
 * 定时任务
 */

@Component
@Slf4j
@Async
public class TaskOutTime {
    
    

   @Autowired
   private ProTaskService proTaskService;

   /*@Autowired
   private Executor taskTimeOutExecutor;

   @Autowired
   private ProTaskMapper proTaskMapper;*/



//   @Scheduled(cron = "0 0 0 * * ?")
   @Scheduled(cron = "0/2 * * * * ?")
   public void proTaskTimeOut(){
    
    

      /**
       * 添加一个线程绑定
       */
      RequestAttributes obj = RequestContextHolder.getRequestAttributes();
      if(obj==null){
    
    
         obj = new ServletRequestAttributes(new MockHttpServletRequest());
      }
      RequestContextHolder.setRequestAttributes(obj);

      Date date = new Date();
      SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
      String date1 = sdf.format(date);
      log.info("date1"+date1);
      LambdaQueryWrapper<ProTask> proTaskLambdaQueryWrapper = new LambdaQueryWrapper<>();
      proTaskLambdaQueryWrapper.eq(ProTask::getEndDate,date1);
      List<ProTask> proTaskList = proTaskService.list(proTaskLambdaQueryWrapper);
      for (ProTask proTask : proTaskList) {
    
    
         proTask.setState(4);
         proTaskService.updateById(proTask);
      }


      /*taskTimeOutExecutor.execute(() -> {
         // 在新的线程中执行的任务
         Date date = new Date();
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
         String date1 = sdf.format(date);
         log.info("date1"+date1);
         LambdaQueryWrapper<ProTask> proTaskLambdaQueryWrapper = new LambdaQueryWrapper<>();
         proTaskLambdaQueryWrapper.eq(ProTask::getEndDate,date1);
         List<ProTask> proTaskList = proTaskService.list(proTaskLambdaQueryWrapper);
         for (ProTask proTask : proTaskList) {
            proTask.setState(4);
            proTaskService.updateById(proTask);
         }
      });*/
   }
}