com.github.kristofa.brave.EndPointSubmitterImpl Maven / Gradle / Ivy

Go to download
package com.github.kristofa.brave;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;

import org.apache.commons.lang3.Validate;

import com.twitter.zipkin.gen.Endpoint;

/**
 * {@link EndPointSubmitter} implementation.
 * 
 * @author kristof
 */
class EndPointSubmitterImpl implements EndPointSubmitter {

    private final ServerAndClientSpanState spanstate;

    /**
     * Creates a new instance.
     * 
     * @param state {@link CommonSpanState}, should not be null.
     */
    EndPointSubmitterImpl(final ServerAndClientSpanState state) {
        Validate.notNull(state);
        spanstate = state;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void submit(final String ip, final int port, final String serviceName) {

        final int ipv4 = ipAddressToInt(ip);
        final Endpoint endpoint = new Endpoint(ipv4, (short)port, serviceName);

        spanstate.setServerEndPoint(endpoint);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean endPointSubmitted() {
        return spanstate.getServerEndPoint() != null;
    }

    private int ipAddressToInt(final String ip) {
        InetAddress inetAddress = null;
        try {
            inetAddress = InetAddress.getByName(ip);
        } catch (final UnknownHostException e) {
            throw new IllegalArgumentException(e);
        }
        return ByteBuffer.wrap(inetAddress.getAddress()).getInt();
    }

}