IndexRebuildJob.java 4.52 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_CONVERSION);
		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 0 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----[end={}][cost={}]", System.currentTimeMillis(), (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);
		}
	}

}