黑马程序员技术交流社区

标题: TCP求教 [打印本页]

作者: 湛添友 时间: 2014-5-19 09:53
标题: TCP求教
写了个tcp 协议客户端上服务端上传文件的代码  里面有一个地方不同请指教

/*
while((len=bisPicture.read())!=-1){

   //我要问的问题就在这里

   System.out.print("");//这里我不知道是什么原因把这行注释了就不能上传成功了服务端存储的文件是没有内容的文件
bos.write(len);
}

*/

package com.zty.net;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.concurrent.Executors;

public class UploadPictureByTcp {
public static void main(String args[]){//主函数需要接受文件路径参数
  if(args==null){
System.out.println("请输入要出入的文件路径");
}
//如果输入了带空格的路径名就应该重新拼接
  String name=null;//完整的路径名
  String fileName = null;
if(args.length>1){
StringBuilder pathName=new StringBuilder();
for(int x=0;x<args.length;x++){
   if(x==(args.length-1)){
   pathName.append(args[x]);
   }
   pathName.append(args[x]+" ");
}
name = pathName.toString();
fileName=name.substring(name.lastIndexOf("\\")+1);
}else{
name=args[0];
fileName=name.substring(name.lastIndexOf("\\")+1);
}

File filePath=new File(name);//把路径名封装成对象
  if(!filePath.exists())
{
System.out.println("路径或文件不存在");
return;
}
if(filePath.length()>50*1024*1024){
System.out.println("文件键过大");
return ;
}

//建立客户端

Socket client=null;
BufferedInputStream bisPicture=null;
try {
//建立客户端
client = new Socket(InetAddress.getLocalHost(),10001);
   //获取流并装饰
InputStream in=client.getInputStream();
BufferedOutputStream bos=new BufferedOutputStream(client.getOutputStream());
//读取本地需要上传的文件的流
bisPicture = new BufferedInputStream(new FileInputStream(filePath));

//先把文件的名字给服务端 \r\n是结束标记
bos.write((fileName+"\r\n").getBytes());
bos.flush();//穿了名字立即刷新
int len=0;//然后开始上传文件
while((len=bisPicture.read())!=-1){

   //我要问的问题就在这里

   System.out.print("");//这里我不知道是什么原因把这行注释了就不能上传成功了服务端存储的文件是没有内容的文件
bos.write(len);
}
bos.flush();
//标记
client.shutdownOutput();
//读取服务端的反馈信息并打印
byte[] buf=new byte[1024];
int l=in.read(buf);
System.out.println(new String(buf,0,l));

} catch (UnknownHostException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}finally{//关流
try {
   if(client!=null)
   client.close();
} catch (IOException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
}try {
   if(bisPicture!=null)
   bisPicture.close();
} catch (IOException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
}
}
}
}

class Servers{
public static void main(String args[]){
//服务端
  ServerSocket server = null;

try {
server = new ServerSocket(10001,50);
while(true)
{
   final Socket s =  server.accept();//阻塞方法
//为每一个客户端分配一个线程
Executors.newSingleThreadExecutor().execute(new Runnable(){
   public void run(){
   BufferedOutputStream bosFile= null;
   //用来读取名字
   Reader reader =null;

   try {
      reader=new InputStreamReader(s.getInputStream());
   } catch (IOException e1) {
      // TODO Auto-generated catch block
      e1.printStackTrace();
   }//读取流
   //获取客户端文件名
   char[] buf = new char[1024];
   String fileName=null;
   int ln=0;
   int count1 = 0;
   try {
      while((ln=reader.read())!=-1){
      if(ln=='\r'){
      continue;
      }
      if(ln=='\n'){
      break;
      }
      buf[count1]=(char) ln;
      count1++;
      }
      //得到了名字
   fileName = new String(buf,0,count1);
   } catch (IOException e1) {
      // TODO Auto-generated catch block
      e1.printStackTrace();
   }
   //分解文件名获取去后缀名和文件名
   int lastPoint = fileName.lastIndexOf(".");
   String postfix = fileName.substring(lastPoint);
   String name = fileName.substring(0,lastPoint);

   //获取客户端主机名
   String clientName = s.getLocalAddress().getHostName();
   System.out.println(clientName+" connection......");//打印连接了

   File fileDir = new File("D:\\javatemp");//上传文件存储的目录
   File filePath = new File(fileDir,fileName);//完整的路径名
   //上传同名文件时防止被覆盖
   int count = 1;
   if(filePath.exists())
   {
      filePath = new File(fileDir,name+"("+count+")"+postfix);
   }while(filePath.exists())
   {
      filePath = new File(fileDir,name+"("+ ++count +")"+"."+postfix);
   }
   try {

      //存储上传文件的输出流
   bosFile = new BufferedOutputStream(new FileOutputStream(filePath));
      int len= 0;
      //读取客户端上传数据
   BufferedInputStream bis=new BufferedInputStream(s.getInputStream());
      while((len=bis.read())!=-1)
      {
      bosFile.write(len);
      }
      bosFile.flush();
      s.getOutputStream().write("文件已上传".getBytes());
   } catch (IOException e) {
      // TODO Auto-generated catch block
      e.printStackTrace();
   } finally{
      try {
      if(bosFile!=null)
      bosFile.close();
      } catch (IOException e) {
      e.printStackTrace();
      }try {
      s.close();
      } catch (IOException e) {
      // TODO Auto-generated catch block
      e.printStackTrace();
      }
   }
   }
   });
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}

作者: skill20 时间: 2014-5-19 10:59
用这么麻烦吗？直接在服务端判断文件是否存在不行嘛？不存在就创建，存在自增再创建一个。用打印流会好些的说，不用刷新。

作者: 湛添友 时间: 2014-5-19 11:26

skill20 发表于 2014-5-19 10:59
用这么麻烦吗？直接在服务端判断文件是否存在不行嘛？不存在就创建，存在自增再创建一个。用打印流会好些的 ...

我是要上传任意类型的文件，要知道上传文件的类型和文件名

作者: skill20 时间: 2014-5-19 12:09

湛添友发表于 2014-5-19 11:26
我是要上传任意类型的文件，要知道上传文件的类型和文件名

可以在服务端判断呀，getName（）和 endsWith（）；

欢迎光临黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2