1 files changed, 5 insertions, 5 deletions
diff --git a/src/main/java/com/juick/www/api/PM.java b/src/main/java/com/juick/www/api/PM.java
index 55c1f2d0..96387dce 100644
--- a/src/main/java/com/juick/www/api/PM.java
+++ b/src/main/java/com/juick/www/api/PM.java
@@ -26,7 +26,7 @@ import com.juick.util.HttpBadRequestException;
 import com.juick.util.HttpForbiddenException;
 import com.juick.util.WebUtils;
 import com.juick.www.WebApp;
-import com.juick.service.PMQueriesService;
+import com.juick.service.ChatService;
 import com.juick.service.UserService;
 import com.juick.service.component.SystemEvent;
 import com.juick.service.security.annotation.Visitor;
@@ -49,7 +49,7 @@ public class PM {
     @Inject
     private UserService userService;
     @Inject
-    private PMQueriesService pmQueriesService;
+    private ChatService chatService;
     @Inject
     private ApplicationEventPublisher applicationEventPublisher;
     @Inject
@@ -68,7 +68,7 @@ public class PM {
             throw new HttpBadRequestException();
         }
 
-        List<Message> msgs = pmQueriesService.getPMMessages(visitor.getUid(), uid);
+        List<Message> msgs = chatService.getChat(visitor.getUid(), uid);
         msgs.forEach(m -> m.getUser().setAvatar(webApp.getAvatarUrl(m.getUser())));
         return msgs;
     }
@@ -91,7 +91,7 @@ public class PM {
             throw new HttpForbiddenException();
         }
 
-        if (pmQueriesService.createPM(visitor.getUid(), userTo.getUid(), body)) {
+        if (chatService.createMessage(visitor.getUid(), userTo.getUid(), body)) {
             Message jmsg = new Message();
             jmsg.setUser(visitor);
             jmsg.setText(body);
@@ -111,7 +111,7 @@ public class PM {
             @RequestParam(defaultValue = "5") int cnt) {
         // TODO: ignore cnt param for now but make sure paging param will not be cnt
 
-        List<Chat> lastconv = pmQueriesService.getLastChats(visitor);
+        List<Chat> lastconv = chatService.getLastChats(visitor);
         lastconv.forEach(c -> c.setAvatar(webApp.getAvatarUrl(c)));
         PrivateChats pms = new PrivateChats();
         pms.setUsers(lastconv);