JavaWeb檔案上傳流程

CREATE TABLE `t_user_info` (
`noid` int(11) NOT NULL AUTO_INCREMENT,
`username` varchar(32) COLLATE utf8mb4_unicode_ci NOT NULL,
`password` varchar(32) COLLATE utf8mb4_unicode_ci NOT NULL,
`head_portrait_path` varchar(64) COLLATE utf8mb4_unicode_ci NOT NULL,
PRIMARY KEY (`noid`)
) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci

public class BaseServlet extends HttpServlet{
@Override
protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
//1.獲取瀏覽器請求的資源
String uri = req.getRequestURI();
//2.獲取請求的方法名，最後斜線後面的內容
String methodName = uri.substring(uri.lastIndexOf("/")+1);
try {
//3.根據方法名獲取方法，通過反射獲取
Method method = this.getClass().getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);
//4.呼叫方法
method.invoke(this, req, resp);
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}

driverClassName=com.mysql.jdbc.Driver
url=jdbc:mysql://127.0.0.1/db_blog?useSSL=true
username=root
password=111111

public class BaseDao {
//採用單例模式實現，防止資料庫連線超時
private static DataSource ds = null;
public QueryRunner initQueryRunner() throws Exception {
if (ds == null) {
String dbFile = this.getClass().getClassLoader().getResource("/").getFile();
dbFile = dbFile.substring(1) + "db.properties";
FileReader fr = new FileReader(dbFile);
Properties pro = new Properties();
pro.load(fr);
ds = DruidDataSourceFactory.createDataSource(pro);
}
QueryRunner qur = new QueryRunner(ds);
return qur;
}
}

@WebServlet("/user/*")
public class UserServlet extends BaseServlet{

//業務層類，用於呼叫業務層方法
UserService userService = new UserServiceImpl();

/**
* 註冊使用者
* @param req
* @param resp
*/
public void register(HttpServletRequest req, HttpServletResponse resp) {
//獲取資料
DiskFileItemFactory factory = new DiskFileItemFactory();
ServletFileUpload fileUpload = new ServletFileUpload(factory);
try {
List<FileItem> fileItemList = fileUpload.parseRequest(req);
//獲取當前專案的路徑
String classesPath = this.getClass().getResource("/").getPath();

File f1 = new File(classesPath);
//專案路徑
String projectPath = f1.getParentFile().getParentFile().getParentFile().getAbsolutePath();

//最後上傳的路徑
String uploadPath = projectPath + "\ROOT\upload\";

File f2 = new File(uploadPath);
if (!f2.exists()) {
f2.mkdirs();
}
//存入資料庫的路徑
String headPortraitPath = "";
for (FileItem fileItem : fileItemList) {
if (!fileItem.isFormField()) {
//是檔案域
String fileName = fileItem.getName();
//獲取原來檔案的字尾
String suffix = fileName.substring(fileName.lastIndexOf("."));
//生成新的檔名，為防止重複，採用亂數
String destFileName = UUID.randomUUID().toString().replace("-", "");

//存入資料庫的路徑拼接完畢，例如格式：隨機檔名.txt
headPortraitPath = destFileName + suffix;
//寫入硬碟的路徑
uploadPath += headPortraitPath;
//獲取輸入流
InputStream is = fileItem.getInputStream();
//輸出流
FileOutputStream fos = new FileOutputStream(uploadPath);
//將上傳的檔案寫入指定路徑
try {
byte[] buf = new byte[10240];
while (true) {
int realLen = is.read(buf, 0, buf.length);
if (realLen < 0) {
break;
}
fos.write(buf, 0, realLen);
}
} finally {
if (fos != null)
fos.close();
if (is != null)
is.close();
}
} else {
//不是檔案域，是普通控制元件
//獲取輸入框的名稱
String fieldName = fileItem.getFieldName();
//獲取輸入框中的值
String fieldVal = fileItem.getString("utf-8");
//加入請求域中
req.setAttribute(fieldName, fieldVal);
}
}
String username = (String) req.getAttribute("username");
String password = (String) req.getAttribute("password");
//驗證引數是否合法，不為空
boolean flag = userService.exam(username, password);
if (flag) {
//將資料存入資料庫
User user = new User();
user.setUsername(username);
user.setPassword(password);
user.setHead_portrait_path(headPortraitPath);
if (userService.save(user)) {
resp.sendRedirect(req.getContextPath()+"/login.jsp");
}
} else {
resp.sendRedirect(req.getContextPath()+"/register.jsp");
}
} catch (Exception e1) {
// TODO Auto-generated catch block
e1.printStackTrace();
}
}

/**
* 使用者登入
* @param req
* @param resp
*/
public void login(HttpServletRequest req, HttpServletResponse resp) {
//獲取資料
String username = req.getParameter("username");
String password = req.getParameter("password");
//驗證是否存在該使用者
User user = new User();
try {
if (userService.exam(username, password)) {
user.setUsername(username);
user.setPassword(password);
user = userService.getByUser(user);
if (user.getHead_portrait_path() != null) {
HttpSession s1 = req.getSession();
s1.setAttribute("user", user);
resp.sendRedirect(req.getContextPath()+"/index.jsp");
} else {
resp.sendRedirect(req.getContextPath()+"/login.jsp");
}
} else {
resp.sendRedirect(req.getContextPath()+"/login.jsp");
}
} catch(Exception e) {
e.printStackTrace();
}
}
}