代码优化

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.Inject;
import org.noear.solon.core.Lifecycle;
import org.noear.solon.core.bean.LifecycleBean;
import java.util.*;
import java.util.function.Function;
@@ -40,7 +41,7 @@ import java.util.stream.Collectors;
* @date: 2022/8/6
*/
@Component
public class LicenseService implements Lifecycle {
public class LicenseService implements LifecycleBean {
@Inject
private MapperFacade mapperFacade;
@Db
@@ -36,6 +36,7 @@ import org.dromara.neutrinoproxy.server.util.ProxyUtil;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import org.noear.solon.core.Lifecycle;
import org.noear.solon.core.bean.LifecycleBean;
import java.util.*;
import java.util.function.Function;
@@ -47,7 +48,7 @@ import java.util.stream.Collectors;
* @date: 2022/8/8
*/
@Component
public class PortMappingService implements Lifecycle {
public class PortMappingService implements LifecycleBean {
@Inject
private MapperFacade mapperFacade;
@Db
@@ -174,6 +175,11 @@ public class PortMappingService implements Lifecycle {
portMappingMapper.updateById(portMappingDO);
// 更新VisitorChannel
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()) &&
StrUtil.isNotBlank(proxyConfig.getServer().getDomainName()) &&