WebSocket (转载)

开发环境:Windows x64,Eclipse 4.6.2,Tomcat 7.0.79,Jdk 1.8。

介绍

现很多网站为了实现即时通讯,所用的技术都是轮询(polling)。轮询是在特定的的时间间隔(如每1秒),由浏览器对服务器发出HTTP request,然后由服务器返回最新的数据给客服端的浏览器。这种传统的HTTP request 的模式带来很明显的缺点 – 浏览器需要不断的向服务器发出请求,然而HTTP request 的header是非常长的,里面包含的数据可能只是一个很小的值,这样会占用很多的带宽。

而最比较新的技术去做轮询的效果是Comet – 用了AJAX。但这种技术虽然可达到全双工通信,但依然需要发出请求。

在 WebSocket API,浏览器和服务器只需要要做一个握手的动作,然后,浏览器和服务器之间就形成了一条快速通道。两者之间就直接可以数据互相传送。

依赖:

Tomcat 7 或者 J2EE7

我用的tomcat 7.0.79和Java jdk1.8.0(1.7Linux服务器架设有问题),我使用的时候,没有需要自己手动引用别的jar包,tomcat默认jar包就可以运行。

遇到的问题:

开始用的eclipse4.3版本,不支持jdk1.8,后改用的4.6.2。

虽然tomcat7.0.47就已支持WebSocket,但是tomcat7.0.47版本只支持jdk1.6,所以改用tomcat7.0.79.

注意:早前业界没有统一的标准,各服务器都有各自的实现,现在J2EE7的JSR356已经定义了统一的标准,请尽量使用支持最新通用标准的服务器。

详见:

http://www.oracle.com/technetwork/articles/java/jsr356-1937161.html

http://jinnianshilongnian.iteye.com/blog/1909962

用nginx做反向代理的需要注意啦,socket请求需要做特殊配置的,切记!

##服务端

服务端使用注解模式后,不需要在web.xml中做额外的配置,Tomcat启动后就可以直接连接了(去掉注解,使用web.xml中配置后,启动服务报错cannot be cast to javax.servlet.Servlet)。

源码:

package com.hdtytech.websocket;

import java.io.IOException;

import javax.websocket.OnClose;

import javax.websocket.OnError;

import javax.websocket.OnMessage;

import javax.websocket.OnOpen;

import javax.websocket.Session;

import javax.websocket.server.ServerEndpoint;

@ServerEndpoint(value = "/websocket") 

public class MyWebSocket{

    //与某个客户端的连接会话,需要通过它来给客户端发送数据

    private Session session;

    /**

    * 连接建立成功调用的方法

    * @param session  可选的参数。session为与某个客户端的连接会话,需要通过它来给客户端发送数据

    * @throws Exception

    */

    @OnOpen

    public void onOpen(Session session) throws Exception{

        this.session = session;

        WebSocketMapUtil.put(session.getQueryString(),this);

    }


    /**

    * 连接关闭调用的方法

    * @throws Exception

    */

    @OnClose

    public void onClose() throws Exception{

    //从map中删除

    WebSocketMapUtil.remove(session.getQueryString());

    }


    /**

    * 收到客户端消息后调用的方法

    * @param message 客户端发送过来的消息

    * @param session 可选的参数

    * @throws IOException

    */

    @OnMessage

    public void onMessage(String message, Session session) throws IOException {

        try {

        MyWebSocket myWebSocket= ((MyWebSocket) WebSocketMapUtil.get(session.getQueryString().replace("service", "client")));

        if(myWebSocket != null){

        myWebSocket.sendMessage(message);

        }

        } catch (IOException e) {

            e.printStackTrace();

        }

    }


    /**

    * 发生错误时调用

    * @param session

    * @param error

    */

    @OnError

    public void onError(Session session, Throwable error){

        error.printStackTrace();

    }



    /**

    * 发送消息方法。

    * @param message

    * @throws IOException

    */

    public void sendMessage(String message) throws IOException{

        this.session.getBasicRemote().sendText(message);

    }


    /**

    * 群发消息方法。

    * @param message

    * @throws IOException

    */

    public void sendMessageAll(String message) throws IOException{

    for(MyWebSocket myWebSocket : WebSocketMapUtil.getValues()){

    myWebSocket.sendMessage(message);

    }

    }

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

##客户端

源码:

<%@ page language="java" import="java.util.*" pageEncoding="UTF-8"%>

<%

String path = request.getContextPath();

String basePath = request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+path+"/";

%>

<!DOCTYPE HTML>

<html>

  <head>

    <base href="<%=basePath%>">

    <title>My WebSocket</title>

  </head>


  <body>

    Welcome<br/>

    <input id="text" type="text" /><button onclick="send()">Send</button>    <button onclick="closeWebSocket()">Close</button>

    <div id="message">

    </div>

  </body>


  <script type="text/JavaScript">

      var websocket = null;

      //判断当前浏览器是否支持WebSocket

      if('WebSocket' in window){

          websocket = new WebSocket("ws://localhost:8080/MyWebSocket/websocket?client123");

      }

      else{

          alert('Not support websocket');

      }


      //连接发生错误的回调方法

      websocket.onerror = function(){

          setMessageInnerHTML("error");

      };


      //连接成功建立的回调方法

      websocket.onopen = function(event){

          setMessageInnerHTML("open");

      };


      //接收到消息的回调方法

      websocket.onmessage = function(event){

          setMessageInnerHTML(event.data);

      };


      //连接关闭的回调方法

      websocket.onclose = function(){

          setMessageInnerHTML("close");

      };


      //监听窗口关闭事件,当窗口关闭时,主动去关闭websocket连接,防止连接还没断开就关闭窗口,server端会抛异常。

      window.onbeforeunload = function(){

          websocket.close();

      };


      //将消息显示在网页上

      function setMessageInnerHTML(innerHTML){

          document.getElementById('message').innerHTML += innerHTML + '<br/>';

      }


      //关闭连接

      function closeWebSocket(){

          websocket.close();

      }


      //发送消息

      function send(){

          var message = document.getElementById('text').value;

          websocket.send(message);

      }

  </script>

</html>

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

##工具类ConcurrentMap

用来存储唯一key和连接(ConcurrentMap是线性安全的Map)

这个是我业务的需要,因为我的业务是服务器需要主动点对点发送消息给客户端,所以需要用Map来存储客户端连接。

遇到的问题:

本来想直接调用服务端的发送接口,从Map中取到需要发送的客户端连接信息,但是只要在外部调用MapUtil,MapUtil就会重新实例化,Map为空,也就是把连接放在单独静态Map中不可行。

后来也考虑把Map放在session中,但是考虑到session容易丢失,就放弃了这个想法。

又考虑使用memcache,但是在往memcache放的时候,报需要序列化的错,也放弃了这个想法。

解决办法:

后来就模仿微信聊天一样,把服务端做为一个中间转发消息的中间件,在服务端定义一个客户端类,在需要给客户端发送消息时,调用客户端类, 发送消息到中间件,通过中间件转发消息到你想发送的客户端,若需要取法消息,则可便利Map客户端进行群发通知公告等。

源码:

package com.hdtytech.websocket;

import java.util.Collection;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.ConcurrentMap;

public class WebSocketMapUtil {

public static ConcurrentMap<String, MyWebSocket> webSocketMap = new ConcurrentHashMap<>();

    public static void put(String key, MyWebSocket myWebSocket){

    webSocketMap.put(key, myWebSocket);

    }


    public static MyWebSocket get(String key){

    return webSocketMap.get(key);

    }


public static void remove(String key){

webSocketMap.remove(key);

}

public static Collection<MyWebSocket> getValues(){

return webSocketMap.values();

}

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

##服务端用于主动发消息的客户端类

源码:

package com.hdtytech.websocket;

import java.io.IOException;

import java.net.URI;

import javax.websocket.ClientEndpoint;

import javax.websocket.ContainerProvider;

import javax.websocket.OnClose;

import javax.websocket.OnError;

import javax.websocket.OnMessage;

import javax.websocket.OnOpen;

import javax.websocket.Session;

import javax.websocket.WebSocketContainer;

@ClientEndpoint

public class MyClient {

private Session session;

@OnOpen

public void onOpen(Session session) throws IOException {

this.session = session;

}

@OnMessage

public void onMessage(String message) {

}

@OnError

public void onError(Throwable t) {

t.printStackTrace();

}

/**

    * 连接关闭调用的方法

    * @throws Exception

    */

    @OnClose

    public void onClose() throws Exception{

    }


    /**

    * 关闭链接方法

    * @param message

    * @throws IOException

    */

    public void closeSocket() throws IOException{

        this.session.close();

    }


    /**

    * 发送消息方法。

    * @param message

    * @throws IOException

    */

    public void sendMessage(String message) throws IOException{

        this.session.getBasicRemote().sendText(message);

    }

    //启动客户端并建立链接

    public void start(String uri) {

WebSocketContainer container = ContainerProvider.getWebSocketContainer();

try {

this.session = container.connectToServer(MyClient.class, URI.create(uri));

} catch (Exception e) {

e.printStackTrace();

}

}

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

##测试main函数

说明:为了测试,写了一个timer,每个两秒,调用发送给客户端发送一条消息;

源码:

package com.hdtytech.websocket;

import java.io.IOException;

import java.util.Timer;

import java.util.TimerTask;

public class MyClientApp {

public static void main(String[] args){

// TODO todo.generated by zoer 

        Timer timer = new Timer(); 

        timer.schedule(new MyTask(), 1000, 2000); 

}

static int num = 0;

static class MyTask extends TimerTask { 


    @Override 

    public void run() { 

    MyClient client = new MyClient();

String uri = "ws://localhost:8080/MyWebSocket/websocket?service123";

client.start(uri);

try {

num++;

client.sendMessage("消息测试"+num);

client.closeSocket();

System.out.println(num);

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

    } 


}

}

————————————————

版权声明:本文为CSDN博主「君子丨莫笑」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。

原文链接:https://blog.csdn.net/gaolele_92/article/details/76158087

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容