java网络编程(7)——利用tcp实现文件上传

package com.seven.tcp;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

public class TcpFileUploadClient {
    
    public static void main(String[] args) throws Exception {
        
        Socket socket = new Socket("127.0.0.1",10001);
        
        FileInputStream fis = new FileInputStream(new File("e:\\test.txt"));
        
        OutputStream out = socket.getOutputStream();
        
        byte[] buf = new byte[1024];
        
        int len = 0;
        
        while((len=fis.read(buf))!=-1){
            out.write(buf);
        }
        
        socket.shutdownOutput();
        
        InputStream in = socket.getInputStream();
        
        len = in.read(buf);
        
        System.out.println("return:"+new String(buf,0,len));
        
        fis.close();
        
        socket.close();
        
        
    }
    
}

package com.seven.tcp;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * 服务端要用多线程来实现，这样才能实现多个用户同时上传
 * 
 * @author Seven
 *
 */
public class TcpFileUploadServer {

    public static void main(String[] args) throws Exception {
        
        ServerSocket serverSocket = new ServerSocket(10001);
        
        //循环，每次有新的连接就开启一个新的线程
        while(true){
            
            Socket socket = serverSocket.accept();
            
            new Thread(new FileUploadThread(socket)).start();
            
        }

    }

}

class FileUploadThread implements Runnable {

    private Socket socket;

    public FileUploadThread(Socket s) {
        this.socket = s;
    }

    // 在run方法中实现文件上传
    public void run() {

        try {

            System.out.println(socket.getInetAddress().getHostAddress()+"....connected");
            
            String filename = socket.getInetAddress().getHostAddress();

            int count = 1;

            File file = new File("e:\\" + filename + count);

            while (file.exists()) {
                file = new File("e:\\" + filename + count++);
            }

            FileOutputStream fileOutputStream = new FileOutputStream(file);

            InputStream in = socket.getInputStream();
            
            byte[] b = new byte[1024];

            int len = 0;
            
            while ((len = in.read(b)) != -1) {
                fileOutputStream.write(b,0,len);
            }
            //返回成功
            OutputStream out = socket.getOutputStream();
            out.write("success!".getBytes());
            
            socket.close();

        }catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }

    }

}