代码优化

This commit is contained in:
aoshiguchen
2023-04-03 11:08:48 +08:00
parent 1be2326873
commit 8a37ae2d4b
2 changed files with 9 additions and 2 deletions
@@ -29,6 +29,7 @@ import org.jetbrains.annotations.Nullable;
import org.noear.solon.annotation.Component; import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject; import org.noear.solon.annotation.Inject;
import org.noear.solon.core.Lifecycle; import org.noear.solon.core.Lifecycle;
import org.noear.solon.core.bean.LifecycleBean;
import java.util.*; import java.util.*;
import java.util.function.Function; import java.util.function.Function;
@@ -40,7 +41,7 @@ import java.util.stream.Collectors;
* @date: 2022/8/6 * @date: 2022/8/6
*/ */
@Component @Component
public class LicenseService implements Lifecycle { public class LicenseService implements LifecycleBean {
@Inject @Inject
private MapperFacade mapperFacade; private MapperFacade mapperFacade;
@Db @Db
@@ -36,6 +36,7 @@ import org.dromara.neutrinoproxy.server.util.ProxyUtil;
import org.noear.solon.annotation.Component; import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject; import org.noear.solon.annotation.Inject;
import org.noear.solon.core.Lifecycle; import org.noear.solon.core.Lifecycle;
import org.noear.solon.core.bean.LifecycleBean;
import java.util.*; import java.util.*;
import java.util.function.Function; import java.util.function.Function;
@@ -47,7 +48,7 @@ import java.util.stream.Collectors;
* @date: 2022/8/8 * @date: 2022/8/8
*/ */
@Component @Component
public class PortMappingService implements Lifecycle { public class PortMappingService implements LifecycleBean {
@Inject @Inject
private MapperFacade mapperFacade; private MapperFacade mapperFacade;
@Db @Db
@@ -174,6 +175,11 @@ public class PortMappingService implements Lifecycle {
portMappingMapper.updateById(portMappingDO); portMappingMapper.updateById(portMappingDO);
// 更新VisitorChannel // 更新VisitorChannel
visitorChannelService.updateVisitorChannelByPortMapping(oldPortMappingDO, portMappingDO); visitorChannelService.updateVisitorChannelByPortMapping(oldPortMappingDO, portMappingDO);
// 删除老的域名映射
if (NetworkProtocolEnum.HTTP.getDesc().equals(oldPortMappingDO.getProtocal()) &&
StrUtil.isNotBlank(oldPortMappingDO.getSubdomain())) {
ProxyUtil.removeSubdomainToServerPort(oldPortMappingDO.getSubdomain());
}
// 更新域名映射 // 更新域名映射
if (NetworkProtocolEnum.HTTP.getDesc().equals(portMappingDO.getProtocal()) && if (NetworkProtocolEnum.HTTP.getDesc().equals(portMappingDO.getProtocal()) &&
StrUtil.isNotBlank(proxyConfig.getServer().getDomainName()) && StrUtil.isNotBlank(proxyConfig.getServer().getDomainName()) &&