feat(sync): 增加pulsar消息最后接收时间

用于判断pulsar有多久没有传入消息了

service-api/src/main/java/com/lanyuanxiaoyao/service/api/controller/ApiController.java

@@ -163,4 +163,14 @@ public class ApiController {
         logger.info(makeMarker(flinkJobId, alias), "Compaction latest operation time: {} {} {}", flinkJobId, alias, latestOpTs);
         syncStateService.saveCompactionLatestOperationTime(flinkJobId, alias, latestOpTs);
     }
+
+    @GetMapping("sync_receive_time")
+    public void syncReceiveTime(
+            @RequestParam("flink_job_id") Long flinkJobId,
+            @RequestParam("alias") String alias,
+            @RequestParam("receive_time") Long receiveTime
+    ) {
+        logger.info(makeMarker(flinkJobId, alias), "Source receive time: {} {} {}", flinkJobId, alias, receiveTime);
+        syncStateService.saveSyncReceiveTime(flinkJobId, alias, receiveTime);
+    }
 }

service-api/src/main/java/com/lanyuanxiaoyao/service/api/service/SyncStateService.java

@@ -250,7 +250,9 @@ public class SyncStateService {
     }
 
     public void saveCompactionLatestOperationTime(Long flinkJobId, String alias, Long latestOperationTime) {
-        Date operationDate = latestOperationTime == 0 ? null : Date.from(Instant.ofEpochMilli(latestOperationTime));
+        Date operationDate = (ObjectUtil.isNull(latestOperationTime) || latestOperationTime == 0)
+                ? null
+                : Date.from(Instant.ofEpochMilli(latestOperationTime));
         jdbcTemplate.update(
                 SqlBuilder
                         .insertInto(
@@ -275,4 +277,24 @@ public class SyncStateService {
                 operationDate
         );
     }
+
+    public void saveSyncReceiveTime(Long flinkJobId, String alias, Long sourceReceiveTime) {
+        Date sourceReceiveDate = (ObjectUtil.isNull(sourceReceiveTime) || sourceReceiveTime == 0)
+                ? null
+                : Date.from(Instant.ofEpochMilli(sourceReceiveTime));
+        jdbcTemplate.update(
+                SqlBuilder
+                        .insertInto(
+                                TbAppHudiSyncState._origin_,
+                                TbAppHudiSyncState.ID_O,
+                                TbAppHudiSyncState.SOURCE_RECEIVE_TIME_O
+                        )
+                        .values()
+                        .addValue(null, null)
+                        .onDuplicateKeyUpdateColumn(TbAppHudiSyncState.SOURCE_RECEIVE_TIME_O)
+                        .precompileSql(),
+                syncStateId(flinkJobId, alias),
+                sourceReceiveDate
+        );
+    }
 }

service-common/src/main/java/com/lanyuanxiaoyao/service/common/SQLConstants.java

@@ -618,6 +618,22 @@ public interface SQLConstants {
              * 字段 type 别名值 tahcm.type 类型
              */
             String TYPE_A = _alias_.getAlias() + "." + TYPE_O;
+            /**
+             * 字段 source_schema 原始值 source_schema 库名
+             */
+            String SOURCE_SCHEMA_O = "source_schema";
+            /**
+             * 字段 source_schema 别名值 tahcm.source_schema 库名
+             */
+            String SOURCE_SCHEMA_A = _alias_.getAlias() + "." + SOURCE_SCHEMA_O;
+            /**
+             * 字段 source_table 原始值 source_table 表名
+             */
+            String SOURCE_TABLE_O = "source_table";
+            /**
+             * 字段 source_table 别名值 tahcm.source_table 表名
+             */
+            String SOURCE_TABLE_A = _alias_.getAlias() + "." + SOURCE_TABLE_O;
             /**
              * 字段 compaction_plan_instant 原始值 compaction_plan_instant 压缩计划时间点
              */
@@ -910,6 +926,14 @@ public interface SQLConstants {
              * 字段 source_start_time 别名值 tahss.source_start_time 同步启动时间
              */
             String SOURCE_START_TIME_A = _alias_.getAlias() + "." + SOURCE_START_TIME_O;
+            /**
+             * 字段 source_receive_time 原始值 source_receive_time pulsar最后接收消息的时间
+             */
+            String SOURCE_RECEIVE_TIME_O = "source_receive_time";
+            /**
+             * 字段 source_receive_time 别名值 tahss.source_receive_time pulsar最后接收消息的时间
+             */
+            String SOURCE_RECEIVE_TIME_A = _alias_.getAlias() + "." + SOURCE_RECEIVE_TIME_O;
             /**
              * 字段 source_checkpoint_time 原始值 source_checkpoint_time 同步检查时间
              */

utils/sync/src/main/java/com/lanyuanxiaoyao/service/sync/functions/PulsarMessage2RecordFunction.java

@@ -30,7 +30,10 @@ import org.slf4j.LoggerFactory;
  */
 public class PulsarMessage2RecordFunction extends RichMapFunction<String, Record> implements CheckpointedFunction {
     private static final Logger logger = LoggerFactory.getLogger(PulsarMessage2RecordFunction.class);
-    private static final AtomicReference<String> lastOperationTime = new AtomicReference<>("");
+    /**
+     * 最后操作时间
+     */
+    private static final AtomicReference<String> latestOperationTime = new AtomicReference<>("");
     private final static DateTimeFormatter FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
     private static final Pattern OPTS_PATTERN = Pattern.compile("^\\d{4}-\\d{2}-\\d{2} \\d{2}:\\d{2}:\\d{2}$");
     private final GlobalConfiguration globalConfiguration;
@@ -50,7 +53,7 @@ public class PulsarMessage2RecordFunction extends RichMapFunction<String, Record
         try {
             record = mapper.readValue(message, Record.class);
             if (RecordHelper.isNotVersionUpdateRecord(record)) {
-                lastOperationTime.set(record.getStatement().getOpTs());
+                latestOperationTime.set(record.getStatement().getOpTs());
             }
         } catch (Exception exception) {
             logger.error("Message json parse failure", exception);
@@ -60,7 +63,7 @@ public class PulsarMessage2RecordFunction extends RichMapFunction<String, Record
 
     @Override
     public void snapshotState(FunctionSnapshotContext context) {
-        String opTs = lastOperationTime.get();
+        String opTs = latestOperationTime.get();
         Long timestamp = null;
         try {
             if (StrUtil.isNotBlank(opTs) && OPTS_PATTERN.matcher(opTs).matches()) {

utils/sync/src/main/java/com/lanyuanxiaoyao/service/sync/functions/PulsarMessageSourceReader.java

@@ -18,6 +18,7 @@ import com.lanyuanxiaoyao.service.sync.utils.StatusUtils;
 import dev.failsafe.Failsafe;
 import dev.failsafe.RetryPolicy;
 import java.time.Duration;
+import java.time.Instant;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.atomic.AtomicLong;
@@ -28,13 +29,20 @@ import org.apache.flink.runtime.state.FunctionInitializationContext;
 import org.apache.flink.runtime.state.FunctionSnapshotContext;
 import org.apache.flink.streaming.api.checkpoint.CheckpointedFunction;
 import org.apache.flink.streaming.api.functions.source.RichParallelSourceFunction;
-import org.apache.pulsar.client.api.*;
+import org.apache.pulsar.client.api.Message;
+import org.apache.pulsar.client.api.MessageId;
+import org.apache.pulsar.client.api.PulsarClient;
+import org.apache.pulsar.client.api.PulsarClientException;
+import org.apache.pulsar.client.api.Reader;
 import org.apache.pulsar.client.impl.schema.StringSchema;
 import org.apache.pulsar.client.internal.DefaultImplementation;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import static com.lanyuanxiaoyao.service.common.utils.LogHelper.LogPoint.*;
+import static com.lanyuanxiaoyao.service.common.utils.LogHelper.LogPoint.CHECKPOINT_COMPLETE;
+import static com.lanyuanxiaoyao.service.common.utils.LogHelper.LogPoint.CHECKPOINT_INITIAL;
+import static com.lanyuanxiaoyao.service.common.utils.LogHelper.LogPoint.CHECKPOINT_INITIAL_MESSAGE_ID;
+import static com.lanyuanxiaoyao.service.common.utils.LogHelper.LogPoint.CHECKPOINT_START;
 
 /**
  * Pulsar Reader Source
@@ -54,7 +62,8 @@ public class PulsarMessageSourceReader extends RichParallelSourceFunction<String
     private final FlinkJob flinkJob;
     private final TableMeta tableMeta;
     private final AtomicReference<MessageId> lastMessageId = new AtomicReference<>();
-    private final AtomicLong lastPublishTime = new AtomicLong(0);
+    private final AtomicLong latestPublishTime = new AtomicLong(0);
+    private final AtomicLong latestReceiveTime = new AtomicLong(0);
     private final RateMetric messageReceiveMetric;
     private final MessageSizeSizeMetric messageSizeReceiveMetric;
     private final Map<Long, MessageId> messageIdMap = new ConcurrentHashMap<>();
@@ -143,7 +152,8 @@ public class PulsarMessageSourceReader extends RichParallelSourceFunction<String
                     }
 
                     if (RecordHelper.isNotVersionUpdateRecord(value)) {
-                        lastPublishTime.set(message.getPublishTime());
+                        latestPublishTime.set(message.getPublishTime());
+                        latestReceiveTime.set(Instant.now().toEpochMilli());
                     }
                     lastMessageId.set(message.getMessageId());
 
@@ -224,7 +234,8 @@ public class PulsarMessageSourceReader extends RichParallelSourceFunction<String
     public void notifyCheckpointComplete(long checkpointId) {
         MessageId messageId = messageIdMap.getOrDefault(checkpointId, MessageId.earliest);
         LogHelper.info(logger, CHECKPOINT_COMPLETE, "Checkpoint complete message id: {}, checkpoint id: {}", messageId, checkpointId);
-        StatusUtils.syncCheckpoint(globalConfiguration, flinkJob, tableMeta, messageId.toString(), lastPublishTime.get());
+        StatusUtils.syncCheckpoint(globalConfiguration, flinkJob, tableMeta, messageId.toString(), latestPublishTime.get());
+        StatusUtils.syncReceive(globalConfiguration, flinkJob, tableMeta, latestReceiveTime.get());
         messageIdMap.remove(checkpointId);
     }
 

utils/sync/src/main/java/com/lanyuanxiaoyao/service/sync/utils/StatusUtils.java

@@ -268,4 +268,24 @@ public class StatusUtils {
                                 .execute()
                 );
     }
+
+    public static void syncReceive(GlobalConfiguration configuration, FlinkJob flinkJob, TableMeta tableMeta, Long receiveTime) {
+        // logger.info("Enter method: syncReceive[configuration, flinkJob, tableMeta, receiveTime]. " + "configuration:" + configuration + "," + "flinkJob:" + flinkJob + "," + "tableMeta:" + tableMeta + "," + "receiveTime:" + receiveTime);
+        Failsafe.with(RetryPolicyProvider.HTTP_RETRY)
+                .run(() ->
+                        HttpUtil.createGet(
+                                        StrUtil.format(
+                                                "{}/api/sync_receive_time?flink_job_id={}&alias={}&receive_time={}",
+                                                LoadBalance.getCustomPublishUrl(configuration),
+                                                flinkJob.getId(),
+                                                tableMeta.getAlias(),
+                                                receiveTime
+                                        )
+                                )
+                                .header(Constants.API_HEADER_NAME, Constants.API_VERSION)
+                                .basicAuth(Constants.SPRING_SECURITY_USERNAME, Constants.SPRING_SECURITY_PASSWORD_PLAIN)
+                                .timeout(HTTP_TIMEOUT)
+                                .execute()
+                );
+    }
 }