inlong-dataproxy/dataproxy-source/src/main/java/org/apache/inlong/dataproxy/utils/AddressUtils.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.inlong.dataproxy.utils;

import org.apache.inlong.dataproxy.consts.SourceConstants;

import io.netty.channel.Channel;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketAddress;
import java.util.Map;

public class AddressUtils {

    private static final Logger logger = LoggerFactory.getLogger(AddressUtils.class);

    private static final String localIp;

    static {
        localIp = getLocalIp();
    }

    public static String getLocalIp() {
        if (localIp != null) {
            return localIp;
        }
        String ip = "127.0.0.1";
        DatagramSocket socket = null;
        try {
            socket = new DatagramSocket();
            socket.connect(InetAddress.getByName("8.8.8.8"), 10002);
            ip = socket.getLocalAddress().getHostAddress();
        } catch (Throwable ex) {
            logger.error("Get local IP failure,", ex);
        } finally {
            if (socket != null) {
                socket.close();
            }
        }
        return ip;
    }

    public static String getChannelLocalIP(Channel channel) {
        return getChannelIP(channel, true);
    }

    public static String getChannelRemoteIP(Channel channel) {
        return getChannelIP(channel, false);
    }

    private static String getChannelIP(Channel channel, boolean isLocal) {
        if (channel == null) {
            return null;
        }
        SocketAddress address = isLocal ? channel.localAddress() : channel.remoteAddress();
        if (address == null) {
            return null;
        }
        String strAddrIP = address.toString();
        try {
            strAddrIP = strAddrIP.substring(1, strAddrIP.indexOf(':'));
            return strAddrIP;
        } catch (Exception ee) {
            if (isLocal) {
                logger.warn("Fail to get the local IP, localAddress = {}", address);
            } else {
                logger.warn("Fail to get the remote IP, remoteAddress = {}", address);
            }
            return null;
        }
    }

    public static String getSelfHost() {
        String localIp = null;
        Map<String, String> envMap = System.getenv();
        if (envMap.containsKey(SourceConstants.SYSENV_HOST_IP)) {
            String tmpVal = envMap.get(SourceConstants.SYSENV_HOST_IP);
            if (ConfStringUtils.isValidIp(tmpVal)) {
                localIp = tmpVal.trim();
            } else {
                logger.error("{}({}) config in system env not valid",
                        SourceConstants.SYSENV_HOST_IP, tmpVal);
            }
        }
        if (StringUtils.isBlank(localIp)
                || SourceConstants.VAL_LOOPBACK_HOST_VALUE.equals(localIp)
                || SourceConstants.VAL_DEF_HOST_VALUE.equals(localIp)) {
            localIp = AddressUtils.getLocalIp();
        }

        return localIp;
    }
}