ChomeNS/chomens-bot-java
3
1
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

refactor: refactor PersistentDataUtilities, hopefully it will work ok

Browse source 
i do have a backup so this should be fine if it fails somehow
This commit is contained in:
Chayapak 2024-11-12 20:04:22 +07:00
parent f322e2dd14
commit 66dc3b97ec
Signed by: ChomeNS
SSH key fingerprint: SHA256:0YoxhdyXsgbc0nfeB2N6FYE60mxMU7DS4uCUMaw2mvA
1 changed files with 16 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
src/main/java/me/chayapak1/chomens_bot/util/PersistentDataUtilities.java
View file

@ -4,7 +4,6 @@ import com.google.gson.Gson;
import com.google.gson.JsonElement; import com.google.gson.JsonElement;
import com.google.gson.JsonObject; import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive; import com.google.gson.JsonPrimitive;
import me.chayapak1.chomens_bot.Main;
import java.io.BufferedReader; import java.io.BufferedReader;
import java.io.BufferedWriter; import java.io.BufferedWriter;
@ -12,10 +11,6 @@ import java.io.IOException;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.StandardOpenOption; import java.nio.file.StandardOpenOption;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
public class PersistentDataUtilities { public class PersistentDataUtilities {
public static final Path path = Path.of("persistent.json"); public static final Path path = Path.of("persistent.json");
@ -24,33 +19,10 @@ public class PersistentDataUtilities {
public static JsonObject jsonObject = new JsonObject(); public static JsonObject jsonObject = new JsonObject();
private static final Map<String, JsonElement> queue = new ConcurrentHashMap<>(); private static boolean stopping = false;
private static final ScheduledFuture<?> future;
static { static {
init(); init();
future = Main.executor.scheduleAtFixedRate(() -> {
try {
if (queue.isEmpty()) return;
final Map.Entry<String, JsonElement> entry = queue.entrySet().iterator().next(); // is this the best way to get the first item of the map?
final String property = entry.getKey();
final JsonElement value = entry.getValue();
Main.executorService.submit(() -> {
jsonObject.add(property, value);
write(jsonObject.toString());
queue.remove(property);
});
} catch (Exception e) {
e.printStackTrace();
}
}, 0, 100, TimeUnit.MILLISECONDS);
} }
private static void init () { private static void init () {
@ -66,19 +38,17 @@ public class PersistentDataUtilities {
jsonObject = gson.fromJson(reader, JsonObject.class); jsonObject = gson.fromJson(reader, JsonObject.class);
} }
writer = Files.newBufferedWriter(path, StandardOpenOption.TRUNCATE_EXISTING); writer = Files.newBufferedWriter(path, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
} catch (IOException e) { } catch (IOException e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
private static void write (String string) { private static void write (String string) {
try { if (stopping) return; // is this necessary?
writer.close();
// ? how do i clear the file contents without making a completely new writer? try {
// or is this the only way? writer = Files.newBufferedWriter(path, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
writer = Files.newBufferedWriter(path, StandardOpenOption.TRUNCATE_EXISTING);
writer.write(string); writer.write(string);
writer.flush(); writer.flush();
@ -86,27 +56,32 @@ public class PersistentDataUtilities {
} }
public static void stop () { public static void stop () {
future.cancel(false); stopping = true;
write(jsonObject.toString()); write(jsonObject.toString());
} }
public static void put (String property, JsonElement value) { public static void put (String property, JsonElement value) {
Main.executorService.submit(() -> queue.put(property, value)); jsonObject.add(property, value);
write(jsonObject.toString());
} }
public static void put (String property, String value) { public static void put (String property, String value) {
Main.executorService.submit(() -> queue.put(property, new JsonPrimitive(value))); jsonObject.add(property, new JsonPrimitive(value));
write(jsonObject.toString());
} }
public static void put (String property, boolean value) { public static void put (String property, boolean value) {
Main.executorService.submit(() -> queue.put(property, new JsonPrimitive(value))); jsonObject.add(property, new JsonPrimitive(value));
write(jsonObject.toString());
} }
public static void put (String property, int value) { public static void put (String property, int value) {
Main.executorService.submit(() -> queue.put(property, new JsonPrimitive(value))); jsonObject.add(property, new JsonPrimitive(value));
write(jsonObject.toString());
} }
public static void put (String property, char value) { public static void put (String property, char value) {
Main.executorService.submit(() -> queue.put(property, new JsonPrimitive(value))); jsonObject.add(property, new JsonPrimitive(value));
write(jsonObject.toString());
} }
} }