11package com .codingapi .txlcn .tm .txmsg ;
22
33import com .codingapi .txlcn .common .runner .TxLcnInitializer ;
4+ import com .codingapi .txlcn .common .util .ApplicationInformation ;
5+ import com .codingapi .txlcn .common .util .Transactions ;
46import com .codingapi .txlcn .common .util .id .IdGenInit ;
57import com .codingapi .txlcn .logger .TxLogger ;
68import com .codingapi .txlcn .tm .config .TxManagerConfig ;
911import com .codingapi .txlcn .txmsg .listener .HeartbeatListener ;
1012import com .codingapi .txlcn .txmsg .listener .RpcConnectionListener ;
1113import org .springframework .beans .factory .annotation .Autowired ;
14+ import org .springframework .boot .autoconfigure .web .ServerProperties ;
15+ import org .springframework .core .env .ConfigurableEnvironment ;
1216import org .springframework .stereotype .Component ;
1317
1418/**
@@ -26,11 +30,18 @@ public class EnsureIdGenEngine implements RpcConnectionListener, HeartbeatListen
2630
2731 private final TxLogger txLogger ;
2832
33+ private final ConfigurableEnvironment environment ;
34+
35+ private final ServerProperties serverProperties ;
36+
2937 @ Autowired
30- public EnsureIdGenEngine (ManagerService managerService , TxManagerConfig managerConfig , TxLogger txLogger ) {
38+ public EnsureIdGenEngine (ManagerService managerService , TxManagerConfig managerConfig , TxLogger txLogger ,
39+ ConfigurableEnvironment environment , ServerProperties serverProperties ) {
3140 this .managerService = managerService ;
3241 this .managerConfig = managerConfig ;
3342 this .txLogger = txLogger ;
43+ this .environment = environment ;
44+ this .serverProperties = serverProperties ;
3445 }
3546
3647 @ Override
@@ -45,6 +56,8 @@ public void disconnect(String remoteKey, String appName) {
4556 @ Override
4657 public void init () throws Exception {
4758 IdGenInit .applySnowFlakeIdGen (managerConfig .getMachineIdLen (), managerService .machineIdSync ());
59+
60+ Transactions .setApplicationIdWhenRunning (ApplicationInformation .modId (environment , serverProperties ));
4861 }
4962
5063 @ Override
0 commit comments