多线程异步

电玩女神 2022-10-05 05:42 231阅读 0赞

### 1. 配置线程池 ###

import com.google.common.util.concurrent.ThreadFactoryBuilder;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    
    import java.util.concurrent.Executor;
    import java.util.concurrent.LinkedBlockingQueue;
    import java.util.concurrent.ThreadFactory;
    import java.util.concurrent.ThreadPoolExecutor;
    import java.util.concurrent.TimeUnit;
    
    @Configuration
    public class ThreadPoolExecutorConfig { 
    
        private static final Logger LOGGER = LoggerFactory.getLogger(ThreadPoolExecutorConfig.class);
    
        private static final int THREADS = Runtime.getRuntime().availableProcessors() + 1;
        private final ThreadFactory threadFactory = new ThreadFactoryBuilder()
                .setNameFormat("async-task-name-%d")
                .setDaemon(true).build();
    
        @Bean
        public Executor taskExecutor() { 
            return new ThreadPoolExecutor(THREADS, 2 * THREADS, 5
                    , TimeUnit.SECONDS, new LinkedBlockingQueue<>(1024)
                    , threadFactory, (r, executor) -> { 
                LOGGER.error("async task is rejected");
            });
        }
    }

### 2. 配置线程池服务 ###

import org.springframework.scheduling.annotation.Async;
    import org.springframework.stereotype.Component;
    
    @Component
    public class AsyncTaskService { 
    
        @Async("taskExecutor")
        public void doTask(Runnable runnable){ 
            runnable.run();
        }
    }

### 3. 使用服务 ###

#### 实现多线程异步 ####

#### 3.1 interface ####

public interface MailService { 
        boolean sendMail(Mail mail);
    }

#### 3.2 impl ####

import com.leinovo.npi.portal.component.mail.MailService;
    import com.leinovo.npi.portal.component.mail.domain.Mail;
    import com.leinovo.platform.common.component.datacollect.AsyncTaskService;
    import lombok.extern.slf4j.Slf4j;
    
    @Slf4j
    public class AsyncLeinovoMailService implements MailService { 
    
        private static final int NUM_RETRY_SEND_MAIL = 5;
    
        private MailService mailService;
    
        private AsyncTaskService asyncTaskService;
    
        public AsyncLeinovoMailService(MailService mailService, AsyncTaskService asyncTaskService) { 
            this.mailService = mailService;
            this.asyncTaskService = asyncTaskService;
        }
    
        @Override
        public boolean sendMail(Mail mail) { 
            asyncTaskService.doTask(() -> { 
                int i = 1;
                do { 
                    try { 
                        mailService.sendMail(mail);
                        //send successfully and exit the loop
                        i += NUM_RETRY_SEND_MAIL;
                    } catch (Exception e) { 
                        log.error("[{}]th async send mail error: ", i, e);
                        //the number of failed retries added by 1
                        i++;
                    }
                } while (i <= NUM_RETRY_SEND_MAIL);
    
            });
            return true;
        }
    }