多子类的设计

顶级接口

/**
 * @author caiyike
 * @version 1.0
 * @date 2019-03-03 16:09
 */
public interface ISearchService {

    IndexAliasVO getIndexAlias(String staticIndexName, List<String> staticAllIndexName);

    void resetIndex(String indexName, String settingFile, String mappingFile, String type);

    void refreshIndex(String indexName);

    boolean indexExists(String indexName);

    boolean switchIndexAlias(List<String> staticAllIndexName, String staticIndexName, String newIndexName);

    String indexItem(List<String> staticAllIndexName, String staticIndexName, String indexName, Object doc);

    boolean indexItems(List<String> staticAllIndexName, String staticIndexName, String indexName, Collection<Object> docs);
}

父类(顶级接口实现类)

/**
 * @author caiyike
 * @version 1.0
 * @date 2019-03-03 16:11
 */
@Slf4j
public abstract class SearchService implements ISearchService {

    private static Logger logger = LoggerFactory.getLogger(SearchService.class);

    @Autowired
    private Client esClient;

    @Autowired
    private ElasticsearchTemplate esTemplate;

    @Override
    public IndexAliasVO getIndexAlias(String staticIndexName, List<String> staticAllIndexName) {
        GetAliasesAction act = GetAliasesAction.INSTANCE;
        GetAliasesRequestBuilder req = act.newRequestBuilder(esClient);
        req.setAliases(staticIndexName);
        ImmutableOpenMap<String, List<AliasMetaData>> resp = req.execute().actionGet().getAliases();
        IndexAliasVO vo = new IndexAliasVO();
        vo.setAlias(staticIndexName);

        List<String> candidateIndices = new ArrayList<>(staticAllIndexName);
        vo.setCandidateIndices(candidateIndices);
        for (String index : staticAllIndexName) {
            if (SoaHelper.isNotNullOrEmpty(resp.get(index))) {
                vo.setCurrentIndex(index);
                candidateIndices.remove(index);
                break;
            }
        }
        return vo;
    }

    @Override
    public void resetIndex(String indexName, String settingFile, String mappingFile, String type) {
        if (esTemplate.indexExists(indexName))
            esTemplate.deleteIndex(indexName);

        createIndexIfNotExists(indexName, settingFile, mappingFile, type);
    }


    @Override
    public void refreshIndex(String indexName) {
        esTemplate.refresh(indexName);
    }

    @Override
    public boolean indexExists(String indexName) {
        return esTemplate.indexExists(indexName);
    }

    @Override
    public boolean switchIndexAlias(List<String> staticAllIndexName, String staticIndexName, String newIndexName) {
        checkIndexName(staticAllIndexName, staticIndexName, newIndexName, false);

        if (!esTemplate.indexExists(newIndexName)) {
            throw new RuntimeException(String.format("index not exists: '%s'", newIndexName));
        }

        IndexAliasVO vo = getIndexAlias(staticIndexName, staticAllIndexName);

        if (newIndexName.equals(vo.getCurrentIndex()))
            return true;

        refreshIndex(newIndexName);

        IndicesAliasesRequestBuilder req = esClient.admin().indices().prepareAliases();
        if (GeneralHelper.isStrNotEmpty(vo.getCurrentIndex())) {
            req.removeAlias(vo.getCurrentIndex(), staticIndexName);
        }
        return req.addAlias(newIndexName, staticIndexName).execute().actionGet().isAcknowledged();
    }


    @Override
    public String indexItem(List<String> staticAllIndexName, String staticIndexName, String indexName, Object doc) {
        checkIndexName(staticAllIndexName, staticIndexName, indexName, true);

        IndexQuery req = new IndexQuery();
        req.setIndexName(indexName);
        req.setObject(doc);
        return esTemplate.index(req);
    }


    @Override
    public boolean indexItems(List<String> staticAllIndexName, String staticIndexName, String indexName, Collection<Object> docs) {

        try {
            checkIndexName(staticAllIndexName, staticIndexName, indexName, true);

            if (SoaHelper.isNullOrEmpty(docs))
                return false;

            List<IndexQuery> list = new ArrayList<>();

            for (Object doc : docs) {
                IndexQuery req = new IndexQuery();
                req.setIndexName(indexName);
                req.setObject(doc);
                list.add(req);
            }

            if (CollectionUtils.isNotEmpty(list)) {
                logger.info("导入es数据量: {}", list.size());
                logger.info("导入批次item起始: {}----item结束: {}", list.get(0).getObject().toString(), list.get(list.size() - 1).getObject().toString());
            }

            esTemplate.bulkIndex(list);
        } catch (RuntimeException e) {
            logger.info("导入商品入es失败index: {}", indexName);

            e.printStackTrace();
        }
        return true;
    }


    private void createIndexIfNotExists(String indexName, String settingFile, String mappingFile, String type) {
        if (!esTemplate.indexExists(indexName)) {
            String setting = ElasticsearchTemplate.readFileFromClasspath(settingFile);
            String mapping = ElasticsearchTemplate.readFileFromClasspath(mappingFile);

            try {
                esTemplate.createIndex(indexName, setting);
                esTemplate.putMapping(indexName, type, mapping);
            } catch (Exception e) {
                if (esTemplate.indexExists(indexName))
                    esTemplate.deleteIndex(indexName);

                String msg = MessageFormat.format("create index {} fail", indexName);
                log.error(msg, e);
                throw new RuntimeException(msg, e);
            }
        }
    }

    private void checkIndexName(List<String> staticAllIndexName, String staticIndexName, String indexName, boolean withAliasName) {
        boolean isOK = staticAllIndexName.contains(indexName);

        if (!isOK && withAliasName)
            isOK = staticIndexName.equals(indexName);

        if (!isOK)
            throw new RuntimeException(String.format("invald indexName: '%s'", indexName));
    }
}

子类接口

 

/**
 * @author caiyike
 * @version 1.0
 * @date 2019-03-03 17:05
 */
public interface IOriginalItemSearchService extends ISearchService {

    IndexAliasVO getIndexAlias();
    void resetIndex(String indexName);
    void refreshIndex();
    void refreshIndex(String indexName);
    boolean indexExists(String indexName);
    boolean switchIndexAlias(String newIndexName);

}

子类实现

/**
 * @author caiyike
 * @version 1.0
 * @date 2019-03-03 17:08
 */
public class OriginalItemSearchService extends SearchService implements IOriginalItemSearchService {

    private static final String INDEX_NAME = "";

    private static final String INDEX_NAME_A = "";

    private static final String INDEX_NAME_B = "";

    private static final List<String> ALL_INDEX_NAME = Arrays.asList(INDEX_NAME_A, INDEX_NAME_B);

    public static final String INDEX_SETTING_FILE = "xxx-setting.json";

    public static final String INDEX_ITEM_TYPE_MAPPING_FILE = "xxx-mapping.json";

    public static final String ITEM_TYPE = "item";

    @Override
    public IndexAliasVO getIndexAlias() {
        return super.getIndexAlias(INDEX_NAME, ALL_INDEX_NAME);
    }

    @Override
    public void resetIndex(String indexName) {
        super.resetIndex(indexName, INDEX_SETTING_FILE, INDEX_ITEM_TYPE_MAPPING_FILE, ITEM_TYPE);
    }


    @Override
    public void refreshIndex() {
        super.refreshIndex(INDEX_NAME);
    }

    @Override
    public boolean switchIndexAlias(String newIndexName) {
        return super.switchIndexAlias(ALL_INDEX_NAME, INDEX_NAME, newIndexName);
    }


}