use of codeu.model.data.Message in project CodeU-Spring-2018 by jwang281.
the class PersistentDataStore method loadMessages.
/**
* Loads all Message objects from the Datastore service and returns them in a List.
*
* @throws PersistentDataStoreException if an error was detected during the load from the
* Datastore service
*/
public List<Message> loadMessages() throws PersistentDataStoreException {
List<Message> messages = new ArrayList<>();
// Retrieve all messages from the datastore.
Query query = new Query("chat-messages");
PreparedQuery results = datastore.prepare(query);
for (Entity entity : results.asIterable()) {
try {
UUID uuid = UUID.fromString((String) entity.getProperty("uuid"));
UUID conversationUuid = UUID.fromString((String) entity.getProperty("conv_uuid"));
UUID authorUuid = UUID.fromString((String) entity.getProperty("author_uuid"));
Instant creationTime = Instant.parse((String) entity.getProperty("creation_time"));
String content = (String) entity.getProperty("content");
Message message = new Message(uuid, conversationUuid, authorUuid, content, creationTime);
messages.add(message);
} catch (Exception e) {
// database entity definition mismatches, or service mismatches.
throw new PersistentDataStoreException(e);
}
}
return messages;
}
use of codeu.model.data.Message in project CodeU-Spring-2018 by jwang281.
the class ChatServlet method doPost.
/**
* This function fires when a user submits the form on the chat page. It gets the logged-in
* username from the session, the conversation title from the URL, and the chat message from the
* submitted form data. It creates a new Message from that data, adds it to the model, and then
* redirects back to the chat page.
*/
@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
String username = (String) request.getSession().getAttribute("user");
if (username == null) {
// user is not logged in, don't let them add a message
response.sendRedirect("/login");
return;
}
User user = userStore.getUser(username);
if (user == null) {
// user was not found, don't let them add a message
response.sendRedirect("/login");
return;
}
String requestUrl = request.getRequestURI();
String conversationTitle = requestUrl.substring("/chat/".length());
Conversation conversation = conversationStore.getConversationWithTitle(conversationTitle);
if (conversation == null) {
// couldn't find conversation, redirect to conversation list
response.sendRedirect("/conversations");
return;
}
String messageContent = request.getParameter("message");
// this removes any HTML from the message content
String cleanedMessageContent = Jsoup.clean(messageContent, Whitelist.none());
Message message = new Message(UUID.randomUUID(), conversation.getId(), user.getId(), cleanedMessageContent, Instant.now());
messageStore.addMessage(message);
// redirect to a GET request
response.sendRedirect("/chat/" + conversationTitle);
}
use of codeu.model.data.Message in project CodeU-Spring-2018 by jwang281.
the class ChatServlet method doGet.
/**
* This function fires when a user navigates to the chat page. It gets the conversation title from
* the URL, finds the corresponding Conversation, and fetches the messages in that Conversation.
* It then forwards to chat.jsp for rendering.
*/
@Override
public void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
String requestUrl = request.getRequestURI();
String conversationTitle = requestUrl.substring("/chat/".length());
Conversation conversation = conversationStore.getConversationWithTitle(conversationTitle);
if (conversation == null) {
// couldn't find conversation, redirect to conversation list
System.out.println("Conversation was null: " + conversationTitle);
response.sendRedirect("/conversations");
return;
}
UUID conversationId = conversation.getId();
List<Message> messages = messageStore.getMessagesInConversation(conversationId);
request.setAttribute("conversation", conversation);
request.setAttribute("messages", messages);
request.getRequestDispatcher("/WEB-INF/view/chat.jsp").forward(request, response);
}
use of codeu.model.data.Message in project CodeU-Spring-2018 by maksymko.
the class PersistentDataStoreTest method testSaveAndLoadMessages.
@Test
public void testSaveAndLoadMessages() throws PersistentDataStoreException {
UUID idOne = UUID.randomUUID();
UUID conversationOne = UUID.randomUUID();
UUID authorOne = UUID.randomUUID();
String contentOne = "test content one";
Instant creationOne = Instant.ofEpochMilli(1000);
Message inputMessageOne = new Message(idOne, conversationOne, authorOne, contentOne, creationOne);
UUID idTwo = UUID.randomUUID();
UUID conversationTwo = UUID.randomUUID();
UUID authorTwo = UUID.randomUUID();
String contentTwo = "test content one";
Instant creationTwo = Instant.ofEpochMilli(2000);
Message inputMessageTwo = new Message(idTwo, conversationTwo, authorTwo, contentTwo, creationTwo);
// save
persistentDataStore.writeThrough(inputMessageOne);
persistentDataStore.writeThrough(inputMessageTwo);
// load
List<Message> resultMessages = persistentDataStore.loadMessages();
// confirm that what we saved matches what we loaded
Message resultMessageOne = resultMessages.get(0);
Assert.assertEquals(idOne, resultMessageOne.getId());
Assert.assertEquals(conversationOne, resultMessageOne.getConversationId());
Assert.assertEquals(authorOne, resultMessageOne.getAuthorId());
Assert.assertEquals(contentOne, resultMessageOne.getContent());
Assert.assertEquals(creationOne, resultMessageOne.getCreationTime());
Message resultMessageTwo = resultMessages.get(1);
Assert.assertEquals(idTwo, resultMessageTwo.getId());
Assert.assertEquals(conversationTwo, resultMessageTwo.getConversationId());
Assert.assertEquals(authorTwo, resultMessageTwo.getAuthorId());
Assert.assertEquals(contentTwo, resultMessageTwo.getContent());
Assert.assertEquals(creationTwo, resultMessageTwo.getCreationTime());
}
use of codeu.model.data.Message in project CodeU-Spring-2018 by maksymko.
the class PersistentStorageAgentTest method testWriteThroughMessage.
@Test
public void testWriteThroughMessage() {
Message message = new Message(UUID.randomUUID(), UUID.randomUUID(), UUID.randomUUID(), "test content", Instant.now());
persistentStorageAgent.writeThrough(message);
Mockito.verify(mockPersistentDataStore).writeThrough(message);
}
Aggregations