im-backed/im-server/src/main/java/com/bx/imserver/task/AbstractPullMessageTask.java

package com.bx.imserver.task;

import com.bx.imserver.netty.IMServerGroup;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;

import javax.annotation.PreDestroy;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

@Slf4j
public abstract class AbstractPullMessageTask  implements CommandLineRunner {

    private int threadNum = 1;
    private ExecutorService executorService;

    @Autowired
    private IMServerGroup serverGroup;

    public AbstractPullMessageTask() {
        this.threadNum = 1;
    }

    public AbstractPullMessageTask(int threadNum) {
        this.threadNum = threadNum;
    }


    @Override
    public void run(String... args) {
        // 初始化定时器
        executorService = Executors.newFixedThreadPool(threadNum);

        for (int i = 0; i < threadNum; i++) {
            executorService.execute(new Runnable() {
                @SneakyThrows
                @Override
                public void run() {
                    try {
                        if (serverGroup.isReady()) {
                            pullMessage();
                        }
                    } catch (Exception e) {
                        log.error("任务调度异常", e);
                        Thread.sleep(200);
                    }
                    if (!executorService.isShutdown()) {
                        executorService.execute(this);
                    }
                }
            });
        }
    }

    @PreDestroy
    public void destroy() {
        log.info("{}线程任务关闭", this.getClass().getSimpleName());
        executorService.shutdown();
    }

    public abstract void pullMessage();
}