Solr Schema Java Api 字段定义等相关操作

Stella981
• 阅读 570

import java.io.IOException;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.apache.commons.collections.CollectionUtils;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.impl.CloudSolrClient;
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.client.solrj.request.CollectionAdminRequest.Delete;
import org.apache.solr.client.solrj.request.schema.SchemaRequest;
import org.apache.solr.client.solrj.response.CollectionAdminResponse;
import org.apache.solr.client.solrj.response.schema.SchemaResponse;
import org.apache.solr.common.cloud.ZkConfigManager;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.util.NamedList;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import com.google.common.collect.Lists;

/**
 * 索引节点管理
 * 
 * @author lican
 * @date 2018-12-11
 */
@Component
public class CollectionManager {

    private static final Logger LOG = LoggerFactory.getLogger(CollectionManager.class);

    @Autowired
    private CloudSolrClient solrCloudClient;
    @Autowired
    private ZkConfigManager zkConfigManager;

    private final String configPath = this.getClass().getClassLoader().getResource("solrconfig").getPath();
    /**
     * 获取schemaFields参数
     */
    private static ModifiableSolrParams schemaFieldSolrParams = new ModifiableSolrParams();
    private static ModifiableSolrParams schemaCopyFieldSolrParams = new ModifiableSolrParams();
    private static ModifiableSolrParams schemaUniqueKeySolrParams = new ModifiableSolrParams();
    static {
        // 是否显示域类型的默认信息
        schemaFieldSolrParams.add("showDefaults", "false");
        // 是否返回动态域的定义信息
        schemaFieldSolrParams.add("includeDynamic", "true");
        // 指定返回那些域的定义信息
        schemaFieldSolrParams.add("f1", "name,_version_");

        schemaCopyFieldSolrParams.add("wt", "json");
        // 设置返回sourcefield信息
        schemaCopyFieldSolrParams.add("source.fl", "name");
        
        schemaUniqueKeySolrParams.add("wt", "json");
    }

    /**
     * 创建字段
     * 
     * @param param
     * @throws Exception
     */
    public boolean createFields(String collection, List schemaParams) {
        LOG.info("CollectionManager.createFields...collection:{},schema:{}", collection, schemaParams);
        try {
            List diffFields = getDiffFields(collection, schemaParams);
            if (CollectionUtils.isEmpty(diffFields)) {
                LOG.info("create fields fail,Fields is exist,collection:{},schemaParams:{}", collection, schemaParams);
                return false;
            }
            for (SchemaParam schemaParam : diffFields) {
                SchemaRequest.UniqueKey uniqueKey = new SchemaRequest.UniqueKey(schemaUniqueKeySolrParams);
                insertSchemaField(collection, schemaParam);
            }
            LOG.info("[createFields]create success");
            return true;
        } catch (Exception e) {
            LOG.error("[createFields] create fail,collection:{},schema:{}", collection, schemaParams, e);
            return false;
        }
    }

    /**
     * 写入单个field到schema
     * 
     * @param collection
     * @param schemaParam
     * @throws Exception
     */
    private void insertSchemaField(String collection, SchemaParam schemaParam) throws Exception {
        String fieldName = schemaParam.getName();
        // 增加到字段到Collection
        Map<String, Object> fieldAttributes = new HashMap<>();
        fieldAttributes = new HashMap<>();
        fieldAttributes.put(SchemaParam.NAME, fieldName);
        fieldAttributes.put(SchemaParam.TYPE, schemaParam.getType());
        fieldAttributes.put(SchemaParam.INDEXED, schemaParam.getIndexed());
        fieldAttributes.put(SchemaParam.STORED, schemaParam.getStored());
        fieldAttributes.put(SchemaParam.REQUIRED, schemaParam.getRequired());
        fieldAttributes.put(SchemaParam.MULTIVALUED, schemaParam.getMultiValued());
        fieldAttributes.put(SchemaParam.OMIT_NORMS, schemaParam.getOmitNorms());
        SchemaRequest.AddField addField = new SchemaRequest.AddField(fieldAttributes);
        addField.process(solrCloudClient, collection);
        LOG.info("[insertField]field :{},insert success", fieldName);
    }

    /**
     * 对比获取到schema中没有的Field
     * 
     * @param collection
     * @param schemaParams
     * @return
     */
    private List getDiffFields(String collection, List schemaParams) throws Exception {
        List diffSchemas = Lists.newArrayList();
        List schemaFields = getSchemaFields(collection);
        for (SchemaParam schemaParam : schemaParams) {
            String fieldName = schemaParam.getName();
            if (CollectionUtils.isNotEmpty(schemaFields) && !schemaFields.contains(fieldName)) {
                diffSchemas.add(schemaParam);
            }
        }
        return diffSchemas;
    }

    /**
     * 创建 copyField, 可用于数据union 主要参数source、dest,复制与只要将多个域组合成一个域
     * 
     * @param collection
     * @param columns
     */
    public void createCopyFields(String collection, List schemaParams) {
        LOG.info("create copy fields...collection:{},schema:{}", collection, schemaParams);
        List destFields = getDiffCopyField(collection, schemaParams);
        SchemaRequest.AddCopyField addCopyField = new SchemaRequest.AddCopyField("union_field", destFields);
        try {
            SchemaResponse.UpdateResponse response = addCopyField.process(solrCloudClient, collection);
            NamedList result = response.getResponse();
            LOG.info("create copy fields results :{}", result);
        } catch (Exception e) {
            LOG.error("create copy fields fail,collection:{},schema:{}", collection, schemaParams, e);
        }
    }

    /**
     * 对比获取到schema中没有的copyField
     * 
     * @return
     */
    private List getDiffCopyField(String collection, List schemaParams) {
        List sorecesCopyFields = getSchemaCopyFields(collection);
        List destFields = new ArrayList<>();
        for (SchemaParam schemaParam : schemaParams) {
            String fieldName = schemaParam.getName();
            if (!sorecesCopyFields.contains(fieldName)) {
                destFields.add(fieldName);
            }
        }
        return destFields;
    }

    /**
     * 查看所有copyFields
     * 
     * @param collection
     * @return
     */
    public List getSchemaCopyFields(String collection) {
        List sources = Lists.newArrayList();
        SchemaRequest.CopyFields allCopyFields = new SchemaRequest.CopyFields(schemaCopyFieldSolrParams);
        try {
            SchemaResponse.CopyFieldsResponse response = allCopyFields.process(solrCloudClient, collection);
            NamedList result = response.getResponse();
            List<NamedList> copyFields = (List<NamedList>) result.get("copyFields");
            for (NamedList copyField : copyFields) {
                for (Entry<String, Object> entry : copyField) {
                    String key = entry.getKey();
                    Object val = entry.getValue();
                    if (key.equals("source")) {
                        sources.add(val.toString());
                    }
                }
            }
        } catch (Exception e) {
            LOG.error("Get Schema CopyFields fail,collection:{} ", collection, e);
            return sources;
        }
        return sources;
    }

    /**
     * 按字段名删除字段
     * 
     * @param connection
     * @param fieldName
     */
    public void deleteField(String collection, String fieldName) {
        SchemaRequest.DeleteField deleteField = new SchemaRequest.DeleteField(fieldName);
        try {
            SchemaResponse.UpdateResponse response = deleteField.process(solrCloudClient, collection);
            if (null != response) {
                LOG.info("[deleteFields]delete success,collection:{},field:{},requestUrl:{}", collection, fieldName,
                        response.getRequestUrl());
            }
        } catch (Exception e) {
            LOG.error("[deleteFields]delete fail,collection:{},field:{}", collection, fieldName, e);
        }
    }

    /**
     * 覆盖字段
     * 
     * @param collection
     */
    public void updateField(String collection, List schemaParams) {
        Map<String, Object> fieldAttributes = new HashMap<>();
        fieldAttributes.put("name", "product_name");
        fieldAttributes.put("type", "date");
        fieldAttributes.put("stored", "true");
        fieldAttributes.put("omitNorms", true);
        SchemaRequest.ReplaceField replaceField = new SchemaRequest.ReplaceField(fieldAttributes);
        try {
            replaceField.process(solrCloudClient, collection);
        } catch (Exception e) {
            LOG.error("[updateFields]update fail,collection:{},schemaParams:{}", collection, schemaParams, e);
        }
    }

    /**
     * 上传默认配置到zk 只有新增索引时调用
     * 
     * @param collection
     * @return
     */
    public boolean uploadDefaultConfig(String collection) {
        LOG.info("[ready]upload default config ----ready----,collection:{}", collection);
        try {
            zkConfigManager.uploadConfigDir(Paths.get(configPath), collection);
            LOG.info("[success]upload default config ----success----,collection:{}", collection);
            return true;
        } catch (IOException e) {
            LOG.error("[fail]upload default config ----fail----,collection:{}", collection, e);
            return false;
        }
    }

    /**
     * 创建索引节点
     * 
     * @param collection
     * @return
     */
    public boolean createCollection(String collection) {
        LOG.info("[ready]create collection ----ready----,collection:{}", collection);
        CollectionAdminRequest.Create create = CollectionAdminRequest.createCollection(collection, 1, 1);
        try {
            CollectionAdminResponse resp = create.process(solrCloudClient);
            if (null == resp) {
                LOG.error("[fail]create collection ----fail----,create response is null,collection:{}", collection);
                return false;
            }
            if (!resp.isSuccess()) {
                LOG.error("[fail]create collection ----fail----,error message:{},collection:{}",
                        resp.getErrorMessages(), collection);
                return false;
            }
            LOG.info("[success]create collection ----success----,collection:{}", collection);
            return true;
        } catch (SolrServerException | IOException e) {
            LOG.error("[fail]create collection ----fail----,collection:{}", collection, e);
            return false;
        }
    }

    /**
     * 刷新节点配置信息（新增字段需要刷新后建索引）
     * 
     * @param collection
     * @return
     */
    public boolean reloadCollection(String collection) {
        LOG.info("[ready]reload collection ----ready----,collection:{}", collection);
        CollectionAdminRequest.Reload reload = CollectionAdminRequest.reloadCollection(collection);
        try {
            CollectionAdminResponse resp = reload.process(solrCloudClient);
            if (null == resp) {
                LOG.error("[fail]reload collection ----fail----,reload response is null,collection:{}", collection);
                return false;
            }
            if (!resp.isSuccess()) {
                LOG.error("[fail]reload collection ----fail----,error message:{},collection:{}",
                        resp.getErrorMessages(), collection);
                return false;
            }
            LOG.info("[success]reload collection ----success----,collection:{}", collection);
            return true;
        } catch (SolrServerException | IOException e) {
            LOG.error("[fail]reload collection ----fail----,collection:{}", collection, e);
            return false;
        }
    }

    /**
     * 删除索引节点
     * 
     * @param collection
     * @throws Exception
     */
    public void deleteCollection(String collection) throws Exception {
        LOG.info("deleteCollection:{},------ready------", collection);
        if (isExist(collection)) {
            Delete delete = CollectionAdminRequest.deleteCollection(collection);
            delete.process(solrCloudClient);
            LOG.info("deleteCollection:{},------success------", collection);
        } else {
            LOG.info("deleteCollection:{},------fail------,找不到要删除的collection", collection);
        }
    }

    /**
     * 索引是否存在
     * 
     * @param collection
     * @return
     * @throws Exception
     */
    public boolean isExist(String collection) throws Exception {
        boolean isConnection = solrCloudClient.getZkStateReader().getClusterState().hasCollection(collection);
        if (isConnection) {
            LOG.info("CollectionManager.collection:{},is exist", collection);
        } else {
            LOG.info("CollectionManager.collection:{},not exist", collection);
        }
        return isConnection;
    }

    /**
     * 查看schema 所有field
     * 
     * @param collection
     * @return
     * @throws Exception
     */
    public List getSchemaFields(String collection) throws Exception {
        List allField = Lists.newArrayList();
        SchemaRequest.Fields allFields = new SchemaRequest.Fields(schemaFieldSolrParams);
        SchemaResponse.FieldsResponse response;
        try {
            response = allFields.process(solrCloudClient, collection);
            NamedList result = response.getResponse();
            @SuppressWarnings("unchecked")
            List<NamedList> fields = (List<NamedList>) result.get("fields");
            for (NamedList field : fields) {
                for (Entry<String, Object> entry : field) {
                    String key = entry.getKey();
                    Object val = entry.getValue();
                    if ("name".equals(key)) {
                        if (null == val) {
                            allField.add("");
                        } else {
                            allField.add(val.toString());
                        }
                    }
                }
            }
        } catch (SolrServerException | IOException e) {
            LOG.error("Get Schema Fields fail,colleciotn:{}", collection, e);
        }
        return allField;
    }
}

stringsolr前端前端
点赞
收藏
评论区
推荐文章
blmius blmius
•
3年前
MySQL:[Err] 1292 - Incorrect datetime value: ‘0000-00-00 00:00:00‘ for column ‘CREATE_TIME‘ at row 1
文章目录问题用navicat导入数据时，报错：原因这是因为当前的MySQL不支持datetime为0的情况。解决修改sql\mode:sql\mode:SQLMode定义了MySQL应支持的SQL语法、数据校验等，这样可以更容易地在不同的环境中使用MySQL。全局s
4070
0
0
helloworld_34035044 helloworld_34035044
•
2年前
皕杰报表之UUID
​在我们用皕杰报表工具设计填报报表时，如何在新增行里自动增加id呢？能新增整数排序id吗？目前可以在新增行里自动增加id，但只能用uuid函数增加UUID编码，不能新增整数排序id。uuid函数说明：获取一个UUID，可以在填报表中用来创建数据ID语法：uuid()或uuid(sep)参数说明：sep布尔值，生成的uuid中是否包含分隔符''，缺省为
1112
0
0
待兔 待兔
•
3个月前
手写Java HashMap源码
HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程22
1046
0
0
Jacquelyn38 Jacquelyn38
•
3年前
2020年前端实用代码段，为你的工作保驾护航
有空的时候，自己总结了几个代码段，在开发中也经常使用，谢谢。1、使用解构获取json数据let jsonData  id: 1,status: "OK",data: 'a', 'b';let  id, status, data: number   jsonData;console.log(id, status, number )
2361
0
0
Wesley13 Wesley13
•
3年前
Java获得今日零时零分零秒的时间（Date型）
publicDatezeroTime()throwsParseException{    DatetimenewDate();    SimpleDateFormatsimpnewSimpleDateFormat("yyyyMMdd00:00:00");    SimpleDateFormatsimp2newS
1228
0
0
Wesley13 Wesley13
•
3年前
mysql设置时区
mysql设置时区mysql\_query("SETtime\_zone'8:00'")ordie('时区设置失败，请联系管理员！');中国在东8区所以加8方法二：selectcount(user\_id)asdevice,CONVERT\_TZ(FROM\_UNIXTIME(reg\_time),'08:00','0
1287
0
0
Wesley13 Wesley13
•
3年前
Java日期时间API系列36
　　十二时辰，古代劳动人民把一昼夜划分成十二个时段，每一个时段叫一个时辰。二十四小时和十二时辰对照表：时辰时间24时制子时深夜11：00凌晨01：0023：0001：00丑时上午01：00上午03：0001：0003：00寅时上午03：00上午0
1194
0
0
Wesley13 Wesley13
•
3年前
00：Java简单了解
浅谈Java之概述Java是SUN(StanfordUniversityNetwork)，斯坦福大学网络公司)1995年推出的一门高级编程语言。Java是一种面向Internet的编程语言。随着Java技术在web方面的不断成熟，已经成为Web应用程序的首选开发语言。Java是简单易学，完全面向对象，安全可靠，与平台无关的编程语言。
1231
0
0
Stella981 Stella981
•
3年前
Django中Admin中的一些参数配置
设置在列表中显示的字段，id为django模型默认的主键list_display('id','name','sex','profession','email','qq','phone','status','create_time')设置在列表可编辑字段list_editable
1146
0
0
Wesley13 Wesley13
•
3年前
MySQL部分从库上面因为大量的临时表tmp_table造成慢查询
背景描述Time:20190124T00:08:14.70572408:00User@Host:@Id:Schema:sentrymetaLast_errno:0Killed:0Query_time:0.315758Lock_
1486
0
0
Python进阶者 Python进阶者
•
9个月前
Excel中这日期老是出来00：00：00，怎么用Pandas把这个去除
大家好，我是皮皮。一、前言前几天在Python白银交流群【上海新年人】问了一个Pandas数据筛选的问题。问题如下：这日期老是出来00：00：00，怎么把这个去除。二、实现过程后来【论草莓如何成为冻干莓】给了一个思路和代码如下：pd.toexcel之前把这
484
0
0
Stella981
Stella981
Lv1
接天莲叶无穷碧，映日荷花别样红。
文章
1.7w
粉丝
42
获赞
28
热门文章
OpenVPN下载、安装、配置及使用详解
GitHub神器，一个可以白嫖全网无损音乐的神器
OpenWrt 路由器过滤广告的N种方法
Python—执行系统命令的四种方法（os.system、os.popen、commands、subprocess）
SS端加密以及obfs混淆