alsace/wu-lazy-cloud-network
1
0
Fork 0
mirror of https://gitee.com/wujiawei1207537021/wu-lazy-cloud-network.git synced 2025-06-06 21:37:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

【fix】 添加流量控制开关

Browse Source
This commit is contained in:
wujiawei 2024-08-26 16:17:20 +08:00
parent 5ddf1337ad
commit fd364c78e7
13 changed files with 96 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/config/ServerFlowConfiguration.java
View File

@ -2,6 +2,7 @@ package org.framework.lazy.cloud.network.heartbeat.server.config;
import org.framework.lazy.cloud.network.heartbeat.server.netty.flow.ServerHandlerInFlowHandler;
import org.framework.lazy.cloud.network.heartbeat.server.netty.flow.ServerHandlerOutFlowHandler;
import org.framework.lazy.cloud.network.heartbeat.server.properties.ServerNodeProperties;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyVisitorPortFlowApplication;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
@ -25,8 +26,8 @@ public class ServerFlowConfiguration {
*/
@Role(BeanDefinition.ROLE_INFRASTRUCTURE)
@Bean
public ServerHandlerInFlowHandler serverHandlerInFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication) {
return new ServerHandlerInFlowHandler(lazyVisitorPortFlowApplication);
public ServerHandlerInFlowHandler serverHandlerInFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication,ServerNodeProperties serverNodeProperties) {
return new ServerHandlerInFlowHandler(lazyVisitorPortFlowApplication, serverNodeProperties);
}
/**
@ -36,8 +37,8 @@ public class ServerFlowConfiguration {
*/
@Role(BeanDefinition.ROLE_INFRASTRUCTURE)
@Bean
public ServerHandlerOutFlowHandler serverHandlerOutFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication) {
return new ServerHandlerOutFlowHandler(lazyVisitorPortFlowApplication);
public ServerHandlerOutFlowHandler serverHandlerOutFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication, ServerNodeProperties serverNodeProperties) {
return new ServerHandlerOutFlowHandler(lazyVisitorPortFlowApplication,serverNodeProperties);
}

23
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/flow/ServerHandlerInFlowHandler.java
View File

@ -1,20 +1,23 @@
package org.framework.lazy.cloud.network.heartbeat.server.netty.flow;
import io.netty.channel.Channel;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyVisitorPortFlowApplication;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.command.visitor.flow.LazyVisitorPortFlowStoryCommand;
import org.framework.lazy.cloud.network.heartbeat.common.advanced.flow.AbstractHandleChannelFlowAdvanced;
import org.framework.lazy.cloud.network.heartbeat.common.advanced.flow.ChannelFlow;
import org.framework.lazy.cloud.network.heartbeat.common.enums.ChannelFlowEnum;
import org.framework.lazy.cloud.network.heartbeat.server.properties.ServerNodeProperties;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyVisitorPortFlowApplication;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.command.visitor.flow.LazyVisitorPortFlowStoryCommand;
/**
* 进口流量处理
*/
public class ServerHandlerInFlowHandler extends AbstractHandleChannelFlowAdvanced {
private final LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication;
private final ServerNodeProperties serverNodeProperties;
public ServerHandlerInFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication) {
public ServerHandlerInFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication, ServerNodeProperties serverNodeProperties) {
this.lazyVisitorPortFlowApplication = lazyVisitorPortFlowApplication;
this.serverNodeProperties = serverNodeProperties;
}
/**
@ -41,11 +44,13 @@ public class ServerHandlerInFlowHandler extends AbstractHandleChannelFlowAdvance
Integer flow = channelFlow.flow();
// 进口流量处理
LazyVisitorPortFlowStoryCommand visitorPortFlow = new LazyVisitorPortFlowStoryCommand();
visitorPortFlow.setInFlow(flow);
visitorPortFlow.setClientId(clientId);
visitorPortFlow.setVisitorPort(port);
visitorPortFlow.setIsDeleted(false);
lazyVisitorPortFlowApplication.flowIncreaseStory(visitorPortFlow);
if (serverNodeProperties.getEnableFlowControl()) {
LazyVisitorPortFlowStoryCommand visitorPortFlow = new LazyVisitorPortFlowStoryCommand();
visitorPortFlow.setInFlow(flow);
visitorPortFlow.setClientId(clientId);
visitorPortFlow.setVisitorPort(port);
visitorPortFlow.setIsDeleted(false);
lazyVisitorPortFlowApplication.flowIncreaseStory(visitorPortFlow);
}
}
}

23
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/flow/ServerHandlerOutFlowHandler.java
View File

@ -1,20 +1,23 @@
package org.framework.lazy.cloud.network.heartbeat.server.netty.flow;
import io.netty.channel.Channel;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyVisitorPortFlowApplication;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.command.visitor.flow.LazyVisitorPortFlowStoryCommand;
import org.framework.lazy.cloud.network.heartbeat.common.advanced.flow.AbstractHandleChannelFlowAdvanced;
import org.framework.lazy.cloud.network.heartbeat.common.advanced.flow.ChannelFlow;
import org.framework.lazy.cloud.network.heartbeat.common.enums.ChannelFlowEnum;
import org.framework.lazy.cloud.network.heartbeat.server.properties.ServerNodeProperties;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyVisitorPortFlowApplication;
import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.command.visitor.flow.LazyVisitorPortFlowStoryCommand;
/**
* 出口流量处理
*/
public class ServerHandlerOutFlowHandler extends AbstractHandleChannelFlowAdvanced {
private final LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication;
private final ServerNodeProperties serverNodeProperties;
public ServerHandlerOutFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication) {
public ServerHandlerOutFlowHandler(LazyVisitorPortFlowApplication lazyVisitorPortFlowApplication, ServerNodeProperties serverNodeProperties) {
this.lazyVisitorPortFlowApplication = lazyVisitorPortFlowApplication;
this.serverNodeProperties = serverNodeProperties;
}
/**
@ -42,11 +45,13 @@ public class ServerHandlerOutFlowHandler extends AbstractHandleChannelFlowAdvanc
// 出口流量处理
LazyVisitorPortFlowStoryCommand visitorPortFlow = new LazyVisitorPortFlowStoryCommand();
visitorPortFlow.setOutFlow(flow);
visitorPortFlow.setClientId(clientId);
visitorPortFlow.setVisitorPort(port);
visitorPortFlow.setIsDeleted(false);
lazyVisitorPortFlowApplication.flowIncreaseStory(visitorPortFlow);
if(serverNodeProperties.getEnableFlowControl()){
LazyVisitorPortFlowStoryCommand visitorPortFlow = new LazyVisitorPortFlowStoryCommand();
visitorPortFlow.setOutFlow(flow);
visitorPortFlow.setClientId(clientId);
visitorPortFlow.setVisitorPort(port);
visitorPortFlow.setIsDeleted(false);
lazyVisitorPortFlowApplication.flowIncreaseStory(visitorPortFlow);
}
}
}

15
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/handler/VisitorHandler.java
View File

@ -9,7 +9,6 @@ import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.util.internal.StringUtil;
import lombok.extern.slf4j.Slf4j;
import org.framework.lazy.cloud.network.heartbeat.common.*;
import org.framework.lazy.cloud.network.heartbeat.common.*;
import org.framework.lazy.cloud.network.heartbeat.common.adapter.ChannelFlowAdapter;
import org.framework.lazy.cloud.network.heartbeat.common.enums.ChannelFlowEnum;
import org.framework.lazy.cloud.network.heartbeat.common.utils.ChannelAttributeKeyUtils;
@ -55,6 +54,9 @@ public class VisitorHandler extends SimpleChannelInboundHandler<ByteBuf> {
nettyProxyMsg.setVisitorId(visitorId);
// 判断是否有可用的通道 如果没有创建新的通道
// 客户端心跳通道
ChannelContext.ClientChannel clientChannel = ChannelContext.get(clientId);
if (clientChannel != null) {
@ -62,7 +64,7 @@ public class VisitorHandler extends SimpleChannelInboundHandler<ByteBuf> {
Channel channel = clientChannel.getChannel();
channel.writeAndFlush(nettyProxyMsg);
} else {
log.error("无法通过客户端ID获取客户端通道");
log.error("客户端：【{}】已经下线，无法通过客户端ID获取客户端通道", clientId);
}
@ -76,19 +78,20 @@ public class VisitorHandler extends SimpleChannelInboundHandler<ByteBuf> {
@Override
public void channelRead0(ChannelHandlerContext ctx, ByteBuf buf) {
Channel realChannel = ctx.channel();
// 访客通道
Channel visitorChannel = ctx.channel();
String clientId = internalNetworkPenetrationRealClient.getClientId();
String clientTargetIp = internalNetworkPenetrationRealClient.getClientTargetIp();
Integer clientTargetPort = internalNetworkPenetrationRealClient.getClientTargetPort();
Integer visitorPort = internalNetworkPenetrationRealClient.getVisitorPort();
String visitorId = ChannelAttributeKeyUtils.getVisitorId(realChannel);
String visitorId = ChannelAttributeKeyUtils.getVisitorId(visitorChannel);
if (StringUtil.isNullOrEmpty(clientId)) {
return;
}
byte[] bytes = new byte[buf.readableBytes()];
buf.readBytes(bytes);
// 获取客户端通道，而后进行数据下发
log.debug("服务端访客端口成功接收数据:{}", new String(bytes));
log.debug("【服务端】访客端口成功接收数据:{}", new String(bytes));
// 使用访客的通信通道
Channel visitorCommunicationChannel = NettyCommunicationIdContext.getVisitor(visitorId);
@ -111,7 +114,7 @@ public class VisitorHandler extends SimpleChannelInboundHandler<ByteBuf> {
.clientId(clientId)
.flow(bytes.length)
.build();
channelFlowAdapter.handler(realChannel, serverChannelFlow);
channelFlowAdapter.handler(visitorChannel, serverChannelFlow);
log.debug("服务端访客端口成功发送数据了");
}

5
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/properties/ServerNodeProperties.java
View File

@ -36,4 +36,9 @@ public class ServerNodeProperties {
* 集群节点端口
*/
private Integer nodePort;
/**
* 开启流量监控
*/
private Boolean enableFlowControl=false;
}

8
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/command/netty/server/visitor/LazyNettyServerVisitorQueryListCommand.java
View File

@ -54,7 +54,13 @@ public class LazyNettyServerVisitorQueryListCommand {
* 访客端口
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort; /**
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/
@Schema(description = "服务端ID", name = "serverId", example = "")

5
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/command/netty/server/visitor/LazyNettyServerVisitorQueryOneCommand.java
View File

@ -55,6 +55,11 @@ public class LazyNettyServerVisitorQueryOneCommand {
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/

5
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/command/netty/server/visitor/LazyNettyServerVisitorRemoveCommand.java
View File

@ -55,6 +55,11 @@ public class LazyNettyServerVisitorRemoveCommand {
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/

5
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/command/netty/server/visitor/LazyNettyServerVisitorStoryCommand.java
View File

@ -55,6 +55,11 @@ public class LazyNettyServerVisitorStoryCommand {
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize=20;
/**
* 服务端ID
*/

5
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/command/netty/server/visitor/LazyNettyServerVisitorUpdateCommand.java
View File

@ -55,6 +55,11 @@ public class LazyNettyServerVisitorUpdateCommand {
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/

8
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/dto/LazyNettyServerVisitorDTO.java
View File

@ -54,7 +54,13 @@ public class LazyNettyServerVisitorDTO {
* 访客端口
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort; /**
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/
@Schema(description = "服务端ID", name = "serverId", example = "")

9
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/domain/model/lazy/netty/server/visitor/LazyNettyServerVisitor.java
View File

@ -4,6 +4,7 @@ import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import lombok.experimental.Accessors;
import org.wu.framework.lazy.orm.core.persistence.reverse.lazy.ddd.DefaultDDDLazyDomain;
import org.wu.framework.lazy.orm.core.stereotype.LazyTableFieldUnique;
import java.time.LocalDateTime;
@ -54,7 +55,13 @@ public class LazyNettyServerVisitor {
* 访客端口
*/
@Schema(description = "访客端口", name = "visitorPort", example = "")
private Integer visitorPort; /**
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
private Integer poolSize;
/**
* 服务端ID
*/
@Schema(description = "服务端ID", name = "serverId", example = "")

7
wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/infrastructure/entity/LazyNettyServerVisitorDO.java
View File

@ -64,6 +64,13 @@ public class LazyNettyServerVisitorDO {
@Schema(description = "访客端口", name = "visitorPort", example = "")
@LazyTableFieldUnique(name = "visitor_port", comment = "访客端口", columnType = "int")
private Integer visitorPort;
/**
* 访客端口池大小
*/
@Schema(description = "访客端口池大小", name = "poolSize", example = "")
@LazyTableFieldUnique(name = "pool_size", comment = "访客端口池大小", columnType = "int",defaultValue = "'20")
private Integer poolSize;
/**
* 服务端ID
*/