Windcorp/JavaProgressia
Windcorp/JavaProgressia
Archived
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TaskManager bug fixes and code cleanup

Browse Source
This commit is contained in:
Евгений Смирнов 2021-07-07 17:02:38 +03:00
parent 5a06788652
commit f520d4b2c6
4 changed files with 120 additions and 106 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/ru/windcorp/progressia/client/audio/AudioSystem.java
View File

@ -18,6 +18,7 @@
package ru.windcorp.progressia.client.audio;
import org.apache.logging.log4j.LogManager;
import ru.windcorp.progressia.common.modules.Module;
import ru.windcorp.progressia.common.modules.Task;
import ru.windcorp.progressia.common.modules.TaskManager;
@ -29,10 +30,12 @@ public class AudioSystem {
AudioManager.initAL();
Thread shutdownHook = new Thread(AudioManager::closeAL, "AL Shutdown Hook");
Runtime.getRuntime().addShutdownHook(shutdownHook);
Task t = new Task("Task:InitializeAudio") {
Task t = new Task("AudioSystem:Initialize") {
@Override
protected void perform() {
loadAudioData();
LogManager.getLogger().info("Audio data is loaded");
}
};
audioModule.addTask(t);

5
src/main/java/ru/windcorp/progressia/common/modules/Module.java
View File

@ -1,12 +1,12 @@
package ru.windcorp.progressia.common.modules;
import ru.windcorp.progressia.common.util.namespaces.Namespaced;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import ru.windcorp.progressia.common.util.namespaces.Namespaced;
public class Module extends Namespaced {
private final List<Task> tasks = new ArrayList<>();
@ -25,6 +25,7 @@ public class Module extends Namespaced {
public List<Task> getTasks() {
return tasks;
}
public void addTask(Task task) {
tasks.add(task);
}

16
src/main/java/ru/windcorp/progressia/common/modules/Task.java
View File

@ -7,13 +7,11 @@ import java.util.List;
public abstract class Task
extends Namespaced
implements Runnable
{
private boolean isDone = false;
private boolean isActive = false;
implements Runnable {
List<Task> requiredTasks = new ArrayList<>();
private boolean isDone = false;
private boolean isActive = false;
protected Task(String id) {
super(id);
@ -33,7 +31,9 @@ public abstract class Task
return isDone;
}
public boolean isActive() { return isActive; }
public boolean isActive() {
return isActive;
}
public boolean canRun() {
if (this.isActive) return false;
@ -46,4 +46,8 @@ public abstract class Task
public List<Task> getRequiredTasks() {
return requiredTasks;
}
public void addRequiredTask(Task task) {
requiredTasks.add(task);
}
}

46
src/main/java/ru/windcorp/progressia/common/modules/TaskManager.java
View File

@ -1,10 +1,14 @@
package ru.windcorp.progressia.common.modules;
import ru.windcorp.progressia.common.state.StateFieldBuilder;
import org.apache.logging.log4j.LogManager;
import ru.windcorp.progressia.common.util.crash.CrashReports;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import static java.util.concurrent.Executors.newFixedThreadPool;
@ -12,14 +16,13 @@ public class TaskManager {
private static final TaskManager instance = new TaskManager();
private final List<Task> tasks = new ArrayList<>();
private final List<Module> modules = new ArrayList<>();
private boolean loadingDone;
private int activeThreadsCount;
private final ExecutorService executorService;
private final AtomicBoolean loadingDone;
private final AtomicInteger activeThreadsCount;
private TaskManager() {
loadingDone = false;
activeThreadsCount = 0;
loadingDone = new AtomicBoolean(false);
activeThreadsCount = new AtomicInteger(0);
executorService = newFixedThreadPool(
Runtime.getRuntime().availableProcessors(), Thread::new);
}
@ -34,25 +37,23 @@ public class TaskManager {
}
public boolean isLoadingDone() {
return loadingDone;
return loadingDone.get();
}
public void startLoading() {
for(int i = 0; i < Runtime.getRuntime().availableProcessors(); i++) {
LogManager.getLogger().info("Loading is started");
for (int i = 0; i < Runtime.getRuntime().availableProcessors(); i++) {
executorService.submit(() -> {
while(!loadingDone) {
while (!loadingDone.get()) {
Task t = getRunnableTask();
if (t != null) {
synchronized (this) {
activeThreadsCount++;
}
activeThreadsCount.incrementAndGet();
t.run();
activeThreadsCount.decrementAndGet();
synchronized (this) {
activeThreadsCount--;
notifyAll();
}
} else if (activeThreadsCount > 0) {
} else if (activeThreadsCount.get() > 0) {
try {
synchronized (this) {
this.wait();
@ -61,8 +62,9 @@ public class TaskManager {
e.printStackTrace();
}
} else {
loadingDone.set(true);
synchronized (this) {
loadingDone = true;
notifyAll();
}
}
}
@ -70,14 +72,18 @@ public class TaskManager {
}
waitForLoadingEnd();
if (!tasks.isEmpty()) {
throw CrashReports.crash(new Exception("Loading is failed"), "");
}
LogManager.getLogger().info("Loading is finished");
executorService.shutdownNow();
}
public synchronized Task getRunnableTask() {
if(!tasks.isEmpty()) {
if (!tasks.isEmpty()) {
for (Task t :
tasks) {
if(t.canRun()) {
if (t.canRun()) {
tasks.remove(t);
return t;
}
@ -88,7 +94,7 @@ public class TaskManager {
private void waitForLoadingEnd() {
synchronized (this) {
while(!loadingDone) {
while (!loadingDone.get()) {
try {
this.wait();
} catch (InterruptedException e) {