diff --git a/src/main/java/org/codiki/core/entities/dto/CategoryDTO.java b/src/main/java/org/codiki/core/entities/dto/CategoryDTO.java index 6509298..addcf10 100644 --- a/src/main/java/org/codiki/core/entities/dto/CategoryDTO.java +++ b/src/main/java/org/codiki/core/entities/dto/CategoryDTO.java @@ -1,11 +1,16 @@ package org.codiki.core.entities.dto; +import java.util.List; +import java.util.stream.Collectors; + import org.codiki.core.entities.persistence.Category; public class CategoryDTO { private Long id; private String name; + + private List listSubCategories; public CategoryDTO() { super(); @@ -15,6 +20,8 @@ public class CategoryDTO { this(); id = pCategory.getId(); name = pCategory.getName(); + listSubCategories = pCategory.getListSubCategories().stream().map(CategoryDTO::new) + .collect(Collectors.toList()); } public Long getId() { @@ -32,5 +39,12 @@ public class CategoryDTO { public void setName(String name) { this.name = name; } - + + public List getListSubCategories() { + return listSubCategories; + } + + public void setListSubCategories(List listSubCategories) { + this.listSubCategories = listSubCategories; + } } diff --git a/src/main/java/org/codiki/core/repositories/CategoryRepository.java b/src/main/java/org/codiki/core/repositories/CategoryRepository.java index 104c8b2..b6fc9a4 100644 --- a/src/main/java/org/codiki/core/repositories/CategoryRepository.java +++ b/src/main/java/org/codiki/core/repositories/CategoryRepository.java @@ -1,10 +1,14 @@ package org.codiki.core.repositories; +import java.util.List; + import org.codiki.core.entities.persistence.Category; +import org.springframework.data.jpa.repository.Query; import org.springframework.data.repository.CrudRepository; import org.springframework.stereotype.Repository; @Repository public interface CategoryRepository extends CrudRepository { - + @Query("SELECT c FROM Category c JOIN FETCH c.listSubCategories") + List findAllWithSubCategories(); }