Skip to content

Commit 5fb963e

Browse files
authored
Export remoting client for dledger server (#172)
1 parent d207481 commit 5fb963e

File tree

2 files changed

+22
-3
lines changed

2 files changed

+22
-3
lines changed

src/main/java/io/openmessaging/storage/dledger/DLedgerRpcNettyService.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -493,4 +493,8 @@ public void setdLedgerServer(DLedgerServer dLedgerServer) {
493493
public NettyRemotingServer getRemotingServer() {
494494
return remotingServer;
495495
}
496+
497+
public NettyRemotingClient getRemotingClient() {
498+
return remotingClient;
499+
}
496500
}

src/main/java/io/openmessaging/storage/dledger/DLedgerServer.java

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959

6060
import org.apache.rocketmq.remoting.ChannelEventListener;
6161
import org.apache.rocketmq.remoting.netty.NettyClientConfig;
62+
import org.apache.rocketmq.remoting.netty.NettyRemotingClient;
6263
import org.apache.rocketmq.remoting.netty.NettyRemotingServer;
6364
import org.apache.rocketmq.remoting.netty.NettyServerConfig;
6465
import org.slf4j.Logger;
@@ -83,11 +84,17 @@ public DLedgerServer(DLedgerConfig dLedgerConfig) {
8384
this(dLedgerConfig, null, null, null);
8485
}
8586

86-
public DLedgerServer(DLedgerConfig dLedgerConfig, NettyServerConfig nettyServerConfig, NettyClientConfig nettyClientConfig) {
87+
public DLedgerServer(DLedgerConfig dLedgerConfig, NettyServerConfig nettyServerConfig) {
88+
this(dLedgerConfig, nettyServerConfig, null, null);
89+
}
90+
91+
public DLedgerServer(DLedgerConfig dLedgerConfig, NettyServerConfig nettyServerConfig,
92+
NettyClientConfig nettyClientConfig) {
8793
this(dLedgerConfig, nettyServerConfig, nettyClientConfig, null);
8894
}
8995

90-
public DLedgerServer(DLedgerConfig dLedgerConfig, NettyServerConfig nettyServerConfig, NettyClientConfig nettyClientConfig, ChannelEventListener channelEventListener) {
96+
public DLedgerServer(DLedgerConfig dLedgerConfig, NettyServerConfig nettyServerConfig,
97+
NettyClientConfig nettyClientConfig, ChannelEventListener channelEventListener) {
9198
this.dLedgerConfig = dLedgerConfig;
9299
this.memberState = new MemberState(dLedgerConfig);
93100
this.dLedgerStore = createDLedgerStore(dLedgerConfig.getStoreType(), this.dLedgerConfig, this.memberState);
@@ -452,8 +459,16 @@ public DLedgerConfig getdLedgerConfig() {
452459

453460
public NettyRemotingServer getRemotingServer() {
454461
if (this.dLedgerRpcService instanceof DLedgerRpcNettyService) {
455-
return ((DLedgerRpcNettyService)this.dLedgerRpcService).getRemotingServer();
462+
return ((DLedgerRpcNettyService) this.dLedgerRpcService).getRemotingServer();
463+
}
464+
return null;
465+
}
466+
467+
public NettyRemotingClient getRemotingClient() {
468+
if (this.dLedgerRpcService instanceof DLedgerRpcNettyService) {
469+
return ((DLedgerRpcNettyService) this.dLedgerRpcService).getRemotingClient();
456470
}
457471
return null;
458472
}
473+
459474
}

0 commit comments

Comments
 (0)