package com.walker.semantics;

import com.walker.semantics.support.AbstractTextSimilarEngine;

import java.util.ArrayList;
import java.util.List;

public class TestTextSimilarEngine extends AbstractTextSimilarEngine {

//    @Override
//    protected List<Variable> doLoadUserLibrary() {
//        List<Variable> data = new ArrayList<>();
//        data.add(this.acquireVariable("1", "项目管理,人员设置"));
//        data.add(this.acquireVariable("2", "项目,过程,管理"));
//        data.add(this.acquireVariable("3", "人员,绩效,考核"));
//        data.add(this.acquireVariable("4", "银行业务,存款,定期"));
//        data.add(this.acquireVariable("5", "银行业务,存款,活期"));
//        data.add(this.acquireVariable("6", "银行业务,存款"));
//        data.add(this.acquireVariable("7", "人员管理,增加"));
//        data.add(this.acquireVariable("8", "基础信息库管理,基础信息库-单位审核,基础信息库-单位,单位审核岗-单位信息2"));
//        data.add(this.acquireVariable("9", "基础信息库管理,供养信息,供养信息区划上报"));
//        return data;
//    }

//    private Variable acquireVariable(String id, String text){
//        DefaultVariable variable = new DefaultVariable();
//        variable.setType(VariableType.String);
//        variable.setId(id);
//        variable.setValue(text);
//        return variable;
//    }

    @Override
    protected List<OwnerTextItem> doLoadUserLibrary() {
        List<OwnerTextItem> data = new ArrayList<>();
        data.add(this.acquireOwnerTextItem("1", "项目管理,人员设置", "test1"));
        data.add(this.acquireOwnerTextItem("2", "项目,过程,管理", "test1"));
        data.add(this.acquireOwnerTextItem("3", "人员,绩效,考核", "test1"));
        data.add(this.acquireOwnerTextItem("4", "银行业务,存款,定期", "test1"));
        data.add(this.acquireOwnerTextItem("5", "银行业务,存款,活期", "test1"));
        data.add(this.acquireOwnerTextItem("6", "银行业务,存款", "test1"));
        data.add(this.acquireOwnerTextItem("7", "人员管理,增加", "test1"));
        data.add(this.acquireOwnerTextItem("8", "基础信息库管理,基础信息库-单位审核,基础信息库-单位,单位审核岗-单位信息2", "test1"));
        data.add(this.acquireOwnerTextItem("9", "基础信息库管理,供养信息,供养信息区划上报", "demo"));
        return data;
    }

    private OwnerTextItem acquireOwnerTextItem(String id, String text, String owner){
        OwnerTextItem item = new OwnerTextItem();
        item.setId(id);
        item.setText(text);
        item.setOwner(owner);
        return item;
    }
}