linuxsir首页 LinuxSir.Org | Linux、BSD、Solaris、Unix | 开源传万世,因有我参与欢迎您!
网站首页 | 设为首页 | 加入收藏
您所在的位置:主页 > Linux基础建设 >

用Java分别实现WebSocket客户端与服务端

时间:2019-02-11  来源:未知  作者:admin666

 

最近公司在搞一个项目刚好需要用到WebSocket技术来实现实时数据的传输,因为之前也没接触过,所以捣鼓了好些天,最近恰巧有空就写写。有误的地方还请大牛们能及时指正。

 

项目背景:基于spring+spring MVC+mybatis框架的maven项目

 

服务端:

 

1、添加Jar包依赖:

 

<dependency>
  <groupId>javax.websocket</groupId>
  <artifactId>javax.websocket-api</artifactId>
  <version>1.1</version>
  <scope>provided</scope>
</dependency>

 

2、创建一个WebSocket服务端类MyWebSocketServer,并在类前添加@ServerEndpoint(value = "/websocket")注解,@ServerEndpoint(value = "/websocket")注释端点表示将 WebSocket 服务端运行在 ws://[Server 端 IP 或域名]:[Server 端口]/项目名/websocket 的访问端点

 

3、实现onOpen、onClose、onMessage、onError等方法

 

实例代码:

 

MyWebSocketServer 类

 

import java.io.IOException;
import javax.websocket.EncodeException;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import net.sf.json.JSONObject;

 

@ServerEndpoint(value = "/websocket")
public class MyWebSocketServer {

 

  private Logger logger = Logger.getLogger(MyWebSocketServer.class);
  private Session session;

 

  /**
   * 连接建立后触发的方法
   */
  @OnOpen
  public void onOpen(Session session){
    this.session = session;
    logger.info("onOpen"+session.getId());
    WebSocketMapUtil.put(session.getId(),this);
  }

 


  /**
   * 连接关闭后触发的方法
   */
  @OnClose
  public void onClose(){
    //从map中删除
    WebSocketMapUtil.remove(session.getId());
    logger.info("====== onClose:"+session.getId()+" ======");
  }

 


  /**
   * 接收到客户端消息时触发的方法
   */
  @OnMessage
  public void onMessage(String params,Session session) throws Exception{
    //获取服务端到客户端的通道
    MyWebSocketServer myWebSocket = WebSocketMapUtil.get(session.getId());
    logger.info("收到来自"+session.getId()+"的消息"+params);
    String result = "收到来自"+session.getId()+"的消息"+params;
    //返回消息给Web Socket客户端(浏览器)
    myWebSocket.sendMessage(1,”成功!”,result);
  }

 


  /**
   * 发生错误时触发的方法
   */
  @OnError
  public void onError(Session session,Throwable error){
    logger.info(session.getId()+"连接发生错误"+error.getMessage());
    error.printStackTrace();
  }

 

  public void sendMessage(int status,String message,Object datas) throws IOException{
    JSONObject result = new JSONObject();
    result.put("status", status);
    result.put("message", message);
    result.put("datas", datas);
        this.session.getBasicRemote().sendText(result.toString());
    }

 

}

 

WebSocketMapUtil 工具类

 

import j麻将牌九ava.util.Collection;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

 

public class WebSocketMapUtil {

 

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

 

  public static void put(String key, MyWebSocketServer myWebSocketServer){
      webSocketMap.put(key, myWebSocketServer);
    }

 

    public static MyWebSocketServer get(String key){
        return webSocketMap.get(key);
    }

 

    public static void remove(String key){
        webSocketMap.remove(key);
    }
 
    public static Collection<MyWebSocketServer> getValues(){
        return webSocketMap.values();
    }
}

 

客户端:

 

1、添加Jar包依赖:

 

<dependency>
  <groupId>org.java-websocket</groupId>
  <artifactId>Java-WebSocket</artifactId>
  <version>1.3.8</version>
</dependency>

 

2、创建Web Socket客户端类MyWebSocketClient,并继承WebSocketClient

 

3、实现构造器,重写onOpen、onClose、onMessage、onError等方法

 

实例代码:

 

MyWebSocketClient 类

 

import java.net.URI;
import org.activiti.engine.impl.util.json.JSONObject;
import org.apache.log4j.Logger;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

 

public class MyWebSocketClient extends WebSocketClient{

 

  Logger logger = Logger.getLogger(MyWebSocketClient.class);
 
  public MyWebSocketClient(URI serverUri) {
     super(serverUri);
  }

 

  @Override
  public void onOpen(ServerHandshake arg0) {
    // TODO Auto-generated method stub
    logger.info("------ MyWebSocket onOpen ------");
  }

 

  @Override
  public void onClose(int arg0, String arg1, boolean arg2) {
     // TODO Auto-generated method stub
     logger.info("------ MyWebSocket onClose ------");
  }

 

  @Override
  public void onError(Exception arg0) {
     // TODO Auto-generated method stub
     logger.info("------ MyWebSocket onError ------");
  }

 

  @Override
  public void onMessage(String arg0) {
     // TODO Auto-generated method stub
     logger.info("-------- 接收到服务端数据: " + arg0 + "--------");
  }
}

 

MyTest 测试类

 

public class MyTest{

 

  public static void main(String[] arg0){
    MyWebSocketClient myClient = new MyWebSocketClient("此处为websocket服务端URI");
    // 往websocket服务端发送数据
    myClient.send("此为要发送的数据内容");
  }

 

}

Linux公社的RSS地址:https://www.linuxidc.com/rssFeed.aspx

上一篇:没有了
友情链接