solr项目实战

绝地灬酷狼 2022-05-18 03:06 181阅读 0赞

小编想要总结一下项目中用到的solr部分所以特意来总结一下。包括和Spring的结合。先说一下需求。

# 【需求】 #

教师添加题的时候，要根据题干内容自动的推荐已经添加的相似的题。

# 【pom.xml】 #

<dependency>
                <groupId>org.apache.solr</groupId>
                <artifactId>solr-solrj</artifactId>
                <version>7.1.0</version>
    </dependency>

# 【spring-context.xml】 #

<bean id="httpSolrClient" class="org.apache.solr.client.solrj.impl.HttpSolrClient">
            <constructor-arg name="builder" value="${solr.address}"/>
        </bean>

# 【solr.properties】 #

solr.address=http://xxx.xxx.xx.xxx:8080/solr/coreitoo

# 【serviceImpl.java】 #

/**
     * 题库管理 solr查询
     *
     * @author 滕柳
     */
    @Service("Service")
    public class ServiceImpl{
         @Autowired
        private HttpSolrClient solrServer;
        /**
         * 抽取添加到索引的方法
         *
         * @param questionMainEntityList 题库集合
         * @author 滕柳
         */
        public void addSolr(List<QuestionMainEntity> questionMainEntityList) {
            //插入到solr中
            try {
                for (int i = 0; questionMainEntityList != null && !questionMainEntityList.isEmpty() && i < questionMainEntityList.size(); i++) {
                    //创建文档对象
                    SolrInputDocument document = new SolrInputDocument();
                    //添加域
                    document.addField("id", questionMainEntityList.get(i).getId());
                    document.addField(QUESTION_CONTENT, questionMainEntityList.get(i).getQuestionContent());
                    //添加到索引库
                    solrServer.add(document);
                    //提交
                    solrServer.commit();
                }
            } catch (SolrServerException e) {
                logger.error("solr插入失败SolrServerException", e);
            } catch (IOException e) {
                logger.error("solr插入失败IOException", e);
            } catch (Exception e) {
                logger.error("solr插入失败Exception", e);
            }
        }
        
         /**
         * 使用solr查询题干内容
         *
         * @param questionContent 题干信息
         * @return 符合条件的题干实体集合
         * @author 滕柳
         */
        @Override
        public List<QuestionMainEntity> getQuestionContentBySolr(String questionContent) throws SolrServerException, IOException {
            List<QuestionMainEntity> questionContentList = new ArrayList<>();
            //执行查询，创建一个查询对象
            SolrQuery query = new SolrQuery();
            query.setQuery("question_content:" + questionContent);
            // 开启高亮组件或用
            query.setHighlight(true);
            // 高亮字段
            query.addHighlightField(QUESTION_CONTENT);
            //标记，高亮关键字前缀
            query.setHighlightSimplePre("<font color='red'>");
            //后缀
            query.setHighlightSimplePost("</font>");
            //执行查询
            QueryResponse response = solrServer.query(query);
            //取查询结果
            SolrDocumentList solrDocumentList = response.getResults();
            for (SolrDocument solrDocument : solrDocumentList) {
                QuestionMainEntity questionMainEntity = new QuestionMainEntity();
                questionMainEntity.setId("id");
                String question_content = solrDocument.get("question_content").toString();
                questionMainEntity.setQuestionContent(question_content);
                questionContentList.add(questionMainEntity);
            }
            return questionContentList;
        }
    
        /**
         * 检查solr中是否有数据
         *
         * @author 滕柳
         */
        @Override
        public void querySolr() throws SolrServerException, IOException {
            //执行查询，创建一个查询对象
            SolrQuery query = new SolrQuery();
            query.set("q", "*:*");
            //执行查询
            QueryResponse response = solrServer.query(query);
            //取查询结果
            SolrDocumentList solrDocumentList = response.getResults();
            if (response.getResults().isEmpty()) {
                importSolr();
            } else {
                QuestionMainExample questionMainExample = new QuestionMainExample();
                QuestionMainCriteria questionMainCriteria = questionMainExample.createCriteria();
                questionMainCriteria.andIsDeleteEqualTo((byte) 0);
                long count = questionMainDao.getCountByExample(questionMainExample);
                //如果相等，则不再需要向solr中导入数据
                if (solrDocumentList.getNumFound() != count) {
                    importSolr();
                }
            }
        }
    
        // deleteSingleProblemMain-删除单个题目
        @Override
        @Transactional(rollbackFor = Exception.class)
        public boolean deleteSingleProblemMain(String problemId, String operator) {
            List<String> listAll = getPidAndAllHisChildByPid(idList, problemId);
            //2.删除题目 同时删除solr
            if (questionMainDao.updateToHistoryFlag(listAll) >= 0 && questionSubDao.deleteByQuestionMainId(listAll, operator) >= 0) {
                try {
                    solrServer.deleteById(listAll);
                    solrServer.commit();
                } catch (SolrServerException e) {
                    e.printStackTrace();
                    logger.error("在solr中删除题库内容失败");
                } catch (IOException e) {
                    e.printStackTrace();
                    logger.error("在solr中删除题库内容失败", e);
                }
    }

删除的方法还有一个根据查询的题型内容进行删除

@Test
    	public void deleteDocumentByQuery() throws Exception {
    		solrServer.deleteByQuery("question_content:我");
    		solrServer.commit();
    	}

好了小编的实战关于添加查询删除就先写到这里了。这个是一个借口用的会后直接调用就好了。