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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.prelude.fastsearch;
import com.yahoo.fs4.BasicPacket;
import com.yahoo.fs4.ChannelTimeoutException;
import com.yahoo.fs4.QueryPacket;
import com.yahoo.fs4.QueryResultPacket;
import com.yahoo.fs4.mplex.FS4Channel;
import com.yahoo.fs4.mplex.InvalidChannelException;
import com.yahoo.search.Query;
import com.yahoo.search.Result;
import com.yahoo.search.dispatch.ResponseMonitor;
import com.yahoo.search.dispatch.SearchInvoker;
import com.yahoo.search.dispatch.searchcluster.Node;
import com.yahoo.search.result.ErrorMessage;
import com.yahoo.search.searchchain.Execution;
import java.io.IOException;
import java.util.Optional;
import java.util.logging.Logger;
/**
* {@link SearchInvoker} implementation for FS4 nodes and fdispatch
*
* @author ollivir
*/
public class FS4SearchInvoker extends SearchInvoker implements ResponseMonitor<FS4Channel> {
private static final Logger log = Logger.getLogger(FS4SearchInvoker.class.getName());
private final VespaBackEndSearcher searcher;
private FS4Channel channel;
private ErrorMessage pendingSearchError = null;
private Query query = null;
private QueryPacket queryPacket = null;
public FS4SearchInvoker(VespaBackEndSearcher searcher, Query query, FS4Channel channel, Optional<Node> node) {
super(node);
this.searcher = searcher;
this.channel = channel;
channel.setQuery(query);
channel.setResponseMonitor(this);
}
@Override
protected void sendSearchRequest(Query query) throws IOException {
log.finest("sending query packet");
this.query = query;
this.queryPacket = searcher.createQueryPacket(searcher.getServerId(), query);
try {
boolean couldSend = channel.sendPacket(queryPacket);
if (!couldSend) {
setPendingError("Could not reach '" + getName() + "'");
}
} catch (InvalidChannelException e) {
setPendingError("Invalid channel " + getName());
} catch (IllegalStateException e) {
setPendingError("Illegal state in FS4: " + e.getMessage());
}
}
private void setPendingError(String message) {
pendingSearchError = ErrorMessage.createBackendCommunicationError(message);
responseAvailable();
}
@Override
protected Result getSearchResult(Execution execution) throws IOException {
if (pendingSearchError != null) {
return errorResult(query, pendingSearchError);
}
BasicPacket[] basicPackets;
try {
basicPackets = channel.receivePackets(query.getTimeLeft(), 1);
} catch (ChannelTimeoutException e) {
return errorResult(query, ErrorMessage.createTimeout("Timeout while waiting for " + getName()));
} catch (InvalidChannelException e) {
return errorResult(query, ErrorMessage.createBackendCommunicationError("Invalid channel for " + getName()));
}
if (basicPackets.length == 0) {
return errorResult(query, ErrorMessage.createBackendCommunicationError(getName() + " got no packets back"));
}
log.finest(() -> "got packets " + basicPackets.length + " packets");
basicPackets[0].ensureInstanceOf(QueryResultPacket.class, getName());
QueryResultPacket resultPacket = (QueryResultPacket) basicPackets[0];
log.finest(() -> "got query packet. " + "docsumClass=" + query.getPresentation().getSummary());
if (query.getPresentation().getSummary() == null)
query.getPresentation().setSummary(searcher.getDefaultDocsumClass());
Result result = new Result(query);
searcher.addMetaInfo(query, queryPacket.getQueryPacketData(), resultPacket, result);
searcher.addUnfilledHits(result, resultPacket.getDocuments(), queryPacket.getQueryPacketData(), distributionKey());
return result;
}
@Override
public void release() {
if (channel != null) {
channel.close();
channel = null;
}
}
private String getName() {
return searcher.getName();
}
@Override
public void responseAvailable(FS4Channel from) {
responseAvailable();
}
}
|