package com.walker.semantics;

import com.walker.infrastructure.utils.StringUtils;
import com.walker.infrastructure.utils.WaitConsoleInput;
import com.walker.semantics.support.DefaultSemanticsManager;
import org.junit.Test;

import java.util.List;

public class TestSimilar {

//    @Test
    public void search(){
        TextSimilarEngine engine = this.acquireTextSimilarEngine();
//        List<TextSimilar> list = engine.search(new String[]{"项目管理", "人员设置"});
//        List<TextSimilar> list = engine.search(new String[]{"银行业务","存款"});
//        List<TextSimilar> list = engine.search(new String[]{"银行业务","存款","定期"});
        List<TextSimilar> list = engine.search(new String[]{"基础信息库管理","基础信息库-单位","单位经办岗-人员信息"}, "demo");
        this.print(list);

//        List<WordMeta> metaList = engine.extract("项目过程管理，人员分派，银行缴费无法查询");
//        printMeta(metaList);

//        metaList = engine.extract("知识库分类解析");
//        printMeta(metaList);
//        WaitConsoleInput.waitInput(new WaitConsoleInput.InputCallback() {
//            @Override
//            public void doInput(String input) {
//                String[] text = StringUtils.commaDelimitedListToStringArray(input);
//                List<TextSimilar> list = engine.search(text);
//                if(list != null){
//                    for(TextSimilar ts : list){
//                        System.out.println(ts);
//                    }
//                }
//            }
//        });
    }

    private void printMeta(List<WordMeta> metaList){
        if(metaList != null){
            for(WordMeta ts : metaList){
                System.out.println(ts);
            }
        }
    }
    private void print(List<TextSimilar> list){
        if(list != null){
            for(TextSimilar ts : list){
                System.out.println(ts);
            }
        }
    }

    private TextSimilarEngine acquireTextSimilarEngine(){
        SemanticsManager manager = new DefaultSemanticsManager();
        TestTextSimilarEngine engine = new TestTextSimilarEngine();
        engine.setSemanticsManager(manager);
        engine.loadLibrary();
        return engine;
    }
}