package com.aliyun.opensearch;

import com.alibaba.fastjson.JSONArray;
import com.aliyun.opensearch.sdk.dependencies.com.google.common.collect.Lists;
import com.aliyun.opensearch.sdk.dependencies.org.json.JSONObject;
import com.aliyun.opensearch.sdk.generated.OpenSearch;
import com.aliyun.opensearch.sdk.generated.commons.OpenSearchClientException;
import com.aliyun.opensearch.sdk.generated.commons.OpenSearchException;
import com.aliyun.opensearch.sdk.generated.search.Config;
import com.aliyun.opensearch.sdk.generated.search.SearchFormat;
import com.aliyun.opensearch.sdk.generated.search.SearchParams;
import com.aliyun.opensearch.sdk.generated.search.general.SearchResult;
import com.aliyun.opensearch.search.SearchParamsBuilder;
import com.aliyun.opensearch.search.SearchResultDebug;

import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

public class testQueryProcessor {

    private static String appName = "替換為opensearch應用名";
    private static String host = "替換應用的API訪問地址";

    public static void main(String[] args) {
      
      	//用戶識別信息
        //從環境變量讀取配置的AccessKey ID和AccessKey Secret，運行代碼示例前必須先配置環境變量
        String accesskey = System.getenv("ALIBABA_CLOUD_ACCESS_KEY_ID");
        String secret = System.getenv("ALIBABA_CLOUD_ACCESS_KEY_SECRET");

        //查看文件和默認編碼格式
        System.out.println(String.format("file.encoding: %s", System.getProperty("file.encoding")));
        System.out.println(String.format("defaultCharset: %s", Charset.defaultCharset().name()));

        //創建并構造OpenSearch對象
        OpenSearch openSearch = new OpenSearch(accesskey, secret, host);

        //創建OpenSearchClient對象，并以OpenSearch對象作為構造參數
        OpenSearchClient serviceClient = new OpenSearchClient(openSearch);

        //創建SearcherClient對象，并以OpenSearchClient對象作為構造參數
        SearcherClient searcherClient = new SearcherClient(serviceClient);

        //定義Config對象，用于設定config子句參數，分頁或數據返回格式，指定應用名等等
        Config config = new Config(Lists.newArrayList(appName));
        config.setStart(0);
        config.setHits(5);
        //設置返回格式為json，3.1.2 sdk只支持返回xml和json格式，暫不支持返回fulljson類型，3.1.3版本支持fulljson
        config.setSearchFormat(SearchFormat.FULLJSON);

        // 創建參數對象
        SearchParams searchParams = new SearchParams(config);

        // 設置查詢子句，若需多個索引組合查詢，需要setQuery處合并，否則若設置多個setQuery后面的會替換前面查詢
        searchParams.setQuery("name:'kfc'");//通過kfc會搜索出“肯德基”的文檔

        List<String> qpname = new ArrayList<String>();
        qpname.add("qptest");  //設置查詢分析名稱
        searchParams.setQueryProcessorNames(qpname);

        // SearchParams的工具類，提供了更為便捷的操作
        SearchParamsBuilder paramsBuilder = SearchParamsBuilder.create(searchParams);

        // 執行返回查詢結果
        SearchResult searchResult;
        try {
            searchResult = searcherClient.execute(paramsBuilder);
            String result = searchResult.getResult();

            JSONObject obj = new JSONObject(result);

            //判斷執行返回是否成功，如果報錯執行下面代碼，否則輸出正確查詢內容
            if(obj.get("errors").toString().length()>2){
                JSONArray jsonarr = JSONArray.parseArray(obj.get("errors").toString());
                System.out.println("error code:"+jsonarr.getJSONObject(0).get("code"));
                System.out.println("error message:"+jsonarr.getJSONObject(0).get("message"));
            }else{
                // 輸出查詢結果
                System.out.println("output:"+obj.toString());
                //個別用戶可能需要debug請求地址信息
                SearchResultDebug srd = searcherClient.executeDebug(searchParams);
                System.out.println(srd.getRequestUrl().toString());
            }

        } catch (OpenSearchException e) {
            e.printStackTrace();
        } catch (OpenSearchClientException e) {
            e.printStackTrace();
        }

    }
}