feat: add entity fetcher for game

llucasreis · Aug 23, 2021 · 705d133 · 705d133
1 parent eb6cd7e
commit 705d133
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/game-dgs/src/main/java/com/llucasreis/gamedgs/datafetchers/GameDataFetcher.java b/game-dgs/src/main/java/com/llucasreis/gamedgs/datafetchers/GameDataFetcher.java
@@ -1,11 +1,11 @@
 package com.llucasreis.gamedgs.datafetchers;
 
 import java.util.List;
+import java.util.Map;
 
 import com.llucasreis.gamedgs.domain.entities.Game;
 import com.llucasreis.gamedgs.repositories.GameRepository;
-import com.netflix.graphql.dgs.DgsComponent;
-import com.netflix.graphql.dgs.DgsQuery;
+import com.netflix.graphql.dgs.*;
 
 import org.springframework.beans.factory.annotation.Autowired;
 
@@ -24,5 +24,12 @@ public List<Game> games() {
     return this.gameRepository.findAll();
   }
 
+  @DgsEntityFetcher(name = "Game")
+  public Game game(Map<String, Object> values) {
+    Long id = Long.parseLong((String) values.get("id"));
+
+    return this.gameRepository.findById(id).orElse(null);
+  }
+
 
 }