Added method to get contacts, although as of now all Users are returned

This commit is contained in:
delvh 2020-01-18 13:28:54 +01:00
parent 3df01532d8
commit c5f8c3c6be
3 changed files with 24 additions and 4 deletions

View File

@ -8,6 +8,7 @@ import javax.persistence.Entity;
import javax.persistence.GeneratedValue; import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType; import javax.persistence.GenerationType;
import javax.persistence.Id; import javax.persistence.Id;
import javax.persistence.NamedQueries;
import javax.persistence.NamedQuery; import javax.persistence.NamedQuery;
import javax.persistence.Table; import javax.persistence.Table;
import javax.persistence.Temporal; import javax.persistence.Temporal;
@ -28,7 +29,11 @@ import javax.persistence.TemporalType;
*/ */
@Entity @Entity
@Table(name = "users") @Table(name = "users")
@NamedQuery(query = "SELECT u FROM User u WHERE u.id = :id", name = "getUserById") @NamedQueries(
{ @NamedQuery(query = "SELECT u FROM User u WHERE u.id = :id", name = "getUserById"),
@NamedQuery(query = "SELECT u.contacts FROM User u WHERE u = :user", name = "getContactsOfUser")// not tested
}
)
public class User { public class User {
@Id @Id

View File

@ -98,7 +98,17 @@ public class PersistenceManager {
*/ */
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public List<Message> getUnreadMessages(User user) { public List<Message> getUnreadMessages(User user) {
// TODO may need to be changed to clientId
return entityManager.createNamedQuery("getUnreadMessages").setParameter("recipient", user).getResultList(); return entityManager.createNamedQuery("getUnreadMessages").setParameter("recipient", user).getResultList();
} }
/**
* @param user the User whose contacts should be retrieved
* @return the contacts of this User - currently everyone using Envoy
* @since Envoy Server Standalone v0.1-alpha
*/
@SuppressWarnings("unchecked")
public List<User> getContacts(User user) { return entityManager.createQuery("FROM User").getResultList(); }
// TODO current solution gets all users, not just contacts. Should be changed to
// entityManager.createNamedQuery("getContactsOfUser").setParameter("user",
// user).getResultList();
} }

View File

@ -1,9 +1,9 @@
package envoy.server.processors; package envoy.server.processors;
import java.io.IOException; import java.io.IOException;
import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.List; import java.util.List;
import java.util.stream.Collectors;
import envoy.data.Contacts; import envoy.data.Contacts;
import envoy.data.LoginCredentials; import envoy.data.LoginCredentials;
@ -42,7 +42,12 @@ public class LoginCredentialProcessor implements ObjectProcessor<LoginCredential
ConnectionManager.getInstance().registerUser(socketId, user.getId()); ConnectionManager.getInstance().registerUser(socketId, user.getId());
// Create contacts // Create contacts
Contacts contacts = new Contacts(user.getId(), new ArrayList<>()); List<User> users = PersistenceManager.getPersistenceManager()
.getContacts(new envoy.server.data.User(user))
.stream()
.map(envoy.server.data.User::toCommonUser)
.collect(Collectors.toList());
Contacts contacts = new Contacts(user.getId(), users);
// Complete handshake // Complete handshake
System.out.println("Sending user..."); System.out.println("Sending user...");