天天看點

Socket檔案傳輸demo

一、Socket本地上傳

用戶端

package com.demo.client;

import java.io.*;
import java.net.Socket;
import java.util.Scanner;

public class Client
{
    public static void main(String[] args) throws IOException
    {
        //1.連接配接伺服器
        Socket socket = new Socket("127.0.0.1", 9999);
        //2.擷取網絡輸出流
        OutputStream netOut = socket.getOutputStream();					//擷取網絡輸入流  OutputStream  socket.getOutputStream
        //3.建立一個File,擷取要上傳的"檔案名"
        File src = new File("E:\\123.txt");
        //4.發送檔案名;
        if (src.exists()) {
            netOut.write(src.getName().getBytes());
        }else{
            System.out.println("要上傳的檔案不存在!");
            return;
        }
        //5.接收回報
        InputStream netIn = socket.getInputStream();     				//擷取網絡輸出流 InputStream  socket.getInputStream
        byte[] byteArray = new byte[1024];
        int len = netIn.read(byteArray);
        String msg = new String(byteArray, 0, len);
        //6.如果回報是:OK,建立檔案輸入流
        if("ok".equals(msg.toLowerCase().trim())) {						//檔案輸入流    FileInputStream
            //7.循環讀取檔案,并且發送到網絡;
            FileInputStream fileIn = new FileInputStream(src);
            len = 0;
            while ((len = fileIn.read(byteArray)) != -1) {
                netOut.write(byteArray,0,len);
            }
            fileIn.close();
        }
        //關閉輸出流
        socket.shutdownOutput();//向網絡發送一個結束符EOF

        System.out.println("【發送完畢】等待回報...");
        //接收回報
        len = netIn.read(byteArray);
        String fk = new String(byteArray, 0, len);
        System.out.println("【用戶端】收到回報:" + fk);
        //8.循環結束,關閉資源
        socket.close();
        System.out.println("【客戶端】上傳完畢!");
    }
}
           

伺服器端

package com.demo.server;



import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;

public class Server
{
    public static void main(String[] args) throws IOException {
        //1.監聽端口
        ServerSocket server = new ServerSocket(9999);
        //2.等待連接配接
        System.out.println("【伺服器】等待連接配接...");
        Socket socket = server.accept();
        //3.擷取網絡輸入流,接收檔案名
        InputStream netIn = socket.getInputStream();
        byte[] byteArray = new byte[1024];
        int len = netIn.read(byteArray);
        String fileName = new String(byteArray, 0, len);
        System.out.println("【伺服器】收到檔案名:" + fileName);
        //4.擷取網絡輸出流,發送OK
        OutputStream netOut = socket.getOutputStream();
        netOut.write("ok".getBytes());
        //5.建立一個"檔案輸出流",使用接收的"檔案名"
        FileOutputStream fileOut = new FileOutputStream("E:\\upload/" + fileName);
        //6.使用"網絡輸入流"循環從網絡讀取byte[]數組,向檔案輸出一個byte[]數組
        System.out.println("【伺服器】開始上傳...");
        while ((len = netIn.read(byteArray)) != -1) {
            fileOut.write(byteArray,0,len);
        }
        System.out.println("【伺服器】處理上傳完畢,發送回報....");
        //給用戶端發送【回報】
        netOut.write("接收完畢".getBytes());
        //7.循環結束,關閉資源
        fileOut.close();
        socket.close();
        server.close();
    }
}
           

二、點對點多線程聊天

用戶端

package com.demo.client;

import java.io.*;
import java.net.Socket;
import java.util.Scanner;

public class Client
{
    public static void main(String[] args) throws IOException
    {
        //1.連接配接伺服器-預設本機
        Socket socket = new Socket("127.0.0.1", 9999);
        //2.開啟線程,用于接收伺服器端資訊
        new ClientThread(socket).start();

        //控制台輸入
        OutputStream out = socket.getOutputStream();
        Scanner scanner = new Scanner(System.in);


        //循環從控制台接受資訊,并發送給伺服器
        while (true)
        {
            System.out.println("請輸入資訊:");
//            從控制台接受消息
            String msg = scanner.next();
            out.write(msg.getBytes());
        }
    }
}
           

用戶端線程

package com.demo.client;

import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;

public class ClientThread extends Thread
{
    private Socket socket;

    public ClientThread(Socket socket)
    {
        this.socket = socket;
    }

    @Override
    public void run()
    {
        //死循環,從網絡讀取資料
        try
        {
            //輸入流
            InputStream in = socket.getInputStream();
            byte[] bytes = new byte[1024];

            int len = 0;
            while (true)
            {
                len = in.read(bytes);
                String msg = new String(bytes, 0, len);
                System.out.println("【伺服器】:" + msg);
                //聊天結束 語句
                if (msg.equals("88"))
                {
                    //強制關閉虛拟機  結束線程
                    System.exit(0);

                }
            }
        } catch (IOException e)
        {
            e.printStackTrace();
        }
    }
}
           

伺服器端

package com.demo.server;



import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Scanner;

public class Server
{
    public static void main(String[] args) throws IOException {

        //1.啟動伺服器, 監聽端口 9999
        ServerSocket server = new ServerSocket(9999);
        //2.等待用戶端連接配接
        Socket socket = server.accept();
        //3.有用戶端連接配接成功 啟動線程
        new ServerThread(socket).start();
        //4.讀取資訊
        OutputStream out = socket.getOutputStream();
        Scanner scanner = new Scanner(System.in);

        while (true)
        {
            System.out.println("請輸入資訊:");
            String msg = scanner.next();
            try
            {
                out.write(msg.getBytes());
            } catch (IOException e)
            {
                System.out.println("聊天結束");
            }
        }
    }
}
           

伺服器端線程

package com.demo.server;


import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;

public class ServerThread extends Thread
{
    private Socket socket;

    public ServerThread(Socket socket)
    {
        this.socket = socket;
    }

    @Override
    public void run()
    {
        try
        {
            //輸入流
            InputStream in = socket.getInputStream();
            byte[] bytes = new byte[1024];
            int len = 0;
            while (true)
            {
                try
                {
                    len = in.read(bytes);
                } catch (IOException e)
                {
                    e.printStackTrace();
                }
                String msg = new String(bytes, 0, len);
                System.out.println("【用戶端】:" + msg);
                if (msg.equals("88"))
                {
                    System.exit(0);

                }
            }
        } catch (IOException e)
        {
            e.printStackTrace();
        }
    }
}