package com.library.controller; import java.io.File; import java.io.IOException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Scope; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import com.library.model.User; import com.library.service.UserService; /** * 登录/注册 */ @Controller @Scope(value = "prototype") public class LoginController { @Autowired private UserService userService; @RequestMapping("login") public String login(HttpServletRequest request) { String username = request.getParameter("username"); String password = request.getParameter("password"); if (username == null || username.isEmpty()) { return "login"; } if (password == null || password.isEmpty()) { request.setAttribute("username", username); request.setAttribute("errorMessage", "密码不能为空"); return "login"; } User user = userService.login(username, password); if (user != null) { HttpSession session = request.getSession(); session.setAttribute("user", user); if ("1".equals(user.getIdent())) { return "redirect:admin_home.cc"; } else if ("0".equals(user.getIdent())) { return "redirect:user_home.cc"; } else { return "login"; } } else { request.setAttribute("username", username); request.setAttribute("errorMessage", "用户名或密码错误"); return "login"; } } @RequestMapping("register") public String register(HttpServletRequest request) { String username = request.getParameter("username"); String password = request.getParameter("password"); String sex = request.getParameter("sex"); String telephone = request.getParameter("telephone"); String address = request.getParameter("address"); String headUrl = "/static/images/touxiang/01.jpg"; if (username == null || username.isEmpty()) { return "register"; } if (userService.checkUsername(username)) { request.setAttribute("errorMessage", "用户名已存在,立即登录"); return "login"; } MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) (request); MultipartFile head = multipartRequest.getFile("head"); if (!head.getOriginalFilename().isEmpty()) { String path = request.getServletContext().getRealPath("/") + "page/static/images/touxiang/"; String name = String.valueOf(System.currentTimeMillis() + "_" + head.getOriginalFilename()); File destFile = new File(path, name); try { head.transferTo(destFile); } catch (IllegalStateException | IOException e) { e.printStackTrace(); } headUrl = "images/touxiang/" + name; } User user = new User(); user.setUsername(username); user.setPassword(password); user.setSex(sex); user.setTelephone(telephone); user.setAddress(address); user.setIdent("0"); user.setPortrait(headUrl); userService.addUser(user); request.setAttribute("errorMessage", "注册成功,立即登录"); return "login"; } @RequestMapping("logout") public String logout(HttpSession session) { session.removeAttribute("user"); return "login"; } }
最近下载更多
pangzhihui LV13
4月11日
weilaizhisheng LV21
3月21日
uid0901 LV2
3月12日
xiaopengzep LV1
3月11日
微信网友_6427471221870592 LV2
3月10日
n2352441 LV4
3月3日
潘潘123456 LV2
3月2日
renyong LV3
3月1日
wanglinddad LV55
2月28日
lo8lukoukoum LV4
2月27日
最近浏览更多
李俊雄 LV3
昨天
2017143155 LV12
5月6日
fawaikuai
4月19日
暂无贡献等级
saaaaaa LV14
4月16日
yeanan LV1
4月12日
pangzhihui LV13
4月11日
第四维
4月1日
暂无贡献等级
2131234536546 LV7
3月31日
陈仁怀 LV1
3月30日
微信网友_6927932988952576 LV12
3月30日