socket网络套节字---聊天室

一：服务端：

1.创建客户端：

package com.ywh.serversocket;
 
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.util.Set;
 
public class LoginThread extends Thread
{
    Socket client;
    boolean bl;
 
    //通过构造传入一个客户socket
    public LoginThread(Socket socket)
    {
        client=socket;
    }
 
    @Override
    public void run()
    {
        // TODO Auto-generated method stub
        try
        {
            //如果登录成功才运行发送消息的线程
            LoginServer();
            if(bl)
            {
                new SendThread(client).start();
            }
        } catch (Exception e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
    }
 
    public void LoginServer()throws Exception
    {
        //将连接后客户端输入传过来的用户名进行比对确认是否存在此用户
 
        //先把传过来的用户名转换为string字符串
        InputStream input=client.getInputStream();
        byte[] bytes=new byte[1024];
        input.read(bytes);
        String str=new String(bytes);
        System.out.println(str);
        //通过Set的add方法的返回值来确认用户名是否存在，
        //若用户名存在返回值为false，不存在则将新用户名加入用户名Set为true，
        boolean b=UserInfo.nickName.add(str);
        OutputStream output=client.getOutputStream();
 
        if(b)
        {
            //当b=true，即新建用户成功时通知所有用户有新用户上线
 
            //告知当前新用户创建成功
            output.write("1".getBytes());
            //将新用户的socket和用户名存入用户名map
            UserInfo.userMap.put(client, str);
            //获取用户socket
            Set<Socket> users=UserInfo.userMap.keySet();
            for(Socket user:users)
            {
                OutputStream op=user.getOutputStream();
                op.write((str+"已上线").getBytes());
            }
            bl=true;
        }else
        {
            //若创建新用户失败，则返回false
            output.write("0".getBytes());
            LoginServer();
            bl=false;
        }
 
    }
 
}

JDBCUtil.java

2.创建一个类用于存储登录用户的信息：

package com.ywh.serversocket;
 
import java.net.Socket;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;
 
public class UserInfo
{
    //用来存储用户和用户名的Map
    public static HashMap<Socket, String> userMap=new HashMap<Socket, String>();
    //用于存储用户名比对的Set
    public static Set<String> nickName=new HashSet<String>();
 
}

3.交互：发送线程

package com.ywh.serversocket;
 
import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;
import java.util.Set;
 
public class SendThread extends Thread
{
 
    Socket client;
 
    public SendThread(Socket socket)
    {
        client = socket;
    }
 
    @Override
    public void run()
    {
        // TODO Auto-generated method stub
        while(true)
        {
            try
            {
                InputStream input=client.getInputStream();
                byte[] bytes=new byte[1024];
                input.read(bytes);
                String str=new String(bytes);
                Set<Socket> users=UserInfo.userMap.keySet();
                for(Socket user:users)
                {
                    if(user!=client)
                    {
                        user.getOutputStream().write
 
((UserInfo.userMap.get(client)+":"+str).getBytes());
                    }
 
                }
 
            } catch (IOException e)
            {
                // TODO Auto-generated catch block
                Set<Socket> users=UserInfo.userMap.keySet();
                for(Socket user:users)
                {
                    if(user!=client)
                    {
                        try
                        {
                            user.getOutputStream
 
().write((UserInfo.userMap.get(client)+"下线").getBytes());
                        } catch (IOException e1)
                        {
                            // TODO Auto-generated 
 
catch block
                            e1.printStackTrace();
                        }
                    }
                    UserInfo.nickName.remove
 
(UserInfo.userMap.get(client));
                    UserInfo.userMap.remove(client);
 
                }
 
            }
 
        }
    }
 
}

4.主方法，用于运行

package com.ywh.serversocket;
 
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
 
public class Main
{
 
    public static void main(String[] args)
    {
        // TODO Auto-generated method stub
 
        try
        {
            @SuppressWarnings("resource")
            ServerSocket serversocket=new ServerSocket(8088);
            System.out.println("服务器已启动...");
 
            while(true)
            {
                Socket socket=serversocket.accept();
                new LoginThread(socket).start();
            }
 
        } catch (IOException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
    }
 
}

服务端完成

二：客户端部分

1.客户端发送信息

package com.ywh.clientsocket;
 
import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;
 
public class ClientReceive extends Thread
{
    Socket client;
    public ClientReceive(Socket socket)
    {
        // TODO Auto-generated constructor stub
        client=socket;
    }
    @Override
    public void run()
    {
        // TODO Auto-generated method stub
        while(true)
        {
            try
            {
                InputStream input=client.getInputStream();
                byte[] bytes=new byte[1024];
                input.read(bytes);
                System.out.println(new String(bytes));
            } catch (IOException e)
            {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
 
        }
    }
 
}

2.客户端接受信息

package com.ywh.clientsocket;
 
import java.io.IOException;
import java.net.Socket;
import java.util.Scanner;
 
public class ClientSend extends Thread
{
    Socket client;
 
    public ClientSend(Socket socket)
    {
        // TODO Auto-generated constructor stub
        client=socket;
    }
 
    @Override
    public void run()
    {
        // TODO Auto-generated method stub
        @SuppressWarnings("resource")
        Scanner sc=new Scanner(System.in);
        while(true)
        {
            String str=sc.nextLine();
            try
            {
                client.getOutputStream().write(str.getBytes());
            } catch (IOException e)
            {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }
 
}

ClientSend.java

3.客户端主方法

package com.ywh.clientsocket;
 
import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;
import java.util.Scanner;
 
public class Main
{
    public static String ip = "192.168.10.32";
    public static int port = 8088;
    static Scanner sc = new Scanner(System.in);
 
    public static void main(String[] args)
    {
        // TODO Auto-generated method stub
        // System.out.println("请输入IP地址");
        // String ip=sc.nextLine().trim();
        // System.out.println("请输入端口号");
        // int port=sc.nextInt();
 
        try
        {
            Socket user = new Socket(ip, port);
            System.out.println("连接服务器成功，请输入登录昵称");
            String nickName = sc.nextLine();
            while (nickName.equals(""))
            {
                System.out.println("昵称不能为空，请重新输入昵称");
                nickName = sc.nextLine();
            }
 
            Login(user,nickName);
 
            new ClientSend(user).start();
            new ClientReceive(user).start();
 
        } catch (IOException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
    }
 
    public static void Login(Socket user, String nickName)
    {
        try
        {
            user.getOutputStream().write(nickName.getBytes());
 
            InputStream input = user.getInputStream();
            byte[] bytes = new byte[1024];
            input.read(bytes);
            String str = new String(bytes).trim();
            while (str.equals("0"))
            {
                System.out.println("昵称重复，请重新输入昵称");
                nickName = sc.nextLine();
                while (nickName.equals(""))
                {
                    System.out.println("昵称不能为空，请重新输入昵称");
                    nickName = sc.nextLine();
                }
                user.getOutputStream().write(nickName.getBytes());
                input = user.getInputStream();
                bytes = new byte[1024];
                input.read(bytes);
                str = new String(bytes);
            }
            if(str.equals("1"))
            {
                System.out.println("登录成功");
            }else
            {
                System.out.println("服务器错误");
            }
        } catch (IOException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
    }
 
}