在SpringBoot中利用Websocket實現(xiàn)一個網(wǎng)頁聊天功能

發(fā)布時間：2020-12-02 17:04:59 來源：億速云閱讀：275 作者：Leah 欄目：編程語言

這篇文章將為大家詳細講解有關(guān)在SpringBoot中利用Websocket實現(xiàn)一個網(wǎng)頁聊天功能，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關(guān)知識有一定的了解。

導入websocket的包。

通過使用SpringBoot導入包的時候，我們可以發(fā)現(xiàn)，很多包都是以 spring-boot-starter 開頭的，對于我這種強迫癥，簡直是福音

<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

配置websocket

服務端

首先新建一個WebSocketConfig的類，添加 @Component 注解注入一個bean

@Component
public class WebSocketConfig {

  @Bean
  public ServerEndpointExporter serverEndpointExporter() {

    return new ServerEndpointExporter();
  }
}

新建一個service，本來覺得是新建一個控制器，但是我覺得這也不是控制器，但是也不太是service。最后還是選擇新建一個servece。這里你如果有更好的想法可以按照你的來。

這個service里面有4個方法，這些方法很簡單，通過名稱就可以看出是什么意思。

@Component
@ServerEndpoint("/webSocket")
@Slf4j
public class WebSocket {

  private Session session;

  private static CopyOnWriteArraySet<WebSocket> webSockets = new CopyOnWriteArraySet<>();

  private MessageVO messageVO = new MessageVO();

  @OnOpen
  public void onOpen(Session session) {
    this.session = session;
    webSockets.add(this);


    messageVO.setType(1);
    messageVO.setUserNum(webSockets.size());
    messageVO.setMessage("有新的連接");

    ObjectMapper mapper = new ObjectMapper();

    String Json = "";
    try {
      Json = mapper.writeValueAsString(messageVO);
    } catch (Exception ex) {
      log.error(ex.getMessage());
    }

    this.sendMessage(Json);
    log.info("【websocket消息】有新的連接, 總數(shù):{}", webSockets.size());
  }


  @OnClose
  public void onClose() {
    webSockets.remove(this);

    messageVO.setType(2);
    messageVO.setUserNum(webSockets.size());
    messageVO.setMessage("有用戶離開");

    ObjectMapper mapper = new ObjectMapper();

    String Json = "";
    try {
      Json = mapper.writeValueAsString(messageVO);
    } catch (Exception ex) {
      log.error(ex.getMessage());
    }

    this.sendMessage(Json);


    log.info("【websocket消息】連接斷開, 總數(shù):{}", webSockets.size());
  }

  @OnMessage
  public void onMessage(String message) {

    messageVO.setType(3);
    messageVO.setUserNum(webSockets.size());
    messageVO.setMessage(message);

    ObjectMapper mapper = new ObjectMapper();

    String Json = "";
    try {
      Json = mapper.writeValueAsString(messageVO);
    } catch (Exception ex) {
      log.error(ex.getMessage());
    }

    this.sendMessage(Json);

    log.info("【websocket消息】收到客戶端發(fā)來的消息:{}", message);
  }

  public void sendMessage(String message) {
    for (WebSocket webSocket : webSockets) {
      log.info("【websocket消息】廣播消息, message={}", message);
      try {
        webSocket.session.getBasicRemote().sendText(message);
      } catch (Exception e) {
        e.printStackTrace();
      }
    }
  }
}

客戶端

客戶端也就是我們的html，我用的是原生html5的websocket ,不需要其他的js。直接在html中寫js就可以

<script type="application/javascript">
  var websocket = null;
  var cahtNum = $('.chat-num').text();
  if ('WebSocket' in window) {
    websocket = new WebSocket('ws://localhost:8080/chat/webSocket');

  } else {
    alert('該瀏覽器不支持websocket');
  }

  websocket.onopen = function (event) {
    console.log('websocket建立連接');
  }

  websocket.onclose = function (event) {
    console.log('websocket關(guān)閉連接');
  }

  websocket.onmessage = function (event) {
    console.log('websocket收到消息' + event.data);
    var result = $.parseJSON(event.data);
    if (result.type == 3) {
      var element = document.getElementById('message-template').innerHTML;
      $('.message-container').append(element);
      $(".message-content:last").html(result.message);
    }
    else {
      $('.chat-num').text(result.userNum);
    }

  }

  websocket.onerror = function (event) {
    console.log('websocket通信發(fā)生錯誤');

  }

  window.onbeforeunload = function (event) {
    websocket.close();
  }

  $('.send-message').click(function () {
    var message = $('.chat-message').val();
    if (message == "") {
      mdui.alert('請輸入要發(fā)送的消息');
      return;
    }
    sendmessage(message);
    $('.chat-message').val("");
  })

  function sendmessage(message) {
    websocket.send(message);
  }
</script>

關(guān)于在SpringBoot中利用Websocket實現(xiàn)一個網(wǎng)頁聊天功能就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。

向AI問一下細節(jié)

在SpringBoot中利用Websocket實現(xiàn)一個網(wǎng)頁聊天功能

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標簽