aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/main/java/com/yahoo/search/debug/OutputSearchChain.java
blob: 4413ea462c8986322d49fd11e5031258e9adc6b6 (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
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.debug;

import static com.yahoo.protect.Validator.ensureNotNull;

import com.yahoo.jrt.Request;
import com.yahoo.jrt.StringValue;
import com.yahoo.jrt.Value;
import com.yahoo.yolean.Exceptions;
import com.yahoo.search.handler.SearchHandler;
import com.yahoo.search.searchchain.SearchChain;
import com.yahoo.search.searchchain.SearchChainRegistry;

/**
 * Outputs a human readable representation of a given search chain.
 *
 * @author tonytv
 */
final class OutputSearchChain implements DebugMethodHandler {
      private String getSearchChainName(Request request) {
        final int numParameters = request.parameters().size();

        if (numParameters == 0)
            return SearchHandler.defaultSearchChainName;
        else if (numParameters == 1)
            return request.parameters().get(0).asString();
        else
            throw new RuntimeException("Too many parameters given.");
    }

    private SearchChain getSearchChain(SearchChainRegistry registry, String searchChainName) {
        SearchChain searchChain = registry.getComponent(searchChainName);
        ensureNotNull("There is no search chain named '" + searchChainName + "'", searchChain);
        return searchChain;
    }

    public JrtMethodSignature getSignature() {
        String returnTypes = "" + (char)Value.STRING;
        String parametersTypes = "*"; //optional string
        return new JrtMethodSignature(returnTypes, parametersTypes);
    }

    public void invoke(Request request) {
        try {
            SearchHandler searchHandler = SearcherUtils.getSearchHandler();
            SearchChainRegistry searchChainRegistry = searchHandler.getSearchChainRegistry();
            SearchChain searchChain = getSearchChain(searchChainRegistry,
                    getSearchChainName(request));

            SearchChainTextRepresentation textRepresentation = new SearchChainTextRepresentation(searchChain, searchChainRegistry);
            request.returnValues().add(new StringValue(textRepresentation.toString()));
        } catch (Exception e) {
            request.setError(1000, Exceptions.toMessageString(e));
        }
    }


}