如何自定义compareTo方法同时考虑方向流动

Question

，如果我需要自定义代码才能使用这个逻辑

if this.srcAddr=other.srcAddr or 
this.src.Addr = other.sdstAddr 
this.srcPort=other.srcPort 
this.srcPort=other.dstPort 

，因为我要考虑的双向流动，从源数据包的目的地和分组从目的地到源头属于一个流程。

我该如何更改我的代码？

package myclassifier; 
public class Flows implements Comparable<Flows> { 

    String srcAddr, dstAddr, srcPort, dstPort, protocol; 

    public Flows(String sIP, String dIP){ 
     this.srcAddr = sIP; 
     this.dstAddr = dIP; 
    } 

    public int compareTo(Flows other) { 
      int res = (this.srcAddr.compareTo(other.srcAddr)); 
      if (res != 0) { 
       return res; 
      } 
      res = this.dstAddr.compareTo(other.dstAddr); 
      if (res != 0) { 
       return res; 
      } 
      res = this.srcPort.compareTo(other.srcPort); 
      if (res != 0) { 
       return res; 
      } 
      res = this.dstPort.compareTo(other.dstPort); 
      if (res != 0) { 
       return res; 
      } 
      return this.protocol.compareTo(other.protocol); 
    } 

    @Override 
    public int hashCode() { 

     final int prime = 31; 
     int result = 1; 
     result = prime * result + ((dstAddr == null) ? 0 : dstAddr.hashCode()); 
     result = prime * result + ((dstPort == null) ? 0 : dstPort.hashCode()); 
     result = prime * result + ((srcAddr == null) ? 0 : srcAddr.hashCode()); 
     result = prime * result + ((srcPort == null) ? 0 : srcPort.hashCode()); 
     return result; 

    } 

    @Override 
    public boolean equals(Object obj) { 
     if (this == obj) 
      return true; 
     if (obj == null) 
      return false; 

     if (getClass() != obj.getClass()) 
      return false; 

     Flows other = (Flows) obj; 

     if (dstAddr == null) { 
      if (other.dstAddr != null) 
       return false; 
     } else if (!dstAddr.equals(other.dstAddr)) 
      return false; 

     if (dstPort == null) { 
      if (other.dstPort != null) 
       return false; 
     } else if (!dstPort.equals(other.dstPort)) 
      return false; 

     if (srcAddr == null) { 
      if (other.srcAddr != null) 
       return false; 
     } else if (!srcAddr.equals(other.srcAddr)) 
      return false; 

     if (srcPort == null) { 
      if (other.srcPort != null) 
       return false; 
     } else if (!srcPort.equals(other.srcPort)) 
      return false; 

     return true; 
    } 

} 

Answer 1

只是一个猜测，但我认为你要寻找的是沿着以下线的东西（空检查，类型检查和错误处理作为练习留给用户）：

return ((this.srcAddr.equals(other.srcAddr) && this.srcPort.equals(other.srcPort) || 
     (this.srcAddr.equals(other.dstAddr) && this.srcPort.equals(other.dstPort)); 

请注意，这是基于以下假设：从机器1端口b到机器2端口a的连接与从机器1端口a到机器2端口b的连接不相同。

Answer 2

您可以大大简化compareTo方法。你只是比较字符串，如果你只是连接所有的字符串和一个单一的比较，你会得到相同的结果。下面的示例添加一个toString（）实现作为奖金：

@Override 
public String toString() { 
    return String.format("[%s, %s, %s, %s, %s]", srcAddr, dstAddr, srcPort, dstPort, protocol); 
} 

public int compareTo(Flows other) { 
    if (other == null) 
    return 0; // the necessary null check was missing in your code 

    return toString().compareTo(other.toString()); 
} 

如果您需要更多的性能，可以考虑构建连接字符串，而你构建一个流并将其存储在私有字段。