IndexRebuildJob.java 5.28 KB
package com.yoho.search.consumer.job;

import com.yoho.error.event.SearchEvent;
import com.yoho.search.base.utils.EventReportEnum;
import com.yoho.search.base.utils.ISearchConstants;
import com.yoho.search.consumer.index.common.IYohoIndexService;
import com.yoho.search.consumer.index.rebuild.RebuildFlagService;
import com.yoho.search.core.es.utils.IgnoreSomeException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.context.ApplicationEventPublisherAware;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

@Component
public class IndexRebuildJob implements ApplicationEventPublisherAware {

    @Autowired
    private IYohoIndexService yohoIndexService;
    @Autowired
    private RebuildFlagService rebuildFlagService;
    @Autowired
    private TplAdaptorJob tplAdaptorJob;

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

    private ApplicationEventPublisher publisher;

    @Override
    public void setApplicationEventPublisher(ApplicationEventPublisher applicationEventPublisher) {
        this.publisher = applicationEventPublisher;
    }

    @Scheduled(cron = "0 0 3 * * ?")
    public void execute() {
        long begin = System.currentTimeMillis();
        logger.info("indexRebuildJob execute start----[begin={}]", begin);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_BRAND);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_SIZE);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_PRODUCT_COLOR);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_PRODUCT_SORT);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_STYLE);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_STANDARD);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_ROBOTQUESTION);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_PRODUCT_PRICE_PLAN);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_HELPER);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_SHOPS);

        try {
            // 重建完storagesku索引,等待1分钟,再执行其他索引的重建
            this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_STORAGE_SKU_INDEX);
            logger.info("indexRebuild storagesku success, wait 1 min for rebuild productindex... ");
            Thread.sleep(1 * 60 * 1000L);
        } catch (Exception e) {
        }

        try {
            this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_PRODUCT_INDEX);
            logger.info("indexRebuild productindex success, wait 2 min for doTplAdaptor ... ");
        } catch (Exception e) {
        }

        logger.info("indexRebuildJob execute end----[end={}][cost={}]", System.currentTimeMillis(), (System.currentTimeMillis() - begin));

    }

    @Scheduled(cron = "0 30 4 * * ?")
    public void rebuildSuggestIndex() {
        // suggest索引由于需要依赖productindex索引计算关键词的count 因此单独放在4点半重建
        long begin = System.currentTimeMillis();
        logger.info("indexRebuildJob rebuildSuggestIndex start----[begin={}]", begin);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_SUGGEST);
        logger.info("indexRebuildJob rebuildSuggestIndex end----[cost={}]", System.currentTimeMillis() - begin);
    }

    @Scheduled(cron = "0 45 4 * * ?")
    public void rebuildConversionIndex() {
        // conversion索引由于需要suggestConvertorJob的执行 因此单独放在5点半重建
        long begin = System.currentTimeMillis();
        logger.info("indexRebuildJob rebuildConversionIndex start----[begin={}]", begin);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_CONVERSION);
        logger.info("indexRebuildJob rebuildConversionIndex end----[cost={}]", System.currentTimeMillis() - begin);
    }

    @Scheduled(cron = "0 50 * * * ?")
    public void rebuildTblProductIndex() {
        long begin = System.currentTimeMillis();
        logger.info("indexRebuildJob rebuildTblProductIndex start----[begin={}]", begin);
        this.rebuildIndexWithlog(ISearchConstants.INDEX_NAME_TBLPRODUCT);
        logger.info("indexRebuildJob rebuildTblProductIndex end----[end={}][cost={}]", System.currentTimeMillis(), (System.currentTimeMillis() - begin));
    }

    private void rebuildIndexWithlog(String indexName) {
        try {
            long begin = System.currentTimeMillis();
            logger.info("rebuild index start----[indexName={}][begin={}]", indexName, begin);
            rebuildFlagService.waitingRebuildingIndex();
            yohoIndexService.rebuild(indexName);
            logger.info("rebuild index end----[indexName={}][end={}][cost={}ms]", indexName, System.currentTimeMillis(), (System.currentTimeMillis() - begin));
        } catch (Exception e) {
            publisher.publishEvent(new SearchEvent(EventReportEnum.INDEXREBUILDJOB_REBUILDINDEXWITHLOG.getEventName(), EventReportEnum.INDEXREBUILDJOB_REBUILDINDEXWITHLOG
                    .getFunctionName(), EventReportEnum.INDEXREBUILDJOB_REBUILDINDEXWITHLOG.getMoudleName(), "exception", IgnoreSomeException.filterSomeException(e), null));
            logger.error(e.getMessage(), e);
        }
    }

}