RemotingUtils.java

/*
 * (C) 2007-2012 Alibaba Group Holding Limited.
 * 
 * Licensed 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 com.taobao.gecko.core.util;

import java.net.Inet6Address;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.NetworkInterface;
import java.net.SocketAddress;
import java.net.SocketException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import java.util.Enumeration;

import com.taobao.gecko.core.buffer.IoBuffer;
import com.taobao.gecko.service.config.WireFormatType;


/**
 * Remoting������
 * 
 * 
 * 
 * @author boyan
 * 
 * @since 1.0, 2009-12-16 ����06:21:51
 */
public class RemotingUtils {

    public static final String getAddrString(final SocketAddress address) {
        if (address == null) {
            return null;
        }
        if (address instanceof InetSocketAddress) {
            final InetSocketAddress socketAddr = (InetSocketAddress) address;
            final InetAddress inetAddr = socketAddr.getAddress();
            return (inetAddr != null ? inetAddr.getHostAddress() : socketAddr.getHostName()) + ":"
                    + socketAddr.getPort();
        }
        return null;
    }


    // ��������������һ���ǻ�·ip��ַ�����أ����û���ҵ����򷵻�InetAddress.getLocalHost()
    public static InetAddress getLocalHostAddress() throws UnknownHostException, SocketException {
        final Enumeration<NetworkInterface> enumeration = NetworkInterface.getNetworkInterfaces();
        InetAddress ipv6Address = null;
        while (enumeration.hasMoreElements()) {
            final NetworkInterface networkInterface = enumeration.nextElement();
            final Enumeration<InetAddress> en = networkInterface.getInetAddresses();
            while (en.hasMoreElements()) {
                final InetAddress address = en.nextElement();
                if (!address.isLoopbackAddress()) {
                    if (address instanceof Inet6Address) {
                        ipv6Address = address;
                    }
                    else {
                        // ����ʹ��ipv4
                        return address;
                    }
                }
            }

        }
        // û��ipv4����ʹ��ipv6
        if (ipv6Address != null) {
            return ipv6Address;
        }
        return InetAddress.getLocalHost();
    }


    public static InetSocketAddress getInetSocketAddressFromURI(final String uriStr) {
        if (uriStr == null) {
            return null;
        }
        try {
            final URI uri = new URI(uriStr);
            return new InetSocketAddress(uri.getHost(), uri.getPort());
        }
        catch (final URISyntaxException e) {
            // ignore
            return null;
        }
    }


    public static final String formatServerUrl(final WireFormatType wireFormatType,
            final InetSocketAddress socketAddress) {
        if (socketAddress == null) {
            throw new IllegalArgumentException("Null socketAddress");
        }
        final InetAddress inetAddress = socketAddress.getAddress();
        return wireFormatType.getScheme() + "://"
                + (inetAddress == null ? socketAddress.getHostName() : inetAddress.getHostAddress()) + ":"
                + socketAddress.getPort();
    }


    public static final String formatServerUrl(final WireFormatType wireFormatType, final String hostNmae,
            final int port) {
        if (StringUtils.isEmpty(hostNmae)) {
            throw new IllegalArgumentException("Blank hostName");
        }
        return wireFormatType.getScheme() + "://" + hostNmae + ":" + port;
    }


    public static String dumpBuffer(final IoBuffer buff) {
        final StringBuilder sb = new StringBuilder("[Buffer \n");
        final int pos = buff.position();
        int count = 0;
        for (int i = pos; i < buff.limit(); i++, count++) {
            RemotingUtils.byte2hex(buff.get(i), sb);
            sb.append(" ");
            if (count > 0 && count % 16 == 0) {
                sb.append("\n");
            }
        }
        sb.append("\n]");
        return sb.toString();
    }


    public static void byte2hex(final byte b, final StringBuilder buf) {
        final char[] hexChars = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };
        final int high = (b & 0xf0) >> 4;
        final int low = b & 0x0f;
        buf.append(hexChars[high]);
        buf.append(hexChars[low]);
    }

}