注册后自动登录

godaner · Jan 4, 2018 · b5f661e · b5f661e
1 parent 94e3ef8
commit b5f661e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/vm-frontend/src/main/java/com/vm/frontend/websocket/OnlineUsersWebSocket.java b/vm-frontend/src/main/java/com/vm/frontend/websocket/OnlineUsersWebSocket.java
@@ -74,10 +74,11 @@ public void close() throws IOException {
     public static void userLogin(Long userId) throws IOException {
 
         OnlineUsersWebSocket client = cacheOfClients.get(userId);
-        cacheOfClients.remove(userId);
 
         onlineClients.put(userId, client);
 
+        cacheOfClients.remove(userId);
+
         //通知客户端登陆成功
         OutMessage message = new OutMessage();
         message.setUserId(userId);

diff --git a/vm-frontend/src/main/resources/static/components/head.jsx b/vm-frontend/src/main/resources/static/components/head.jsx
@@ -45,6 +45,11 @@ var Head = React.createClass({
     },
     onRegistSuccess: function (user) {
         this.updateStateUser(user);
+        //open ws
+        this.wsOpen(user.id, function () {
+            //ajax ws
+            this.wsLogin();
+        }.bind(this));
     },
     updateStateUser(user){
         //when login success reset user

diff --git a/vm-frontend/src/main/resources/static/dist/bundle.js b/vm-frontend/src/main/resources/static/dist/bundle.js
@@ -30659,6 +30659,11 @@ var Head = _react2.default.createClass({
     },
     onRegistSuccess: function onRegistSuccess(user) {
         this.updateStateUser(user);
+        //open ws
+        this.wsOpen(user.id, function () {
+            //ajax ws
+            this.wsLogin();
+        }.bind(this));
     },
     updateStateUser: function updateStateUser(user) {
         //when login success reset user