package cc.mrbird.febs.system.controller; import cc.mrbird.febs.common.annotation.Limit; import cc.mrbird.febs.common.controller.BaseController; import cc.mrbird.febs.common.entity.FebsResponse; import cc.mrbird.febs.common.enumerates.AgentLevelEnum; import cc.mrbird.febs.common.enumerates.OrderPayMethodEnum; import cc.mrbird.febs.common.enumerates.OrderStatusEnum; import cc.mrbird.febs.common.exception.FebsException; import cc.mrbird.febs.common.service.ValidateCodeService; import cc.mrbird.febs.common.utils.Md5Util; import cc.mrbird.febs.common.utils.RedisUtils; import cc.mrbird.febs.mall.entity.ChatUser; import cc.mrbird.febs.mall.entity.MallMember; import cc.mrbird.febs.mall.mapper.ChatUserMapper; import cc.mrbird.febs.mall.mapper.MallAchieveRecordMapper; import cc.mrbird.febs.mall.mapper.MallMemberMapper; import cc.mrbird.febs.mall.mapper.MallOrderInfoMapper; import cc.mrbird.febs.monitor.entity.LoginLog; import cc.mrbird.febs.monitor.service.ILoginLogService; import cc.mrbird.febs.system.entity.User; import cc.mrbird.febs.system.service.IUserService; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.date.DateUtil; import cn.hutool.core.util.NumberUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import lombok.RequiredArgsConstructor; import org.apache.shiro.authc.UsernamePasswordToken; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RestController; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpSession; import javax.validation.constraints.NotBlank; import java.io.IOException; import java.math.BigDecimal; import java.text.NumberFormat; import java.text.ParseException; import java.util.*; /** * @author MrBird */ @Validated @RestController @RequiredArgsConstructor public class LoginController extends BaseController { private final IUserService userService; private final ValidateCodeService validateCodeService; private final ILoginLogService loginLogService; private final MallMemberMapper mallMemberMapper; private final MallOrderInfoMapper mallOrderInfoMapper; private final MallAchieveRecordMapper mallAchieveRecordMapper; private final ChatUserMapper chatUserMapper; private final RedisUtils redisUtils; @PostMapping("login") @Limit(key = "login", period = 60, count = 10, name = "登录接口", prefix = "limit") public FebsResponse login( @NotBlank(message = "{required}") String username, @NotBlank(message = "{required}") String password, @NotBlank(message = "{required}") String verifyCode, boolean rememberMe, HttpServletRequest request) throws FebsException { HttpSession session = request.getSession(); validateCodeService.check(session.getId(), verifyCode); password = Md5Util.encrypt(username.toLowerCase(), password); UsernamePasswordToken token = new UsernamePasswordToken(username, password, rememberMe); super.login(token); // 保存登录日志 LoginLog loginLog = new LoginLog(); loginLog.setUsername(username); loginLog.setSystemBrowserInfo(); this.loginLogService.saveLoginLog(loginLog); return new FebsResponse().success(); } @PostMapping("regist") public FebsResponse regist( @NotBlank(message = "{required}") String username, @NotBlank(message = "{required}") String password) throws FebsException { User user = userService.findByName(username); if (user != null) { throw new FebsException("该用户名已存在"); } this.userService.regist(username, password); return new FebsResponse().success(); } @GetMapping("index/{username}") public FebsResponse index(@NotBlank(message = "{required}") @PathVariable String username) { // 更新登录时间 this.userService.updateLoginTime(username); Map data = new HashMap<>(5); // 获取近期系统访问记录 List> lastSevenVisitCount = this.loginLogService.findLastSevenDaysVisitCount(null); data.put("lastSevenVisitCount", lastSevenVisitCount); User param = new User(); param.setUsername(username); List> lastSevenUserVisitCount = this.loginLogService.findLastSevenDaysVisitCount(param); data.put("lastSevenUserVisitCount", lastSevenUserVisitCount); data.put("totalAmount", NumberUtil.decimalFormatMoney(chatUserMapper.selectTotalAmount().doubleValue())); List chatUsers = chatUserMapper.selectUserList(0,null,null); List chatRobots = chatUserMapper.selectUserList(1,null,null); data.put("totalUser", (CollUtil.isEmpty(chatUsers) ? 0:chatUsers.size()) +"-"+(CollUtil.isEmpty(chatRobots) ? 0:chatRobots.size())); data.put("totalSystemPay", NumberUtil.decimalFormatMoney(chatUserMapper.selectTotalType(6,null,null).doubleValue())); data.put("totalIncome", NumberUtil.decimalFormatMoney(chatUserMapper.selectTotalType(8,null,null).doubleValue())); data.put("todaySystemPay", NumberUtil.decimalFormatMoney(chatUserMapper.selectTotalType(6,"D",DateUtil.date()).doubleValue())); data.put("todayIncome", NumberUtil.decimalFormatMoney(chatUserMapper.selectTotalType(8,"D",DateUtil.date()).doubleValue())); List chatUsersToday = chatUserMapper.selectUserList(0,"D",DateUtil.date()); List chatRobotsToday = chatUserMapper.selectUserList(1,"D",DateUtil.date()); data.put("todayUser", (CollUtil.isEmpty(chatUsersToday) ? 0:chatUsersToday.size()) +"-"+(CollUtil.isEmpty(chatRobotsToday) ? 0:chatRobotsToday.size())); return new FebsResponse().success().data(data); } public static BigDecimal formatCurrency(BigDecimal amount) { // 设置本地化格式为当前默认地区 Locale locale = Locale.getDefault(); // 创建一个货币格式化对象 NumberFormat currencyFormat = NumberFormat.getCurrencyInstance(locale); // 将 BigDecimal 格式化为货币格式的字符串 String formattedAmount = currencyFormat.format(amount); // 将格式化的字符串转换回 BigDecimal 对象 BigDecimal formattedBigDecimal = null; try { formattedBigDecimal = new BigDecimal(currencyFormat.parse(formattedAmount).doubleValue()); } catch (ParseException e) { e.printStackTrace(); } // 返回格式化后的 BigDecimal 对象 return formattedBigDecimal; } @GetMapping("images/captcha") @Limit(key = "get_captcha", period = 60, count = 10, name = "获取验证码", prefix = "limit") public void captcha(HttpServletRequest request, HttpServletResponse response) throws IOException, FebsException { validateCodeService.create(request, response); } }