
tech.powerjob.worker.common.utils.WorkerNetUtils Maven / Gradle / Ivy
package tech.powerjob.worker.common.utils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import tech.powerjob.common.response.ResultDTO;
import tech.powerjob.common.serialize.JsonUtils;
import tech.powerjob.common.utils.HttpUtils;
import tech.powerjob.common.utils.NetUtils;
import tech.powerjob.common.utils.net.PingPongServer;
import tech.powerjob.common.utils.net.PingPongSocketServer;
import java.util.List;
/**
* PowerJob Worker 专用的网络工具类
*
* @author tjq
* @since 2024/2/8
*/
@Slf4j
public class WorkerNetUtils {
private static final String SERVER_CONNECTIVITY_CHECK_URL_PATTERN = "http://%s/server/checkConnectivity?targetIp=%s&targetPort=%d";
/**
* 多网卡情况下,解析可与 server 通讯的本地 IP 地址
* @param port 目标端口
* @param serverAddress server 服务地址
* @return 本机IP
*/
public static String parseLocalBindIp(int port, List serverAddress) {
PingPongServer pingPongServer = null;
try {
pingPongServer = new PingPongSocketServer();
pingPongServer.initialize(port);
log.info("[WorkerNetUtils] initialize PingPongSocketServer successfully~");
return NetUtils.getLocalHostWithNetworkInterfaceChecker(((networkInterface, inetAddress) -> {
if (inetAddress == null) {
log.info("[WorkerNetUtils] [networkInterface:{}] skip due to inetAddress is null!", networkInterface);
return false;
}
String workerIp = inetAddress.getHostAddress();
for (String address : serverAddress) {
String url = String.format(SERVER_CONNECTIVITY_CHECK_URL_PATTERN, address, workerIp, port);
try {
String resp = HttpUtils.get(url);
log.info("[WorkerNetUtils] [networkInterface:{},inetAddress:{}] check connectivity by url[{}], response: {}", networkInterface, inetAddress, url, resp);
if (StringUtils.isNotEmpty(resp)) {
ResultDTO> resultDTO = JsonUtils.parseObject(resp, ResultDTO.class);
boolean ret = Boolean.TRUE.toString().equalsIgnoreCase(String.valueOf(resultDTO.getData()));
if (ret) {
return true;
}
}
} catch (Exception ignore) {
}
}
return false;
}));
} catch (Exception e) {
log.warn("[WorkerNetUtils] PingPongSocketServer failed to start, which may result in an incorrectly bound IP, please pay attention to the initialize log.", e);
} finally {
if (pingPongServer != null) {
try {
pingPongServer.close();
log.info("[WorkerNetUtils] close PingPongSocketServer successfully~");
} catch (Exception e) {
log.warn("[WorkerNetUtils] close PingPongSocketServer failed!", e);
}
}
}
return NetUtils.getLocalHostWithNetworkInterfaceChecker(null);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy