package net.rubyeye.xmemcached.utils;

import java.net.InetSocketAddress;

/**
 * InetSocketAddress wrapper,encapsulate an order number.
 * 
 * @author dennis
 * 
 */
public class InetSocketAddressWrapper {
  private volatile InetSocketAddress inetSocketAddress;
  private int order; // The address order in list
  private int weight; // The weight of this address
  private volatile String remoteAddressStr;
  private volatile String hostName;
  private volatile String mainNodeHostName;
  private boolean resolve;
  /**
   * Main memcached node address,if this is a main node,then this value is null.
   */
  private volatile InetSocketAddress mainNodeAddress;

  public InetSocketAddressWrapper(InetSocketAddress inetSocketAddress, int order, int weight,
      InetSocketAddress mainNodeAddress) {
    this(inetSocketAddress, order, weight, mainNodeAddress, true);
  }

  public InetSocketAddressWrapper(InetSocketAddress inetSocketAddress, int order, int weight,
      InetSocketAddress mainNodeAddress, boolean resolve) {
    super();
    setInetSocketAddress(inetSocketAddress);
    setMainNodeAddress(mainNodeAddress);
    this.order = order;
    this.weight = weight;
    this.resolve = resolve;
  }

  public String getRemoteAddressStr() {
    return this.remoteAddressStr;
  }

  public void setRemoteAddressStr(String remoteAddressStr) {
    this.remoteAddressStr = remoteAddressStr;
  }

  public final InetSocketAddress getInetSocketAddress() {
    if (resolve && ByteUtils.isValidString(this.hostName)) {
      // If it has a hostName, we try to resolve it again.
      return new InetSocketAddress(this.hostName, this.inetSocketAddress.getPort());
    } else {
      return this.inetSocketAddress;
    }
  }

  public final InetSocketAddress getResolvedSocketAddress() {
    return this.inetSocketAddress;
  }

  public final void setResolvedSocketAddress(InetSocketAddress addr) {
    this.inetSocketAddress = addr;
  }

  public final InetSocketAddress getResolvedMainNodeSocketAddress() {
    return this.mainNodeAddress;
  }

  public final void setResolvedMainNodeSocketAddress(InetSocketAddress addr) {
    this.mainNodeAddress = addr;
  }

  private final void setInetSocketAddress(InetSocketAddress inetSocketAddress) {
    this.inetSocketAddress = inetSocketAddress;
    if (resolve && inetSocketAddress != null) {
      this.hostName = inetSocketAddress.getHostName();
    }
  }

  public final int getOrder() {
    return this.order;
  }

  public int getWeight() {
    return this.weight;
  }

  public void setWeight(int weight) {
    this.weight = weight;
  }

  public InetSocketAddress getMainNodeAddress() {
    if (resolve && ByteUtils.isValidString(this.mainNodeHostName)) {
      return new InetSocketAddress(this.mainNodeHostName, this.mainNodeAddress.getPort());
    } else {
      return this.mainNodeAddress;
    }
  }

  private void setMainNodeAddress(InetSocketAddress mainNodeAddress) {
    this.mainNodeAddress = mainNodeAddress;
    if (resolve && mainNodeAddress != null) {
      this.mainNodeHostName = mainNodeAddress.getHostName();
    }
  }

  public final void setOrder(int order) {
    this.order = order;
  }

}

@@ -132,17 +132,16 @@
 mber.%0A *
- 
 %0A * @aut
@@ -153,17 +153,16 @@
 ennis%0A *
- 
 %0A */%0Apub
@@ -1027,24 +1027,52 @@
    super();%0A
+    this.resolve = resolve;%0A
     setInetS
@@ -1199,36 +1199,8 @@
 ht;%0A
-    this.resolve = resolve;%0A
   %7D%0A

fix https://github.com/killme2008/xmemcached/issues/68 which already appeared somewhere in april 2020? when allowing disabling address resolution