gtsoft/gtsoft-snail-job-server
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects 1 Releases Wiki Activity

feat:(1.3.0-beta1.1): 新增grpc 重连事件

Browse Source
This commit is contained in:
opensnail 2025-01-04 00:19:42 +08:00
parent 8fde82c067
commit f35eddc806
2 changed files with 37 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pom.xml
View File

@ -21,7 +21,7 @@
<java.version>17</java.version> <java.version>17</java.version>
<maven.compiler.source>17</maven.compiler.source> <maven.compiler.source>17</maven.compiler.source>
<maven.compiler.target>17</maven.compiler.target> <maven.compiler.target>17</maven.compiler.target>
<revision>1.3.0-beta1</revision> <revision>1.3.0-beta1.1</revision>
<netty-all.version>4.1.114.Final</netty-all.version> <netty-all.version>4.1.114.Final</netty-all.version>
<hutool-all.version>5.8.32</hutool-all.version> <hutool-all.version>5.8.32</hutool-all.version>
<mybatis-plus.version>3.5.9</mybatis-plus.version> <mybatis-plus.version>3.5.9</mybatis-plus.version>

54
snail-job-client/snail-job-client-common/src/main/java/com/aizuda/snailjob/client/common/rpc/client/SnailJobGrpcClient.java
View File

@ -4,20 +4,20 @@ import com.aizuda.snailjob.client.common.Lifecycle;
import com.aizuda.snailjob.client.common.config.SnailJobProperties; import com.aizuda.snailjob.client.common.config.SnailJobProperties;
import com.aizuda.snailjob.client.common.config.SnailJobProperties.RpcClientProperties; import com.aizuda.snailjob.client.common.config.SnailJobProperties.RpcClientProperties;
import com.aizuda.snailjob.client.common.config.SnailJobProperties.ThreadPoolConfig; import com.aizuda.snailjob.client.common.config.SnailJobProperties.ThreadPoolConfig;
import com.aizuda.snailjob.client.common.event.SnailChannelReconnectEvent;
import com.aizuda.snailjob.common.core.context.SnailSpringContext;
import com.aizuda.snailjob.common.core.enums.RpcTypeEnum; import com.aizuda.snailjob.common.core.enums.RpcTypeEnum;
import com.aizuda.snailjob.common.log.SnailJobLog; import com.aizuda.snailjob.common.log.SnailJobLog;
import com.google.common.util.concurrent.ThreadFactoryBuilder; import com.google.common.util.concurrent.ThreadFactoryBuilder;
import io.grpc.DecompressorRegistry; import io.grpc.*;
import io.grpc.ManagedChannel; import io.grpc.netty.shaded.io.grpc.netty.NettyChannelBuilder;
import io.grpc.ManagedChannelBuilder;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.core.Ordered; import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order; import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.util.concurrent.LinkedBlockingQueue; import java.util.concurrent.*;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
/** /**
* @author: opensnail * @author: opensnail
@ -29,27 +29,31 @@ import java.util.concurrent.TimeUnit;
public class SnailJobGrpcClient implements Lifecycle { public class SnailJobGrpcClient implements Lifecycle {
private ManagedChannel channel; private ManagedChannel channel;
private final SnailJobProperties snailJobProperties; private final SnailJobProperties snailJobProperties;
private static final ScheduledExecutorService SCHEDULE_EXECUTOR = Executors.newSingleThreadScheduledExecutor(
r -> new Thread(r, "sj-client-check"));
@Override @Override
public void start() { public void start() {
if (RpcTypeEnum.GRPC != snailJobProperties.getRpcType()) { if (RpcTypeEnum.GRPC != snailJobProperties.getRpcType()) {
return; return;
} }
RpcClientProperties clientRpc = snailJobProperties.getClientRpc(); channel = connection();
// 创建 gRPC 频道
String serverHost = GrpcChannel.getServerHost();
channel = ManagedChannelBuilder.forAddress(serverHost, GrpcChannel.getServerPort())
.executor(createGrpcExecutor(serverHost))
.decompressorRegistry(DecompressorRegistry.getDefaultInstance())
.maxInboundMessageSize(clientRpc.getMaxInboundMessageSize())
.keepAliveTime(clientRpc.getKeepAliveTime().toMillis(), TimeUnit.MILLISECONDS)
.keepAliveTimeout(clientRpc.getKeepAliveTimeout().toMillis(), TimeUnit.MILLISECONDS)
.usePlaintext()
.build();
GrpcChannel.setChannel(channel); GrpcChannel.setChannel(channel);
SnailJobLog.LOCAL.info("grpc client started connect to server"); SnailJobLog.LOCAL.info("grpc client started connect to server");
// 连接检测
SCHEDULE_EXECUTOR.scheduleAtFixedRate(() -> {
ConnectivityState state = channel.getState(true);
if (state == ConnectivityState.TRANSIENT_FAILURE) {
try {
// 抛出重连事件
SnailSpringContext.getContext().publishEvent(new SnailChannelReconnectEvent());
} catch (Exception e) {
SnailJobLog.LOCAL.error("reconnect error ", e);
}
}
}, 0, 10, TimeUnit.SECONDS);
} }
@Override @Override
@ -59,6 +63,20 @@ public class SnailJobGrpcClient implements Lifecycle {
} }
} }
public ManagedChannel connection() {
RpcClientProperties clientRpc = snailJobProperties.getClientRpc();
// 创建 gRPC 频道
String serverHost = GrpcChannel.getServerHost();
return NettyChannelBuilder.forAddress(serverHost, GrpcChannel.getServerPort())
.executor(createGrpcExecutor(serverHost))
.decompressorRegistry(DecompressorRegistry.getDefaultInstance())
.maxInboundMessageSize(clientRpc.getMaxInboundMessageSize())
.keepAliveTime(clientRpc.getKeepAliveTime().toMillis(), TimeUnit.MILLISECONDS)
.keepAliveTimeout(clientRpc.getKeepAliveTimeout().toMillis(), TimeUnit.MILLISECONDS)
.usePlaintext().enableRetry().maxRetryAttempts(16)
.build();
}
private ThreadPoolExecutor createGrpcExecutor(String serverIp) { private ThreadPoolExecutor createGrpcExecutor(String serverIp) {
RpcClientProperties clientRpc = snailJobProperties.getClientRpc(); RpcClientProperties clientRpc = snailJobProperties.getClientRpc();
ThreadPoolConfig threadPool = clientRpc.getClientTp(); ThreadPoolConfig threadPool = clientRpc.getClientTp();