|
|
|
@ -2,16 +2,20 @@ package com.supervision;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
|
|
import cn.hutool.core.map.MapUtil;
|
|
|
|
|
import cn.hutool.core.util.ArrayUtil;
|
|
|
|
|
import cn.hutool.core.util.NumberUtil;
|
|
|
|
|
import cn.hutool.json.JSONUtil;
|
|
|
|
|
import cn.hutool.poi.excel.ExcelReader;
|
|
|
|
|
import cn.hutool.poi.excel.ExcelUtil;
|
|
|
|
|
import cn.hutool.poi.excel.ExcelWriter;
|
|
|
|
|
import com.supervision.domain.QaSimilarityQuestionAnswer;
|
|
|
|
|
import com.supervision.model.AskPatientAnswer;
|
|
|
|
|
import com.supervision.model.AskTemplateQuestionLibrary;
|
|
|
|
|
import com.supervision.model.AskTemplateQuestionSimilarity;
|
|
|
|
|
import com.supervision.model.CommonDic;
|
|
|
|
|
import com.supervision.service.AskPatientAnswerService;
|
|
|
|
|
import com.supervision.service.AskTemplateQuestionLibraryService;
|
|
|
|
|
import com.supervision.service.AskTemplateQuestionSimilarityService;
|
|
|
|
|
import com.supervision.service.CommonDicService;
|
|
|
|
|
import com.supervision.util.SimilarityUtil;
|
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
@ -21,6 +25,7 @@ import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
|
|
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
|
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
import java.util.Optional;
|
|
|
|
@ -88,4 +93,32 @@ public class RedisVecTest {
|
|
|
|
|
writer.write(maps, true);
|
|
|
|
|
writer.close();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
|
private AskTemplateQuestionSimilarityService askTemplateQuestionSimilarityService;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 迁移相似问
|
|
|
|
|
*
|
|
|
|
|
*/
|
|
|
|
|
@Test
|
|
|
|
|
public void similarityQuestionMigrateTest() {
|
|
|
|
|
|
|
|
|
|
ArrayList<Long> dicIds = CollUtil.newArrayList(180L, 181L, 182L, 183L, 1114L);
|
|
|
|
|
|
|
|
|
|
List<AskTemplateQuestionLibrary> questionLibraryList = askTemplateQuestionLibraryService.lambdaQuery().notIn(AskTemplateQuestionLibrary::getDictId, dicIds).list();
|
|
|
|
|
|
|
|
|
|
for (AskTemplateQuestionLibrary library : questionLibraryList) {
|
|
|
|
|
List<String> questionList = library.getQuestion();
|
|
|
|
|
for (String question : questionList) {
|
|
|
|
|
AskTemplateQuestionSimilarity similarity = new AskTemplateQuestionSimilarity();
|
|
|
|
|
similarity.setLibraryId(library.getId());
|
|
|
|
|
similarity.setSimilarityQuestion(question);
|
|
|
|
|
askTemplateQuestionSimilarityService.save(similarity);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|