blob: 63302bee8c1bd453a46bb147e6d43cc13cf91d6a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
package com.yahoo.search.dispatch;
import com.yahoo.fs4.QueryPacketData;
import java.util.Arrays;
public class LeanHit implements Comparable<LeanHit> {
private final byte [] gid;
private final double relevance;
private final byte [] sortData;
private final int partId;
private final int distributionKey;
//TODO Remove when FS4 is gone
private QueryPacketData queryPacketData;
public LeanHit(byte [] gid, int partId, int distributionKey, double relevance) {
this.gid = gid;
this.relevance = Double.isNaN(relevance) ? Double.NEGATIVE_INFINITY : relevance;
this.sortData = null;
this.partId = partId;
this.distributionKey = distributionKey;
}
public LeanHit(byte [] gid, int partId, int distributionKey, byte [] sortData) {
this.gid = gid;
this.relevance = 0.0;
this.sortData = sortData;
this.partId = partId;
this.distributionKey = distributionKey;
}
public double getRelevance() { return relevance; }
public byte [] getGid() { return gid; }
public byte [] getSortData() { return sortData; }
public boolean hasSortData() { return sortData != null; }
public int getPartId() { return partId; }
public int getDistributionKey() { return distributionKey; }
QueryPacketData getQueryPacketData() { return queryPacketData; }
public void setQueryPacketData(QueryPacketData queryPacketData) {
this.queryPacketData = queryPacketData;
}
@Override
public int compareTo(LeanHit o) {
int res = (sortData != null)
? compareData(sortData, o.sortData)
: Double.compare(o.relevance, relevance);
return (res != 0) ? res : compareData(gid, o.gid);
}
private static int compareData(byte [] left, byte [] right) {
int i = Arrays.mismatch(left, right);
if (i < 0) {
return 0;
}
int max = Integer.min(left.length, right.length);
if (i >= max) {
return left.length - right.length;
}
int vl = (int) left[i] & 0xFF;
int vr = (int) right[i] & 0xFF;
return vl - vr;
}
}
|