UserServiceImpl.java 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package com.example.onlinejudge.service.impl;
  2. import cn.dev33.satoken.stp.SaTokenInfo;
  3. import cn.dev33.satoken.stp.StpUtil;
  4. import cn.dev33.satoken.util.SaResult;
  5. import cn.hutool.core.util.StrUtil;
  6. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  7. import com.example.onlinejudge.model.entity.User;
  8. import com.example.onlinejudge.mapper.UserMapper;
  9. import com.example.onlinejudge.model.entity.result.Result;
  10. import com.example.onlinejudge.model.entity.result.ResultCode;
  11. import com.example.onlinejudge.service.UserService;
  12. import org.springframework.beans.factory.annotation.Autowired;
  13. import org.springframework.stereotype.Service;
  14. import sun.security.krb5.internal.PAData;
  15. import java.util.LinkedHashMap;
  16. /**
  17. * 实现用户服务
  18. */
  19. @Service
  20. public class UserServiceImpl implements UserService {
  21. @Autowired
  22. private UserMapper userMapper;
  23. /**
  24. * 用户登录服务
  25. *
  26. * @param username 用户名
  27. * @param password 密码
  28. * @return 登录结果
  29. */
  30. @Override
  31. public Result login(String username, String password) {
  32. QueryWrapper<User> queryWrapper = new QueryWrapper<>();
  33. queryWrapper.eq("username", username).eq("password", password);
  34. User user = userMapper.selectOne(queryWrapper);
  35. if (user != null) {
  36. StpUtil.login(user.getId());
  37. return Result.success();
  38. }
  39. return Result.error(ResultCode.USER_LOGIN_ERROR);
  40. }
  41. @Override
  42. public String getLoginUser() {
  43. return StpUtil.getLoginId().toString();
  44. }
  45. }