多端并用
数据随身携带
21 |
8 | * 用来获取数据库连接
9 | */
10 |
11 |
12 | public class DbUtil {
13 | public static Connection connection = null;
14 |
15 | static String dbName = "netdisc";
16 |
17 | static {
18 | try {
19 | //加载驱动
20 | Class.forName("com.mysql.cj.jdbc.Driver");
21 | } catch (ClassNotFoundException e) {
22 | e.printStackTrace();
23 | }
24 | }
25 |
26 | /**
27 | * 用 数据库名 做参数调用该方法，返回一个创建好的数据库连接
28 | *
29 | * @param db
30 | * @return Connection
31 | */
32 | public static Connection getConnection(String db) {
33 | try {
34 | //获得数据库链接
35 | connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/" + db + "?serverTimezone=Asia/Shanghai", "root", "");
36 | } catch (SQLException e) {
37 | e.printStackTrace();
38 | }
39 | return connection;
40 | }
41 |
42 | /**
43 | * 执行查询语句，选择某些列 返回该列的值 （int 类型）
44 | *
45 | * @param SQL
46 | * @param colName
47 | * @return
48 | */
49 | public static int selectToInt(String SQL, String colName) {
50 | Connection connection = getConnection(dbName);
51 | int data = 0;
52 | try {
53 | Statement statement = connection.createStatement();
54 | ResultSet resultSet = statement.executeQuery(SQL);
55 |
56 | while (resultSet.next()) {
57 | data = resultSet.getInt(colName);
58 | }
59 | connection.close();
60 | statement.close();
61 | resultSet.close();
62 | } catch (SQLException e) {
63 | e.printStackTrace();
64 | } finally {
65 | return data;
66 | }
67 | }
68 |
69 | /**
70 | * 执行查询语句，选择某些列 返回该列的值 （String 类型）
71 | *
72 | * @param SQL
73 | * @param colName
74 | * @return
75 | */
76 | public static String selectToString(String SQL, String colName) {
77 | Connection connection = getConnection(dbName);
78 | try {
79 | Statement statement = connection.createStatement();
80 | } catch (SQLException e) {
81 | e.printStackTrace();
82 | }
83 | return "";
84 | }
85 |
86 | public static ResultSet selectToSet(String SQL) {
87 | Connection connection = getConnection(dbName);
88 | ResultSet resultSet=null;
89 | try {
90 | Statement statement = connection.createStatement();
91 | resultSet = statement.executeQuery(SQL);
92 |
93 | } catch (SQLException e) {
94 | e.printStackTrace();
95 | } finally {
96 | return resultSet;
97 | }
98 | }
99 |
100 | public static boolean execute(String SQL) {
101 | Connection connection = getConnection(dbName);
102 | boolean successful = false;
103 | try {
104 | Statement statement = connection.createStatement();
105 | successful = statement.execute(SQL);
106 | connection.close();
107 | statement.close();
108 | } catch (SQLException e) {
109 | e.printStackTrace();
110 | } finally {
111 | return successful;
112 |
113 | }
114 | }
115 | }
116 |
--------------------------------------------------------------------------------
/src/Model/FileDownload.java:
--------------------------------------------------------------------------------
1 | package model;
2 |
3 | import utils.Authentication;
4 |
5 | import javax.servlet.ServletException;
6 | import javax.servlet.annotation.WebServlet;
7 | import javax.servlet.http.Cookie;
8 | import javax.servlet.http.HttpServlet;
9 | import javax.servlet.http.HttpServletRequest;
10 | import javax.servlet.http.HttpServletResponse;
11 | import java.io.*;
12 | import java.net.URLEncoder;
13 | import java.sql.Connection;
14 | import java.sql.ResultSet;
15 | import java.sql.SQLException;
16 | import java.sql.Statement;
17 |
18 | @WebServlet(name = "FileDownload", urlPatterns = "/downloadFile")
19 | public class FileDownload extends HttpServlet {
20 | protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
21 |
22 |
23 | if(!Authentication.isLogin(request)){
24 | return;
25 | }
26 |
27 | //获取文件名
28 | String filename = request.getParameter("filename");
29 |
30 | if (filename == null) {
31 | System.out.println("无 filename 参数");
32 | return;
33 | }
34 |
35 | System.out.println("filename = " + filename);
36 |
37 |
38 | //查找该用户的数据库，在数据库中找到文件对应的下载地址
39 |
40 | Connection connection=DbUtil.getConnection("netdisc");
41 |
42 | try {
43 | Statement statement=connection.createStatement();
44 |
45 | ResultSet resultSet=statement.executeQuery(
46 | "select filename,path from "+request.getSession().getAttribute("username")+
47 | " where filename='"+filename+"';");
48 |
49 | String path=null;
50 | while (resultSet.next()){
51 | path=resultSet.getString("path");
52 | }
53 |
54 | File file=new File(path);
55 |
56 | if (file.exists()) {
57 | // 配置响应头，指定响应类型为附件，对中文文件名进行编码
58 | response.addHeader("Content-Disposition",
59 | "attachment;filename=" + URLEncoder.encode(filename, "utf-8"));
60 |
61 | //添加文件大小信息
62 | response.addHeader("Content-Length", "" + file.length());
63 |
64 | InputStream inputStream = new FileInputStream(file);
65 |
66 | byte[] chars = new byte[1024];
67 | int len = -1;
68 | OutputStream out = response.getOutputStream();
69 | while ((len = inputStream.read(chars)) != -1) {
70 | out.write(chars, 0, len);
71 | }
72 | inputStream.close();
73 | }
74 |
75 | } catch (SQLException e) {
76 | e.printStackTrace();
77 | }
78 |
79 | }
80 |
81 | }
82 |
83 |
84 |
85 |
--------------------------------------------------------------------------------
/src/Model/FileUpload.java:
--------------------------------------------------------------------------------
1 | package model;
2 |
3 | import utils.Authentication;
4 |
5 | import javax.servlet.ServletException;
6 | import javax.servlet.annotation.MultipartConfig;
7 | import javax.servlet.annotation.WebServlet;
8 | import javax.servlet.http.*;
9 | import java.io.IOException;
10 | import java.sql.Connection;
11 | import java.sql.SQLException;
12 | import java.sql.Statement;
13 |
14 | @MultipartConfig
15 | @WebServlet(name = "FileUpload",urlPatterns = "/upload")
16 | public class FileUpload extends HttpServlet {
17 | @Override
18 | protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
19 |
20 | if(!Authentication.isLogin(request)){
21 | return;
22 | }
23 |
24 |
25 | //获取文件对象
26 | Part part=request.getPart("file");
27 |
28 | String filename=part.getSubmittedFileName();
29 |
30 | String path="D://0testFile/"+filename;
31 |
32 | //保存
33 | part.write(path);
34 |
35 | //----------------------------------- 更新数据库信息 ---------------------
36 |
37 | //获取数据库连接
38 | Connection connection=DbUtil.getConnection("netdisc");
39 |
40 | try {
41 | Statement statement=connection.createStatement();
42 | statement.execute("insert into "+request.getSession().getAttribute("username")+" values(0,'"+filename+"',"+part.getSize()+",now(),'"+path+"');");
43 |
44 | } catch (SQLException e) {
45 | e.printStackTrace();
46 | response.getWriter().print(e.getMessage());
47 | return;
48 | }
49 | response.sendRedirect("disc.html");
50 | }
51 |
52 | }
53 |
--------------------------------------------------------------------------------
/src/Utils/Authentication.java:
--------------------------------------------------------------------------------
1 | package utils;
2 |
3 | import javax.servlet.http.HttpServletRequest;
4 | import javax.servlet.http.HttpSession;
5 |
6 |
7 | /**
8 | * 鉴权模块，验证用户是否有操作权限
9 | */
10 | public class Authentication {
11 |
12 | /**
13 | * 传入请求对象做参数，从请求所属的会话中，获取用户名属性
14 | * @param request
15 | * @return
16 | */
17 | public static boolean isLogin(HttpServletRequest request){
18 | String username=(String)request.getSession().getAttribute("username");
19 |
20 | //判断用户是否存在
21 | return username != null;
22 | }
23 | }
24 |
--------------------------------------------------------------------------------
/src/Utils/Encoder.java:
--------------------------------------------------------------------------------
1 | package utils;
2 |
3 | import java.security.MessageDigest;
4 | import java.security.NoSuchAlgorithmException;
5 | import java.util.Base64;
6 |
7 | public class Encoder {
8 |
9 |
10 | public static String encodeBase64(String data) {
11 | MessageDigest md= null;
12 | try {
13 | md = MessageDigest.getInstance("md5");
14 | } catch (NoSuchAlgorithmException e) {
15 | e.printStackTrace();
16 | }
17 | byte[] dataEncode= md.digest(data.getBytes());
18 |
19 | return Base64.getEncoder().encodeToString(dataEncode);
20 | }
21 |
22 | }
23 |
--------------------------------------------------------------------------------
/src/filters/EncodeFilter.java:
--------------------------------------------------------------------------------
1 | package filters;
2 |
3 | import javax.servlet.*;
4 | import javax.servlet.Filter;
5 | import javax.servlet.annotation.WebFilter;
6 | import javax.servlet.http.HttpServletRequest;
7 | import javax.servlet.http.HttpServletResponse;
8 | import java.io.IOException;
9 |
10 | /**
11 | * 编码过滤器，匹配所有请求
12 | */
13 | @WebFilter(filterName = "Filter1",urlPatterns = "/*")
14 | public class EncodeFilter implements Filter {
15 | public void destroy() {
16 | }
17 |
18 | /**
19 | * 将 请求编码 修改为utf-8编码
20 | *
21 | * @param req
22 | * @param resp
23 | * @param chain
24 | * @throws ServletException
25 | * @throws IOException
26 | */
27 | public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain) throws ServletException, IOException {
28 | HttpServletRequest request=(HttpServletRequest) req;
29 | HttpServletResponse response=(HttpServletResponse)resp;
30 | System.out.println("编码过滤");
31 | request.setCharacterEncoding("utf-8");
32 |
33 | chain.doFilter(request, response);
34 | }
35 |
36 | public void init(FilterConfig config) throws ServletException {
37 |
38 | }
39 |
40 | }
41 |
--------------------------------------------------------------------------------
/src/filters/Filter.java:
--------------------------------------------------------------------------------
1 | package filters;
2 |
3 | import javax.servlet.*;
4 | import javax.servlet.annotation.WebFilter;
5 | import javax.servlet.http.HttpServletRequest;
6 | import javax.servlet.http.HttpServletResponse;
7 | import java.io.IOException;
8 |
9 | /**
10 | * 首页访问过滤
11 | */
12 | @WebFilter(filterName = "Filter2",urlPatterns = "/")
13 | public class Filter implements javax.servlet.Filter {
14 | //销毁
15 | public void destroy() {
16 | }
17 |
18 | /**
19 | * 从会话中获取 用户名 参数，跳转到对应用户的网盘页面
20 | * 若不存在，则跳转到登录页面
21 | * @param req
22 | * @param resp
23 | * @param chain
24 | * @throws ServletException
25 | * @throws IOException
26 | */
27 | public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain) throws ServletException, IOException {
28 | HttpServletRequest request=(HttpServletRequest) req;
29 | HttpServletResponse response=(HttpServletResponse)resp;
30 |
31 | String username=(String)request.getSession().getAttribute("username");
32 |
33 | System.out.println("首页跳转过滤");
34 | if(username==null){
35 | response.sendRedirect("user/login.html");
36 | }else {
37 | response.sendRedirect("disc.html");
38 | }
39 | }
40 |
41 | //初始化
42 | public void init(FilterConfig config) throws ServletException {
43 |
44 | }
45 |
46 | }
47 |
--------------------------------------------------------------------------------
/src/user/LoginController.java:
--------------------------------------------------------------------------------
1 | package user;
2 |
3 | import model.DbUtil;
4 | import utils.Encoder;
5 |
6 | import javax.servlet.ServletException;
7 | import javax.servlet.annotation.WebServlet;
8 | import javax.servlet.http.HttpServlet;
9 | import javax.servlet.http.HttpServletRequest;
10 | import javax.servlet.http.HttpServletResponse;
11 | import java.io.IOException;
12 | import java.sql.Connection;
13 | import java.sql.ResultSet;
14 | import java.sql.SQLException;
15 | import java.sql.Statement;
16 |
17 | @WebServlet(name = "LoginController", urlPatterns = "/user/userLogin")
18 | public class LoginController extends HttpServlet {
19 |
20 | protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
21 | String username = request.getParameter("username");
22 | String password = request.getParameter("password");
23 | System.out.println("username = " + username);
24 | System.out.println("password = " + password);
25 |
26 | //密码加密
27 | password = Encoder.encodeBase64(password);
28 |
29 | int userCount = DbUtil.selectToInt("select count(username) from user where username=\""
30 | + username + "\" and password=\"" + password + "\";", "count(username)");
31 | if (userCount == 0) {
32 | System.out.println("用户名或密码错误");
33 | return;
34 | }
35 |
36 | //保存登录状态
37 | request.getSession().setAttribute("username", username);
38 | response.sendRedirect("../disc.html");
39 | }
40 | }
41 |
--------------------------------------------------------------------------------
/src/user/RegisterController.java:
--------------------------------------------------------------------------------
1 | package user;
2 |
3 | import model.DbUtil;
4 | import utils.Encoder;
5 |
6 | import javax.servlet.ServletException;
7 | import javax.servlet.annotation.WebServlet;
8 | import javax.servlet.http.HttpServlet;
9 | import javax.servlet.http.HttpServletRequest;
10 | import javax.servlet.http.HttpServletResponse;
11 | import java.io.IOException;
12 | import java.sql.Connection;
13 | import java.sql.ResultSet;
14 | import java.sql.SQLException;
15 | import java.sql.Statement;
16 |
17 | @WebServlet(name = "RegisterController", urlPatterns = "/user/userRegister")
18 | public class RegisterController extends HttpServlet {
19 | @Override
20 | protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
21 | String username = request.getParameter("username");
22 | String password = request.getParameter("password");
23 |
24 | //前端验证：密码跟用户名都有时，才能发送请求
25 |
26 |
27 | //查找是否有重复用户，有则报错
28 | int userCount = DbUtil.selectToInt("select count(username) from user where username=\""
29 | + username + "\";", "count(username)");
30 |
31 | if (userCount > 0) {
32 | System.out.println("用户已存在");
33 | return;
34 | }
35 |
36 | //加密密码
37 | password = Encoder.encodeBase64(password);
38 |
39 | //将注册用户的信息插入数据库
40 | DbUtil.execute("insert into user values(0,\"" + username + "\",\"" + password + "\");");
41 |
42 | //用户创建成功后，为其创建文件表
43 | DbUtil.execute("create table " + username + "(id int auto_increment not null,filename varchar(50) not null," +
44 | "filesize bigint not null,time timestamp default current_timestamp,path varchar(100) not null,primary key(id));");
45 |
46 | //filesize long bigint
47 |
48 | System.out.println("ok");
49 | }
50 | }
51 |
--------------------------------------------------------------------------------
/web/WEB-INF/web.xml:
--------------------------------------------------------------------------------
1 |
2 |