Adding database persistence and User models

pom.xml

@@ -25,6 +25,23 @@
 	</properties>
 
 	<dependencies>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-data-jpa</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-data-rest</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-web</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>com.h2database</groupId>
+			<artifactId>h2</artifactId>
+			<scope>runtime</scope>
+		</dependency>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-websocket</artifactId>
@@ -41,7 +58,28 @@
 		    <groupId>org.springframework.boot</groupId>
 		    <artifactId>spring-boot-starter-reactor-netty</artifactId>
 		</dependency>
-
+		<dependency>
+			<groupId>org.webjars</groupId>
+			<artifactId>sockjs-client</artifactId>
+			<version>1.1.2</version>
+		</dependency>
+		<dependency>
+			<groupId>org.webjars</groupId>
+			<artifactId>stomp-websocket</artifactId>
+			<version>2.3.3</version>
+		</dependency>
+		<!--  jQuery  -->
+		<dependency>
+			<groupId>org.webjars</groupId>
+			<artifactId>jquery</artifactId>
+			<version>3.1.1</version>
+		</dependency>
+		<!-- Bootstrap -->
+		<dependency>
+			<groupId>org.webjars</groupId>
+			<artifactId>bootstrap</artifactId>
+			<version>3.3.5</version>
+		</dependency>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-devtools</artifactId>

src/main/java/com/example/websocketdemo/api/RestController.java

@@ -0,0 +1,34 @@
+package com.example.websocketdemo.api;
+
+import com.example.websocketdemo.models.Message;
+import com.example.websocketdemo.models.User;
+import com.example.websocketdemo.services.MessageService;
+import com.example.websocketdemo.services.UserService;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import java.util.List;
+
+@org.springframework.web.bind.annotation.RestController
+@RequestMapping("/api")
+public class RestController {
+
+    private final UserService userService;
+    private final MessageService messageService;
+
+    public RestController(UserService userService, MessageService messageService) {
+        this.userService = userService;
+        this.messageService=messageService;
+    }
+
+    @GetMapping("/users")
+    List<User> getAllUsers(){
+        return userService.findAll();
+    }
+
+    @GetMapping("/messages")
+    List<Message> getMessages(){
+        return messageService.findAll();
+    }
+
+}

src/main/java/com/example/websocketdemo/controller/ChatController.java

@@ -1,6 +1,12 @@
 package com.example.websocketdemo.controller;
 
-import com.example.websocketdemo.model.ChatMessage;
+import com.example.websocketdemo.models.Message;
+import com.example.websocketdemo.models.MessageDTO;
+import com.example.websocketdemo.models.User;
+import com.example.websocketdemo.services.MessageService;
+import com.example.websocketdemo.services.UserService;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.messaging.handler.annotation.MessageMapping;
 import org.springframework.messaging.handler.annotation.Payload;
 import org.springframework.messaging.handler.annotation.SendTo;
@@ -12,20 +18,40 @@
  */
 @Controller
 public class ChatController {
+    private final UserService userService;
+    private final MessageService messageService;
+    private static Logger logger= LoggerFactory.getLogger(ChatController.class);
+
+    public ChatController(UserService userService, MessageService messageService) {
+        this.userService = userService;
+        this.messageService = messageService;
+    }
 
     @MessageMapping("/chat.sendMessage")
     @SendTo("/topic/public")
-    public ChatMessage sendMessage(@Payload ChatMessage chatMessage) {
-        return chatMessage;
+    public Message sendMessage(@Payload MessageDTO messageDTO) {
+        Message message=new Message();
+        User user=userService.findByUsername(messageDTO.getUsername());
+        message.setAuthor(user);
+        message.setContent(messageDTO.getContent());
+        message.setMessageType(messageDTO.getType());
+        System.out.println(messageDTO);
+        messageService.save(message);
+        logger.info(">>>"+message.getAuthor().getUsername()+ " sent a message");
+        return message;
     }
 
     @MessageMapping("/chat.addUser")
     @SendTo("/topic/public")
-    public ChatMessage addUser(@Payload ChatMessage chatMessage,
+    public MessageDTO addUser(@Payload MessageDTO message,
                                SimpMessageHeaderAccessor headerAccessor) {
         // Add username in web socket session
-        headerAccessor.getSessionAttributes().put("username", chatMessage.getSender());
-        return chatMessage;
+        headerAccessor.getSessionAttributes().put("username", message.getUsername());
+        User user=new User();
+        user.setUsername(message.getUsername());
+        userService.save(user);
+        logger.info(message.getUsername()+" joined the chat ! Great !");
+        return message;
     }
 
 }

src/main/java/com/example/websocketdemo/controller/WebSocketEventListener.java

@@ -1,6 +1,10 @@
 package com.example.websocketdemo.controller;
 
-import com.example.websocketdemo.model.ChatMessage;
+
+import com.example.websocketdemo.models.MessageDTO;
+import com.example.websocketdemo.models.MessageType;
+import com.example.websocketdemo.models.User;
+import com.example.websocketdemo.services.UserService;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -34,10 +38,9 @@ public void handleWebSocketDisconnectListener(SessionDisconnectEvent event) {
         String username = (String) headerAccessor.getSessionAttributes().get("username");
         if(username != null) {
             logger.info("User Disconnected : " + username);
-
-            ChatMessage chatMessage = new ChatMessage();
-            chatMessage.setType(ChatMessage.MessageType.LEAVE);
-            chatMessage.setSender(username);
+            MessageDTO chatMessage = new MessageDTO();
+            chatMessage.setType(MessageType.LEAVE);
+            chatMessage.setUsername(username);
 
             messagingTemplate.convertAndSend("/topic/public", chatMessage);
         }

src/main/java/com/example/websocketdemo/models/Message.java

@@ -0,0 +1,74 @@
+package com.example.websocketdemo.models;
+
+import com.fasterxml.jackson.annotation.JsonManagedReference;
+
+import javax.persistence.Entity;
+import javax.persistence.*;
+
+import java.util.Date;
+import java.util.UUID;
+
+import static javax.persistence.FetchType.EAGER;
+
+@Entity
+public class Message {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.AUTO)
+    private UUID id;
+
+    private String content;
+    private Date created=new Date();
+
+    @JsonManagedReference
+    @ManyToOne(fetch=EAGER)
+    @JoinColumn(name="fk_author")
+    private User author;
+
+    @Enumerated(value=EnumType.STRING)
+    private MessageType messageType;
+
+    public MessageType getMessageType() {
+        return messageType;
+    }
+
+    public void setMessageType(MessageType messageType) {
+        this.messageType = messageType;
+    }
+
+    public UUID getId() {
+        return id;
+    }
+
+    public String getContent() {
+        return content;
+    }
+
+    public void setContent(String content) {
+        this.content = content;
+    }
+
+    public Date getCreated() {
+        return created;
+    }
+
+
+    public User getAuthor() {
+        return author;
+    }
+
+    public void setAuthor(User author) {
+        this.author = author;
+    }
+
+    @Override
+    public String toString() {
+        return "Message{" +
+                "id=" + id +
+                ", content='" + content + '\'' +
+                ", created=" + created +
+                ", author=" + author +
+                ", messageType=" + messageType +
+                '}';
+    }
+}

src/main/java/com/example/websocketdemo/models/MessageDTO.java

@@ -0,0 +1,43 @@
+package com.example.websocketdemo.models;
+
+public class MessageDTO {
+
+
+    private String username;
+    private String content;
+    private MessageType type;
+
+    public String getUsername() {
+        return username;
+    }
+
+    public void setUsername(String username) {
+        this.username = username;
+    }
+
+    public String getContent() {
+        return content;
+    }
+
+    public void setContent(String content) {
+        this.content = content;
+    }
+
+
+    public MessageType getType() {
+        return type;
+    }
+
+    public void setType(MessageType type) {
+        this.type = type;
+    }
+
+    @Override
+    public String toString() {
+        return "MessageDTO{" +
+                "username='" + username + '\'' +
+                ", content='" + content + '\'' +
+                ", type=" + type +
+                '}';
+    }
+}

src/main/java/com/example/websocketdemo/models/MessageType.java

@@ -0,0 +1,5 @@
+package com.example.websocketdemo.models;
+
+public enum MessageType {
+    JOIN, CHAT, LEAVE
+}

src/main/java/com/example/websocketdemo/models/User.java

@@ -0,0 +1,62 @@
+package com.example.websocketdemo.models;
+
+import com.fasterxml.jackson.annotation.JsonBackReference;
+
+import javax.persistence.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+
+@Entity
+public class User {
+
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.AUTO)
+    private UUID id;
+
+    private String username;
+
+    @JsonBackReference
+    @OneToMany(cascade=CascadeType.ALL, mappedBy="author")
+    private List<Message> messages=new ArrayList<>();
+
+    public User(){};
+
+    public User(String username) {
+        this.username = username;
+    }
+
+    public UUID getId() {
+        return id;
+    }
+
+    public String getUsername() {
+        return username;
+    }
+
+    public void setUsername(String username) {
+        this.username = username;
+    }
+
+    public List<Message> getMessages() {
+        return messages;
+    }
+
+    public void setMessages(List<Message> messages) {
+        this.messages = messages;
+    }
+
+    public void newMessage(Message message){
+        this.messages.add(message);
+    }
+
+    @Override
+    public String toString() {
+        return "User{" +
+                "id=" + id +
+                ", username='" + username + '\'' +
+                ", messages=" + messages +
+                '}';
+    }
+}

src/main/java/com/example/websocketdemo/repository/MessageRepository.java

@@ -0,0 +1,9 @@
+package com.example.websocketdemo.repository;
+
+import com.example.websocketdemo.models.Message;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+import java.util.UUID;
+
+public interface MessageRepository extends JpaRepository<Message, UUID> {
+}

src/main/java/com/example/websocketdemo/repository/UserRepository.java

@@ -0,0 +1,13 @@
+package com.example.websocketdemo.repository;
+
+import com.example.websocketdemo.models.User;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import java.util.Optional;
+import java.util.UUID;
+
+@Repository
+public interface UserRepository extends JpaRepository<User, UUID> {
+    Optional<User> findByUsername(String username);
+}

src/main/java/com/example/websocketdemo/services/MessageService.java

@@ -0,0 +1,10 @@
+package com.example.websocketdemo.services;
+
+import com.example.websocketdemo.models.Message;
+
+import java.util.List;
+
+public interface MessageService {
+    void save(Message message);
+    List<Message> findAll();
+}

src/main/java/com/example/websocketdemo/services/MessageServiceImpl.java

@@ -0,0 +1,26 @@
+package com.example.websocketdemo.services;
+
+import com.example.websocketdemo.models.Message;
+import com.example.websocketdemo.repository.MessageRepository;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+@Service
+public class MessageServiceImpl implements MessageService{
+    private final MessageRepository messageRepository;
+
+    public MessageServiceImpl(MessageRepository messageRepository) {
+        this.messageRepository = messageRepository;
+    }
+
+    @Override
+    public void save(Message message) {
+        messageRepository.save(message);
+    }
+
+    @Override
+    public List<Message> findAll() {
+        return messageRepository.findAll();
+    }
+}

src/main/java/com/example/websocketdemo/services/UserService.java

@@ -0,0 +1,11 @@
+package com.example.websocketdemo.services;
+
+import com.example.websocketdemo.models.User;
+
+import java.util.List;
+
+public interface UserService {
+    void save(User user);
+    User findByUsername(String username);
+    List<User> findAll();
+}

src/main/java/com/example/websocketdemo/services/UserServiceImpl.java

@@ -0,0 +1,38 @@
+package com.example.websocketdemo.services;
+
+import com.example.websocketdemo.models.User;
+import com.example.websocketdemo.repository.UserRepository;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.Optional;
+
+@Service
+public class UserServiceImpl implements UserService{
+
+    private final UserRepository userRepository;
+
+    public UserServiceImpl(UserRepository userRepository) {
+        this.userRepository = userRepository;
+    }
+
+    @Override
+    public void save(User user) {
+        userRepository.save(user);
+    }
+
+    @Override
+    public User findByUsername(String username) {
+        Optional<User> userOptional=userRepository.findByUsername(username);
+        if(! userOptional.isPresent()){
+            throw new RuntimeException("User not found");
+        }
+        User user=userOptional.get();
+        return user;
+    }
+
+    @Override
+    public List<User> findAll() {
+        return userRepository.findAll();
+    }
+}

src/main/resources/application.properties

@@ -0,0 +1,10 @@
+#JPA
+spring.jpa.generate-ddl=true
+spring.jpa.hibernate.dll-auto=create-drop
+
+#H2
+spring.h2.console.enabled=true
+spring.datasource.url=jdbc:h2:mem:testdb
+
+#Rest Repositories
+spring.data.rest.base-path=/api

src/main/resources/static/js/main.js

@@ -39,7 +39,7 @@ function onConnected() {
     // Tell your username to the server
     stompClient.send("/app/chat.addUser",
         {},
-        JSON.stringify({sender: username, type: 'JOIN'})
+        JSON.stringify({username: username, type: 'JOIN'})
     )
 
     connectingElement.classList.add('hidden');
@@ -57,7 +57,7 @@ function sendMessage(event) {
 
     if(messageContent && stompClient) {
         var chatMessage = {
-            sender: username,
+            username: username,
             content: messageInput.value,
             type: 'CHAT'
         };
@@ -76,22 +76,22 @@ function onMessageReceived(payload) {
 
     if(message.type === 'JOIN') {
         messageElement.classList.add('event-message');
-        message.content = message.sender + ' joined!';
+        message.content = message.username + ' joined!';
     } else if (message.type === 'LEAVE') {
         messageElement.classList.add('event-message');
-        message.content = message.sender + ' left!';
+        message.content = message.username + ' left!';
     } else {
         messageElement.classList.add('chat-message');
 
         var avatarElement = document.createElement('i');
-        var avatarText = document.createTextNode(message.sender[0]);
+        var avatarText = document.createTextNode(message.author.username[0]);
         avatarElement.appendChild(avatarText);
-        avatarElement.style['background-color'] = getAvatarColor(message.sender);
+        avatarElement.style['background-color'] = getAvatarColor(message.author.username);
 
         messageElement.appendChild(avatarElement);
 
         var usernameElement = document.createElement('span');
-        var usernameText = document.createTextNode(message.sender);
+        var usernameText = document.createTextNode(message.author.username);
         usernameElement.appendChild(usernameText);
         messageElement.appendChild(usernameElement);
     }