| | |
| | | import com.ruoyi.system.model.TbUser; |
| | | import com.ruoyi.system.query.UserMessageQuery; |
| | | import com.ruoyi.system.service.TbMessageService; |
| | | import com.ruoyi.system.service.TbOrderService; |
| | | import com.ruoyi.system.service.TbUserService; |
| | | import com.ruoyi.system.vo.InviteUserListVo; |
| | | import io.swagger.annotations.Api; |
| | |
| | | import org.springframework.web.bind.annotation.*; |
| | | |
| | | import java.util.ArrayList; |
| | | import java.util.List; |
| | | |
| | | @Slf4j |
| | | @RestController |
| | |
| | | @Autowired |
| | | private TokenService tokenService; |
| | | |
| | | @Autowired |
| | | private TbOrderService orderService; |
| | | |
| | | |
| | | @ApiOperation(value = "获取用户消息",tags = {"消息模块"}) |
| | | @GetMapping("/getUserMessage") |
| | | public R<Page<TbMessage>> getUserMessage(UserMessageQuery query) { |
| | | LoginUser loginUser = tokenService.getLoginUser(); |
| | | Long userId = loginUser.getUserId(); |
| | | Page<TbMessage> page = messageService.page(new Page<>(query.getPageNum(), query.getPageSize()), new LambdaQueryWrapper<TbMessage>().eq(TbMessage::getUserId, userId)); |
| | | Page<TbMessage> page = messageService.page(new Page<>(query.getPageNum(), query.getPageSize()), new LambdaQueryWrapper<TbMessage>().eq(query.getIsRead()!=null, TbMessage::getIsRead, query.getIsRead()).eq(TbMessage::getUserId, userId).orderByDesc(TbMessage::getCreateTime)); |
| | | page.getRecords().forEach(item -> { |
| | | String companyId = orderService.getById(item.getOrderId()).getCompanyId(); |
| | | item.setCompanyId(companyId); |
| | | }); |
| | | long count = messageService.count(new LambdaQueryWrapper<TbMessage>().eq(TbMessage::getIsRead, 0).eq(TbMessage::getUserId, userId)); |
| | | page.setCurrent(count); |
| | | return R.ok(page); |
| | | } |
| | | |
| | |
| | | |
| | | |
| | | |
| | | @ApiOperation(value = "分享有礼列表",tags = {"分享有礼列表"}) |
| | | @GetMapping("/getInviteUserList") |
| | | public R<Page<InviteUserListVo>> getInviteUserList(BasePage page) { |
| | | LoginUser loginUser = tokenService.getLoginUser(); |
| | | Long userId = loginUser.getUserId(); |
| | | Page<TbUser> page1 = userService.page(new Page<>(page.getPageNum(), page.getPageSize()), new LambdaQueryWrapper<TbUser>().eq(TbUser::getInviteId, userId).orderByDesc(TbUser::getCreateTime)); |
| | | Page<InviteUserListVo> inviteUserListVoPage = new Page<>(); |
| | | if(page1.getRecords().isEmpty()){ |
| | | return R.ok(inviteUserListVoPage); |
| | | } |
| | | ArrayList<InviteUserListVo> inviteUserListVos = new ArrayList<>(); |
| | | for (TbUser record : page1.getRecords()) { |
| | | InviteUserListVo inviteUserListVo = new InviteUserListVo(); |
| | | inviteUserListVo.setUsername(record.getUserName()); |
| | | inviteUserListVo.setCreateTime(record.getCreateTime()); |
| | | inviteUserListVo.setInviteNum(record.getInviteNum()); |
| | | inviteUserListVos.add(inviteUserListVo); |
| | | } |
| | | BeanUtils.copyProperties(page1,inviteUserListVoPage); |
| | | inviteUserListVoPage.setRecords(inviteUserListVos); |
| | | return R.ok(inviteUserListVoPage); |
| | | } |
| | | // @ApiOperation(value = "分享有礼列表",tags = {"分享有礼列表"}) |
| | | // @GetMapping("/getInviteUserList") |
| | | // public R<Page<InviteUserListVo>> getInviteUserList(BasePage page) { |
| | | // LoginUser loginUser = tokenService.getLoginUser(); |
| | | // Long userId = loginUser.getUserId(); |
| | | // Page<TbUser> page1 = userService.page(new Page<>(page.getPageNum(), page.getPageSize()), new LambdaQueryWrapper<TbUser>().eq(TbUser::getInviteId, userId).orderByDesc(TbUser::getCreateTime)); |
| | | // Page<InviteUserListVo> inviteUserListVoPage = new Page<>(); |
| | | // if(page1.getRecords().isEmpty()){ |
| | | // return R.ok(inviteUserListVoPage); |
| | | // } |
| | | // ArrayList<InviteUserListVo> inviteUserListVos = new ArrayList<>(); |
| | | // for (TbUser record : page1.getRecords()) { |
| | | // InviteUserListVo inviteUserListVo = new InviteUserListVo(); |
| | | // inviteUserListVo.setUsername(record.getUserName()); |
| | | // inviteUserListVo.setCreateTime(record.getCreateTime()); |
| | | // inviteUserListVo.setInviteNum(record.getInviteNum()); |
| | | // inviteUserListVos.add(inviteUserListVo); |
| | | // } |
| | | // BeanUtils.copyProperties(page1,inviteUserListVoPage); |
| | | // inviteUserListVoPage.setRecords(inviteUserListVos); |
| | | // return R.ok(inviteUserListVoPage); |
| | | // } |
| | | |
| | | |
| | | } |