Create resolver and mutator

AnoshkoAlexey · AnoshkoAlexey · commit 9ef9ec5530d4 · 2024-07-03T17:04:52.000Z
diff --git a/src/main/java/com/udacity/doggraphqlapi/mutator/Mutation.java b/src/main/java/com/udacity/doggraphqlapi/mutator/Mutation.java
@@ -0,0 +1,59 @@
+package com.udacity.doggraphqlapi.mutator;
+
+import java.util.Optional;
+
+import org.springframework.stereotype.Component;
+
+import com.coxautodev.graphql.tools.GraphQLMutationResolver;
+import com.udacity.doggraphqlapi.entity.Dog;
+import com.udacity.doggraphqlapi.repository.DogRepository;
+
+@Component
+public class Mutation implements GraphQLMutationResolver {
+    private DogRepository dogRepository;
+    
+    public  Mutation(DogRepository dogRepository) {this.dogRepository = dogRepository; }
+
+    public Dog newDog(String name, String breed) {
+        Dog dog = new Dog(name, breed);
+        dogRepository.save(dog);
+        return dog;        
+    }
+
+    public boolean deleteDog(Long id) {
+        dogRepository.deleteById(id);
+        return true;
+    }
+
+    public boolean deleteDogBreed(String breed) {
+        boolean deleted = false;
+        Iterable<Dog> allDogs = dogRepository.findAll();
+        // Loop through all dogs to check their breed
+        for (Dog d:allDogs) {
+            if (d.getBreed().equals(breed)) {
+                // Delete if the breed is found
+                dogRepository.delete(d);
+                deleted = true;
+            }
+        }
+        // Throw an exception if the breed doesn't exist
+        if (!deleted) {
+            throw new BreedNotFoundException("Breed Not Found", breed);
+        }
+        return deleted;
+    }
+
+    public Dog updateDogName(String newName, Long id) {
+        Optional<Dog> optionalDog = dogRepository.findById(id);
+        if (optionalDog.isPresent()) {
+            Dog dog = optionalDog.get();
+            // Set the new name and save the updated dog
+            dog.setName(newName);
+            dogRepository.save(dog);
+            return dog;
+        } else {
+            throw new DogNotFoundException("Dog Not Found", id);
+        }
+        
+    }
+}
diff --git a/src/main/java/com/udacity/doggraphqlapi/resolver/Query.java b/src/main/java/com/udacity/doggraphqlapi/resolver/Query.java
@@ -0,0 +1,33 @@
+package com.udacity.doggraphqlapi.resolver;
+
+import com.udacity.doggraphqlapi.entity.Dog;
+
+import java.util.Optional;
+
+import org.springframework.stereotype.Component;
+
+import com.coxautodev.graphql.tools.GraphQLQueryResolver;
+import com.udacity.doggraphqlapi.repository.DogRepository;
+
+
+@Component
+public class Query implements GraphQLQueryResolver {
+    private DogRepository dogRepository;
+
+    public Query(DogRepository dogRepository) {
+        this.dogRepository = dogRepository;
+    }
+    
+    public Iterable<Dog> findAllDogs() {
+        return dogRepository.findAll();
+    }
+
+    public Dog findDogById(Long id) {
+        Optional<Dog> optionalDog = dogRepository.findById(id);
+        if (optionalDog.isPresent()) {
+            return optionalDog.get();
+        } else {
+            throw new DogNotFoundException("Dog Not Found", id);
+        } 
+    }
+}
