diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/impl/LazyInternalNetworkClientPermeateServerMappingApplicationImpl.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/impl/LazyInternalNetworkClientPermeateServerMappingApplicationImpl.java index 2b0e136..f0140cc 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/impl/LazyInternalNetworkClientPermeateServerMappingApplicationImpl.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/application/impl/LazyInternalNetworkClientPermeateServerMappingApplicationImpl.java @@ -1,5 +1,6 @@ package org.framework.lazy.cloud.network.heartbeat.server.standalone.application.impl; +import org.framework.lazy.cloud.network.heartbeat.server.properties.ServerNodeProperties; import org.wu.framework.database.lazy.web.plus.stereotype.LazyApplication; import org.framework.lazy.cloud.network.heartbeat.server.standalone.application.LazyInternalNetworkClientPermeateServerMappingApplication; import org.springframework.web.bind.annotation.*; @@ -31,6 +32,10 @@ public class LazyInternalNetworkClientPermeateServerMappingApplicationImpl imple @Resource LazyInternalNetworkClientPermeateServerMappingRepository lazyInternalNetworkClientPermeateServerMappingRepository; + + @Resource + ServerNodeProperties serverNodeProperties; + /** * describe 新增客户端渗透服务端映射 * @@ -44,6 +49,8 @@ public class LazyInternalNetworkClientPermeateServerMappingApplicationImpl imple @Override public Result story(LazyInternalNetworkClientPermeateServerMappingStoryCommand lazyInternalNetworkClientPermeateServerMappingStoryCommand) { LazyInternalNetworkClientPermeateServerMapping lazyInternalNetworkClientPermeateServerMapping = LazyInternalNetworkClientPermeateServerMappingDTOAssembler.INSTANCE.toLazyInternalNetworkClientPermeateServerMapping(lazyInternalNetworkClientPermeateServerMappingStoryCommand); + String serverId = serverNodeProperties.getNodeId(); + lazyInternalNetworkClientPermeateServerMapping.setServerId(serverId); return lazyInternalNetworkClientPermeateServerMappingRepository.story(lazyInternalNetworkClientPermeateServerMapping); } /** @@ -59,6 +66,10 @@ public class LazyInternalNetworkClientPermeateServerMappingApplicationImpl imple @Override public Result> batchStory(List lazyInternalNetworkClientPermeateServerMappingStoryCommandList) { List lazyInternalNetworkClientPermeateServerMappingList = lazyInternalNetworkClientPermeateServerMappingStoryCommandList.stream().map( LazyInternalNetworkClientPermeateServerMappingDTOAssembler.INSTANCE::toLazyInternalNetworkClientPermeateServerMapping).collect(Collectors.toList()); + for (LazyInternalNetworkClientPermeateServerMapping lazyInternalNetworkClientPermeateServerMapping : lazyInternalNetworkClientPermeateServerMappingList) { + String serverId = serverNodeProperties.getNodeId(); + lazyInternalNetworkClientPermeateServerMapping.setServerId(serverId); + } return lazyInternalNetworkClientPermeateServerMappingRepository.batchStory(lazyInternalNetworkClientPermeateServerMappingList); } /** @@ -74,6 +85,8 @@ public class LazyInternalNetworkClientPermeateServerMappingApplicationImpl imple @Override public Result updateOne(LazyInternalNetworkClientPermeateServerMappingUpdateCommand lazyInternalNetworkClientPermeateServerMappingUpdateCommand) { LazyInternalNetworkClientPermeateServerMapping lazyInternalNetworkClientPermeateServerMapping = LazyInternalNetworkClientPermeateServerMappingDTOAssembler.INSTANCE.toLazyInternalNetworkClientPermeateServerMapping(lazyInternalNetworkClientPermeateServerMappingUpdateCommand); + String serverId = serverNodeProperties.getNodeId(); + lazyInternalNetworkClientPermeateServerMapping.setServerId(serverId); return lazyInternalNetworkClientPermeateServerMappingRepository.story(lazyInternalNetworkClientPermeateServerMapping); } diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/infrastructure/persistence/LazyInternalNetworkClientPermeateServerMappingRepositoryImpl.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/infrastructure/persistence/LazyInternalNetworkClientPermeateServerMappingRepositoryImpl.java index ccf7898..a4c5534 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/infrastructure/persistence/LazyInternalNetworkClientPermeateServerMappingRepositoryImpl.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/standalone/infrastructure/persistence/LazyInternalNetworkClientPermeateServerMappingRepositoryImpl.java @@ -1,19 +1,19 @@ package org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.persistence; -import org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.entity.LazyInternalNetworkClientPermeateServerMappingDO; -import org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.converter.LazyInternalNetworkClientPermeateServerMappingConverter; -import org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.mapper.LazyInternalNetworkClientPermeateServerMappingMapper; +import jakarta.annotation.Resource; +import org.framework.lazy.cloud.network.heartbeat.server.standalone.domain.model.lazy.internal.network.client.permeate.server.mapping.LazyInternalNetworkClientPermeateServerMapping; import org.framework.lazy.cloud.network.heartbeat.server.standalone.domain.model.lazy.internal.network.client.permeate.server.mapping.LazyInternalNetworkClientPermeateServerMappingRepository; +import org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.converter.LazyInternalNetworkClientPermeateServerMappingConverter; +import org.framework.lazy.cloud.network.heartbeat.server.standalone.infrastructure.entity.LazyInternalNetworkClientPermeateServerMappingDO; import org.springframework.stereotype.Repository; -import java.util.stream.Collectors; +import org.wu.framework.lazy.orm.database.lambda.domain.LazyPage; +import org.wu.framework.lazy.orm.database.lambda.stream.lambda.LazyLambdaStream; import org.wu.framework.lazy.orm.database.lambda.stream.wrapper.LazyWrappers; import org.wu.framework.web.response.Result; import org.wu.framework.web.response.ResultFactory; -import jakarta.annotation.Resource; -import org.framework.lazy.cloud.network.heartbeat.server.standalone.domain.model.lazy.internal.network.client.permeate.server.mapping.LazyInternalNetworkClientPermeateServerMapping; -import org.wu.framework.lazy.orm.database.lambda.stream.lambda.LazyLambdaStream; + import java.util.List; -import org.wu.framework.lazy.orm.database.lambda.domain.LazyPage; +import java.util.stream.Collectors; /** * describe 客户端渗透服务端映射 * @@ -128,7 +128,7 @@ public class LazyInternalNetworkClientPermeateServerMappingRepositoryImpl imple @Override public Result remove(LazyInternalNetworkClientPermeateServerMapping lazyInternalNetworkClientPermeateServerMapping) { LazyInternalNetworkClientPermeateServerMappingDO lazyInternalNetworkClientPermeateServerMappingDO = LazyInternalNetworkClientPermeateServerMappingConverter.INSTANCE.fromLazyInternalNetworkClientPermeateServerMapping(lazyInternalNetworkClientPermeateServerMapping); - // lazyLambdaStream.delete(LazyWrappers.lambdaWrapperBean(lazyInternalNetworkClientPermeateServerMappingDO)); + lazyLambdaStream.delete(LazyWrappers.lambdaWrapperBean(lazyInternalNetworkClientPermeateServerMappingDO)); return ResultFactory.successOf(); }