com.palantir.dialogue.core.TargetUri Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dialogue-core Show documentation
Palantir open source project
The newest version!
/*
 * (c) Copyright 2024 Palantir Technologies Inc. All rights reserved.
 *
 * 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.palantir.dialogue.core;

import com.google.errorprone.annotations.CheckReturnValue;
import com.palantir.logsafe.Preconditions;
import java.net.InetAddress;
import java.util.Arrays;
import java.util.Optional;
import javax.annotation.Nullable;

public final class TargetUri implements Comparable {

    private final String uri;
    private final Optional resolvedAddress;

    private TargetUri(String uri, Optional resolvedAddress) {
        this.uri = MeshMode.stripMeshPrefix(Preconditions.checkNotNull(uri, "uri"));
        this.resolvedAddress = Preconditions.checkNotNull(resolvedAddress, "resolvedAddress");
    }

    /** Original service URI. */
    public String uri() {
        return uri;
    }

    /** Resolved IP address of the {@link #uri()}, or the IP address from the URI if it is not a hostname. */
    public Optional resolvedAddress() {
        return resolvedAddress;
    }

    @Override
    public int compareTo(TargetUri other) {
        int result = uri.compareTo(other.uri);
        if (result != 0) {
            return result;
        }

        result = Arrays.compare(
                resolvedAddress.map(InetAddress::getAddress).orElse(null),
                other.resolvedAddress.map(InetAddress::getAddress).orElse(null));
        if (result != 0) {
            return result;
        }

        return 0;
    }

    @Override
    public String toString() {
        return "TargetUri{uri='" + uri + "', resolvedAddress=" + resolvedAddress + '}';
    }

    @Override
    public boolean equals(Object other) {
        if (this == other) {
            return true;
        }
        if (other == null || getClass() != other.getClass()) {
            return false;
        }
        TargetUri targetUri = (TargetUri) other;
        return uri.equals(targetUri.uri) && resolvedAddress.equals(targetUri.resolvedAddress);
    }

    @Override
    public int hashCode() {
        int result = uri.hashCode();
        result = 31 * result + resolvedAddress.hashCode();
        return result;
    }

    /** Note that this does not retain service-mesh prefixes. */
    public static TargetUri of(String uri) {
        return new TargetUri(uri, Optional.empty());
    }

    /** Note that this does not retain service-mesh prefixes. */
    public static TargetUri of(String uri, @Nullable InetAddress resolvedAddress) {
        return new TargetUri(uri, Optional.ofNullable(resolvedAddress));
    }

    public static Builder builder() {
        return new Builder();
    }

    public static final class Builder {

        @Nullable
        private String uri;

        @Nullable
        private InetAddress resolvedAddress;

        private Builder() {}

        /** Sets the {@link #uri} field. Note that this does not retain service-mesh prefixes. */
        public Builder uri(String value) {
            this.uri = MeshMode.stripMeshPrefix(Preconditions.checkNotNull(value, "uri"));
            return this;
        }

        public Builder resolvedAddress(InetAddress value) {
            this.resolvedAddress = Preconditions.checkNotNull(value, "resolvedAddress");
            return this;
        }

        public Builder resolvedAddress(Optional value) {
            this.resolvedAddress =
                    Preconditions.checkNotNull(value, "resolvedAddress").orElse(null);
            return this;
        }

        @CheckReturnValue
        public TargetUri build() {
            return TargetUri.of(Preconditions.checkNotNull(uri, "uri"), resolvedAddress);
        }
    }
}