博客
关于我
强烈建议你试试无所不能的chatGPT，快点击我
SOLR组合条件查询
阅读量：7042 次
发布时间：2019-06-28

本文共 13188 字，大约阅读时间需要 43 分钟。

hot3.png

package com.paic.icore.ph.search.biz.service.impl;

import java.util.ArrayList;

import java.util.Date;
import java.util.List;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.params.SolrParams;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import com.paic.icore.ph.search.biz.service.SearchUtilService;

import com.paic.icore.ph.search.dto.Condition;
import com.paic.icore.ph.search.dto.SearchResultItem;

import com.paic.icore.ph.util.DateUtil;

import com.paic.pafa.app.biz.service.BaseService;
import com.sun.org.apache.bcel.internal.generic.NEW;

/**

 * 
 * EX-GOUDINGCHENG001
 * 
 */
@Component("assemebleSearchResultUtil")
public class AssemebleSearchResultUtil extends BaseService {

    // /**

    // * 根据传入的Condition对象封装成为一个solrQuery
    // *
    // * con
    // * SolrQuery
    // */
    public static ModifiableSolrParams assembleSolrCondtion(Condition con,
            ArrayList<String> deptList) {
        ModifiableSolrParams query = new ModifiableSolrParams();
        SolrQuery filterQuery = new SolrQuery();
        if (null == con.getKeyword()) {
            query.add("q", "keyword:*");
        } else {
            query.add("q", "keyword:" + con.getKeyword() + "");
        }
        if (null != con.getProffession()) {
            String string = con.getProffession().substring(0, con.getProffession().length()-1);
            filterQuery.addFilterQuery("pproffession:"
                    + string + "");
        }

        if (null != con.getFiletype()) {

            filterQuery.addFilterQuery("filetype:"
                    + con.getFiletype() + "");
        }

        if (null != con.getProffession1()) {

            String string = con.getProffession1().substring(0, con.getProffession1().length()-1);
            filterQuery.addFilterQuery("proffession:"
                    + string + "");
        }
        
        if (null == con.getOrganization()
                || "".equalsIgnoreCase(con.getOrganization())) {

        } else {

        
            filterQuery.addFilterQuery("organization:"
                    + con.getOrganization() + "");
        }
        
        // 查询一天内的数据
        if ("week".equalsIgnoreCase(con.getDuring())) {
            filterQuery.addFilterQuery("day:[NOW/DAY-7DAY TO NOW/DAY]");
        }
        // 查询一周内的数据
        if ("month".equalsIgnoreCase(con.getDuring())) {
            filterQuery.addFilterQuery("day:[NOW/DAY-1MONTH TO NOW/DAY]");

        }

        // 查询近一年的数据
        if ("year".equalsIgnoreCase(con.getDuring())) {
            filterQuery.addFilterQuery("day:[NOW/DAY-1YEAR TO NOW/DAY]");
        }

        if ("heated".equalsIgnoreCase(con.getOderBy())) {

            query.add("sort", "day desc");
        }
        if ("atime".equalsIgnoreCase(con.getOderBy())) {
            query.add("sort", "day desc");
        }
        if ("algorithm".equalsIgnoreCase(con.getOderBy())) {
            query.add("sort", "day desc");
        }
        query.add("sort", "day desc");
        int startRow = ((int) con.getPage() == 0) ? 0 : (con.getPage() - 1)
                * con.getMaxResults();
        query.add("start", String.valueOf(startRow));
        query.add("rows", String.valueOf(con.getMaxResults()));
        query.add(filterQuery);
        // 设置高亮信息

//        query.add("hl", "true");// 高亮开关

//        query.add("hl.fl", "title,content");// 高亮返回字段
//        query.add("hl.fragsize", "150");// 高亮字段长度,0代表返回全部
//        query.add("hl.encoder", "html");
//        query.add("hl.fragmenter", "regex");
//        query.add("hl.useFastVectorHighlighter", "true");
//        query.add("f.content.hl.alternateField", "description");
//        query.add("f.content.hl.maxAlternateFieldLength", "150");
        // 对经验的可见性进行处理

        if ("ROLE_PH_SYS_MANAGER".equalsIgnoreCase(con.getRolecode())) {

            // 如果是超级管理员,全部可见
            if ("all".equalsIgnoreCase(con.getType()) || null == con.getType()) {
                filterQuery.addFilterQuery("type:*");
            } else {
                filterQuery.addFilterQuery("type:" + con.getType());
            }
        } else if("ROLE_PH_DEPT_MANAGER".equalsIgnoreCase(con.getRolecode())){//
            
            String string="(NOT visiflag:1 AND (";
            StringBuffer visibilitycontent = new StringBuffer();
            if (deptList.size() > 0) {
                //如果指定的人可见中包含有自己的UMCODE        
                visibilitycontent.append("visiInstitution:"+con.getUmCode()+" OR ");
                for (int i = 0; i < deptList.size(); i++) {
                    visibilitycontent.append("visiInstitution:"
                            + deptList.get(i).toString() + " OR ");
                }
            } else {
                //如果指定的人可见中包含有自己的UMCODE        
                visibilitycontent.append("visiInstitution:"+con.getUmCode()+" OR ");
                visibilitycontent.append("visiInstitution:' '");
            }
            if (deptList.size() > 0) {
                visibilitycontent.delete(visibilitycontent.length() - 3,
                        visibilitycontent.length());
            }
            visibilitycontent.append("))");
            visibilitycontent.append("OR (visiflag:1)");
        
            // 对用户所处的机构进行查询 处理结束
            string += visibilitycontent.toString();
            System.out.println("String:"+string);

            filterQuery.addFilterQuery(string);

            if (null != con.getType()) {
                if ("all".equalsIgnoreCase(con.getType())) {
                    filterQuery.addFilterQuery( "type:*");
                } else {
                    filterQuery.addFilterQuery("type:" + con.getType() + "");
                }
            }

        }else if("ROLE_PH_KNG_MANAGER".equalsIgnoreCase(con.getRolecode())){

            String string="(NOT visiflag:1 AND (";
            StringBuffer visibilitycontent = new StringBuffer();
            if (deptList.size() > 0) {
                //如果指定的人可见中包含有自己的UMCODE        
                visibilitycontent.append("visiInstitution:"+con.getUmCode()+" OR ");
                for (int i = 0; i < deptList.size(); i++) {
                    visibilitycontent.append("visiInstitution:"
                            + deptList.get(i).toString() + " OR ");
                }
            } else {
                //如果指定的人可见中包含有自己的UMCODE        
                visibilitycontent.append("visiInstitution:"+con.getUmCode()+" OR ");
                visibilitycontent.append("visiInstitution:' '");
            }
            if (deptList.size() > 0) {
                visibilitycontent.delete(visibilitycontent.length() - 3,
                        visibilitycontent.length());
            }
            visibilitycontent.append("))");
            visibilitycontent.append("OR (visiflag:1)");
        
            // 对用户所处的机构进行查询 处理结束
            string += visibilitycontent.toString();
            System.out.println("String:"+string);
            String type=con.getType();
            filterQuery.addFilterQuery(string);
            if (null != type) {
                if ("all".equalsIgnoreCase(con.getType())||null==con.getType()) {
                    filterQuery.addFilterQuery( "type:wenwen OR type:yunpan OR (type:jingyan AND visiInstitution:"+con.getUmCode()+") OR user:"+con.getUmCode()+"");
                } else if("wenwen".equalsIgnoreCase(con.getType())) {
                    filterQuery.addFilterQuery("type:"+con.getType());
                }else if("jingyan".equalsIgnoreCase(con.getType())){
                    filterQuery.addFilterQuery("type:jingyan AND visiInstitution:"+con.getUmCode());
                }else if("yunpan".equalsIgnoreCase(con.getType())){
                    filterQuery.addFilterQuery("type:"+con.getType());
                }
            }
        }else if ("ROLE_PH_NORMAL".equalsIgnoreCase(con.getRolecode())) {
            if ("all".equalsIgnoreCase(con.getType()) || null == con.getType()) {
                filterQuery.addFilterQuery("type:*");
                String string="visiInstitution:" + con.getUmCode() + " OR visiflag:1";
                
                if (deptList.size() > 0) {
                    //如果指定的人可见中包含有自己的UMCODE        
                    for (int i = 0; i < deptList.size(); i++) {
                        string+=" OR visiInstitution:"+deptList.get(i);
                    }
                } 
                filterQuery.addFilterQuery(string);
                
                
            } else {
                filterQuery.addFilterQuery("type:" + con.getType());
                String string="visiInstitution:" + con.getUmCode() + " OR visiflag:1";
                if (deptList.size() > 0) {
                    //如果指定的人可见中包含有自己的UMCODE        
                    for (int i = 0; i < deptList.size(); i++) {
                        string+=" OR visiInstitution:"+deptList.get(i);
                    }
                } 
                //filterQuery.addFilterQuery("visiInstitution:" + con.getUmCode() + " OR visiflag:1");
                filterQuery.addFilterQuery(string);
            }
            
            //    filterQuery.addFilterQuery("type:" + con.getType() + "");
        //    filterQuery.addFilterQuery("visiInstitution:" + con.getUmCode() + " OR visiflag:1");
            
            
        }
        query.add(filterQuery);
        System.out.println("query:" + query.toString());
        return query;
    }

    /**

     * 传入一个solrDocument封装成为一个SearchResultItem
     * 
     * doc
     * @return SearchResultItem
     */
    public SearchResultItem assembleWenwenResult(SolrDocument doc,
            QueryResponse response) {
        SearchResultItem item = new SearchResultItem();
        if (null != doc.get("id")) {
            item.setId(doc.get("id").toString());// 设置ID
        }
        if (null != doc.get("title")) {
            item.setTitle(doc.get("title").toString());// 设置标题栏
        }
        if (null != doc.get("user")) {
            item.setAskedBy(doc.get("user").toString());// 设置提问人
        }

        if (null != doc.get("acontent")) {

            item.setContent(doc.get("acontent").toString());

        }

        if (null != doc.get("type")) {

            item.setType(doc.get("type").toString());
        }
        String proffesString = "";
        if (null == doc.get("pproffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("pproffession").toString();
        }
        if (null == doc.get("proffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("proffession").toString();
        }
        item.setProffession(proffesString);

        Date showDate = (Date) doc.get("day");

        String ttString = DateUtil.getShowDate(showDate);
        item.setPublishDateStr(DateUtil.getShowDate(showDate));
        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("title");
            if (highlightSnippets != null) {
                item.setTitle(StringUtils.join(highlightSnippets.iterator(), ""));
            }

        }

        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("acontent");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }

        }

        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("desc");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }

        }

        return item;

    }

    public SearchResultItem assembleJinyanResult(SolrDocument doc,

            QueryResponse response) {
        SearchResultItem item = new SearchResultItem();
        if (null != doc.get("id")) {
            item.setId(doc.get("id").toString());// 设置ID
        }
        if (null != doc.get("title")) {
        
            item.setTitle(doc.get("title").toString());// 设置标题栏
        }
        if (null != doc.get("user")) {
        
            item.setAskedBy(doc.get("user").toString());// 设置提问人
        }

        if (null != doc.get("desc")) {

    
            item.setContent(doc.get("desc").toString());
        }

        if (null != doc.get("type")) {

            item.setType(doc.get("type").toString());

        }
        if (null != doc.get("day")) {
            Date showDate = (Date) doc.get("day");
            String ttString = DateUtil.getShowDate(showDate);
            item.setPublishDateStr(DateUtil.getShowDate(showDate));

        }

        String proffesString = "";
        if (null == doc.get("pproffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("pproffession").toString();
        }
        if (null == doc.get("proffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("proffession").toString();
        }
        item.setProffession(proffesString);
        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("title");
            if (highlightSnippets != null) {
                item.setTitle(StringUtils.join(highlightSnippets.iterator(), ""));
            }

        }

        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("acontent");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }
        }
        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("desc");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }
        }

        return item;

    }

    
    
    public SearchResultItem assembleYunpanResult(SolrDocument doc,
            QueryResponse response) {
        SearchResultItem item = new SearchResultItem();
        if (null != doc.get("id")) {
            item.setId(doc.get("id").toString());// 设置ID
        }
        if (null != doc.get("title")) {

            item.setTitle(doc.get("title").toString());// 设置标题栏

        }
        if (null != doc.get("user")) {

            item.setAskedBy(doc.get("user").toString());// 设置提问人

        }

        if (null != doc.get("desc")) {

            item.setContent(doc.get("desc").toString());

        }

        if (null != doc.get("type")) {

    
            item.setType(doc.get("type").toString());
        }
        if (null != doc.get("day")) {
            Date showDate = (Date) doc.get("day");
            String ttString = DateUtil.getShowDate(showDate);
            item.setPublishDateStr(DateUtil.getShowDate(showDate));

        }

        String proffesString = "";
        if (null == doc.get("pproffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("pproffession").toString();
        }
        if (null == doc.get("proffession")) {
            proffesString += "";
        } else {
            proffesString += doc.get("proffession").toString();
        }
        item.setProffession(proffesString);
        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("title");
            if (highlightSnippets != null) {
                item.setTitle(StringUtils.join(highlightSnippets.iterator(), ""));
            }

        }

        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("acontent");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }
        }
        if (response.getHighlighting() != null
                && response.getHighlighting().get(item.getId()) != null) {
            List<String> highlightSnippets = response.getHighlighting()
                    .get(item.getId()).get("desc");
            if (highlightSnippets != null) {
                item.setContent(StringUtils.join(highlightSnippets.iterator(),
                        ""));
            }
        }

        return item;

    }

}

 

转载于:https://my.oschina.net/goudingcheng/blog/1036450

你可能感兴趣的文章
我的友情链接
查看>>
css3基础学习笔记
查看>>
Ubuntu 16.04 使用小结
查看>>
AngularJS 后退按钮
查看>>
远程管理 KVM 虚机 - 每天5分钟玩转 OpenStack（5）
查看>>
python实战系列之通过libvirt操作KVM(六)
查看>>
高可用术语解释：Split-Brain
查看>>
Laravel 队列系统
查看>>
自动FTP 登录备份脚本
查看>>
view
查看>>
我的友情链接
查看>>
tomcat配置文件server.xml详解
查看>>
ipython的两种安装方式
查看>>
有流媒体功能的lnmp部署练习，强化练习
查看>>
android消息机制,异步和多线程
查看>>
Linux下抓包工具tcpdump以及分析包的工具wireshark
查看>>
设置新建maven项目的jdk版本
查看>>
用copy命令将两个文件绑定，上传asp马
查看>>
FIR.im:iOS 8.1.3 “各路助手挺尸”、“封杀一切助手”的背后
查看>>
我的友情链接
查看>>
喝酒易醉，品茶养心，人生如梦，品茶悟道，何以解忧？唯有杜康！-- 愿君每日到此一游！
当前时间: 2025-02-07 03:02:52   当前IP: 3.129.63.108   联系邮箱:javaeecc@qq.com     Copyright © 2020 - 2022 baihongyu.com 京ICP备2021015314号-2
强烈建议你试试无所不能的CHAT-GPT，快点击我
强烈建议你试试无所不能的CHAT-GPT，快点击我