更改六合边缘服务器动态连接

liusuyi

2025-10-14 8aa9e121554b84735c8ec6139bd095ea3acb05e7

 ard-work/src/main/java/com/ard/work/service/impl/ArdServerServiceImpl.java

@@ -2,6 +2,8 @@

import com.ard.common.utils.uuid.IdUtils;
import com.ard.work.component.forest.TyClient;
import com.ard.work.component.netty.TyTcpClientManager;
import com.ard.work.domain.cache.TyTcpServerCache;
import com.ard.work.domain.dto.TokenDTO;
import com.ard.work.domain.entity.ArdServerInfo;
import com.ard.work.domain.receiver.AccessTokenReceiver;
@@ -31,6 +33,8 @@
    private ArdServerInfoMapper ardServerInfoMapper;
    @Resource
    private TyClient tyClient;
    @Resource
    private TyTcpClientManager tyTcpClientManager;

    /**
     * 查询服务器管理
@@ -63,7 +67,11 @@
    @Override
    public int insertArdServerManagement(ArdServerInfo ardServerInfo) {
        ardServerInfo.setId(IdUtils.fastUUID());
        return ardServerInfoMapper.insertArdServerManagement(ardServerInfo);
        int i = ardServerInfoMapper.insertArdServerManagement(ardServerInfo);
        if (i > 0) {
            tyTcpClientManager.connectServer(ardServerInfo);
        }
        return i;
    }

    /**
@@ -74,7 +82,11 @@
     */
    @Override
    public int updateArdServerManagement(ArdServerInfo ardServerInfo) {
        return ardServerInfoMapper.updateArdServerManagement(ardServerInfo);
        int i = ardServerInfoMapper.updateArdServerManagement(ardServerInfo);
        if (i > 0) {
            tyTcpClientManager.connectServer(ardServerInfo);
        }
        return i;
    }

    /**
@@ -85,7 +97,14 @@
     */
    @Override
    public int deleteArdServerManagementByIds(String[] ids) {
        return ardServerInfoMapper.deleteArdServerManagementByIds(ids);
        int i = ardServerInfoMapper.deleteArdServerManagementByIds(ids);
        if (i > 0) {
            for (String id : ids) {
                //移除tcp连接
                tyTcpClientManager.disconnectServer(id);
            }
        }
        return i;
    }

    /**
@@ -114,8 +133,7 @@
                .timeStamp(timeStamp)
                .build();
        try {
            String baseUrl = "http://" + serverInfo.getHost() + ":" + serverInfo.getHttpPort();
            TyHttpCommonReceiver<AccessTokenReceiver> getTokenRes = tyClient.getToken(baseUrl, gtd);
            TyHttpCommonReceiver<AccessTokenReceiver> getTokenRes = tyClient.getToken(id, gtd);
            accessToken = getTokenRes.getContent();
        } catch (Exception e) {
            log.error("获取token失败:{}", e.getMessage());
@@ -133,19 +151,4 @@
        return new Date().getTime();
    }

    /**
     * 根据 baseUrl 获取 服务器http接口token（给拦截器使用）
     */
    @Override
    public AccessTokenReceiver getAccessTokenByBaseUrl(String baseUrl) {
        // 找到对应的 serverId
        List<ArdServerInfo> servers = selectArdServerManagementList(new ArdServerInfo());
        for (ArdServerInfo server : servers) {
            String _baseUrl = "http://" + server.getHost() + ":" + server.getHttpPort();
            if (_baseUrl.equals(baseUrl)) {
                return getAccessToken(server.getId());
            }
        }
        return null;
    }
}