MixJobBuilder.java

/**
 * 
 */
package com.taobao.top.analysis.node.component;

import java.util.HashMap;
import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.taobao.top.analysis.config.MasterConfig;
import com.taobao.top.analysis.exception.AnalysisException;
import com.taobao.top.analysis.node.IJobBuilder;
import com.taobao.top.analysis.node.job.Job;
import com.taobao.top.analysis.util.AnalyzerUtil;

/**
 * 支持多个扩展的JobBuilder,默认集成了本地文件的builder
 * 此处有陷阱，支持多种类型的jobBuilder切换时注意
 * @author fangweng
 * @Email fangweng@taobao.com
 * 2011-11-28
 *
 */
public class MixJobBuilder implements IJobBuilder {
    private static final Log log = LogFactory.getLog(MixJobBuilder.class);

	Map<String,IJobBuilder> jobBuilders;
	private MasterConfig config;
	private IJobBuilder jobBuilder;
	private String jobResource;
	
	@Override
	public boolean isNeedRebuild() {
		if (jobBuilder == null)
			return false;
		else
			return jobBuilder.isNeedRebuild();
	}

	@Override
	public void setNeedRebuild(boolean needRebuild) {
		if (jobBuilder != null)
			jobBuilder.setNeedRebuild(true);
	}


	@Override
	public MasterConfig getConfig() {
		return config;
	}


	@Override
	public void setConfig(MasterConfig config) {
		this.config = config;
	}
	
	public void addBuilder(String protocol,IJobBuilder jobBuilder)
	{
		jobBuilders.put(protocol, jobBuilder);
	}
	
	public void removeBuilder(String protocol)
	{
		jobBuilders.remove(protocol);
	}
	
	public Map<String,Job> build() throws AnalysisException 
	{
		if(config == null)
			throw new AnalysisException("master config is null!");
			
		return build(config.getJobsSource());
	}
	
	@Override
	public Map<String,Job> build(String config) throws 
			AnalysisException {
				
		String protocol = "file";
//		String conf = config;
		jobResource = config;
		jobBuilder = jobBuilders.get(protocol);
		
		if (config.indexOf(":") > 0)
		{
			protocol = config.substring(0,config.indexOf(":"));
//			conf = config.substring(config.indexOf(":")+1);
			
			if (jobBuilders.containsKey(protocol))
				jobBuilder = jobBuilders.get(protocol);		
		}
		
		return jobBuilder.build(config);
	}

	@Override
	public void init() {
		jobBuilders = new HashMap<String,IJobBuilder>();
		FileJobBuilder fileJobBuilder = new FileJobBuilder();
		fileJobBuilder.init();
		jobBuilders.put("file", fileJobBuilder);
		jobBuilders.get("file").setConfig(config);
		if(log.isInfoEnabled()) {
		    log.info("mixJobBuilder init complete.");
		}
	}

	@Override
	public void releaseResource() {
		jobBuilders.clear();
	}

	@Override
	public Map<String,Job> rebuild(Map<String,Job> jobs) throws AnalysisException {
	    //此处有陷阱，支持多种类型的jobBuilder切换时注意
		if (this.isNeedRebuild())
		{
		    return jobBuilder.rebuild(jobs);
		}
		else
			return null;
	}

	@Override
	public void buildTasks(Job job) throws AnalysisException {
		// TODO Auto-generated method stub
		
	}

    /* (non-Javadoc)
     * @see com.taobao.top.analysis.node.IJobBuilder#isModified()
     */
    @Override
    public boolean isModified() {
        //此处有陷阱，支持多种类型的jobBuilder切换时注意
        if(!AnalyzerUtil.covertNullToEmpty(config.getJobsSource()).equals(AnalyzerUtil.covertNullToEmpty(this.jobResource)))
            return true;
        if(this.jobBuilders.get("file").isModified())
            return true;
        return false;
    }

    /* (non-Javadoc)
     * @see com.taobao.top.analysis.node.IJobBuilder#getJobResource()
     */
    @Override
    public String getJobResource() {
        return jobResource;
    }

}