HorseMapper.java 1.08 KB
Newer Older
1 2 3 4 5 6
package at.ac.tuwien.sepm.assignment.individual.endpoint.mapper;

import at.ac.tuwien.sepm.assignment.individual.endpoint.dto.HorseDto;
import at.ac.tuwien.sepm.assignment.individual.entity.Horse;
import org.springframework.stereotype.Component;

7 8 9
import java.util.ArrayList;
import java.util.List;

10 11 12
@Component
public class HorseMapper {
    public HorseDto entityToDto(Horse horse) {
13
        return new HorseDto(horse.getId(), horse.getName(), horse.getDescription(), horse.getScore(), horse.getBirthday(), horse.getRace(), horse.getImagePath(), horse.getCreatedAt(), horse.getUpdatedAt(), horse.getOwner());
14 15 16
    }

    public Horse dtoToEntity(HorseDto horse) {
17
        return new Horse(horse.getId(), horse.getName(), horse.getDescription(), horse.getScore(), horse.getBirthday(), horse.getRace(), horse.getImagePath(), horse.getOwner());
18
    }
19 20 21 22 23 24 25 26 27

    public List<HorseDto> entityListToDtoList(List<Horse> horseEntities) {
        List<HorseDto> horseDtos = new ArrayList<>();

        for(Horse horse: horseEntities) {
            horseDtos.add(entityToDto(horse));
        }
        return horseDtos;
    }
28
}