com.taobao.tddl.interact.rule.TableRule.java Source code

Introduction

Here is the source code for com.taobao.tddl.interact.rule.TableRule.java

Source

/*(C) 2007-2012 Alibaba Group Holding Limited.   
 *This program is free software; you can redistribute it and/or modify   
*it under the terms of the GNU General Public License version 2 as   
* published by the Free Software Foundation.   
* Authors:   
*   junyu <junyu@taobao.com> , shenxun <shenxun@taobao.com>,   
*   linxuan <linxuan@taobao.com> ,qihao <qihao@taobao.com>    
*/
package com.taobao.tddl.interact.rule;

import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

import com.taobao.tddl.interact.rule.ruleimpl.DbVirtualNodeRule;
import com.taobao.tddl.interact.rule.ruleimpl.GroovyRule;
import com.taobao.tddl.interact.rule.ruleimpl.TableVirtualNodeRule;
import com.taobao.tddl.interact.rule.ruleimpl.WrappedGroovyRule;
import com.taobao.tddl.interact.rule.util.SimpleNamedMessageFormat;
import com.taobao.tddl.interact.rule.virtualnode.DBTableMap;
import com.taobao.tddl.interact.rule.virtualnode.TableSlotMap;

/**
 * 
 * 
 * @author linxuan
 * 
 */
public class TableRule extends VirtualTable {
    Log logger = LogFactory.getLog(TableRule.class);

    private String[] dbRules; // string expression or mappingrule
    private String[] tbRules; // string expression or mappingrule
    private List<String> extraPackages;//add by junyu:

    /**
     * dbRulestbRules
     * dbRuleParamestbRuleParamesruleParames
     */
    private String[] ruleParames;
    private String[] dbRuleParames;
    private String[] tbRuleParames;

    public void init() {
        //
        super.setExtraPackagesStr(extraPackages);
        initDbIndexes();
        initVnodeMap();
        replaceWithParam(this.dbRules, dbRuleParames != null ? dbRuleParames : ruleParames);
        replaceWithParam(this.tbRules, tbRuleParames != null ? tbRuleParames : ruleParames);
        super.setDbShardRules(convertToRuleArray(dbRules, dbNamePattern, tableSlotMap, dbTableMap, false));
        super.setTbShardRules(convertToRuleArray(tbRules, tbNamePattern, tableSlotMap, dbTableMap, true));
        super.init();
    }

    protected final void initDbIndexes() {
        //TODO
    }

    protected static void replaceWithParam(Object[] rules, String[] params) {
        if (params == null || rules == null) {
            return;
        }
        for (int i = 0; i < rules.length; i++) {
            if (rules[i] instanceof String) {
                rules[i] = replaceWithParam((String) rules[i], params);
            }
        }
    }

    private static String replaceWithParam(String template, String[] params) {
        if (params == null || template == null) {
            return template;
        }
        if (params.length != 0 && params[0].indexOf(":") != -1) {
            // paramsNamedParam
            return replaceWithNamedParam(template, params);
        }
        return new MessageFormat(template).format(params);
    }

    private static String replaceWithNamedParam(String template, String[] params) {
        Map<String, String> args = new HashMap<String, String>();
        for (String param : params) {
            int index = param.indexOf(":");
            if (index == -1) {
                throw new IllegalArgumentException(
                        " params:" + Arrays.asList(params));
            }
            args.put(param.substring(0, index).trim(), param.substring(index + 1).trim());
        }
        return new SimpleNamedMessageFormat(template).format(args);
    }

    protected static List<Rule<String>> convertToRuleArray(Object[] rules, String keyPattern,
            TableSlotMap tableSlotMap, DBTableMap dbTableMap, boolean isTableRule) {
        List<Rule<String>> ruleList = new ArrayList<Rule<String>>(1);
        if (null == rules) {
            //tableRule
            if (tableSlotMap != null && dbTableMap != null && !isTableRule) {
                ruleList.add(new DbVirtualNodeRule(String.valueOf(""), dbTableMap, extraPackagesStr));
                return ruleList;
            } else {
                return null;
            }
        }

        for (Object rule : rules) {
            if (keyPattern != null && keyPattern.length() != 0) {
                ruleList.add(new WrappedGroovyRule(String.valueOf(rule), keyPattern, extraPackagesStr));
            } else {
                if (tableSlotMap != null && dbTableMap != null && isTableRule) {
                    ruleList.add(new TableVirtualNodeRule(String.valueOf(rule), tableSlotMap, extraPackagesStr));
                } else {
                    ruleList.add(new GroovyRule<String>(String.valueOf(rule), extraPackagesStr));
                }
            }
        }

        return ruleList;
    }

    public void setDbRuleArray(List<String> dbRules) {
        //String[],spring
        this.dbRules = dbRules.toArray(new String[dbRules.size()]);
    }

    public void setTbRuleArray(List<String> tbRules) {
        //String[],spring
        this.tbRules = tbRules.toArray(new String[tbRules.size()]);
    }

    public void setDbRules(String dbRules) {
        if (this.dbRules == null) {
            // dbRuleArray
            //this.dbRules = dbRules.split("\\|");
            this.dbRules = new String[] { dbRules }; //|
        }
    }

    public void setTbRules(String tbRules) {
        if (this.tbRules == null) {
            // tbRuleArray
            //this.tbRules = tbRules.split("\\|");
            this.tbRules = new String[] { tbRules }; //|
        }
    }

    public void setRuleParames(String ruleParames) {
        if (ruleParames.indexOf('|') != -1) {
            // |,
            this.ruleParames = ruleParames.split("\\|");
        } else {
            this.ruleParames = ruleParames.split(",");
        }
    }

    public void setRuleParameArray(String[] ruleParames) {
        this.ruleParames = ruleParames;
    }

    public void setDbRuleParames(String dbRuleParames) {
        this.dbRuleParames = dbRuleParames.split(",");
    }

    public void setDbRuleParameArray(String[] dbRuleParames) {
        this.dbRuleParames = dbRuleParames;
    }

    public void setTbRuleParames(String tbRuleParames) {
        this.tbRuleParames = tbRuleParames.split(",");
    }

    public void setTbRuleParameArray(String[] tbRuleParames) {
        this.tbRuleParames = tbRuleParames;
    }

    public void setExtraPackages(List<String> extraPackages) {
        this.extraPackages = extraPackages;
    }
}