Chatto/chatto/src/main/java/org/ros/chatto/service/MyConversionService.java

package org.ros.chatto.service;

import java.util.List;
import java.util.stream.Collectors;

import org.modelmapper.ModelMapper;
import org.ros.chatto.dto.ChatMessageDTO;
import org.ros.chatto.model.ChatMessage;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.stereotype.Service;


//private PostDto convertToDto(Post post) {
//    PostDto postDto = modelMapper.map(post, PostDto.class);
//    postDto.setSubmissionDate(post.getSubmissionDate(), 
//        userService.getCurrentUser().getPreference().getTimezone());
//    return postDto;
//}
//
//	
//private Post convertToEntity(PostDto postDto) throws ParseException {
//    Post post = modelMapper.map(postDto, Post.class);
//    post.setSubmissionDate(postDto.getSubmissionDateConverted(
//      userService.getCurrentUser().getPreference().getTimezone()));
//  
//    if (postDto.getId() != null) {
//        Post oldPost = postService.getPostById(postDto.getId());
//        post.setRedditID(oldPost.getRedditID());
//        post.setSent(oldPost.isSent());
//    }
//    return post;
//}
@Service
public class MyConversionService {
	@Autowired
	ModelMapper modelMapper;
	
	public ChatMessageDTO convertToChatMessageDTO(ChatMessage chatMessage)
	{
		ChatMessageDTO chatMessageDTO = modelMapper.map(chatMessage, ChatMessageDTO.class);
		return chatMessageDTO;
	}
	
	public List<ChatMessageDTO> convertToChatMessageDTOs(List<ChatMessage> chatMessages)
	{
		return chatMessages.stream()
				.map(chatMessage -> convertToChatMessageDTO(chatMessage))
				.collect(Collectors.toList());
	}
	
	public List<ChatMessageDTO> convertToChatMessageDTOs(Page<ChatMessage> chatMessages)
	{
		return chatMessages.stream()
				.map(chatMessage -> convertToChatMessageDTO(chatMessage))
				.collect(Collectors.toList());
	}
}