线上实习 奋勇向前——博客登陆

//LoginController

@Controller
@RequestMapping("/admin")
public class LoginController {

@Autowired
private IUserService userService;

@GetMapping
public String toLogin(){
return “admin/login”;
}

@PostMapping("/login")
public String login(String username, String password, HttpSession session, RedirectAttributes redirectAttributes){
User user = userService.checkUser(username,password);
if(user!=null){
session.setAttribute(“user”,user);
return “admin/index”;
}else{
redirectAttributes.addFlashAttribute(“message”,“用户名和密码错误”);
return “redirect:/admin”;
}
}

@GetMapping(“logout”)
public String logout(HttpSession session){
session.removeAttribute(“user”);
return “admin/login”;
}
}

//UserServiceImpl

package com.zr.service.impl;

import com.zr.dao.UserDao;
import com.zr.po.User;
import com.zr.service.IUserService;
import com.zr.util.MD5Util;
import org.springframework.beans.factory.annotation.Autowired;

public class UserServiceImpl implements IUserService {

@Autowired
private UserDao userDao;
@Override
public User checkUser(String username, String password) {
return userDao.findByUsernameAndPassword(username,MD5Util.code(password));
}
}

//MD5Util
public class MD5Util {

/**
 * MD5加密类
 * @param str 要加密的字符串
 * @return    加密后的字符串
 */

public static String code(String str){
try {
MessageDigest md = MessageDigest.getInstance(“MD5”);
md.update(str.getBytes());
byte[]byteDigest = md.digest();
int i;
StringBuffer buf = new StringBuffer("");
for (int offset = 0; offset < byteDigest.length; offset++) {
i = byteDigest[offset];
if (i < 0)
i += 256;
if (i < 16)
buf.append(“0”);
buf.append(Integer.toHexString(i));
}
//32位加密
return buf.toString();
// 16位的加密
//return buf.toString().substring(8, 24);
} catch (NoSuchAlgorithmException e) {
e.printStackTrace();
return null;
}

}

public static void main(String[] args) {
System.out.println(code(“1”));
}
}