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

package org.ros.chatto.service;

import java.util.Date;
import java.util.List;

import org.ros.chatto.dto.ChatMessageDTO;
import org.ros.chatto.model.ChatMessage;
import org.ros.chatto.model.ChatUser;
import org.ros.chatto.model.MessageCipher;
import org.ros.chatto.repository.ChatMessageRepository;
import org.ros.chatto.repository.MessageCipherRepository;
import org.ros.chatto.repository.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Service;

@Service
public class ChatServiceImpl implements ChatService {

//	@Autowired
//	UserService userService;
	@Autowired
	UserRepository userRepository;
	@Autowired
	MessageCipherRepository messageCipherRepository;
	@Autowired
	ChatMessageRepository chatMessageRepository;
	@Autowired
	MyConversionService myConversionService;

	public void saveNewMessage(String fromUserName, String toUserName, MessageCipher messageCipher) {

		ChatUser fromUser = userRepository.findByUserName(fromUserName);
		ChatUser toUser = userRepository.findByUserName(toUserName);
		ChatMessage chatMessage = new ChatMessage();
		messageCipher = messageCipherRepository.save(messageCipher);
		chatMessage.setMessageCipher(messageCipher);
		chatMessage.setFromUser(fromUser);
		chatMessage.setToUser(toUser);

		chatMessageRepository.save(chatMessage);

	}

	@Override
	public List<ChatMessageDTO> getAllMessages(String fromUser, String toUser) {
		List<ChatMessage> chatMessages = chatMessageRepository.getAllMessages(fromUser, toUser);
		List<ChatMessageDTO> chatMessageDTOs = myConversionService.convertToChatMessageDTOs(chatMessages);
 		return chatMessageDTOs;
	}

	@Override
	public List<ChatMessageDTO> getMessagePage(int page, int size) {
//		Sort sort = Sort
		Page<ChatMessage> chatMessages = chatMessageRepository.findAll(PageRequest.of(page, size));
		List<ChatMessageDTO> chatMessageDTOs = myConversionService.convertToChatMessageDTOs(chatMessages);
		return chatMessageDTOs;
	}

	@Override
	public List<ChatMessageDTO> getNewMessages(String fromUser, String toUser, Date lastMessageTime) {
		// TODO Auto-generated method stub
		List<ChatMessage> chatMessages = chatMessageRepository.getNewMessages(fromUser, toUser, lastMessageTime);
//		List<ChatMessageDTO> chatMessageDTOs 
		return myConversionService.convertToChatMessageDTOs(chatMessages);
	}
}