diff --git a/src/main/java/ru/kayashov/bar/controller/dto/cocktail/CocktailSimpleResponseDto.java b/src/main/java/ru/kayashov/bar/controller/dto/cocktail/CocktailSimpleResponseDto.java index 24d500f..42fea2d 100644 --- a/src/main/java/ru/kayashov/bar/controller/dto/cocktail/CocktailSimpleResponseDto.java +++ b/src/main/java/ru/kayashov/bar/controller/dto/cocktail/CocktailSimpleResponseDto.java @@ -9,12 +9,14 @@ import ru.kayashov.bar.model.entity.CocktailEntity; public class CocktailSimpleResponseDto { private Long id; private String name; + private String image; private Boolean hasError; public static CocktailSimpleResponseDto mapToDto(CocktailEntity e) { CocktailSimpleResponseDto d = new CocktailSimpleResponseDto(); d.setId(e.getId()); d.setName(e.getName()); + d.setImage(e.getImage()); boolean hasError = e.getReceipt().stream() .anyMatch(receiptEntity -> receiptEntity.getUnit() == null || receiptEntity.getCount() == null); d.setHasError(hasError);