62,614
社区成员
发帖
与我相关
我的任务
分享
<!DOCTYPE html>
<html dir="ltr" lang="zh" xmlns="http://www.w3.org/1999/xhtml">
<head>
<title>websocket聊天室</title>
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
<link rel="stylesheet" type="text/css" href="css/bootstrap.min.css">
</head>
<body>
<div class="container">
<p>在线人数<span id="userNum"></span></p>
<textarea id="message" rows="10"></textarea>
<div class="send">
<p><textarea id="input" placeholder="请输入要发送的内容"></textarea></p>
<p><button type="button" class="btn btn-primary" id="sub">发送</button></p>
</div>
</div>
<script type="text/javascript">
(function(){
var $ = function(id){return document.getElementById(id) || null;}
var wsServer = 'ws://127.0.0.1:8888';
var ws = new WebSocket(wsServer);
var isConnect = false;
ws.onopen = function (evt) { onOpen(evt) };
ws.onclose = function (evt) { onClose(evt) };
ws.onmessage = function (evt) { onMessage(evt) };
ws.onerror = function (evt) { onError(evt) };
function onOpen(evt) {
alert("连接服务器成功");
console.log("连接服务器成功");
isConnect = true;
}
function onClose(evt) {
//console.log("Disconnected");
}
function onMessage(evt) {
var data = JSON.parse(evt.data);
switch (data.type) {
case 'text':
addMsg(data.msg);
break;
case 'num' :
updataUserNum(data.msg);
break;
}
console.log('Retrieved data from server: ' + evt.data);
}
function onError(evt) {
console.log('Error occured: ' + evt.data);
}
function sendMsg() {
if(isConnect){
ws.send($('input').value);
$('input').value = '';
}
}
function addMsg(msg) {
msg = JSON.parse(msg);
var text = '用户' + msg.user + '说:\n' + msg.text + '\n';
$('message').value += text;
$('message').scrollTop = $('message').scrollHeight;
}
function updataUserNum(msg) {
$('userNum').innerText = msg;
}
$('sub').addEventListener('click',sendMsg,false);
})();
</script>
</body>
</html>
package com.hj.websocket.server;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.security.MessageDigest;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import sun.misc.BASE64Encoder;
public class WebSocket {
private int port=8080;
private int maxUser = 10;
public WebSocket(int port, int maxUser){
this.port = port;
this.maxUser = maxUser;
}
public void startServer(){
ServerSocket serverSocket = null;
try {
serverSocket = new ServerSocket(this.port, this.maxUser);
System.out.println("服务器启动");
while(true){
Socket accept = serverSocket.accept();
new Thread(new SocketThread(accept)).start();
}
} catch (IOException e) {
e.printStackTrace();
} finally{
try {
if(serverSocket != null && !serverSocket.isClosed() ){
serverSocket.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
class SocketThread implements Runnable {
private Socket socket = null;
public SocketThread(Socket socket){
this.socket = socket;
}
@Override
public void run(){
String data = "";
InputStream inStream = null;
try {
inStream = this.socket.getInputStream();
while(true){
data = new String(SocketThread.readStream(inStream), "UTF-8");
if(!data.equals("")){
this.upgrade(data);
}
Thread.sleep(50);
}
} catch (Exception e) {
e.printStackTrace();
}
}
private String upgrade(String data) throws Exception{
String strPat = "Sec-WebSocket-Key: (.*)\r\n";
Pattern regPat = Pattern.compile(strPat);
Matcher m = regPat.matcher(data);
String key = "";
String upgrade = "";
OutputStream ouStream = null;
while(m.find()){
ouStream = this.socket.getOutputStream();
String base_key = m.group(1)+"258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
//String test = new String(MessageDigest.getInstance("SHA-1").digest("123456".getBytes()));
key = new BASE64Encoder().encode(MessageDigest.getInstance("SHA-1").digest(base_key.getBytes()));
upgrade = "HTTP/1.1 101 Switching Protocol\r\n" +
"Upgrade: websocket\r\n" +
"Connection: Upgrade\r\n" +
"Sec-WebSocket-Accept: " + key + "\r\n\r\n"; //一定要两个\r\n
byte[] b = upgrade.getBytes("UTF-8");
ouStream.write(b, 0, b.length);
ouStream.flush();
ouStream.close();
break;
}
return upgrade;
}
public static byte[] readStream(InputStream inStream) {
ByteArrayOutputStream outSteam = new ByteArrayOutputStream();
byte[] buffer = new byte[1024];
int len = -1;
try {
while ((len = inStream.read(buffer)) != -1) {
outSteam.write(buffer, 0, len);
}
outSteam.close();
} catch (IOException e) {
e.printStackTrace();
}
return outSteam.toByteArray();
}
}