Se realizan cambios en el UzyTavInstProyMapper

parent 080d86c5
package ec.edu.espe.movilidad.MovilidadWS.Mapper.Components_Class; package ec.edu.espe.movilidad.MovilidadWS.Mapper.Components_Class;
import ec.edu.espe.movilidad.MovilidadWS.Dto.DtoUzyTavInstProy; import ec.edu.espe.movilidad.MovilidadWS.Dto.DtoUzyTavInstProy;
import ec.edu.espe.movilidad.MovilidadWS.Model.ModelUzyTavInstProy; import ec.edu.espe.movilidad.MovilidadWS.Model.ModelUzyTavInstProy;
import ec.edu.espe.movilidad.MovilidadWS.Model.ModelUzyTavParametros_Det;
import org.modelmapper.ModelMapper; import org.modelmapper.ModelMapper;
import org.modelmapper.PropertyMap;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import javax.persistence.EntityManager;
import java.util.List; import java.util.List;
import java.util.stream.Collectors; import java.util.stream.Collectors;
...@@ -13,8 +17,12 @@ public class UzyTavInstProyMapper { ...@@ -13,8 +17,12 @@ public class UzyTavInstProyMapper {
private final ModelMapper mapper; private final ModelMapper mapper;
public UzyTavInstProyMapper(ModelMapper mapper) { private final EntityManager entityManager;
public UzyTavInstProyMapper(ModelMapper mapper, EntityManager entityManager) {
this.mapper = mapper; this.mapper = mapper;
this.entityManager = entityManager;
configureMappings();
} }
public DtoUzyTavInstProy entityToDto(ModelUzyTavInstProy entity) { public DtoUzyTavInstProy entityToDto(ModelUzyTavInstProy entity) {
...@@ -22,7 +30,18 @@ public class UzyTavInstProyMapper { ...@@ -22,7 +30,18 @@ public class UzyTavInstProyMapper {
} }
public ModelUzyTavInstProy dtoToEntity(DtoUzyTavInstProy dto) { public ModelUzyTavInstProy dtoToEntity(DtoUzyTavInstProy dto) {
return mapper.map(dto, ModelUzyTavInstProy.class); ModelUzyTavInstProy entity = mapper.map(dto, ModelUzyTavInstProy.class);
Long idEntidad = dto.getUzytparametros_det_id_entidad();
Long idOrganizacion = dto.getUzytparametros_det_id_organizacion();
if (idEntidad != null) {
ModelUzyTavParametros_Det parametrosDetEntidad = entityManager.find(ModelUzyTavParametros_Det.class, idEntidad);
entity.setUzytparametrosDetIdEntidad(parametrosDetEntidad);
}
if (idOrganizacion != null) {
ModelUzyTavParametros_Det parametrosDetOrganizacion = entityManager.find(ModelUzyTavParametros_Det.class, idOrganizacion);
entity.setUzytparametrosDetIdOrganizacion(parametrosDetOrganizacion);
}
return entity;
} }
public List<DtoUzyTavInstProy> entitiesToDtos(List<ModelUzyTavInstProy> entities) { public List<DtoUzyTavInstProy> entitiesToDtos(List<ModelUzyTavInstProy> entities) {
...@@ -30,4 +49,15 @@ public class UzyTavInstProyMapper { ...@@ -30,4 +49,15 @@ public class UzyTavInstProyMapper {
.map(this::entityToDto) .map(this::entityToDto)
.collect(Collectors.toList()); .collect(Collectors.toList());
} }
private void configureMappings() {
mapper.addMappings(new PropertyMap<ModelUzyTavInstProy, DtoUzyTavInstProy>() {
@Override
protected void configure() {
map().setUzytparametros_det_id_organizacion(source.getUzytparametrosDetIdOrganizacion().getUzytparametros_det_id());
map().setUzytparametros_det_id_entidad(source.getUzytparametrosDetIdEntidad().getUzytparametros_det_id());
}
});
}
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment