third_codes/PowerJob
1
0
Fork 1
mirror of https://github.com/PowerJob/PowerJob.git synced 2025-07-17 00:00:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: prevent worker from being repeatedly initialized

Browse Source
This commit is contained in:
tjq 2021-03-11 23:28:33 +08:00
parent 48860e3172
commit 4c89a1e69c
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
powerjob-worker/src/main/java/com/github/kfcfans/powerjob/worker/OhMyWorker.java
View File

@ -43,6 +43,7 @@ import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService; import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory; import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
/** /**
* 客户端启动类 * 客户端启动类
@ -55,6 +56,7 @@ public class OhMyWorker implements ApplicationContextAware, InitializingBean, Di
private ScheduledExecutorService timingPool; private ScheduledExecutorService timingPool;
private final WorkerRuntime workerRuntime = new WorkerRuntime(); private final WorkerRuntime workerRuntime = new WorkerRuntime();
private final AtomicBoolean initialized = new AtomicBoolean();
@Override @Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
@ -68,6 +70,11 @@ public class OhMyWorker implements ApplicationContextAware, InitializingBean, Di
public void init() throws Exception { public void init() throws Exception {
if (!initialized.compareAndSet(false, true)) {
log.warn("[OhMyWorker] please do not repeat the initialization");
return;
}
Stopwatch stopwatch = Stopwatch.createStarted(); Stopwatch stopwatch = Stopwatch.createStarted();
log.info("[OhMyWorker] start to initialize OhMyWorker..."); log.info("[OhMyWorker] start to initialize OhMyWorker...");