Files
OpenIsle/backend/src/main/java/com/openisle/mapper/DraftMapper.java
2025-08-04 21:10:09 +08:00

25 lines
698 B
Java

package com.openisle.mapper;
import com.openisle.dto.DraftDto;
import com.openisle.model.Draft;
import org.springframework.stereotype.Component;
import java.util.stream.Collectors;
/** Mapper for draft entities. */
@Component
public class DraftMapper {
public DraftDto toDto(Draft draft) {
DraftDto dto = new DraftDto();
dto.setId(draft.getId());
dto.setTitle(draft.getTitle());
dto.setContent(draft.getContent());
if (draft.getCategory() != null) {
dto.setCategoryId(draft.getCategory().getId());
}
dto.setTagIds(draft.getTags().stream().map(tag -> tag.getId()).collect(Collectors.toList()));
return dto;
}
}