对部分代码进行了修改,将一些代码封装成方法,实现了头像功能,创建一个本地字节输入流 fileinputSteam 对象,构造方法中绑定读取的数据源,创建一个socket对象,构造方法中绑定服务器的IP地址和端口号使用socket中的方法 getoutputstream 获取网络字节输出流,使用本地字节输入流中的read方法,读取本地文件和网络输出流write方法,把读取到的文件上传到服务器
服务端:
package server;
import utils.SQLUtils;
import java.io.*;
import java.net.Socket;
import java.net.URI;
import java.net.URISyntaxException;
import java.sql.SQLException;
class TcpServer {
public static Socket socket;
public static void upLoad(String user) throws IOException, URISyntaxException, SQLException {
//1、创建一个服务器的serversocket 对象
//ServerSocket server = new ServerSocket(6666);
//2、等待客户端连接
socket = Server.server.accept();
//3、获取输入流
InputStream is = socket.getInputStream();
String fileName = "avatar_" + user + ".png"; // 生成唯一文件名
String uploadDir = "file:/E:/pro/SERVER/avatar/"; // 指定存储用户头像文件的目录*/
URI uri = new URI(uploadDir + fileName);
File fl = new File(uri);
FileOutputStream fos = new FileOutputStream(fl);
int len = 0;
byte[] bytes = new byte[1024];
while((len = is.read(bytes)) != -1){
fos.write(bytes, 0, len);
}
SQLUtils.Change("avatar", user, String.valueOf(fl));
socket.getOutputStream().write("图片上传成功".getBytes());
fos.close();
socket.close();
}
}
客户端:
//1、创建一个本地字节输入流 fileinputSteam 对象,构造方法中绑定读取的数据源
FileInputStream fis = new FileInputStream(file);
//2、创建一个socket对象,构造方法中绑定服务器的IP地址和端口号
Socket socket = new Socket("127.0.0.1",6666);
//3、使用socket中的方法 getoutputstream 获取网络字节输出流
OutputStream os = socket.getOutputStream();
int len = 0;
byte[] bytes = new byte[1002];
//4、使用本地字节输入流中的read方法,读取本地文件
//5、使用网络输出流write方法,把读取到的文件上传到服务器
while ((len = fis.read(bytes)) != -1) {
os.write(bytes,0,len);
}
//告诉服务器,文件已经发送完毕
socket.shutdownOutput();
//获取网络输入流
InputStream is = socket.getInputStream();
//7、通过网络输入流对象中的read方法,读取回写数据
while ((len = is.read(bytes)) != -1){
System.out.println(new String(bytes,0,len));
}
//释放资源
fis.close();
socket.close();