| | |
| | | package com.ruoyi.demo.controller.queue; |
| | | |
| | | import cn.hutool.core.util.RandomUtil; |
| | | import com.ruoyi.common.core.domain.AjaxResult; |
| | | import com.ruoyi.common.core.domain.R; |
| | | import com.ruoyi.common.utils.redis.QueueUtils; |
| | | import io.swagger.annotations.Api; |
| | | import io.swagger.annotations.ApiOperation; |
| | |
| | | |
| | | @ApiOperation("添加队列数据") |
| | | @GetMapping("/add") |
| | | public AjaxResult<Void> add(@ApiParam("队列名") String queueName) { |
| | | public R<Void> add(@ApiParam("队列名") String queueName) { |
| | | // 用完了一定要销毁 否则会一直存在 |
| | | boolean b = QueueUtils.destroyPriorityQueueObject(queueName); |
| | | log.info("通道: {} , 删除: {}", queueName, b); |
| | |
| | | log.info("通道: {} , 设置比较器成功", queueName); |
| | | } else { |
| | | log.info("通道: {} , 设置比较器失败", queueName); |
| | | return AjaxResult.error("操作失败"); |
| | | return R.fail("操作失败"); |
| | | } |
| | | for (int i = 0; i < 10; i++) { |
| | | int randomNum = RandomUtil.randomInt(10); |
| | | PriorityDemo data = new PriorityDemo().setName("data-" + i).setOrderNum(randomNum); |
| | | PriorityDemo data = new PriorityDemo(); |
| | | data.setName("data-" + i); |
| | | data.setOrderNum(randomNum); |
| | | if (QueueUtils.addPriorityQueueObject(queueName, data)) { |
| | | log.info("通道: {} , 发送数据: {}", queueName, data); |
| | | } else { |
| | | log.info("通道: {} , 发送数据: {}, 发送失败", queueName, data); |
| | | } |
| | | } |
| | | return AjaxResult.success("操作成功"); |
| | | return R.ok("操作成功"); |
| | | } |
| | | |
| | | @ApiOperation("删除队列数据") |
| | | @GetMapping("/remove") |
| | | public AjaxResult<Void> remove(@ApiParam("队列名") String queueName, |
| | | public R<Void> remove(@ApiParam("队列名") String queueName, |
| | | @ApiParam("对象名") String name, |
| | | @ApiParam("排序号") Integer orderNum) { |
| | | PriorityDemo data = new PriorityDemo().setName(name).setOrderNum(orderNum); |
| | | PriorityDemo data = new PriorityDemo(); |
| | | data.setName(name); |
| | | data.setOrderNum(orderNum); |
| | | if (QueueUtils.removePriorityQueueObject(queueName, data)) { |
| | | log.info("通道: {} , 删除数据: {}", queueName, data); |
| | | } else { |
| | | return AjaxResult.error("操作失败"); |
| | | return R.fail("操作失败"); |
| | | } |
| | | return AjaxResult.success("操作成功"); |
| | | return R.ok("操作成功"); |
| | | } |
| | | |
| | | @ApiOperation("获取队列数据") |
| | | @GetMapping("/get") |
| | | public AjaxResult<Void> get(@ApiParam("队列名") String queueName) { |
| | | public R<Void> get(@ApiParam("队列名") String queueName) { |
| | | PriorityDemo data; |
| | | do { |
| | | data = QueueUtils.getPriorityQueueObject(queueName); |
| | | log.info("通道: {} , 获取数据: {}", queueName, data); |
| | | } while (data != null); |
| | | return AjaxResult.success("操作成功"); |
| | | return R.ok("操作成功"); |
| | | } |
| | | |
| | | } |