package org.dbsyncer.manager.puller; import org.dbsyncer.common.event.Event; import org.dbsyncer.common.event.RowChangedEvent; import org.dbsyncer.common.scheduled.ScheduledTaskJob; import org.dbsyncer.common.scheduled.ScheduledTaskService; import org.dbsyncer.common.util.CollectionUtils; import org.dbsyncer.common.util.StringUtil; import org.dbsyncer.connector.ConnectorFactory; import org.dbsyncer.connector.config.ConnectorConfig; import org.dbsyncer.connector.model.Field; import org.dbsyncer.connector.model.Table; import org.dbsyncer.connector.constant.ConnectorConstant; import org.dbsyncer.listener.AbstractExtractor; import org.dbsyncer.listener.Extractor; import org.dbsyncer.listener.Listener; import org.dbsyncer.listener.config.ListenerConfig; import org.dbsyncer.listener.enums.ListenerTypeEnum; import org.dbsyncer.listener.quartz.AbstractQuartzExtractor; import org.dbsyncer.manager.Manager; import org.dbsyncer.manager.ManagerException; import org.dbsyncer.manager.config.FieldPicker; import org.dbsyncer.parser.Parser; import org.dbsyncer.parser.logger.LogService; import org.dbsyncer.parser.logger.LogType; import org.dbsyncer.parser.model.Connector; import org.dbsyncer.parser.model.Mapping; import org.dbsyncer.parser.model.Meta; import org.dbsyncer.parser.model.TableGroup; import org.dbsyncer.parser.util.PickerUtil; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; import org.springframework.stereotype.Component; import org.springframework.util.Assert; import javax.annotation.PostConstruct; import java.time.Instant; import java.time.LocalDateTime; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.Executor; import java.util.concurrent.atomic.AtomicInteger; import java.util.stream.Collectors; /** * 增量同步 * * @author AE86 * @version 1.0.0 * @date 2020/04/26 15:28 */ @Component public class IncrementPuller extends AbstractPuller implements ScheduledTaskJob { private final Logger logger = LoggerFactory.getLogger(getClass()); @Autowired private Parser parser; @Autowired private Listener listener; @Autowired private Manager manager; @Autowired private LogService logService; @Autowired private ScheduledTaskService scheduledTaskService; @Autowired private ConnectorFactory connectorFactory; @Qualifier("taskExecutor") @Autowired private Executor taskExecutor; private Map map = new ConcurrentHashMap<>(); @PostConstruct private void init() { scheduledTaskService.start(3000, this); } @Override public void asyncStart(Mapping mapping) { final String mappingId = mapping.getId(); final String metaId = mapping.getMetaId(); try { Connector connector = manager.getConnector(mapping.getSourceConnectorId()); Assert.notNull(connector, "连接器不能为空."); List list = manager.getTableGroupAll(mappingId); Assert.notEmpty(list, "映射关系不能为空."); Meta meta = manager.getMeta(metaId); Assert.notNull(meta, "Meta不能为空."); AbstractExtractor extractor = getExtractor(mapping, connector, list, meta); long now = Instant.now().toEpochMilli(); meta.setBeginTime(now); meta.setEndTime(now); manager.editMeta(meta); map.putIfAbsent(metaId, extractor); // 执行任务 logger.info("启动成功:{}", metaId); map.get(metaId).start(); } catch (Exception e) { close(metaId); logService.log(LogType.TableGroupLog.INCREMENT_FAILED, e.getMessage()); logger.error("运行异常，结束任务{}:{}", metaId, e.getMessage()); } } @Override public void close(String metaId) { Extractor extractor = map.get(metaId); if (null != extractor) { extractor.clearAllListener(); extractor.close(); } map.remove(metaId); publishClosedEvent(metaId); logger.info("关闭成功:{}", metaId); } @Override public void run() { // 定时同步增量信息 map.forEach((k, v) -> v.flushEvent()); } private AbstractExtractor getExtractor(Mapping mapping, Connector connector, List list, Meta meta) throws InstantiationException, IllegalAccessException { ConnectorConfig connectorConfig = connector.getConfig(); ListenerConfig listenerConfig = mapping.getListener(); // timing/log final String listenerType = listenerConfig.getListenerType(); // 默认定时抽取 if (ListenerTypeEnum.isTiming(listenerType)) { AbstractQuartzExtractor extractor = listener.getExtractor(ListenerTypeEnum.TIMING, connectorConfig.getConnectorType(), AbstractQuartzExtractor.class); List> commands = list.stream().map(t -> t.getCommand()).collect(Collectors.toList()); extractor.setCommands(commands); setExtractorConfig(extractor, connectorConfig, listenerConfig, meta.getMap(), new QuartzListener(mapping, list)); return extractor; } // 基于日志抽取 if (ListenerTypeEnum.isLog(listenerType)) { AbstractExtractor extractor = listener.getExtractor(ListenerTypeEnum.LOG, connectorConfig.getConnectorType(), AbstractExtractor.class); Set filterTable = new HashSet<>(); LogListener logListener = new LogListener(mapping, list, extractor); logListener.getTablePicker().forEach((k, fieldPickers) -> filterTable.add(k)); extractor.setFilterTable(filterTable); setExtractorConfig(extractor, connectorConfig, listenerConfig, meta.getMap(), logListener); return extractor; } throw new ManagerException("未知的监听配置."); } private void setExtractorConfig(AbstractExtractor extractor, ConnectorConfig connector, ListenerConfig listener, Map snapshot, AbstractListener event) { extractor.setTaskExecutor(taskExecutor); extractor.setConnectorFactory(connectorFactory); extractor.setScheduledTaskService(scheduledTaskService); extractor.setConnectorConfig(connector); extractor.setListenerConfig(listener); extractor.setSnapshot(snapshot); extractor.addListener(event); extractor.setMetaId(event.metaId); } abstract class AbstractListener implements Event { private static final int FLUSH_DELAYED_SECONDS = 30; protected Mapping mapping; protected String metaId; private LocalDateTime updateTime = LocalDateTime.now(); @Override public void flushEvent(Map map) { // 30s内更新，执行写入 if (updateTime.isAfter(LocalDateTime.now().minusSeconds(FLUSH_DELAYED_SECONDS))) { if (!CollectionUtils.isEmpty(map)) { logger.debug("{}", map); } forceFlushEvent(map); } } @Override public void forceFlushEvent(Map map) { Meta meta = manager.getMeta(metaId); if (null != meta) { meta.setMap(map); manager.editMeta(meta); } } @Override public void refreshFlushEventUpdateTime() { updateTime = LocalDateTime.now(); } @Override public void errorEvent(Exception e) { logService.log(LogType.TableGroupLog.INCREMENT_FAILED, e.getMessage()); } @Override public void interruptException(Exception e) { errorEvent(e); close(metaId); } } /** *

定时模式 *
    *
  1. 根据过滤条件筛选
    2. *
*

同步关系： *

数据源表 >> 目标源表 *
    *
  • A >> B
    • *
  • A >> C
    • *
  • E >> F
    • *
*

PS： *
    *
  1. 依次执行同步关系A >> B 然后 A >> C ...
    2. *
*/ final class QuartzListener extends AbstractListener { private List tablePicker; public QuartzListener(Mapping mapping, List list) { this.mapping = mapping; this.metaId = mapping.getMetaId(); this.tablePicker = new LinkedList<>(); list.forEach(t -> tablePicker.add(new FieldPicker(PickerUtil.mergeTableGroupConfig(mapping, t)))); } @Override public void changedEvent(RowChangedEvent rowChangedEvent) { final FieldPicker picker = tablePicker.get(rowChangedEvent.getTableGroupIndex()); TableGroup tableGroup = picker.getTableGroup(); rowChangedEvent.setSourceTableName(tableGroup.getSourceTable().getName()); // 处理过程有异常向上抛 parser.execute(mapping, tableGroup, rowChangedEvent); // 标记有变更记录 refreshFlushEventUpdateTime(); } } /** *

日志模式 *
    *
  1. 监听表增量数据
    2. *
  2. 根据过滤条件筛选
    3. *
*

同步关系： *

数据源表 >> 目标源表 *
    *
  • A >> B
    • *
  • A >> C
    • *
  • E >> F
    • *
*

PS： *
    *
  1. 为减少开销而选择复用监听器实例, 启动时只需创建一个数据源连接器.
    2. *
  2. 关系A >> B和A >> C会复用A监听的数据, A监听到增量数据，会发送给B和C.
    3. *
  3. 该模式下，会监听表所有字段.
    4. *
*/ final class LogListener extends AbstractListener { private Extractor extractor; private Map> tablePicker; private AtomicInteger eventCounter; private static final int MAX_LOG_CACHE_SIZE = 128; public LogListener(Mapping mapping, List list, Extractor extractor) { this.mapping = mapping; this.metaId = mapping.getMetaId(); this.extractor = extractor; this.tablePicker = new LinkedHashMap<>(); this.eventCounter = new AtomicInteger(); list.forEach(t -> { final Table table = t.getSourceTable(); final String tableName = table.getName(); List pkList = t.getTargetTable().getColumn().stream().filter(field -> field.isPk()).collect(Collectors.toList()); tablePicker.putIfAbsent(tableName, new ArrayList<>()); TableGroup group = PickerUtil.mergeTableGroupConfig(mapping, t); tablePicker.get(tableName).add(new FieldPicker(group, pkList, group.getFilter(), table.getColumn(), group.getFieldMapping())); }); } @Override public void changedEvent(RowChangedEvent rowChangedEvent) { // 处理过程有异常向上抛 List pickers = tablePicker.get(rowChangedEvent.getSourceTableName()); if (!CollectionUtils.isEmpty(pickers)) { pickers.forEach(picker -> { final Map dataMap = picker.getColumns(rowChangedEvent.getDataList()); if (picker.filter(dataMap)) { rowChangedEvent.setDataMap(dataMap); parser.execute(mapping, picker.getTableGroup(), rowChangedEvent); } }); // 标记有变更记录 refreshFlushEventUpdateTime(); eventCounter.set(0); return; } // 防止挤压无效的增量数据，刷新最新的有效记录点 if (eventCounter.incrementAndGet() >= MAX_LOG_CACHE_SIZE) { extractor.forceFlushEvent(); eventCounter.set(0); } } public Map> getTablePicker() { return tablePicker; } } }