近一个月没敲JAVA代码了,最近老师布置了一个写JAVA网盘的作业,总共花了十几个小时,总算写完了,debug真的累,感觉自己还是菜了,没有那种有一个想法就能马上用代码实现的能力。。。。不扯了,下面开始正题。

功能介绍

  • 支持1个客户端,1个服务器端。服务器提供网盘空间。
  • 首先运行服务器。服务器运行之后,客户端运行网盘客户端。
  • 运行客户端。用户能够输入昵称。确定,则连接到服务器。连接成功,即可出现客户端面。
  • 可以在网盘中新建文件夹,删除空文件夹,重命名文件夹;可以将自己电脑上某个文件上传到网盘中的某个文件夹下(支持单文件),可以删除单个文件、重命名文件、下载单个文件。
  • 可实现大文件传输

整体思路

JAVA Socket通信 打造属于自己的网盘-LMLPHP

大概分了这么几个类

服务器端

MainServer:

原来是想做个服务器界面的,但还是有点懒,就算了,所以这个类现在就用来创建Panserver对象

public class MainServer {

    private PanServer panServer;//服务器对象

    public static void main(String[] args){
MainServer ms =new MainServer();
ms.actionServer();
} // 开启服务器
public void actionServer() {
// 1.要得到服务器状态
if (null == panServer) {
panServer = new PanServer(8888);
panServer.start();
} else if (panServer.isRunning()) {// 己经在运行
panServer.stopPanServer();
panServer = null;
} }
}

Panserver:

用于建立服务器SocketServer的类

package PanServer;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket; /**
* 〈服务器socketserver创建〉
*
* @author ITryagain
* @create 2018/12/5
* @since 1.0.0
*/ public class PanServer extends Thread {
private ServerSocket ss;//服务器对象
private int port;//端口
private boolean running=false;//服务器是否在运行中 PanServer(int port){
this.port=port;
} public void run(){
setupServer();
} //在指定端口上启动服务器
private void setupServer(){
try{
ss=new ServerSocket(this.port);
running=true;
System.out.println("服务器创建成功:"+this.port);
while(running){
Socket client = ss.accept();
System.out.println("进入了一个客户机对象:"+client.getRemoteSocketAddress());
ServerThread ct = new ServerThread(client);
ct.start();
}
}catch(IOException e){
// TODO Auto-generated catch block
e.printStackTrace();
}
} /*
* 查询服务器是否在运行
*
* @return: true为运行中
*/
public boolean isRunning() {
return this.running;
} // 关闭服务器
public void stopPanServer() {
this.running = false;
try {
ss.close();
} catch (Exception e) {}
}
}

else

剩下的三个类就是服务器实现的关键了

其中最重要的是ServerThread类,该类用于实现与客户端的通信,通过接收客户端的指令进行不同的操作,其中函数如下图所示

JAVA Socket通信 打造属于自己的网盘-LMLPHP

首先,我们在建立连接时,需要输入用户名,并创建一个文件夹,文件名为用户名,因此,我们需要一个接收一开始发送的用户名信息,写在processSocket内

Socket sc=this.client;
ins=sc.getInputStream();
ous=sc.getOutputStream();
//将输入流ins封装为可以读取一行字符串也就是以\r\n结尾的字符串
BufferedReader brd=new BufferedReader(new InputStreamReader(ins));
sendMsg2Me("欢迎您使用!请输入你的用户名:");
User_name=brd.readLine();
System.out.println(User_name);

这样我们就读取了用户名,读取用户名后,马上就能创建文件夹

File directory = new File("D:\\"+User_name);
if(!directory.exists()){
directory.mkdir();
}

然后就进入while循环,不断从客户端读取用户操作信息

String input=brd.readLine();//一行一行的读取客户机发来的消息
while(true) {
System.out.println("服务器收到的是"+input);
if((!this.upLoad)&&(!this.downLoad)){
check(input);
}
if(this.upLoad){//上传中
UpLoad(input);
}
if(this.downLoad){//下载中
DownLoad(input);
}
input=brd.readLine();//读取下一条
System.out.println(input);
}

这里我用了三个函数来分别处理三种状态,其中check函数用来解码,这里我给出其中新建文件夹的写法示例,删除和重命名与之类似

private void check(String input){
if(input.charAt(0)=='~'){
String tmp=input.substring(input.indexOf("~")+1,input.indexOf("#"));
System.out.println(tmp);
if(tmp.equals("downLoad")){
this.downLoad=true;
}else if(tmp.equals("upLoad")){
this.upLoad=true;
}else if(tmp.equals("new")){
       //新建文件夹
System.out.println(input.substring(input.indexOf("#")+1));
File directory = new File(input.substring(input.indexOf("#")+1));
if(!directory.exists()){
directory.mkdir();
}
}else if(tmp.equals("delete")){
//删除文件夹
}else if(tmp.equals("change")){
//重命名文件夹
}
}
}

然后剩下的就是UpLoad和DownLoad函数了,这两个函数分别对应了上传和下载功能,我一开始把这两个功能都放在一开始建立的SockerServer里面了,结果发现文件上传了之后关闭流时把我线程也关了orz。。。还是太菜了,这种错都能写出来,百度了一番,看到好多人都是再开几个端口解决的。。。一开始就想到这方法了,但不想这么干,总觉的应该还有更好的办法,可最终还是决定用这种方法了(真香)。这里就给出其中一个函数的写法吧

    private void UpLoad(String input){
System.out.println("上传文件");
UpLoadThread upLoadThread = new UpLoadThread(8889,input);
upLoadThread.start();
this.upLoad=false;
}

既然给了UoLoad的写法,就顺便讲讲upLoadThread吧

/**
* 〈服务器接受文件线程〉
*
* @author ITryagain
* @create 2018/12/8
* @since 1.0.0
*/ public class UpLoadThread extends Thread{
private ServerSocket UpLoadServer;
private int port;
private String input;
private FileOutputStream fos; UpLoadThread(int port,String input){
this.port=port;
this.input=input;
} public void run(){ } private static DecimalFormat df = null; static {
// 设置数字格式,保留一位有效小数
df = new DecimalFormat("#0.0");
df.setRoundingMode(RoundingMode.HALF_UP);
df.setMinimumFractionDigits(1);
df.setMaximumFractionDigits(1);
} /**
* 格式化文件大小
* @param length
* @return
*/
private String getFormatFileSize(long length) { }
}

大致函数就是这样的,其中run方法里面就是文件接收了,(如果发现缺了什么自己补一补,就一个变量的申明没加上去)

    try{
UpLoadServer = new ServerSocket(port);
socket = UpLoadServer.accept();
dis = new DataInputStream(socket.getInputStream());
//文件名和长度
String fileName = input.substring(input.indexOf("#")+1);
long fileLength = dis.readLong();
File file = new File(fileName);
fos = new FileOutputStream(file); //开始接收文件
byte[] bytes = new byte[1024];
int length=0;
while((length = dis.read(bytes, 0, bytes.length)) != -1) {
fos.write(bytes, 0, length);
fos.flush();
}
System.out.println("======== 文件接收成功 [File Name:" + fileName + "] [Size:" + getFormatFileSize(fileLength) + "] ========"); try {
if(fos!=null)
fos.close();
if(dis != null)
dis.close();
if(socket !=null)
socket.close();
if(UpLoadServer!=null)
UpLoadServer.close(); } catch (Exception e) {} }catch(IOException e){ }

然后就是 getFormatFileSize() 函数了,这个函数是用来干嘛的呢?就是用来转化一下文件大小单位的,不然到时候一个几 GB 的文件显示的就是 *****

******B了,那么长一串,看着也不舒服。

    private String getFormatFileSize(long length) {
double size = ((double) length) / (1 << 30);
if(size >= 1) {
return df.format(size) + "GB";
}
size = ((double) length) / (1 << 20);
if(size >= 1) {
return df.format(size) + "MB";
}
size = ((double) length) / (1 << 10);
if(size >= 1) {
return df.format(size) + "KB";
}
return length + "B";
}

服务器端剩下没讲的代码其实都差不多,就自己去实现吧

客户端

客户端就比较麻烦了,尤其是这个界面,花了我老半天时间

MainClient:

这个类就是客户端用于创建界面以及管理线程的类了,界面我是用JTree来实现的,看下函数先吧

public class MainClient extends JFrame implements ActionListener, TreeModelListener {

    private JTree tree;
private int ServerIP = 8888;
private JLabel statusLabel;
private DefaultTreeModel treeModel;
private String oldNodeName;
private OutputStream ous;
private Socket client;
private String name;
private String stress = "D:\\";
private String downLoadStress="D:\\下载\\"; public static void main(String[] args){
MainClient mc=new MainClient();
mc.showLoginUI();
} public void showLoginUI(){ } // 登陆事件处理
private void loginAction() { } //显示操作窗口
private void showMainUI() { } @Override
public void actionPerformed(ActionEvent e) {
if (e.getActionCommand().equals("新建文件夹")) { }
if (e.getActionCommand().equals("删除文件夹")) { }
if (e.getActionCommand().equals("上传文件")) { }
if(e.getActionCommand().equals("下载文件")){ } } @Override
/*
* 修改文件名字
*/
public void treeNodesChanged(TreeModelEvent e) { } //选择上传文件
public void chooseSendFile(){ } public void sendFile(File file,String path) throws IOException{ } private String getSendPath(TreePath parentPath){ } //选择下载文件
public void chooseDownLoadFile() { }
//下载文件
public void downloadFile(String path) throws IOException{ } private String getDownLoadPath(TreePath parentPath){ } private void newFile(String path){ } private void deleteFile(String path){ } private void ChangeFileName(String NewName,String path){ }
@Override
public void treeNodesInserted(TreeModelEvent e) {
} @Override
public void treeNodesRemoved(TreeModelEvent e) {
} @Override
public void treeStructureChanged(TreeModelEvent e) {
} }

先来看看界面吧,打开后你可能只能看到三个按钮,拉动一下框框就能看到第四个了,大小设置有点问题。

 private void showMainUI() {
JFrame frame=new JFrame("网盘");
Container contentPane = frame.getContentPane(); DefaultMutableTreeNode root = new DefaultMutableTreeNode(name);
tree = new JTree(root);
tree.addMouseListener(new MyTreeMouseListener(oldNodeName));
treeModel = (DefaultTreeModel)tree.getModel();
treeModel.addTreeModelListener(this);
tree.setEditable(true);
tree.getCellEditor().addCellEditorListener(new MyTreeCellEditorListener(tree));
JScrollPane scrollPane = new JScrollPane();
scrollPane.setViewportView(tree); JPanel toolBarPanel = new JPanel();
JButton b = new JButton("新建文件夹");
b.addActionListener(this);
toolBarPanel.add(b);
b = new JButton("删除文件夹");
b.addActionListener(this);
toolBarPanel.add(b);
b = new JButton("上传文件");
b.addActionListener(this);
toolBarPanel.add(b);
b = new JButton("下载文件");
b.addActionListener(this);
toolBarPanel.add(b); statusLabel = new JLabel("Action");
contentPane.add(toolBarPanel, BorderLayout.NORTH);
contentPane.add(scrollPane, BorderLayout.CENTER);
contentPane.add(statusLabel, BorderLayout.SOUTH);
frame.pack();
frame.setVisible(true);
frame.requestFocus();
frame.setSize(400, 400);
frame.setLocationRelativeTo(null);
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
tree.setRootVisible(false);
}

里面两监听类的实现(提醒一下,可能到后面你做重命名文件夹会遇到坑,跟其中一个监听类有关,提醒一下oldNodeName是用来记录修改前的文件夹名称的,想办法获取这个变量的值就能很快实现文件夹重命名的功能),还有文件路径的实现也自己好好想想

至于界面的其它一些功能实现可以去看这篇博客(包括两个监听类的实现以及 actionPerformed() 和 treeNodesChanged() 等函数的实现)https://www.cnblogs.com/freshier/p/4614811.html

然后这里给出NewFile()函数写法

    private void newFile(String path){
String fileName = "~new#"+path+"\r\n";
System.out.println(fileName);
try {
this.ous.write(fileName.getBytes());
this.ous.flush();
} catch (IOException e) {
e.printStackTrace();
}
}

至于 delete 和 change 的写法与此类似

前面我给出了服务器上传文件的类,这里就给出客户端上传文件的类的写法

先看下 sendFile() 函数

 public void sendFile(File file,String path) throws IOException{
String fileName = "~upLoad#"+path+"\\"+file.getName()+"\r\n";
System.out.println(fileName);
this.ous.write(fileName.getBytes());
this.ous.flush();
FileSendThread send_socket = new FileSendThread(8889,file,path);
send_socket.start();
}

FileSendThread:

客户端上传文件的类

public class FileSendThread extends Thread {

    private Socket fileSendSocket;
private int port;
private String path;
private File file;
private OutputStream ous;
private FileInputStream fis;
private DataOutputStream dos; FileSendThread(int port, File file,String path){
this.port=port;
this.file=file;
this.path=path;
}
public void run(){
try{
fileSendSocket = new Socket("localhost",this.port);
// 发送: 文件名称 文件长度
this.ous = fileSendSocket.getOutputStream();
dos = new DataOutputStream(this.ous);
dos.writeLong(file.length()); //开始传输文件
System.out.println("======开始传输文件=======");
byte[] bytes = new byte[1024];
int length;
long progress = 0;
fis = new FileInputStream(file);
while((length=fis.read(bytes,0,bytes.length))!=-1){
dos.write(bytes,0,length);
dos.flush();
progress = progress + length;
System.out.print("| " + (100*progress/file.length()) + "% |");
}
System.out.println();
System.out.println("======== 文件传输成功 ========");
}catch(IOException e1){ }finally {
try {
if (fis != null)
fis.close();
if (dos != null)
dos.close();
}catch(IOException e){
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}

至于下载功能,与上传类似,把服务器和客户端上传类中的部分代码换一换就好了。

到了这里你们是否还记得我们在通信连接建立时需要传输用户名?实现方式如下

登陆框的实现,我这里算是偷懒了,直接利用了 JOptionPane

    public void showLoginUI(){

        name = JOptionPane.showInputDialog("请输入用户名");
System.out.println(name);
loginAction();
}
    // 登陆事件处理
private void loginAction() { try {
this.client = new Socket("localhost",ServerIP);
if(loginServer()){
showMainUI();
}else{
JOptionPane.showMessageDialog(null,"登陆失败","确定",JOptionPane.WARNING_MESSAGE);
}
} catch (IOException e) {
JOptionPane.showMessageDialog(null,"连接失败","确定",JOptionPane.WARNING_MESSAGE);
} }
    private boolean loginServer(){
try{
this.ous = this.client.getOutputStream();
String _name=name+"\r\n";
this.ous.write(_name.getBytes());
this.ous.flush();
return true;
}catch(IOException e){
return false;
}
}

顺便介绍几个方法

文件夹重命名

File directory = new File("D:\a.txt");
directory.renameTo(new File("D:\b.txt"));

这里实现的是将a.txt重命名为b.txt,对文件夹也有效

文件夹删除

 boolean success = (new File("D:\a.txt").delete();
if(success){
  System.out.println("删除成功");
}else{
ystem.out.println("删除失败");
}

还有String的几个操作

1、我的编码

  • 上传文件  ~upLoad#文件路径+文件
  • 下载文件  ~downLoad#文件路径+文件
  • 文件重命名 ~change#文件路径+原文件名@文件路径+新文件名
  • 新建文件夹 ~new#文件路径+文件名
  • 删除文件夹 ~delete#文件路径+文件名

2.String中的 subString() 和 indexOf() 函数

总结

到这里,我的网盘介绍算是完了,因为这是作业并且还没检查的缘故,就没有把所有代码放出来了,就介绍了下思路,顺便给将要写网盘或者正在写网盘的你们一点思路,相信有了这些思路,你们能很快地写完网盘,或者有了写的思路。太久没写代码果然不行,搞了一个多月的强化学习、机器学习和数据挖掘,差点就不会写代码了(面向搜索引擎编程感觉还行。。。百度有时很强大,可能只是你没学会正确的搜索的姿势)


检查完了,开源第一版代码https://github.com/leo6033/JAVA-Pan

05-11 13:31