WebSocket协议-Socket.io聊天室demo

聊天室demo,技术栈:React+Node.js+Express+Socket.io

配置环境

1、首先需要创建一个React项目,略过
2、安装Node.js,略过,使用了基于Node.js的web框架express,安装express

npm install --save express

3、安装Socket.io,Socket.io 由两部分组成:

  • 一个服务端用于集成 (或挂载) 到 Node.JS HTTP 服务器: socket.io
  • 一个加载到浏览器中的客户端: socket.io-client
//服务端安装
npm install --save socket.io

//客户端安装
npm install --save socket.io-client

//使用标准化JS库,在 index.html用script引入
<script src="/socket.io/socket.io.js"></script>

客户端加载了 socket.io-client,socket.io-client 暴露了一个 io 全局变量,然后连接服务器。

至此,依赖的环境全部安装完了。有兴趣可以试一下下面的demo,欢迎指正。

客户端代码:

//ChatRoom.js
import React, { Component } from 'react';
import io from 'socket.io-client';  //客户端引入socket.io
import './ChatRoom.css';

//连接Node服务,后台服务为http://localhost:8000,生成一个socket对象
const socket = io('http://localhost:8000')

//没有做登录页面,在url中输入的第一个参数作为用户名
const userName = window.location.search.substr(1)

class ChatRoom extends Component {
  constructor(props) {
    super(props)
  }

  state = {
    count: 0,
    message: '',
    content: [],
  }

  componentDidMount() {
    const self = this;

    socket.on('connect', function() {
       //客户端发送进入聊天室
      socket.emit('add user', {
        userName: userName
      })
    })

    //客户端监听用户加入聊天室
    socket.on('user joined', function({userName, count}) { 
      const content = [...self.state.content, {userName, message: `${userName}加入`, type: 'tip'}]
      self.setState({content, count})
    })

    //客户端监听新消息
    socket.on('new message', ({userName, message}) => {
      const content = [...self.state.content, {userName, message}]
      self.setState({content})
      this.chatContent.scrollTop = 400
    })

     //客户端监听用户离开聊天室
     socket.on('user leave', function({userName, count}) { 
      const content = [...self.state.content, {userName, message: `${userName}离开`, type: 'tip'}]
      self.setState({content, count})
    })
  }

  handleChange = event => {
    this.setState({message: event.target.value})
  }

  handleSend = event => {
    if (event.key === 'Enter') {
      //客户端发送新消息
      socket.emit('new message', {
        userName: userName,
        message: this.state.message
      })
      this.setState({message: ''})
      this.chatContent.scrollTop = 400
    }
  }

  handleQuit = () => {
    //断开连接
    socket.disconnect();
  }

  render() {
    const {count, content, message} = this.state
    const contentNodes = content.map(({userName, message, type}, index) => {
      return (<div className="ChatRoom-content-item" key={index}>
        {type === 'tip' ? (<div className="ChatRoom-content-tip">{message}</div>) : 
        (<React.Fragment>
          <div className="ChatRoom-content-username">{userName}</div>
          <div className="ChatRoom-content-message">{message}</div> 
        </React.Fragment>
        )}
      </div>)
    })
    return (
      <div className="ChatRoom">
        <div className="ChatRoom-header">
          Chat Room · {count}人
          <button className="ChatRoom-header-quit" onClick={this.handleQuit}>退出</button>
        </div>
        <div className="ChatRoom-content" ref={ref => this.chatContent = ref}>
          {contentNodes}
        </div>
        <textarea className="ChatRoom-message" cols="77" rows="10" value={message} onChange={this.handleChange} onKeyPress={this.handleSend}></textarea>
      </div>
    );
  }
}

export default ChatRoom;
//ChatRoom.css
.ChatRoom {
  width: 500px;
  margin: 80px auto 0 auto;
  border: 1px solid #eee;
  border-radius: 4px;
}

.ChatRoom-header {
  padding: 10px;
  border-bottom: 1px solid #eee;
}

.ChatRoom-header-quit {
  float: right
}

.ChatRoom-content {
  height: 400px;
  padding: 10px;
  overflow: auto;
  border-bottom: 1px solid #eee;
}

.ChatRoom-content-item {
  padding: 5px 0px;
}

.ChatRoom-content-username {
  font-weight: bold
}

.ChatRoom-content-message {
  color: #444;
  font-size: 13px;
}

.ChatRoom-content-tip {
  font-size: 12px;
  color: #ddd;
  text-align: center;
}

.ChatRoom-message {
  padding: 10px;
  border: none;
  outline: none;
}

接下来是服务端代码:

//server.js
var app = require('express')()
var http = require('http').Server(app)
var io = require('socket.io')(http)

const sessionList = []

io.on('connection', function(socket) {
  //服务端监听用户进入聊天室
  socket.on('add user', function({userName}) {
    socket.userName = userName
    if(!sessionList.includes(userName)) {
      sessionList.push(userName)
    }

    //服务端广播用户加入聊天室信息
    io.emit('user joined', {
      userName,
      count: sessionList.length
    })

  })

  //服务端监听客户端发送的新消息
  socket.on('new message', function({userName, message}) {
    io.emit('new message', {
      userName,
      message
    })
  })

  socket.on('disconnect', function() {
    sessionList.splice(sessionList.indexOf(socket.userName), 1 );
    socket.broadcast.emit('user leave', {
      userName: socket.userName,
      count: sessionList.length
    })
  })
})

http.listen(8000, function() {
  console.log('listening on *:8000')
})

代码写完,执行一下试试:
在命令行切换到server.js文件所在目录,执行node server.js,命令行会打印:

listening on *:8000

至此,node服务已经启动。
然后启动React,会自动打开浏览器,地址默认为http://localhost:3000/,进入聊天室需要加个参数作为用户名,如:http://localhost:3000?asan
这时聊天室有1个人了,如图:

1个人.png

多个人在聊天室,如图:
多个人.png

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,117评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,963评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,897评论 0 240
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,805评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,208评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,535评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,797评论 2 311
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,493评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,215评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,477评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,988评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,325评论 2 252
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,971评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,807评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,544评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,455评论 2 266

推荐阅读更多精彩内容