fix: entity serialization and deserialization

GeorgeV220 · May 29, 2023 · 574d94c · 574d94c
1 parent b148d37
commit 574d94c
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/database/src/main/java/com/georgev22/library/utilities/EntityManager.java b/database/src/main/java/com/georgev22/library/utilities/EntityManager.java
@@ -119,9 +119,14 @@ public CompletableFuture<T> load(UUID entityId) {
                                 case MONGODB -> {
                                     Document document = mongoDB.getCollection(collection).find(Filters.eq("entityId", entityId.toString())).first();
                                     if (document != null) {
-                                        T entity = (T) document.get("entity");
-                                        loadedEntities.put(entityId, entity);
-                                        return entity;
+                                        String serializedEntity = document.getString("entity");
+                                        try {
+                                            T entity = (T) Utils.deserializeObjectFromString(serializedEntity);
+                                            loadedEntities.put(entityId, entity);
+                                            return entity;
+                                        } catch (IOException | ClassNotFoundException e) {
+                                            throw new RuntimeException(e);
+                                        }
                                     } else {
                                         throw new RuntimeException("No entity found with id: " + entityId);
                                     }
@@ -163,12 +168,12 @@ public CompletableFuture<Void> save(Entity entity) {
                     String query = result ? "UPDATE " + collection + " SET entity = ? WHERE entity_id =  ?;" : "INSERT INTO " + collection + " (entity, entity_id) VALUES (?, ?)";
                     try {
                         PreparedStatement statement = Objects.requireNonNull(connection).prepareStatement(query);
-                        String serializedEntity = Utils.serialize(entity);
+                        String serializedEntity = Utils.serializeObjectToString(entity);
                         statement.setString(1, serializedEntity);
                         statement.setString(2, entity.getId().toString());
                         statement.executeUpdate();
                         statement.close();
-                    } catch (SQLException e) {
+                    } catch (SQLException | IOException e) {
                         throw new RuntimeException(e);
                     }
                 });