Java Socket+mysql实现简易文件上传器的代码
最近跟着某网站学习了一个小项目,因为白天有课,所以都是晚上写的,今天把它完成了。
项目主要是实现一个文件上传器,通过客户端的登陆,把本地文件上传到服务器的数据库(本地的)。
首先建两个表如下:
一个文件信息表
CREATE TABLE `fileinfo` ( `Fname` char(50) NOT NULL, `FInfo` blob NOT NULL, `FId` int(10) NOT NULL AUTO_INCREMENT, PRIMARY KEY (`FId`) ) ENGINE=InnoDB AUTO_INCREMENT=4 DEFAULT CHARSET=utf8;
一个用户信息表
CREATE TABLE `user` ( `username` char(25) NOT NULL, `password` char(25) NOT NULL, `useid` int(10) NOT NULL AUTO_INCREMENT, PRIMARY KEY (`useid`) ) ENGINE=InnoDB AUTO_INCREMENT=19 DEFAULT CHARSET=utf8;
User类
package fileUpload; import java.io.Serializable; /* * 用户实体类 */ public class User implements Serializable{ private static final long serialVersionUID = -7279093338433393181L; private int useid;//用户编号 private String usename;//用户名 private String usepsd;//用户密码 public User(){ } public User(String usename,String usepsd){ super(); this.usename = usename; this.usepsd = usepsd; } public User(int useid,String usename,String usepsd){ super(); this.useid = useid; this.usename = usename; this.usepsd = usepsd; } public int getUseid() { return useid; } public void setUseid(int useid) { this.useid = useid; } public String getUsename() { return usename; } public void setUsename(String usename) { this.usename = usename; } public String getUsepsd() { return usepsd; } public void setUsepsd(String usepsd) { this.usepsd = usepsd; } }
UserService类
package fileUpload; /** * 用户登陆与注册操作类 */ import java.sql.Connection; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; public class UserService { private Connection conn = null; private PreparedStatement ptmt = null; private ResultSet rs = null; // 用户登录 public boolean login(User user) { String url = "jdbc:mysql://localhost:3306/fileupload";// 数据库连接字符串 String sql = "select * from user where username=? and password=?"; try { Class.forName("org.gjt.mm.mysql.Driver").newInstance(); conn = DriverManager.getConnection(url, "root", "1995520"); ptmt = conn.prepareStatement(sql); ptmt.setString(1, user.getUsename()); ptmt.setString(2, user.getUsepsd()); rs = ptmt.executeQuery(); if (rs.next()) { return true; } } catch (Exception e) { e.printStackTrace(); } finally { try { rs.close(); ptmt.close(); } catch (SQLException e) { e.printStackTrace(); } } return false; } //用户注册 public boolean register(User user){ System.out.println("hello"); String url = "jdbc:mysql://localhost:3306/fileupload";// 数据库连接字符串 String sql = "insert into user(username,password) values (?,?)"; try { Class.forName("org.gjt.mm.mysql.Driver").newInstance(); conn = DriverManager.getConnection(url, "root", "1995520"); ptmt = conn.prepareStatement(sql); ptmt.setString(1, user.getUsename()); ptmt.setString(2, user.getUsepsd()); ptmt.executeUpdate(); } catch (Exception e) { e.printStackTrace(); return false; } finally { try { //rs.close(); ptmt.close(); } catch (SQLException e) { e.printStackTrace(); } } return true; } }
FileIofo类
package fileUpload; import java.io.Serializable; /* * 文件实体类 */ public class FileInfo implements Serializable{ private static final long serialVersionUID = 2554622626228481455L; private int fid;//文件编号 private String fname;//文件名 private byte[] fcontent;//文件内容 public FileInfo(){ } public FileInfo(String fname,byte[] fcontent){ super(); this.fname = fname; this.fcontent = fcontent; } public FileInfo(int fid,String fname,byte[] fcontent){ super(); this.fid = fid; this.fname = fname; this.fcontent = fcontent; } public int getFid() { return fid; } public void setFid(int fid) { this.fid = fid; } public String getFname() { return fname; } public void setFname(String fname) { this.fname = fname; } public byte[] getFcontent() { return fcontent; } public void setFcontent(byte[] fcontent) { this.fcontent = fcontent; } }
FileService类
package fileUpload; /** * 文件上传操作类 */ import java.sql.Connection; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.SQLException; public class FileService { private Connection conn = null; private PreparedStatement ptmt = null; //将文件保存到数据库中 public boolean save(FileInfo file){ String url = "jdbc:mysql://localhost:3306/fileupload";//数据库连接字符串 String sql = "insert into fileinfo(Fname,Finfo) value (?,?)"; try{ Class.forName("org.gjt.mm.mysql.Driver").newInstance(); conn = DriverManager.getConnection(url, "root", "1995520"); ptmt = conn.prepareStatement(sql); ptmt.setString(1, file.getFname()); ptmt.setBytes(2,file.getFcontent()); ptmt.executeUpdate(); }catch(Exception e){ e.printStackTrace(); return false; }finally{ try { ptmt.close(); } catch (SQLException e) { e.printStackTrace(); } } return true; } }
package fileUpload;
import java.io.IOException; import java.net.ServerSocket; import java.net.Socket; /* * 启动服务器类 */ public class StartServer { @SuppressWarnings("resource") public static void main(String[] args){ try { // 1.创建一个服务器端Socket,即ServerSocket,指定绑定的端口,并监听此端口 ServerSocket serverSocket = new ServerSocket(8888); Socket socket = null; System.out.println("服务器已启动,等待客户端的连接..."); //循环监听等待客户端的连接 while (true) { // 2.调用accept()方法开始监听,等待客户端的连接 socket = serverSocket.accept(); //创建一个新的线程 ServerThread serverThread = new ServerThread(socket); //启动线程 serverThread.start(); } } catch (IOException e) { // TODO 自动生成的 catch 块 e.printStackTrace(); } } } [java] view plain copy 在CODE上查看代码片派生到我的代码片 package fileUpload; import java.io.IOException; /** * 启动客户端类 */ public class StartClient { public static void main(String[] args) throws IOException{ SocketClient client = new SocketClient(); client.showMainMenu();//显示主菜单 } }
package fileUpload;
import java.io.IOException; import java.io.ObjectInputStream; import java.io.ObjectOutputStream; import java.net.Socket; import javax.swing.JOptionPane; /* * 服务器端多线程处理类 */ public class ServerThread extends Thread { static Socket socket = null; private static ObjectInputStream ois = null;// 对象输入流 private ObjectOutputStream oos = null;// 对象输出流 private UserService us = new UserService();// 用户业务对象 private FileService fs = new FileService();// 文件业务对象 // 通过构造方法,初始化socket public ServerThread(Socket socket) { ServerThread.socket = socket; } public void run() { try { ois = new ObjectInputStream(socket.getInputStream());//接收客户端发来的消息 oos = new ObjectOutputStream(socket.getOutputStream());//用来向客户端发送消息 CommandTransfer transfer = (CommandTransfer) ois.readObject(); // 读取客户端发送到服务器的指令操作 transfer = execute(transfer); // 执行客户端送到服务器的指令操作 oos.writeObject(transfer);// 响应客户端 } catch (IOException | ClassNotFoundException e) { e.printStackTrace(); } } /* * 执行客户端发送到服务器的指令操作 */ private CommandTransfer execute(CommandTransfer transfer) { String cmd = transfer.getCmd();// 获取当前操作的指令 if (cmd.equals("login")) { // 用户登录 User user =(User)transfer.getData(); boolean flag = us.login(user); if(flag) JOptionPane.showMessageDialog(null, "登陆成功!"); transfer.setFlag(flag); if(flag){//判断登陆是否成功 transfer.setResult("登陆成功!"); }else{ transfer.setResult("用户名或密码不正确,请重新登陆!"); } }else if(cmd.equals("register")){//用户注册 User user = (User) transfer.getData(); us.register(user);//注册用户 boolean flag = us.register(user); if(flag) JOptionPane.showMessageDialog(null, "注册成功!"); transfer.setFlag(flag); if(flag){ transfer.setResult("注册成功!"); }else{ transfer.setResult("注册失败!"); } }else if(cmd.equals("uploadFile")){ FileInfo file =(FileInfo)transfer.getData(); fs.save(file); boolean flag = fs.save(file); if(flag) JOptionPane.showMessageDialog(null, "上传成功!"); transfer.setFlag(flag); if(flag){ transfer.setResult("上传成功!"); }else{ transfer.setResult("上传失败!"); } } return transfer; } } package fileUpload; import java.io.BufferedInputStream; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.io.ObjectInputStream; import java.io.ObjectOutputStream; import java.net.Socket; import java.net.UnknownHostException; import java.util.Scanner; /** * 主菜单类 */ public class SocketClient { Scanner input = new Scanner(System.in); private static Socket socket = null;// 客户端Socket // 主菜单 public void showMainMenu() throws IOException { System.out.println("******欢迎使用xx文件上传器******"); System.out.println("1.登录\n2.注册\n3.退出"); System.out.println("****************************"); System.out.println("请选择:"); int choice = input.nextInt();// 获取用户的选择 socket = new Socket("localhost", 8888); switch (choice) { case 1: showLogin();// 登录 break; case 2: showRegister();// 注册 break; case 3: System.out.println("欢迎你对本系统的支持"); System.exit(0); default: System.out.println("输入有误!"); System.exit(0); } } // 用户注册 private void showRegister() throws IOException { User user = new User(); CommandTransfer transfer = new CommandTransfer(); int count = 0;// 登陆次数 while (true) { count++; if (count > 3) { System.out.println("你已连续3次登陆失败,程序退出!"); System.exit(0); } System.out.print("请输入用户名:"); user.setUsename(input.next()); System.out.print("请输入密码:"); user.setUsepsd(input.next()); System.out.print("请再次输入密码:"); String rePassword = input.next(); if (!user.getUsepsd().equals(rePassword)) { System.out.println("两次输入的密码不一致!"); System.out.println("**********************"); continue; } transfer.setCmd("register"); transfer.setData(user); try { sendData(transfer);// 将数据发送到服务器 transfer = getData();// 获取服务器返回的数据 System.out.println(transfer.getResult());// 输出显示结果 System.out.println("*****************************"); if (transfer.isFlag()) { break;// 如果注册成功,则不再重复执行注册 } } catch (UnknownHostException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } finally { } } showLogin();// 注册成功后显示登陆 } // 获得服务器的数据 public static CommandTransfer getData() throws IOException { // TODO Auto-generated method stub ObjectInputStream ois = null; CommandTransfer res = null; try { if(socket.isClosed()) socket = new Socket("localhost", 8888); ois = new ObjectInputStream(socket.getInputStream()); res = (CommandTransfer) ois.readObject(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (ClassNotFoundException e) { // TODO Auto-generated catch block e.printStackTrace(); } finally { if (ois != null) { try { ois.close(); //socket.close(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } return res; } /* * 上传文件 */ private void showUploadFile() throws IOException { System.out.println("请输入上传文件的绝对路径:"); String path = input.next(); FileInfo file = null; FileInputStream fis = null; BufferedInputStream bis = null; String fname = path.substring(path.lastIndexOf("/") + 1); try { fis = new FileInputStream(path); byte[] fcontent = new byte[fis.available()]; bis = new BufferedInputStream(fis); bis.read(fcontent); file = new FileInfo(fname, fcontent); } catch (FileNotFoundException e) { e.printStackTrace(); } finally { bis.close(); fis.close(); } CommandTransfer transfer = new CommandTransfer(); transfer.setCmd("uploadFile"); transfer.setData(file); try { sendData(transfer);// 将数据发送到服务器 transfer = getData();// 获取服务器返回的数据 System.out.println(transfer.getResult());// 输出显示结果 } catch (UnknownHostException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } finally { } } // 向服务器发送数据 public static void sendData(CommandTransfer cmd) throws IOException { // TODO Auto-generated method stub ObjectOutputStream oos = null; try { if(socket.isClosed())//判断socket是否被关闭,如果关闭,就打开 socket = new Socket("localhost", 8888); oos = new ObjectOutputStream(socket.getOutputStream()); oos.writeObject(cmd); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } // 用户登陆 private void showLogin() throws IOException { User user = new User(); CommandTransfer transfer = new CommandTransfer(); while (true) { System.out.print("请输入用户名:"); user.setUsename(input.next()); System.out.print("请输入密码:"); user.setUsepsd(input.next()); transfer.setCmd("login"); transfer.setData(user); try { sendData(transfer);// 将数据发送到服务器 transfer = getData();// 获取服务器返回的数据 System.out.println(transfer.getResult());// 输出显示结果 System.out.println("*****************************"); if (transfer.isFlag()) { break;// 如果注册成功,则不再重复执行注册 } } catch (UnknownHostException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } finally { } } showUploadFile();// 登陆成功后进行文件的上传 } }
运行效果图如下:
文件已保存到数据库中了,这个项目对于想练练Socket通信和mysql的朋友来说,还是挺不错的。
以上所述是小编给大家介绍的Java Socket+mysql实现简易文件上传器,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持!
相关文章
详解SpringCloud LoadBalancer 新一代负载均衡器
这篇文章主要为大家介绍了SpringCloud LoadBalancer新一代负载均衡器详解使用详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-01-01使用MappingJackson2XmlView实现JSON到XML的视图转换
MappingJackson2XmlView来实现从JSON到XML格式的响应转换,本文将通过案例,将展示如何将JSON格式的数据转换为XML格式,以满足不同客户端的数据交换需求,需要的朋友可以参考下2024-07-07
最新评论