|
|
@@ -0,0 +1,201 @@
|
|
|
+package org.dromara.server.mq.service.impl;
|
|
|
+
|
|
|
+import org.dromara.common.core.utils.MapstructUtils;
|
|
|
+import org.dromara.common.core.utils.StringUtils;
|
|
|
+import org.dromara.common.mybatis.core.page.TableDataInfo;
|
|
|
+import org.dromara.common.mybatis.core.page.PageQuery;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import lombok.RequiredArgsConstructor;
|
|
|
+import org.springframework.stereotype.Service;
|
|
|
+import org.dromara.server.mq.domain.bo.XfOffsetBo;
|
|
|
+import org.dromara.server.mq.domain.vo.XfOffsetVo;
|
|
|
+import org.dromara.server.mq.domain.XfOffset;
|
|
|
+import org.dromara.server.mq.mapper.XfOffsetMapper;
|
|
|
+import org.dromara.server.mq.service.IXfOffsetService;
|
|
|
+
|
|
|
+import java.util.Date;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.Collection;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 记录kafka消费的offsetService业务层处理
|
|
|
+ *
|
|
|
+ * @author LionLi
|
|
|
+ * @date 2025-03-12
|
|
|
+ */
|
|
|
+@RequiredArgsConstructor
|
|
|
+@Service
|
|
|
+public class XfOffsetServiceImpl implements IXfOffsetService {
|
|
|
+
|
|
|
+ private final XfOffsetMapper baseMapper;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询记录kafka消费的offset
|
|
|
+ *
|
|
|
+ * @param offsetId 主键
|
|
|
+ * @return 记录kafka消费的offset
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public XfOffsetVo queryById(String offsetId){
|
|
|
+ return baseMapper.selectVoById(offsetId);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 分页查询记录kafka消费的offset列表
|
|
|
+ *
|
|
|
+ * @param bo 查询条件
|
|
|
+ * @param pageQuery 分页参数
|
|
|
+ * @return 记录kafka消费的offset分页列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public TableDataInfo<XfOffsetVo> queryPageList(XfOffsetBo bo, PageQuery pageQuery) {
|
|
|
+ LambdaQueryWrapper<XfOffset> lqw = buildQueryWrapper(bo);
|
|
|
+ Page<XfOffsetVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
|
|
|
+ return TableDataInfo.build(result);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询符合条件的记录kafka消费的offset列表
|
|
|
+ *
|
|
|
+ * @param bo 查询条件
|
|
|
+ * @return 记录kafka消费的offset列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<XfOffsetVo> queryList(XfOffsetBo bo) {
|
|
|
+ LambdaQueryWrapper<XfOffset> lqw = buildQueryWrapper(bo);
|
|
|
+ return baseMapper.selectVoList(lqw);
|
|
|
+ }
|
|
|
+
|
|
|
+ private LambdaQueryWrapper<XfOffset> buildQueryWrapper(XfOffsetBo bo) {
|
|
|
+ LambdaQueryWrapper<XfOffset> lqw = Wrappers.lambdaQuery();
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getOffsetId()), XfOffset::getOffsetId, bo.getOffsetId());
|
|
|
+ lqw.eq(bo.getOffSet() != null, XfOffset::getOffSet, bo.getOffSet());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getTopic()), XfOffset::getTopic, bo.getTopic());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getGroupId()), XfOffset::getGroupId, bo.getGroupId());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getRecord()), XfOffset::getRecord, bo.getRecord());
|
|
|
+ return lqw;
|
|
|
+ }
|
|
|
+
|
|
|
+ private QueryWrapper<XfOffset> buildQueryWrapper(XfOffsetBo bo,String tableAlias) {
|
|
|
+ QueryWrapper<XfOffset> lqw = new QueryWrapper<>();
|
|
|
+ String columnPrefix = "";
|
|
|
+ if(StringUtils.isNotBlank(tableAlias)){
|
|
|
+ columnPrefix = tableAlias + ".";
|
|
|
+ }
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getOffsetId()), columnPrefix+"offset_id", bo.getOffsetId());
|
|
|
+ lqw.eq(bo.getOffSet() != null, columnPrefix+"offset", bo.getOffSet());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getTopic()), columnPrefix+"topic", bo.getTopic());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getGroupId()), columnPrefix+"group_id", bo.getGroupId());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getRecord()), columnPrefix+"record", bo.getRecord());
|
|
|
+ return lqw;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 新增记录kafka消费的offset
|
|
|
+ *
|
|
|
+ * @param bo 记录kafka消费的offset
|
|
|
+ * @return 是否新增成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean insertByBo(XfOffsetBo bo) {
|
|
|
+ XfOffset add = MapstructUtils.convert(bo, XfOffset.class);
|
|
|
+ validEntityBeforeSave(add);
|
|
|
+ boolean flag = baseMapper.insert(add) > 0;
|
|
|
+ if (flag) {
|
|
|
+ bo.setOffsetId(add.getOffsetId());
|
|
|
+ }
|
|
|
+ return flag;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 新增记录kafka消费的offset
|
|
|
+ * @param offsetId
|
|
|
+ * @param offSet
|
|
|
+ * @param topic
|
|
|
+ * @param groupId
|
|
|
+ * @param record
|
|
|
+ * @param createTime
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean insert(String offsetId, Long offSet, String topic, String groupId, String record, Date createTime) {
|
|
|
+ XfOffset add = new XfOffset();
|
|
|
+ add.setOffsetId(offsetId);
|
|
|
+ add.setOffSet(offSet);
|
|
|
+ add.setTopic(topic);
|
|
|
+ add.setGroupId(groupId);
|
|
|
+ add.setRecord(record);
|
|
|
+ add.setCreateTime(createTime);
|
|
|
+ return baseMapper.insert(add) > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 修改记录kafka消费的offset
|
|
|
+ *
|
|
|
+ * @param bo 记录kafka消费的offset
|
|
|
+ * @return 是否修改成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean updateByBo(XfOffsetBo bo) {
|
|
|
+ XfOffset update = MapstructUtils.convert(bo, XfOffset.class);
|
|
|
+ validEntityBeforeSave(update);
|
|
|
+ return baseMapper.updateById(update) > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 保存前的数据校验
|
|
|
+ */
|
|
|
+ private void validEntityBeforeSave(XfOffset entity){
|
|
|
+ // 做一些数据校验,如唯一约束
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 校验并批量删除记录kafka消费的offset信息
|
|
|
+ *
|
|
|
+ * @param ids 待删除的主键集合
|
|
|
+ * @param isValid 是否进行有效性校验
|
|
|
+ * @return 是否删除成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean deleteWithValidByIds(Collection<String> ids, Boolean isValid) {
|
|
|
+ if(isValid){
|
|
|
+ // 做一些业务上的校验,判断是否需要校验
|
|
|
+ }
|
|
|
+ return baseMapper.deleteByIds(ids) > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据topic和groupId查询offset
|
|
|
+ *
|
|
|
+ * @param topic
|
|
|
+ * @param groupId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Long selectOffsetByTopicAndGroupId(String topic, String groupId) {
|
|
|
+ return baseMapper.selectOffsetByTopicAndGroupId(topic, groupId);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 判断是否可以消费
|
|
|
+ *
|
|
|
+ * @param topic
|
|
|
+ * @param groupId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean judgeCanConsume(String topic, String groupId, Long offset) {
|
|
|
+ Long consumeOffset = baseMapper.selectOffsetByTopicAndGroupId(topic, groupId);
|
|
|
+ if (consumeOffset != null) {
|
|
|
+ return offset > consumeOffset;
|
|
|
+ }
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+}
|