|
|
@@ -14,7 +14,9 @@ import org.springframework.stereotype.Controller;
|
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
|
|
|
|
+import javax.servlet.http.Cookie;
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.Map;
|
|
|
@@ -44,6 +46,7 @@ public class UserController extends BaseController<User> {
|
|
|
* @param name 名称
|
|
|
* @param password 密码
|
|
|
* @param request request
|
|
|
+ * @param response response
|
|
|
* @return 登陆信息
|
|
|
* <table border=1 cellpadding=5 cellspacing=0 summary="Fields and types">
|
|
|
* <tr>
|
|
|
@@ -71,22 +74,40 @@ public class UserController extends BaseController<User> {
|
|
|
@RequestMapping("/login")
|
|
|
@ResponseBody
|
|
|
public Map<String, Object> login(@NotEmpty("name") String name, @NotEmpty("password") String password,
|
|
|
- HttpServletRequest request) {
|
|
|
+ HttpServletRequest request, HttpServletResponse response) {
|
|
|
User user = userService.login(name, password);
|
|
|
if (user != null) {
|
|
|
sessionHelper.saveSession(request, user);
|
|
|
String userCode = user.codeNotEmpty();
|
|
|
RememberKey rememberKey = rememberKeyService.generate(userCode);
|
|
|
+ // 生成的 key 保存到 cookie 里
|
|
|
+ Integer maxInactiveInterval = rememberKey.getMaxInactiveInterval();
|
|
|
+ addCookie("code", userCode, maxInactiveInterval, response);
|
|
|
+ addCookie("key", rememberKey.getKey(), maxInactiveInterval, response);
|
|
|
+
|
|
|
Map<String, Object> map = new HashMap<>();
|
|
|
- map.put("code", userCode);
|
|
|
map.put("name", name);
|
|
|
map.put("role", user.getRole());
|
|
|
- map.put("key", rememberKey.getKey());
|
|
|
return map;
|
|
|
}
|
|
|
return null;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 添加 cookie
|
|
|
+ *
|
|
|
+ * @param name cookie 名
|
|
|
+ * @param value cookie 值
|
|
|
+ * @param maxAge 有效时间(秒)
|
|
|
+ * @param response response
|
|
|
+ */
|
|
|
+ private void addCookie(String name, String value, Integer maxAge, HttpServletResponse response) {
|
|
|
+ Cookie cookie = new Cookie(name, value);
|
|
|
+ cookie.setMaxAge(maxAge);
|
|
|
+ cookie.setPath("/");
|
|
|
+ response.addCookie(cookie);
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 退出登陆
|
|
|
*
|