как настроить метод CompareTo для учета обоих направлений потока - PullRequest
Новая
       50

как настроить метод CompareTo для учета обоих направлений потока

0 голосов
/ 13 августа 2010

, если мне нужно настроить свой код с помощью этой логики 

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;
    }

}

Ответы [ 2 ]

0 голосов
/ 13 августа 2010

Вы можете значительно упростить метод сравнения. Вы просто сравниваете строки и получите тот же результат, если просто объедините все строки и одно сравнение. В следующем примере добавляется реализация 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());
}

Если вам нужна более высокая производительность, рассмотрите возможность создания объединенной строки во время создания потока и сохранения его в закрытом поле.

0 голосов
/ 13 августа 2010

Просто предположение, но я думаю, что вы ищете что-то вроде следующего (проверка нуля, проверка типа и обработка ошибок, оставленные пользователю в качестве упражнения): 

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

Обратите внимание, что это основано на предположении, что соединение от порта 1 машины b к порту 2 машины 2 не совпадает с подключением порта 1 машины a к порту 2 машины b.

...