blob: b134b7c3d3eca98537c4409b2c3f1f81a1c1a0b2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
/*
* Dungeons Guide - The most intelligent Hypixel Skyblock Dungeons Mod
* Copyright (C) 2021 cyoung06
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package kr.syeyoung.dungeonsguide.wsresource;
import kr.syeyoung.dungeonsguide.DungeonsGuide;
import kr.syeyoung.dungeonsguide.events.StompConnectedEvent;
import kr.syeyoung.dungeonsguide.stomp.StompInterface;
import kr.syeyoung.dungeonsguide.stomp.StompMessageHandler;
import kr.syeyoung.dungeonsguide.stomp.StompPayload;
import kr.syeyoung.dungeonsguide.stomp.StompSubscription;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import org.json.JSONObject;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.CompletableFuture;
public class StaticResourceCache implements StompMessageHandler {
public static final StaticResourceCache INSTANCE = new StaticResourceCache();
private Map<UUID, StaticResource> staticResourceMap = new HashMap<>();
private Map<UUID, CompletableFuture<StaticResource>> staticResourceRequest = new HashMap<>();
public void purgeCache() {
staticResourceRequest.clear();
staticResourceMap.clear();
}
public CompletableFuture<StaticResource> getResource(UUID resourceID) {
if (staticResourceMap.containsKey(resourceID)) return CompletableFuture.completedFuture(staticResourceMap.get(resourceID));
if (staticResourceRequest.containsKey(resourceID)) return staticResourceRequest.get(resourceID);
DungeonsGuide.getDungeonsGuide().getStompConnection().send(new StompPayload().header("destination", "/app/staticresource.get").payload(resourceID.toString()));
CompletableFuture<StaticResource> comp = new CompletableFuture<>();
staticResourceRequest.put(resourceID, comp);
return comp;
}
@Override
public void handle(StompInterface stompInterface, StompPayload stompPayload) {
JSONObject object = new JSONObject(stompPayload.payload());
StaticResource staticResource = new StaticResource();
staticResource.setResourceID(UUID.fromString(object.getString("resourceID")));
staticResource.setExists(object.getBoolean("exists"));
staticResource.setValue((!object.has("value") || object.isNull("value")) ? null : object.getString("value"));
staticResourceMap.put(staticResource.getResourceID(), staticResource);
CompletableFuture<StaticResource> completed = staticResourceRequest.remove(staticResource.getResourceID());
if (completed != null) completed.complete(staticResource);
}
@SubscribeEvent
public void stompConnect(StompConnectedEvent stompConnectedEvent) {
stompConnectedEvent.getStompInterface().subscribe(StompSubscription.builder()
.stompMessageHandler(this).ackMode(StompSubscription.AckMode.AUTO).destination("/user/queue/staticresource.get").build());
getResource(BONUS_SCORE);
getResource(TRIVIA_ANSWERS);
}
public static final UUID BONUS_SCORE = UUID.fromString("13f10001-66b5-46e5-94f9-1a5161a23429");
public static final UUID TRIVIA_ANSWERS = UUID.fromString("5657f2cc-2bb8-4fcd-b55c-ffc0a35b9349");
}
|
