使用WebSocket实现多人实时聊天

一 应用

本应用使用WebSocket实现多人实时聊天

 

二 代码

客户端代码

    "author" content="Yeeku.H.Lee(CrazyIt.org)" />
    "Content-Type" content="text/html; charset=UTF-8" />
     使用WebSocket通信
    "text/javascript">
        // 创建Web Socket对象
        var webSocket = new WebSocket("ws://127.0.0.1:30000");
        webSocket.onopen = function()
        {
            // 为onmessage事件绑定监听器，接收消息
            webSocket.onmessage= function(event)
            {
                // 接收、并显示消息
                document.getElementById('show').innerHTML
                    += event.data + "";
            }
        };
        var sendMsg = function(val)
        {
            var inputElement = document.getElementById('msg');
            // 发送消息
            webSocket.send(inputElement.value);
            // 清空单行文本框
            inputElement.value = "";
        }




    overflow-y:auto;border:1px solid #333;" id="show">

  

"text" size="80" id="msg" name="msg"/>  

"button" value="发送" οnclick="sendMsg();"/>  

  

  

 

服务器代码

import java.io.*;
import java.net.*;
import java.nio.charset.Charset;
import java.security.MessageDigest;
import java.util.regex.*;
import java.util.*;
import sun.misc.BASE64Encoder;
/**
 * Description:
 * 网站: 疯狂Java联盟
 * Copyright (C), 2001-2012, Yeeku.H.Lee
 * This program is protected by copyright laws.
 * Program Name:
 * Date:
 * @author Yeeku.H.Lee [email protected]
 * @version 1.0
 */
public class ChatServer
{
    // 记录所有的客户端Soccket
    public static List clientSockets
        = new ArrayList();
    public ChatServer()throws IOException
    {
        // 创建ServerSocket，准备接受客户端连接
        ServerSocket ss = new ServerSocket(30000);
        while(true)
        {
            // 接收到客户端连接
            Socket socket = ss.accept();
            // 将客户端Socket添加到clientSockets集合中
            clientSockets.add(socket);
            // 启动线程
            new ServerThread(socket).start();
        }
    }
    public static void main(String[] args)
        throws Exception
    {
        new ChatServer();
    }
}
class ServerThread extends Thread
{
    private Socket socket;
    public ServerThread(Socket socket)
    {
        this.socket = socket;
    }
    public void run()
    {
        try
        {
            // 得到Socket对应的输入流
            InputStream in = socket.getInputStream();
            // 得到Socket对应的输出流
            OutputStream out = socket.getOutputStream();
            byte[] buff = new byte[1024];
            String req = "";
            // 读取数据，此时建立与WebSocket的"握手"。
            int count = in.read(buff);
            // 如果读取的数据长度大于0
            if(count > 0)
            {
                // 将读取的数据转化为字符串
                req = new String(buff , 0 , count);
                System.out.println("握手请求：" + req);
                // 获取WebSocket的key
                String secKey = getSecWebSocketKey(req);
                System.out.println("secKey = " + secKey);
                String response = "HTTP/1.1 101 Switching Protocols\r\nUpgrade: "
                    + "websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: "
                        + getSecWebSocketAccept(secKey) + "\r\n\r\n";
                System.out.println("secAccept = " + getSecWebSocketAccept(secKey));
                out.write(response.getBytes());
            }
            int hasRead = 0;
            // 不断读取WebSocket发送过来的数据
            while((hasRead = in.read(buff)) > 0)
            {
                System.out.println("接收的字节数：" + hasRead);
                /*
                    因为WebSocket发送过来的数据遵循了一定的协议格式，
                    其中第3个～第6个字节是数据掩码。
                    从第7个字节开始才是真正的有效数据。
                    因此程序使用第3个～第6个字节对后面的数据进行了处理
                */
                for (int i = 0 ; i < hasRead - 6 ; i++ )
                {
                    buff[i + 6] = (byte) (buff[i % 4 + 2] ^ buff[i + 6]);
                }
                // 获得从浏览器发送过来的数据
                String pushMsg = new String(buff
                    , 6 , hasRead - 6 , "UTF-8");
                // 遍历Socket集合，依次向每个Socket发送数据
                for (Iterator it = ChatServer.clientSockets.iterator()
                    ; it.hasNext() ;)
                {
                    try
                    {
                        Socket s = it.next();
                        // 发送数据时，第一个字节必须与读到的第一个字节相同
                        byte[] pushHead = new byte[2];
                        pushHead[0] = buff[0];
                        // 发送数据时，第二个字节记录发送数据的长度
                        pushHead[1] = (byte) pushMsg.getBytes("UTF-8").length;
                        // 发送前两个字节
                        s.getOutputStream().write(pushHead);
                        // 发送有效数据
                        s.getOutputStream().write(pushMsg.getBytes("UTF-8"));
                    }
                    catch (SocketException ex)
                    {
                        // 如果捕捉到异常，表明该Socket已经关闭
                        // 将该Socket从Socket集合中删除
                        it.remove();
                    }
                }
            }
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }
        finally
        {
            try
            {
                // 关闭Socket
                socket.close();
            }
            catch (IOException ex)
            {
                ex.printStackTrace();
            }
        }
    }
    // 获取WebSocket请求的SecKey
    private String getSecWebSocketKey(String req)
    {
        //构建正则表达式，获取Sec-WebSocket-Key: 后面的内容
        Pattern p = Pattern.compile("^(Sec-WebSocket-Key:).+",
                Pattern.CASE_INSENSITIVE | Pattern.MULTILINE);
        Matcher m = p.matcher(req);
        if (m.find())
        {
            // 提取Sec-WebSocket-Key
            String foundstring = m.group();
            return foundstring.split(":")[1].trim();
        }
        else
        {
            return null;
        }
    }
    // 根据WebSocket请求的SecKey计算SecAccept
    private String getSecWebSocketAccept(String key)
        throws Exception
    {
        String guid = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
        key += guid;
        MessageDigest md = MessageDigest.getInstance("SHA-1");
        md.update(key.getBytes("ISO-8859-1") , 0 , key.length());
        byte[] sha1Hash = md.digest();
        BASE64Encoder encoder = new BASE64Encoder();
        return encoder.encode(sha1Hash);
    }
}

 

 

三 运行结果