基于Spring mvc 的Websocket 开发

时间：2023-03-09 01:10:04
基于Spring mvc 的Websocket 开发

1、Pom.xml

 <dependency>

            <groupId>org.springframework</groupId>

            <artifactId>spring-websocket</artifactId>

            <version>4.3.2.RELEASE</version>

        </dependency>

        <dependency>

            <groupId>org.springframework</groupId>

            <artifactId>spring-messaging</artifactId>

            <version>4.3.2.RELEASE</version>

        </dependency>

        <dependency>

            <groupId>javax.websocket</groupId>

            <artifactId>javax.websocket-api</artifactId>

            <version>1.0</version>

            <scope>provided</scope>

        </dependency>

2、Spring-servlet,xml

 <context:component-scan base-package="com.websocket"/>

3、WebSocketConfig.java

package com.websocket;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.web.servlet.config.annotation.EnableWebMvc;

import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

import org.springframework.web.socket.config.annotation.EnableWebSocket;

import org.springframework.web.socket.config.annotation.WebSocketConfigurer;

import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

/**

 * Created by HendSame on 11/29/16.

 */

@Configuration

@EnableWebMvc

@EnableWebSocket

public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {

    @Override

    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {

        //允许连接的域,只能以http或https开头

        String[] allowsOrigins = {"*"};

        //WebIM WebSocket通道

        registry.addHandler(allBedWebSocketHandler(),"/websocket/get/all/data/rt").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor());

        registry.addHandler(allBedWebSocketHandler(), "/sockjs/websocket/get/all/data/rt").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor()).withSockJS();

        registry.addHandler(bedWebSocketHandler(),"/websocket/get/by/time").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor());

        registry.addHandler(bedWebSocketHandler(), "/sockjs/websocket/get/by/time").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor()).withSockJS();

        registry.addHandler(alarmsWebSocketHandler(),"/websocket/get/alarms").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor());

        registry.addHandler(alarmsWebSocketHandler(), "/sockjs/websocket/get/alarms").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor()).withSockJS();

        registry.addHandler(overViewWebSocketHandler(),"/websocket/get/overview").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor());

        registry.addHandler(overViewWebSocketHandler(), "/sockjs/websocket/get/overview").setAllowedOrigins(allowsOrigins).addInterceptors(myInterceptor()).withSockJS();

    }

    @Bean

    public AllBedWebSocketHandler allBedWebSocketHandler() {

        return new AllBedWebSocketHandler();

    }

    @Bean

    public BedWebSocketHandler bedWebSocketHandler() {

        return new BedWebSocketHandler();

    }

    @Bean

    public AlarmsWebSocketHandler alarmsWebSocketHandler() {

        return new AlarmsWebSocketHandler();

    }

    @Bean

    public OverViewWebSocketHandler overViewWebSocketHandler() {

        return new OverViewWebSocketHandler();

    }

    @Bean

    public WebSocketHandshakeInterceptor myInterceptor(){

        return new WebSocketHandshakeInterceptor();

    }

}

4、BedWebSocketHandler.java

package com.websocket;

import com.alibaba.fastjson.JSON;

import com.alibaba.fastjson.JSONObject;

import com.bbkj.nursestation.common.ApiResult;

import com.bbkj.nursestation.common.UserVO;

import com.bbkj.nursestation.site.service.DeviceService;

import com.bbkj.nursestation.site.vi.HospitalBedDataVI;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import org.springframework.web.socket.CloseStatus;

import org.springframework.web.socket.TextMessage;

import org.springframework.web.socket.WebSocketSession;

import org.springframework.web.socket.handler.TextWebSocketHandler;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.Map;

/**

 * Created by hy on 11/29/16.

 */

@Component

public class BedWebSocketHandler extends TextWebSocketHandler {

    private final static List<WebSocketSession> sessions = Collections.synchronizedList(new ArrayList<WebSocketSession>());

    @Autowired

    private DeviceService deviceService;

    //接收文本消息，并发送出去

    @Override

    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {

        super.handleTextMessage(session, message);

        byte[] bytes = message.asBytes();

        String params = new String(bytes);

        JSONObject object=JSON.parseObject(params);

        UserVO userVO = new UserVO();

        userVO = (UserVO) session.getAttributes().get("websocketUser");

        ApiResult<Map<String, Integer>> apiResult = new ApiResult<>(1, "error");

        HospitalBedDataVI hospitalBedDataVI = new HospitalBedDataVI();

        hospitalBedDataVI.setEn(Long.parseLong(object.get("en").toString()));

        hospitalBedDataVI.setPaitenid(Long.parseLong(object.get("paitenid").toString()));

        apiResult = deviceService.getHosptialBedRealTimeData4Websocket(hospitalBedDataVI);

        apiResult.setUser(userVO);

        session.sendMessage(new TextMessage(apiResult.toString()));

    }

    //连接建立后处理

    @SuppressWarnings("unchecked")

    @Override

    public void afterConnectionEstablished(WebSocketSession session) throws Exception {

        sessions.add(session);

        //处理离线消息

    }

    //抛出异常时处理

    @Override

    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {

        if(session.isOpen()){

            session.close();

        }

        sessions.remove(session);

    }

    //连接关闭后处理

    @Override

    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {

        sessions.remove(session);

    }

    @Override

    public boolean supportsPartialMessages() {

        return false;

    }

    /**

     * 给所有在线用户发送消息

     *

     * @param message

     */

    public void sendMessageToUsers(TextMessage message) {

        for (WebSocketSession user : sessions) {

            try {

                if (user.isOpen()) {

                    user.sendMessage(message);

                }

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

    }

    /**

     * 给某个用户发送消息

     *

     * @param userName

     * @param message

     */

    public void sendMessageToUser(String userName, TextMessage message) {

        for (WebSocketSession user : sessions) {

            if (user.getAttributes().get("username").equals(userName)) {

                try {

                    if (user.isOpen()) {

                        user.sendMessage(message);

                    }

                } catch (IOException e) {

                    e.printStackTrace();

                }

                break;

            }

        }

    }

}

5、DeviceService.java

package com.site.service;

import com.bbkj.nursestation.common.ApiResult;

import com.bbkj.nursestation.site.entity.Device;

import com.bbkj.nursestation.site.vi.HospitalBedDataVI;

import com.bbkj.nursestation.site.vo.HosptialBedDataNow;

import com.bbkj.nursestation.site.vo.OverviewVO;

import java.util.List;

import java.util.Map;

/**

 * Created by hy on 11/21/16.

 */

public interface DeviceService {

    ApiResult<Map<String,Integer>> getHosptialBedRealTimeData4Websocket(HospitalBedDataVI hospitalBedDataVI);

}

7、Javascript

<script type="text/javascript" src="js/sockjs-1.1.1.min.js"></script>

    <script>

            var websocket;

            if ('WebSocket' in window) {

              //某张床位实时信息接口

              websocket = new WebSocket("ws://192.168.1.114:8099/websocket/get/by/time");

             } else if ('MozWebSocket' in window) {

             websocket = new MozWebSocket("ws://192.168.1.114:8099/websocket/get/by/time");

            } else {

             websocket = new SockJS("http://192.168.1.114:8099/sockjs/websocket/get/by/time");

            }

            websocket.onopen = function (evnt) {

                alert("建立连接");

            };

            websocket.onmessage = function (evnt) {

                alert(evnt.data)

            };

            websocket.onerror = function (evnt) {

            };

            websocket.onclose = function (evnt) {

        }

        $('#send').click(function(e) {

            var params ="{'en':1,'paitenid':1}";

            websocket.send(params);

    });

 </script>

相关文章