上篇文章已经完成了用户注册功能,这篇文章将详细讲解后续功能的实现原理和展示实现代码,朋友们可以在我的博客里面查看《web登录注册案例(上)》
从代码中我们可以发现,实现了对于表单中的非文件上传控件的数据接收,并且封装为一个user对象,存入list集合中,并且更新了ServletContext中的list集合。至此,我们还剩下上传文件的控件没有处理。
首先,新建一个存放上传文件的文件夹
此处我们需要解决同名文件上传的问题,这里就需要我们对于文件名做一些操作,首先新建一个工具类UploadUtils.java
public class UploadUtils {
/**
* 生成唯一的文件名:
*/
public static String getUUIDFileName(String fileName) {
//将文件名前面的部分进行截取:xx.jpg ----> .jpg
int idx = fileName.lastIndexOf(".");
String extention = fileName.substring(idx);
String uuidFileName = UUID.randomUUID().toString().replace("-", "")+extention;
return uuidFileName;
}
}
这个工具类就可以生成一个唯一的文件名,下面是文件上传的代码
// 文件上传项:
// 文件上传功能:
// 获得文件上传的名称:
String fileName = fileItem.getName();
if (fileName!=null && !"".equals(fileName)) {
//通过工具类获得唯一文件名:
String uuidFileName = UploadUtils.getUUIDFileName(fileName);
//获得文件上传的数据
InputStream is = fileItem.getInputStream();
//获得文件上传的路径:
String path = this.getServletContext().getRealPath("/upload");
//对接输入流到输出流
url = path+"\\"+uuidFileName;
OutputStream os = new FileOutputStream(url);
int len = 0;
byte[] b = new byte[1024];
while((len = is.read(b)) != -1) {
os.write(b, 0, len);
}
is.close();
os.close();
}
此处就将文件写入到了我们项目下了upload文件夹下。可以在tomcat项目目录中查看。此处我们需要在user对象中加入我们的url路径。
user.setPath(url);
至此,用户注册功能全部完成,重定向到我们的登录页面,同时使用session将用户姓名传递过去
// 注册成功,跳转到登录页面:
request.getSession().setAttribute("username", user.getUsername());
response.sendRedirect(request.getContextPath()+"/login.jsp");
在login.jsp中显示用户姓名,首先拿到session。使用jsp表达式显示(代码隐去)
if(session.getAttribute("username")!=null){
username = (String)session.getAttribute("username");
}
注册失败让用户回到注册页面重新注册,用request返回错误信息
// 获得ServletContext对象:
List<User> userList = (List<User>) this.getServletContext().getAttribute("list");
//校验用户名
for (User u : userList) {
if (u.getUsername().equals(map.get("username"))) {
request.setAttribute("msg", "用户名已经存在");
request.getRequestDispatcher("/regist.jsp").forward(request, response);
return;
}
}
String msg = "";
if(request.getAttribute("msg")!=null){
msg = (String)request.getAttribute("msg");
}
登录功能实现
用户注册完成,跳转至登录页面,并且登录页面我们已经写好了用户的姓名,当用户写上password之后,提交到LoginServlet.java。
在LoginServlet.java中,判断用户信息是否正确,正确重定向至success.jsp页面,使用session将user对象传递至success.jsp。错误,转发至login.jsp。同时传递错误信息。(此处与注册之后的操作相同)
贴出success.jsp 页面
<body>
<%
if(session.getAttribute("user")!=null){
User user = (User)session.getAttribute("user");
//获得文件路径中的最后一个"\"
int idx = user.getPath().lastIndexOf("\\");
//得到唯一路径
String fileName = user.getPath().substring(idx+1);
%>
<div class="login">
<div class="header">
<h1>登录成功</h1>
</div>
<div class="content">
<table align="center">
<tr>
<td align="center"><img src="/reg_login/upload/<%= fileName %>" /></td>
</tr>
<tr>
<td align="center">欢迎<%= user.getNickname() %>,登录成功!</td>
</tr>
</table>
</div>
</div>
<%
}else{
%>
<div class="login">
<div class="header">
<h1>您还没有登录!请先去<a href="/reg_login/login.jsp">登录</a>!</h1>
</div>
</div>
<%
}
%>
</body>
Cookie保存用户信息
此处新建一个工具类,这个类的功能是返回cookie名等于name的cookie.
public class CookieUtils {
public static Cookie findCookie(Cookie[] cookies,String name) {
if(cookies == null) {
//说明客户端没有携带Cookie
return null;
}else {
//说明客户端携带Cookie
for (Cookie cookie : cookies) {
if(name.equals(cookie.getName())) {
return cookie;
}
}
return null;
}
}
}
判断是否需要保存用户信息,设置cookie有效路径(项目路径)和cookie的有效时间,此处设置为1day。
//判断复选框是否勾选
String remember = request.getParameter("remember");
if("true".equals(remember)) {
//记住用户名
Cookie cookie = new Cookie("username", user.getUsername());
//设置有效路径和有效时间
cookie.setPath("/reg_login");
cookie.setMaxAge(60*60*24);
//将cookie回写到浏览器
response.addCookie(cookie);
}
最后,将cookie信息显示在登录页面中。当用户关闭浏览器,一天只能再次打开,登录页面中应该存有用户的用户名信息。
<%
String username ="";
//获得从客户端携带过来的所有Cookie
Cookie[] cookies = request.getCookies();
//从cookies中查找指定名称的cookie
Cookie cookie = CookieUtils.findCookie(cookies,"username");
if(cookie != null){
username = cookie.getValue();
}
if(session.getAttribute("username")!=null){
username = (String)session.getAttribute("username");
}
String msg = "";
if(request.getAttribute("msg")!=null){
msg = (String)request.getAttribute("msg");
}
%>
此处把cookie放在session前面,使session可以覆盖掉cookie的值。