【优化】IO流应用之网络传输

javase已经看到网络编程了，快结束啦！掌声鼓励下。。。呵呵

经过一个月的自学，基本把java基础语法掌握的差不多了。我发现学的越多，就越喜欢java代码的简洁，清晰，很享受写代码的过程！加油吧，骚年！接下来就是通过小项目练手了！

废话不多说，下面是昨天写的代码，通过这个例子，我发现优化代码真的很重要！！！！

需求是这样子的，

客户端通过键盘录入用户名
服务端对这个用户名进行校验

如果该用户存在 在服务端显示xxx已登录
并在客户端显示xxx 欢迎光临

如果该用户不存在，在服务端显示xxx，尝试登陆
并在客户端显示xxx，该用户不存在

最多就登陆三次。

环境，控制台。

下面是我的代码。

用户数据存放在data.txt中

 

 

/*
 *本程序为方便调试，将客户端和服务端放在同一个.java文件中,
 *另外将ip地址设置为127.0.0.1，仅限本机登陆,如果连有局域网，可以
 *进行局域网登陆测试，本人亲测通过。（需先获取对方ip）
*/
import java.net.*;
import java.io.*;
import java.util.*;
//客户端
class Client
{
	//为保持代码简洁，这里抛出IO异常
	public static void main(String[] args)throws IOException
	{
		//建立socket服务
		Socket s=new Socket(InetAddress.getByName("127.0.0.1"),10006);
	
		BufferedReader bufr=new BufferedReader(new InputStreamReader(System.in));//键盘录入
		BufferedWriter bufw=new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
		BufferedReader bufIn=new BufferedReader(new InputStreamReader(s.getInputStream()));
	
		String line=null;
		for(int i=0;i<3;i++)
		{
			line=bufr.readLine();
			if(line==null)
				break;
			//将登录名发给服务端
			bufw.write(line);
			bufw.newLine();
			bufw.flush();	
			//接收服务端反馈
			String dataRe=bufIn.readLine();
			System.out.println(dataRe);	
			if(dataRe.contains("欢迎"))
				break;
		}	
		s.close();
	}
}
//服务端
class Server
{
	public static void main(String[] args)throws IOException
	{	
		//开启ServerSocket服务，监听一个端口
		ServerSocket ss=new ServerSocket(10006);		
		while(true)
		{		
			Socket s=ss.accept();
			//多线程处理并发登陆
			new Thread(new LoginThread(s)).start();
		}	
	}
}
class LoginThread implements Runnable
{
	private Socket socket;
	private ArrayList<String> arr;
	LoginThread(Socket socket)
	{
		this.socket=socket;
	}
	public void run()
	{
		try
		{	
			System.out.println(socket.getInetAddress().getHostName()+"...connected");
			File file=new File("data.txt");
			//动态访问数据库，并遍历用户名,存入集合
			BufferedReader buda=new BufferedReader(new FileReader(file));
			String temp=null;
			arr=new ArrayList<String>();
			while((temp=buda.readLine())!=null)
			{	
				arr.add(temp);
			}
			//通过缓冲类包装getInputStream()和getOutputStream()方法
			BufferedReader bufr=new BufferedReader(new InputStreamReader(socket.getInputStream()));
			BufferedWriter bufw=new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
			String line=null;
			//这里要求循环获取信息并反馈
			for(int i=0;i<3;i++)
			{
				line=bufr.readLine();//获取客户端发来的信息
				if(line==null)
					break;
					//遍历集合
				Iterator<String> it=arr.iterator();
				int count=0;
				while(it.hasNext())
				{
					//检索到该用户名
					if(it.next().equals(line))
					{
						System.out.println(line+"已登录");
						//反馈信息
						bufw.write("尊敬的"+line+",欢迎光临");
						bufw.newLine();
						bufw.flush();
						break;
					}
					else
						count++;		
				}		
				if(count==arr.size())
				{
					System.out.println(line+"尝试登陆");
					bufw.write(line+"用户不存在");
					bufw.newLine();
					bufw.flush();
				}
			}
			socket.close();//超过三次，自动断开连接
		}
		catch(IOException e)
		{
			throw new RuntimeException();
		}	
	}
}

以上代码通过不够优化，特别是将文件中的数据存到集合中，每开启一个线程都会执行一次，效率很低，而且是通过计数器来判断返回信息的，比较低效。

下面将代码稍微优化一下。

import java.io.*;
import java.net.*;
//客户端
class NClient
{
	public static void main(String[] args)throws IOException
	{
		//建立服务
		Socket s=new Socket(InetAddress.getByName("127.0.0.1"),10006);
		
		BufferedReader bufr=new BufferedReader(new InputStreamReader(System.in));	//键盘录入
		BufferedWriter bufw=new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
		BufferedReader bufIn=new BufferedReader(new InputStreamReader(s.getInputStream()));
		
		String line=null;
		String temp=null;
		for(int i=0;i<3;i++)
		{
			//读取键盘录入，写入流中，注意，因为有缓冲，必须换行并刷新
			line=bufr.readLine();
			if(line==null)
				break;//如果录入为空，则退出循环
			bufw.write(line);
			bufw.newLine();
			bufw.flush();
			//接收服务端反馈
			temp=bufIn.readLine();
			System.out.println(temp);
			if(temp.contains("欢迎"))//登陆成功就退出循环
				break;
		}
		s.close();
	}
}

//服务端
class NServer
{
	public static void main(String[] args)throws IOException
	{
		ServerSocket ss=new ServerSocket(10006);
		while(true)
		{
			Socket s=ss.accept();
			new Thread(new LoginThread(s)).start();
		}
	}
}
class LoginThread implements Runnable
{
	private Socket s;
	LoginThread(Socket s)
	{
		this.s=s;
	}
	public void run()
	{
		System.out.println(s.getInetAddress().getHostAddress()+"....connected");
		try
		{
			File file=new File("data.txt");
			
			BufferedReader bufr=new BufferedReader(new InputStreamReader(s.getInputStream()));
			BufferedWriter bufw=new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
		
			boolean flag=false;
			String line=null;		
			for(int i=0;i<3;i++)
			{
				
				line=bufr.readLine();
				 if(line==null)
					 break;
				
				BufferedReader buf=new BufferedReader(new FileReader(file));
				String temp=null;
				while((temp=buf.readLine())!=null)
					if(line.equals(temp))
					{
						flag=true;
						break;
					}	
				if(flag)
				{
					System.out.println(line+"已登录");
					bufw.write("尊敬的"+line+"，欢迎光临");
					bufw.newLine();
					bufw.flush();
				}
				else
				{
					System.out.println(line+"尝试登陆");
					bufw.write(line+",该用户名不存在");
					bufw.newLine();
					bufw.flush();
				}
			}
			s.close();
		}
		catch(IOException e)
		{
			//throw new RuntimeException();
			e.printStackTrace();
		}
	}
}

优化后的代码通过判断标记flag的boolean值向客户端返回信息，很高效。

运行结果如下（同时打开两个控制台）：

 

提示：需要先打开服务端！

这里注意一个细节，客户端和服务端都有这样一行代码：

if(line==null)
			
    break;

这是因为如果客户端第一次登陆已经登陆成功的话，就不必要继续登陆了，这时按ctrl+c退出时，服务端并不结束，且其readLine方法会读到null值，所以

我们在客户端加上这句判断。

思考：优化后的代码其实还可以继续优化，例如我们的BufferedWriter类每次write之后都需要换行并刷新缓冲区，

很麻烦，可以用PrinterWriter类，自动刷新。