A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

 找回密码
 加入黑马

QQ登录

只需一步,快速开始

© hello_world! 中级黑马   /  2012-10-23 19:18  /  1786 人查看  /  0 人回复  /   0 人收藏 转载请遵从CC协议 禁止商业使用本文

在TCP上传文件的练习中。毕老师讲课的时候,演示的时候没有发生数据丢失的情况。但是,我将代码多运行几次,发生数据丢失的情况。这种情况是避免不了的。但是可以补救?求方法?
这里有上传文件的代码。给大家实验之用。
客户端:
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Scanner;

public class TransClient {


        public static void main(String[] args) {
                try {
                        Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 10010);
                        OutputStream out = socket.getOutputStream();
                                               
                        Scanner scanner = new Scanner(System.in);
                        File file = null;
                        while (true) {
                                System.out.println("Please put in the name of file:");
                                String fileName = scanner.nextLine();
                                file = new File(fileName);
                               
                                if (!file.exists()) {
                                        System.out.println(fileName + " isn't exists");
                                }else{
                                        break;
                                }
                        }
                       
                        PrintWriter pw = new PrintWriter(out, true);
                        pw.println(file.getName());                                        //发送文件名
                       
                        BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
                        BufferedOutputStream bos = new BufferedOutputStream(out);
                        byte[] buf = new byte[1024*16];
                        int len;
                        while (-1 != (len = bis.read(buf))) {
                                bos.write(buf, 0, len);
                                bos.flush();//注意这里一定要刷新,要不然数据没有出缓冲区。更谈不上发送了
                        }
                       
                        socket.shutdownOutput();
                       
                        BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                        System.out.println(br.readLine());
                       
                        bis.close();
                        socket.close();//是不是与socket相关的流,都被关闭了                       
                } catch (UnknownHostException e) {
                        e.printStackTrace();
                } catch (IOException e) {
                        e.printStackTrace();
                }
        }               
}

服务器端:

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Arrays;
import java.util.List;

public class TransServer {

        public static void main(String[] args) {
                try {
                        ServerSocket ss = new ServerSocket(10010);
                        while (true) {
                                Socket socket = ss.accept();
                                new Thread(new ServerRunnable(socket)).start();
                        }
                } catch (IOException e) {
                        e.printStackTrace();
                }
               
        }
       
       
}

class ServerRunnable implements Runnable{

        private Socket s;
       
        public ServerRunnable(Socket socket) {
                super();
                this.s = socket;
        }

        @Override
        public void run() {
                try {
                       
                        System.out.println(s.getInetAddress().getHostAddress() + "...contected");
                        BufferedReader br = new BufferedReader(new InputStreamReader(s.getInputStream()));
                        File file = new File(br.readLine());                                //获得文件名
                       
                        BufferedInputStream bis = new BufferedInputStream(s.getInputStream());
                       
                        String fileName = null;
                        File dest = new File("D:\\");
                        String[] names = dest.list();
                        List<String> list = Arrays.asList(names);
                        int count = 1;
                        if (list.contains(file.getName())) {//这段代码作用是,如果存在相同名字的文件,那就这样处理fileName(count).***;fileName代表文件名(代表不带后缀的文件名),count重复的个数,***代表文件名的后缀
                                String shortName = file.getName();
                                while (list.contains(shortName.substring(0, shortName.lastIndexOf('.')) + "(" + count + ")" + shortName.substring(shortName.lastIndexOf('.')))) {       
                                        count++;
                                }
                                fileName = shortName.substring(0, shortName.lastIndexOf('.')) + "(" + count + ")" + shortName.substring(shortName.lastIndexOf('.'));
                        }else{
                                fileName = file.getName();
                        }
                        System.out.println(fileName);
                       
                        BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream("D:\\" + fileName));
                       
                        byte[] buf = new byte[1024*16];
                        int len;
                        while (-1 != (len = bis.read(buf))) {
                                bos.write(buf, 0, len);                               
                        }
                       
                        bos.flush();
                       
                        PrintWriter pw = new PrintWriter(s.getOutputStream(), true);
                        pw.println("上传成功");//这里就不要用write(),因为它不会自动刷新的
                       
                        s.shutdownOutput();
                       
                        bos.close();
                        s.close();
                } catch (IOException e) {
                        e.printStackTrace();
                }
        }       
}

等待高手解答。。。



评分

参与人数 1技术分 +1 收起 理由
刘芮铭 + 1

查看全部评分

0 个回复

您需要登录后才可以回帖 登录 | 加入黑马